k9s/internal/views/resource.go

package views

import (
	"context"
	"fmt"
	"strconv"
	"time"

	"github.com/derailed/k9s/internal/config"
	"github.com/derailed/k9s/internal/resource"
	"github.com/gdamore/tcell"
	"github.com/rs/zerolog/log"
)

type (
	updatable interface {
		restartUpdates()
		stopUpdates()
		update(context.Context)
	}

	resourceView struct {
		*masterDetail

		namespaces map[int]string
		list       resource.List
		cancelFn   context.CancelFunc
		parentCtx  context.Context
		path       *string
		colorerFn  colorerFn
		decorateFn decorateFn
	}
)

func newResourceView(title string, app *appView, list resource.List) resourceViewer {
	v := resourceView{
		masterDetail: newMasterDetail(title, app, list.GetNamespace()),
		list:         list,
	}
	v.masterPage().filterChanged(v.filterResource)

	return &v
}

// Init watches all running pods in given namespace
func (v *resourceView) init(ctx context.Context, ns string) {
	v.masterDetail.init(ns, v.backCmd)

	v.parentCtx = ctx
	var vctx context.Context
	vctx, v.cancelFn = context.WithCancel(ctx)

	colorer := defaultColorer
	if v.colorerFn != nil {
		colorer = v.colorerFn
	}
	v.masterPage().setColorer(colorer)

	v.update(vctx)
	v.app.clusterInfoRefresh()
	v.refresh()

	tv := v.masterPage()
	r, _ := tv.GetSelection()
	if r == 0 && tv.GetRowCount() > 0 {
		tv.Select(1, 0)
	}
}

func (v *resourceView) setColorerFn(f colorerFn) {
	v.colorerFn = f
	v.masterPage().setColorer(f)
}

func (v *resourceView) setDecorateFn(f decorateFn) {
	v.decorateFn = f
}

func (v *resourceView) filterResource(sel string) {
	v.list.SetLabelSelector(sel)
	v.refresh()
}

func (v *resourceView) stopUpdates() {
	if v.cancelFn != nil {
		v.cancelFn()
	}
}

func (v *resourceView) restartUpdates() {
	if v.cancelFn != nil {
		v.cancelFn()
	}

	var vctx context.Context
	vctx, v.cancelFn = context.WithCancel(v.parentCtx)
	v.update(vctx)
}

func (v *resourceView) update(ctx context.Context) {
	go func(ctx context.Context) {
		for {
			select {
			case <-ctx.Done():
				log.Debug().Msgf("%s updater canceled!", v.list.GetName())
				return
			case <-time.After(time.Duration(v.app.config.K9s.RefreshRate) * time.Second):
				v.app.QueueUpdateDraw(func() {
					v.refresh()
				})
			}
		}
	}(ctx)
}

// ----------------------------------------------------------------------------
// Actions...

func (v *resourceView) backCmd(*tcell.EventKey) *tcell.EventKey {
	v.switchPage("master")
	return nil
}

func (v *resourceView) enterCmd(evt *tcell.EventKey) *tcell.EventKey {
	// If in command mode run filter otherwise enter function.
	if v.masterPage().filterCmd(evt) == nil || !v.rowSelected() {
		return nil
	}

	if v.enterFn != nil {
		v.enterFn(v.app, v.list.GetNamespace(), v.list.GetName(), v.selectedItem)
	} else {
		v.defaultEnter(v.list.GetNamespace(), v.list.GetName(), v.selectedItem)
	}

	return nil
}

func (v *resourceView) refreshCmd(*tcell.EventKey) *tcell.EventKey {
	v.app.flash().info("Refreshing...")
	v.refresh()
	return nil
}

func (v *resourceView) deleteCmd(evt *tcell.EventKey) *tcell.EventKey {
	if !v.rowSelected() {
		return evt
	}

	sel := v.getSelectedItem()
	msg := fmt.Sprintf("Delete %s %s?", v.list.GetName(), sel)
	showDeleteDialog(v.Pages, msg, func(cascade, force bool) {
		v.masterPage().setDeleted()
		v.app.flash().infof("Deleting %s %s", v.list.GetName(), sel)
		if err := v.list.Resource().Delete(sel, cascade, force); err != nil {
			v.app.flash().errf("Delete failed with %s", err)
		} else {
			v.refresh()
		}
	}, func() {
		v.switchPage("master")
	})
	return nil
}

func (v *resourceView) defaultEnter(ns, resource, selection string) {
	yaml, err := v.list.Resource().Describe(v.masterPage().baseTitle, selection)
	if err != nil {
		v.app.flash().errf("Describe command failed %s", err)
		return
	}

	details := v.detailsPage()
	details.setCategory("Describe")
	details.setTitle(selection)
	details.SetTextColor(v.app.styles.FgColor())
	details.SetText(colorizeYAML(v.app.styles.Views().Yaml, yaml))
	details.ScrollToBeginning()

	v.switchPage("details")
}

func (v *resourceView) describeCmd(evt *tcell.EventKey) *tcell.EventKey {
	if !v.rowSelected() {
		return evt
	}

	v.defaultEnter(v.list.GetNamespace(), v.list.GetName(), v.selectedItem)
	return nil
}

func (v *resourceView) viewCmd(evt *tcell.EventKey) *tcell.EventKey {
	if !v.rowSelected() {
		return evt
	}

	sel := v.getSelectedItem()
	raw, err := v.list.Resource().Marshal(sel)
	if err != nil {
		v.app.flash().errf("Unable to marshal resource %s", err)
		return evt
	}
	details := v.detailsPage()
	details.setCategory("YAML")
	details.setTitle(sel)
	details.SetTextColor(v.app.styles.FgColor())
	details.SetText(colorizeYAML(v.app.styles.Views().Yaml, raw))
	details.ScrollToBeginning()
	v.switchPage("details")

	return nil
}

func (v *resourceView) editCmd(evt *tcell.EventKey) *tcell.EventKey {
	if !v.rowSelected() {
		return evt
	}

	kcfg := v.app.conn().Config().Flags().KubeConfig
	v.stopUpdates()
	{
		ns, po := namespaced(v.selectedItem)
		args := make([]string, 0, 10)
		args = append(args, "edit")
		args = append(args, v.list.GetName())
		args = append(args, "-n", ns)
		args = append(args, "--context", v.app.config.K9s.CurrentContext)
		if kcfg != nil && *kcfg != "" {
			args = append(args, "--kubeconfig", *kcfg)
		}
		runK(true, v.app, append(args, po)...)
	}
	v.restartUpdates()

	return evt
}

func (v *resourceView) setNamespace(ns string) {
	if v.list.Namespaced() {
		v.currentNS = ns
		v.list.SetNamespace(ns)
	}
}

func (v *resourceView) switchNamespaceCmd(evt *tcell.EventKey) *tcell.EventKey {
	i, _ := strconv.Atoi(string(evt.Rune()))
	ns := v.namespaces[i]
	if ns == "" {
		ns = resource.AllNamespace
	}
	if v.currentNS == ns {
		return nil
	}

	v.setNamespace(ns)
	v.app.flash().infof("Viewing `%s namespace...", ns)
	v.refresh()
	v.masterPage().resetTitle()
	v.masterPage().Select(1, 0)
	v.selectItem(1, 0)
	v.app.cmdBuff.reset()
	v.app.config.SetActiveNamespace(v.currentNS)
	v.app.config.Save()

	return nil
}

func (v *resourceView) refresh() {
	if _, ok := v.CurrentPage().Item.(*tableView); !ok {
		return
	}

	v.refreshActions()
	if v.list.Namespaced() {
		v.list.SetNamespace(v.currentNS)
	}
	if err := v.list.Reconcile(v.app.informer, v.path); err != nil {
		v.app.flash().errf("Reconciliation for %s failed - %s", v.list.GetName(), err)
	}
	data := v.list.Data()
	if v.decorateFn != nil {
		data = v.decorateFn(data)
	}
	v.masterPage().update(data)
	v.selectItem(v.selectedRow, 0)
}

func (v *resourceView) switchPage(p string) {
	log.Debug().Msgf("Switching page to %s", p)
	if _, ok := v.CurrentPage().Item.(*tableView); ok {
		v.stopUpdates()
	}

	v.SwitchToPage(p)
	v.currentNS = v.list.GetNamespace()
	if vu, ok := v.GetPrimitive(p).(hinter); ok {
		v.app.setHints(vu.hints())
	}

	if _, ok := v.CurrentPage().Item.(*tableView); ok {
		v.restartUpdates()
	}
}

func (v *resourceView) namespaceActions() {
	if !v.list.Access(resource.NamespaceAccess) {
		return
	}
	v.namespaces = make(map[int]string, config.MaxFavoritesNS)
	// User can't list namespace. Don't offer a choice.
	if v.app.conn().CheckListNSAccess() != nil {
		return
	}
	v.actions[tcell.Key(numKeys[0])] = newKeyAction(resource.AllNamespace, v.switchNamespaceCmd, true)
	v.namespaces[0] = resource.AllNamespace
	index := 1
	for _, n := range v.app.config.FavNamespaces() {
		if n == resource.AllNamespace {
			continue
		}
		v.actions[tcell.Key(numKeys[index])] = newKeyAction(n, v.switchNamespaceCmd, true)
		v.namespaces[index] = n
		index++
	}
}

func (v *resourceView) refreshActions() {
	v.namespaceActions()
	v.defaultActions()
	v.actions[tcell.KeyEnter] = newKeyAction("Enter", v.enterCmd, false)
	v.actions[tcell.KeyCtrlR] = newKeyAction("Refresh", v.refreshCmd, false)

	if v.list.Access(resource.EditAccess) {
		v.actions[KeyE] = newKeyAction("Edit", v.editCmd, true)
	}
	if v.list.Access(resource.DeleteAccess) {
		v.actions[tcell.KeyCtrlD] = newKeyAction("Delete", v.deleteCmd, true)
	}
	if v.list.Access(resource.ViewAccess) {
		v.actions[KeyY] = newKeyAction("YAML", v.viewCmd, true)
	}
	if v.list.Access(resource.DescribeAccess) {
		v.actions[KeyD] = newKeyAction("Describe", v.describeCmd, true)
	}

	t := v.masterPage()
	t.setActions(v.actions)
	v.app.setHints(t.hints())
}