304 lines
7.8 KiB
Go
304 lines
7.8 KiB
Go
package view
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/derailed/k9s/internal"
|
|
"github.com/derailed/k9s/internal/client"
|
|
"github.com/derailed/k9s/internal/dao"
|
|
"github.com/derailed/k9s/internal/model"
|
|
"github.com/derailed/k9s/internal/render"
|
|
"github.com/derailed/k9s/internal/ui"
|
|
"github.com/fatih/color"
|
|
"github.com/gdamore/tcell"
|
|
"github.com/rs/zerolog/log"
|
|
v1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
)
|
|
|
|
// Pod represents a pod viewer.
|
|
type Pod struct {
|
|
ResourceViewer
|
|
}
|
|
|
|
// NewPod returns a new viewer.
|
|
func NewPod(gvr client.GVR) ResourceViewer {
|
|
p := Pod{
|
|
ResourceViewer: NewPortForwardExtender(
|
|
NewLogsExtender(NewBrowser(gvr), nil),
|
|
),
|
|
}
|
|
p.SetBindKeysFn(p.bindKeys)
|
|
p.GetTable().SetEnterFn(p.showContainers)
|
|
p.GetTable().SetColorerFn(render.Pod{}.ColorerFunc())
|
|
|
|
return &p
|
|
}
|
|
|
|
func (p *Pod) bindDangerousKeys(aa ui.KeyActions) {
|
|
aa.Add(ui.KeyActions{
|
|
tcell.KeyCtrlK: ui.NewKeyAction("Kill", p.killCmd, true),
|
|
ui.KeyS: ui.NewKeyAction("Shell", p.shellCmd, true),
|
|
ui.KeyA: ui.NewKeyAction("Attach", p.attachCmd, true),
|
|
})
|
|
}
|
|
|
|
func (p *Pod) bindKeys(aa ui.KeyActions) {
|
|
if !p.App().Config.K9s.GetReadOnly() {
|
|
p.bindDangerousKeys(aa)
|
|
}
|
|
|
|
aa.Add(ui.KeyActions{
|
|
ui.KeyShiftR: ui.NewKeyAction("Sort Ready", p.GetTable().SortColCmd(readyCol, true), false),
|
|
ui.KeyShiftT: ui.NewKeyAction("Sort Restart", p.GetTable().SortColCmd("RESTART", false), false),
|
|
ui.KeyShiftS: ui.NewKeyAction("Sort Status", p.GetTable().SortColCmd(statusCol, true), false),
|
|
ui.KeyShiftC: ui.NewKeyAction("Sort CPU", p.GetTable().SortColCmd(cpuCol, false), false),
|
|
ui.KeyShiftM: ui.NewKeyAction("Sort MEM", p.GetTable().SortColCmd(memCol, false), false),
|
|
ui.KeyShiftX: ui.NewKeyAction("Sort %CPU (REQ)", p.GetTable().SortColCmd("%CPU", false), false),
|
|
ui.KeyShiftZ: ui.NewKeyAction("Sort %MEM (REQ)", p.GetTable().SortColCmd("%MEM", false), false),
|
|
tcell.KeyCtrlX: ui.NewKeyAction("Sort %CPU (LIM)", p.GetTable().SortColCmd("%CPU/L", false), false),
|
|
tcell.KeyCtrlQ: ui.NewKeyAction("Sort %MEM (LIM)", p.GetTable().SortColCmd("%MEM/L", false), false),
|
|
ui.KeyShiftI: ui.NewKeyAction("Sort IP", p.GetTable().SortColCmd("IP", true), false),
|
|
ui.KeyShiftO: ui.NewKeyAction("Sort Node", p.GetTable().SortColCmd("NODE", true), false),
|
|
})
|
|
}
|
|
|
|
func (p *Pod) showContainers(app *App, model ui.Tabular, gvr, path string) {
|
|
co := NewContainer(client.NewGVR("containers"))
|
|
co.SetContextFn(p.coContext)
|
|
if err := app.inject(co); err != nil {
|
|
app.Flash().Err(err)
|
|
}
|
|
}
|
|
|
|
func (p *Pod) coContext(ctx context.Context) context.Context {
|
|
return context.WithValue(ctx, internal.KeyPath, p.GetTable().GetSelectedItem())
|
|
}
|
|
|
|
// Commands...
|
|
|
|
func (p *Pod) killCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
sels := p.GetTable().GetSelectedItems()
|
|
if len(sels) == 0 {
|
|
return evt
|
|
}
|
|
|
|
res, err := dao.AccessorFor(p.App().factory, p.GVR())
|
|
if err != nil {
|
|
p.App().Flash().Err(err)
|
|
return nil
|
|
}
|
|
nuker, ok := res.(dao.Nuker)
|
|
if !ok {
|
|
p.App().Flash().Err(fmt.Errorf("expecting a nuker for %q", p.GVR()))
|
|
return nil
|
|
}
|
|
p.GetTable().ShowDeleted()
|
|
for _, res := range sels {
|
|
p.App().Flash().Infof("Delete resource %s -- %s", p.GVR(), res)
|
|
if err := nuker.Delete(res, true, true); err != nil {
|
|
p.App().Flash().Errf("Delete failed with %s", err)
|
|
} else {
|
|
p.App().factory.DeleteForwarder(res)
|
|
}
|
|
}
|
|
p.Refresh()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *Pod) shellCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
path := p.GetTable().GetSelectedItem()
|
|
if path == "" {
|
|
return evt
|
|
}
|
|
|
|
if !podIsRunning(p.App().factory, path) {
|
|
p.App().Flash().Errf("%s is not in a running state", path)
|
|
return nil
|
|
}
|
|
|
|
if err := containerShellin(p.App(), p, path, ""); err != nil {
|
|
p.App().Flash().Err(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (p *Pod) attachCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
path := p.GetTable().GetSelectedItem()
|
|
if path == "" {
|
|
return evt
|
|
}
|
|
|
|
if !podIsRunning(p.App().factory, path) {
|
|
p.App().Flash().Errf("%s is not in a happy state", path)
|
|
return nil
|
|
}
|
|
|
|
if err := containerAttachIn(p.App(), p, path, ""); err != nil {
|
|
p.App().Flash().Err(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// Helpers...
|
|
|
|
func containerShellin(a *App, comp model.Component, path, co string) error {
|
|
if co != "" {
|
|
resumeShellIn(a, comp, path, co)
|
|
return nil
|
|
}
|
|
|
|
cc, err := fetchContainers(a.factory, path, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(cc) == 1 {
|
|
resumeShellIn(a, comp, path, cc[0])
|
|
return nil
|
|
}
|
|
picker := NewPicker()
|
|
picker.populate(cc)
|
|
picker.SetSelectedFunc(func(_ int, co, _ string, _ rune) {
|
|
resumeShellIn(a, comp, path, co)
|
|
})
|
|
if err := a.inject(picker); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func resumeShellIn(a *App, c model.Component, path, co string) {
|
|
c.Stop()
|
|
defer c.Start()
|
|
|
|
shellIn(a, path, co)
|
|
}
|
|
|
|
func shellIn(a *App, path, co string) {
|
|
args := computeShellArgs(path, co, a.Config.K9s.CurrentContext, a.Conn().Config().Flags().KubeConfig)
|
|
|
|
c := color.New(color.BgGreen).Add(color.FgBlack).Add(color.Bold)
|
|
if !runK(a, shellOpts{clear: true, banner: c.Sprintf(bannerFmt, path, co), args: args}) {
|
|
a.Flash().Err(errors.New("Shell exec failed"))
|
|
}
|
|
}
|
|
|
|
func containerAttachIn(a *App, comp model.Component, path, co string) error {
|
|
if co != "" {
|
|
resumeAttachIn(a, comp, path, co)
|
|
return nil
|
|
}
|
|
|
|
cc, err := fetchContainers(a.factory, path, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(cc) == 1 {
|
|
resumeAttachIn(a, comp, path, cc[0])
|
|
return nil
|
|
}
|
|
picker := NewPicker()
|
|
picker.populate(cc)
|
|
picker.SetSelectedFunc(func(_ int, co, _ string, _ rune) {
|
|
resumeAttachIn(a, comp, path, co)
|
|
})
|
|
if err := a.inject(picker); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func resumeAttachIn(a *App, c model.Component, path, co string) {
|
|
c.Stop()
|
|
defer c.Start()
|
|
|
|
attachIn(a, path, co)
|
|
}
|
|
|
|
func attachIn(a *App, path, co string) {
|
|
args := buildShellArgs("attach", path, co, a.Config.K9s.CurrentContext, a.Conn().Config().Flags().KubeConfig)
|
|
c := color.New(color.BgGreen).Add(color.FgBlack).Add(color.Bold)
|
|
if !runK(a, shellOpts{clear: true, banner: c.Sprintf(bannerFmt, path, co), args: args}) {
|
|
a.Flash().Err(errors.New("Attach exec failed"))
|
|
}
|
|
}
|
|
|
|
func computeShellArgs(path, co, context string, kcfg *string) []string {
|
|
args := buildShellArgs("exec", path, co, context, kcfg)
|
|
return append(args, "--", "sh", "-c", shellCheck)
|
|
}
|
|
|
|
func buildShellArgs(cmd, path, co, context string, kcfg *string) []string {
|
|
args := make([]string, 0, 15)
|
|
args = append(args, cmd, "-it")
|
|
args = append(args, "--context", context)
|
|
ns, po := client.Namespaced(path)
|
|
args = append(args, "-n", ns)
|
|
args = append(args, po)
|
|
if kcfg != nil && *kcfg != "" {
|
|
args = append(args, "--kubeconfig", *kcfg)
|
|
}
|
|
if co != "" {
|
|
args = append(args, "-c", co)
|
|
}
|
|
|
|
return args
|
|
}
|
|
|
|
func fetchContainers(f dao.Factory, path string, includeInit bool) ([]string, error) {
|
|
pod, err := fetchPod(f, path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
nn := make([]string, 0, len(pod.Spec.Containers)+len(pod.Spec.InitContainers))
|
|
for _, c := range pod.Spec.Containers {
|
|
nn = append(nn, c.Name)
|
|
}
|
|
if !includeInit {
|
|
return nn, nil
|
|
}
|
|
for _, c := range pod.Spec.InitContainers {
|
|
nn = append(nn, c.Name)
|
|
}
|
|
|
|
return nn, nil
|
|
}
|
|
|
|
func fetchPod(f dao.Factory, path string) (*v1.Pod, error) {
|
|
o, err := f.Get("v1/pods", path, true, labels.Everything())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var pod v1.Pod
|
|
err = runtime.DefaultUnstructuredConverter.FromUnstructured(o.(*unstructured.Unstructured).Object, &pod)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &pod, nil
|
|
}
|
|
|
|
func podIsRunning(f dao.Factory, path string) bool {
|
|
po, err := fetchPod(f, path)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("unable to fetch pod")
|
|
return false
|
|
}
|
|
|
|
var re render.Pod
|
|
log.Debug().Msgf("Phase %#v", re.Phase(po))
|
|
return re.Phase(po) == render.Running
|
|
}
|