package view

import (
	"strings"

	"github.com/derailed/k9s/internal/resource"
	"github.com/derailed/k9s/internal/ui"
)

// Context presents a context viewer.
type Context struct {
	*Resource
}

// NewContext return a new context viewer.
func NewContext(title, gvr string, list resource.List) ResourceViewer {
	c := Context{
		Resource: NewResource(title, gvr, list),
	}
	c.extraActionsFn = c.extraActions
	c.enterFn = c.useCtx
	c.masterPage().SetSelectedFn(c.cleanser)

	return &c
}

func (c *Context) extraActions(aa ui.KeyActions) {
	c.masterPage().RmAction(ui.KeyShiftA)
}

func (c *Context) useCtx(app *App, _, res, sel string) {
	if err := c.useContext(sel); err != nil {
		app.Flash().Err(err)
		return
	}
	app.gotoResource("po", true)
}

func (*Context) cleanser(s string) string {
	name := strings.TrimSpace(s)
	if strings.HasSuffix(name, "*") {
		name = strings.TrimRight(name, "*")
	}
	if strings.HasSuffix(name, "(𝜟)") {
		name = strings.TrimRight(name, "(𝜟)")
	}
	return name
}

func (c *Context) useContext(name string) error {
	ctx := c.cleanser(name)
	if err := c.list.Resource().(*resource.Context).Switch(ctx); err != nil {
		return err
	}

	c.app.switchCtx(name, false)
	c.refresh()
	if tv, ok := c.GetPrimitive("ctx").(*Table); ok {
		tv.Select(1, 0)
	}

	return nil
}