136 lines
2.4 KiB
Go
136 lines
2.4 KiB
Go
package views
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/gdamore/tcell"
|
|
"github.com/k8sland/k9s/resource"
|
|
"github.com/k8sland/tview"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type logsView struct {
|
|
*tview.Pages
|
|
|
|
pv *podView
|
|
containers []string
|
|
actions keyActions
|
|
cancelFunc context.CancelFunc
|
|
}
|
|
|
|
func newLogsView(pv *podView) *logsView {
|
|
v := logsView{Pages: tview.NewPages(), pv: pv, containers: []string{}}
|
|
v.SetInputCapture(v.keyboard)
|
|
|
|
return &v
|
|
}
|
|
|
|
func (v *logsView) keyboard(evt *tcell.EventKey) *tcell.EventKey {
|
|
if m, ok := v.actions[evt.Key()]; ok {
|
|
m.action(evt)
|
|
return nil
|
|
}
|
|
|
|
if evt.Key() == tcell.KeyRune {
|
|
i, err := strconv.Atoi(string(evt.Rune()))
|
|
if err != nil {
|
|
log.Error("Boom!", err)
|
|
return evt
|
|
}
|
|
if _, ok := numKeys[i]; ok {
|
|
v.load(i - 1)
|
|
v.pv.app.resetCmd()
|
|
return nil
|
|
}
|
|
}
|
|
return evt
|
|
}
|
|
|
|
// SetActions to handle keyboard events.
|
|
func (v *logsView) setActions(aa keyActions) {
|
|
v.actions = aa
|
|
}
|
|
|
|
// Hints show action hints
|
|
func (v *logsView) hints() hints {
|
|
if len(v.containers) > 1 {
|
|
for i, c := range v.containers {
|
|
v.actions[tcell.Key(numKeys[i+1])] = keyAction{description: c}
|
|
}
|
|
}
|
|
return v.actions.toHints()
|
|
}
|
|
|
|
func (v *logsView) addContainer(n string) {
|
|
v.containers = append(v.containers, n)
|
|
l := newLogView(n, v.pv)
|
|
l.SetInputCapture(v.keyboard)
|
|
v.AddPage(n, l, true, false)
|
|
}
|
|
|
|
func (v *logsView) init() {
|
|
v.load(0)
|
|
}
|
|
|
|
func (v *logsView) clearLogs() {
|
|
p := v.CurrentPage()
|
|
if p != nil {
|
|
p.Item.(*logView).Clear()
|
|
}
|
|
}
|
|
|
|
func (v *logsView) deleteAllPages() {
|
|
for i, c := range v.containers {
|
|
v.RemovePage(c)
|
|
delete(v.actions, tcell.Key(numKeys[i+1]))
|
|
}
|
|
v.containers = []string{}
|
|
}
|
|
|
|
func (v *logsView) stop() {
|
|
v.killLogIfAny()
|
|
}
|
|
|
|
func (v *logsView) load(i int) {
|
|
if i < 0 || i > len(v.containers)-1 {
|
|
return
|
|
}
|
|
v.SwitchToPage(v.containers[i])
|
|
if err := v.doLoad(v.pv.selectedItem, v.containers[i]); err != nil {
|
|
v.pv.app.flash(flashErr, err.Error())
|
|
return
|
|
}
|
|
v.pv.app.SetFocus(v)
|
|
}
|
|
|
|
func (v *logsView) killLogIfAny() {
|
|
if v.cancelFunc != nil {
|
|
v.cancelFunc()
|
|
v.cancelFunc = nil
|
|
}
|
|
}
|
|
|
|
func (v *logsView) doLoad(path, co string) error {
|
|
v.killLogIfAny()
|
|
|
|
c := make(chan string)
|
|
go func() {
|
|
l := v.CurrentPage().Item.(*logView)
|
|
for s := range c {
|
|
fmt.Fprintln(l, s)
|
|
}
|
|
}()
|
|
|
|
ns, po := namespaced(path)
|
|
cancelFn, err := v.pv.list.Resource().(resource.Tailable).Logs(c, ns, po, co)
|
|
if err != nil {
|
|
cancelFn()
|
|
return err
|
|
}
|
|
v.cancelFunc = cancelFn
|
|
|
|
return nil
|
|
}
|