392 lines
9.6 KiB
Go
392 lines
9.6 KiB
Go
package view
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/derailed/k9s/internal/client"
|
|
"github.com/derailed/k9s/internal/color"
|
|
"github.com/derailed/k9s/internal/config"
|
|
"github.com/derailed/k9s/internal/dao"
|
|
"github.com/derailed/k9s/internal/model"
|
|
"github.com/derailed/k9s/internal/ui"
|
|
"github.com/derailed/tview"
|
|
"github.com/gdamore/tcell"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
const (
|
|
logTitle = "logs"
|
|
logMessage = "Waiting for logs..."
|
|
logFmt = " Logs([hilite:bg:]%s[-:bg:-])[[green:bg:b]%s[-:bg:-]] "
|
|
logCoFmt = " Logs([hilite:bg:]%s:[hilite:bg:b]%s[-:bg:-])[[green:bg:b]%s[-:bg:-]] "
|
|
flushTimeout = 50 * time.Millisecond
|
|
)
|
|
|
|
// Log represents a generic log viewer.
|
|
type Log struct {
|
|
*tview.Flex
|
|
|
|
app *App
|
|
logs *Details
|
|
indicator *LogIndicator
|
|
ansiWriter io.Writer
|
|
model *model.Log
|
|
}
|
|
|
|
var _ model.Component = (*Log)(nil)
|
|
|
|
// NewLog returns a new viewer.
|
|
func NewLog(gvr client.GVR, path, co string, prev bool) *Log {
|
|
l := Log{
|
|
Flex: tview.NewFlex(),
|
|
model: model.NewLog(
|
|
gvr,
|
|
buildLogOpts(path, co, prev, true, config.DefaultLoggerTailCount),
|
|
flushTimeout,
|
|
),
|
|
}
|
|
|
|
return &l
|
|
}
|
|
|
|
// Init initializes the viewer.
|
|
func (l *Log) Init(ctx context.Context) (err error) {
|
|
if l.app, err = extractApp(ctx); err != nil {
|
|
return err
|
|
}
|
|
l.model.Configure(l.app.Config.K9s.Logger)
|
|
|
|
l.SetBorder(true)
|
|
l.SetDirection(tview.FlexRow)
|
|
|
|
l.indicator = NewLogIndicator(l.app.Config, l.app.Styles)
|
|
l.AddItem(l.indicator, 1, 1, false)
|
|
l.indicator.Refresh()
|
|
|
|
l.logs = NewDetails(l.app, "", "", false)
|
|
if err = l.logs.Init(ctx); err != nil {
|
|
return err
|
|
}
|
|
l.logs.SetBorderPadding(0, 0, 1, 1)
|
|
l.logs.SetText(logMessage)
|
|
l.logs.SetWrap(l.app.Config.K9s.Logger.TextWrap)
|
|
l.logs.SetMaxBuffer(l.app.Config.K9s.Logger.BufferSize)
|
|
l.logs.cmdBuff.AddListener(l)
|
|
|
|
l.ansiWriter = tview.ANSIWriter(l.logs, l.app.Styles.Views().Log.FgColor.String(), l.app.Styles.Views().Log.BgColor.String())
|
|
l.AddItem(l.logs, 0, 1, true)
|
|
l.bindKeys()
|
|
|
|
l.StylesChanged(l.app.Styles)
|
|
l.app.Styles.AddListener(l)
|
|
l.goFullScreen()
|
|
|
|
l.model.Init(l.app.factory)
|
|
l.model.AddListener(l)
|
|
l.updateTitle()
|
|
|
|
return nil
|
|
}
|
|
|
|
// LogCleared clears the logs.
|
|
func (l *Log) LogCleared() {
|
|
l.app.QueueUpdateDraw(func() {
|
|
l.logs.Clear()
|
|
l.logs.ScrollTo(0, 0)
|
|
})
|
|
}
|
|
|
|
// LogFailed notifies an error occurred.
|
|
func (l *Log) LogFailed(err error) {
|
|
l.app.QueueUpdateDraw(func() {
|
|
l.app.Flash().Err(err)
|
|
if l.logs.GetText(true) == logMessage {
|
|
l.logs.Clear()
|
|
}
|
|
fmt.Fprintln(l.ansiWriter, tview.Escape(color.Colorize(err.Error(), color.Red)))
|
|
})
|
|
}
|
|
|
|
// LogChanged updates the logs.
|
|
func (l *Log) LogChanged(lines dao.LogItems) {
|
|
l.app.QueueUpdateDraw(func() {
|
|
l.Flush(lines)
|
|
})
|
|
}
|
|
|
|
// BufferChanged indicates the buffer was changed.
|
|
func (l *Log) BufferChanged(s string) {
|
|
l.model.Filter(l.logs.cmdBuff.GetText())
|
|
l.updateTitle()
|
|
}
|
|
|
|
// BufferActive indicates the buff activity changed.
|
|
func (l *Log) BufferActive(state bool, k model.BufferKind) {
|
|
l.app.BufferActive(state, k)
|
|
}
|
|
|
|
// StylesChanged reports skin changes.
|
|
func (l *Log) StylesChanged(s *config.Styles) {
|
|
l.SetBackgroundColor(s.Views().Log.BgColor.Color())
|
|
l.logs.SetTextColor(s.Views().Log.FgColor.Color())
|
|
l.logs.SetBackgroundColor(s.Views().Log.BgColor.Color())
|
|
}
|
|
|
|
// GetModel returns the log model.
|
|
func (l *Log) GetModel() *model.Log {
|
|
return l.model
|
|
}
|
|
|
|
// Hints returns a collection of menu hints.
|
|
func (l *Log) Hints() model.MenuHints {
|
|
return l.logs.Actions().Hints()
|
|
}
|
|
|
|
// ExtraHints returns additional hints.
|
|
func (l *Log) ExtraHints() map[string]string {
|
|
return nil
|
|
}
|
|
|
|
// Start runs the component.
|
|
func (l *Log) Start() {
|
|
l.model.Start()
|
|
}
|
|
|
|
// Stop terminates the component.
|
|
func (l *Log) Stop() {
|
|
l.model.Stop()
|
|
l.model.RemoveListener(l)
|
|
l.app.Styles.RemoveListener(l)
|
|
l.logs.cmdBuff.RemoveListener(l)
|
|
l.logs.cmdBuff.RemoveListener(l.app.Prompt())
|
|
}
|
|
|
|
// Name returns the component name.
|
|
func (l *Log) Name() string { return logTitle }
|
|
|
|
func (l *Log) bindKeys() {
|
|
l.logs.Actions().Set(ui.KeyActions{
|
|
ui.Key0: ui.NewKeyAction("all", l.sinceCmd(-1), true),
|
|
ui.Key1: ui.NewKeyAction("1m", l.sinceCmd(60), true),
|
|
ui.Key2: ui.NewKeyAction("5m", l.sinceCmd(5*60), true),
|
|
ui.Key3: ui.NewKeyAction("15m", l.sinceCmd(15*60), true),
|
|
ui.Key4: ui.NewKeyAction("30m", l.sinceCmd(30*60), true),
|
|
ui.Key5: ui.NewKeyAction("1h", l.sinceCmd(60*60), true),
|
|
tcell.KeyEnter: ui.NewSharedKeyAction("Filter", l.filterCmd, false),
|
|
ui.KeyC: ui.NewKeyAction("Clear", l.clearCmd, true),
|
|
ui.KeyM: ui.NewKeyAction("Mark", l.markCmd, true),
|
|
ui.KeyS: ui.NewKeyAction("Toggle AutoScroll", l.toggleAutoScrollCmd, true),
|
|
ui.KeyF: ui.NewKeyAction("Toggle FullScreen", l.toggleFullScreenCmd, true),
|
|
ui.KeyT: ui.NewKeyAction("Toggle Timestamp", l.toggleTimestampCmd, true),
|
|
ui.KeyW: ui.NewKeyAction("Toggle Wrap", l.toggleTextWrapCmd, true),
|
|
tcell.KeyCtrlS: ui.NewKeyAction("Save", l.SaveCmd, true),
|
|
})
|
|
}
|
|
|
|
// SendStrokes (testing only!)
|
|
func (l *Log) SendStrokes(s string) {
|
|
l.app.Prompt().SendStrokes(s)
|
|
}
|
|
|
|
// SendKeys (testing only!)
|
|
func (l *Log) SendKeys(kk ...tcell.Key) {
|
|
for _, k := range kk {
|
|
l.logs.keyboard(tcell.NewEventKey(k, ' ', tcell.ModNone))
|
|
}
|
|
}
|
|
|
|
// Indicator returns the scroll mode viewer.
|
|
func (l *Log) Indicator() *LogIndicator {
|
|
return l.indicator
|
|
}
|
|
|
|
func (l *Log) updateTitle() {
|
|
sinceSeconds, since := l.model.SinceSeconds(), "all"
|
|
if sinceSeconds > 0 && sinceSeconds < 60*60 {
|
|
since = fmt.Sprintf("%dm", sinceSeconds/60)
|
|
}
|
|
if sinceSeconds >= 60*60 {
|
|
since = fmt.Sprintf("%dh", sinceSeconds/(60*60))
|
|
}
|
|
var title string
|
|
path, co := l.model.GetPath(), l.model.GetContainer()
|
|
if co == "" {
|
|
title = ui.SkinTitle(fmt.Sprintf(logFmt, path, since), l.app.Styles.Frame())
|
|
} else {
|
|
title = ui.SkinTitle(fmt.Sprintf(logCoFmt, path, co, since), l.app.Styles.Frame())
|
|
}
|
|
|
|
buff := l.logs.cmdBuff.GetText()
|
|
if buff != "" {
|
|
title += ui.SkinTitle(fmt.Sprintf(ui.SearchFmt, buff), l.app.Styles.Frame())
|
|
}
|
|
l.SetTitle(title)
|
|
}
|
|
|
|
// Logs returns the log viewer.
|
|
func (l *Log) Logs() *Details {
|
|
return l.logs
|
|
}
|
|
|
|
var EOL = []byte("\n")
|
|
|
|
// Flush write logs to viewer.
|
|
func (l *Log) Flush(lines dao.LogItems) {
|
|
showTime := l.Indicator().showTime
|
|
ll := make([][]byte, len(lines))
|
|
lines.Render(showTime, ll)
|
|
if _, err := l.ansiWriter.Write(bytes.Join(ll, EOL)); err != nil {
|
|
log.Error().Err(err).Msgf("write log failed")
|
|
}
|
|
l.logs.ScrollToEnd()
|
|
l.indicator.Refresh()
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// Actions()...
|
|
|
|
func (l *Log) sinceCmd(a int) func(evt *tcell.EventKey) *tcell.EventKey {
|
|
return func(evt *tcell.EventKey) *tcell.EventKey {
|
|
opts := l.model.LogOptions()
|
|
opts.SinceSeconds = int64(a)
|
|
l.model.SetLogOptions(opts)
|
|
l.updateTitle()
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (l *Log) filterCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
if !l.logs.cmdBuff.IsActive() {
|
|
return evt
|
|
}
|
|
|
|
l.logs.cmdBuff.SetActive(false)
|
|
l.model.Filter(l.logs.cmdBuff.GetText())
|
|
l.updateTitle()
|
|
|
|
return nil
|
|
}
|
|
|
|
// SaveCmd dumps the logs to file.
|
|
func (l *Log) SaveCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
if path, err := saveData(l.app.Config.K9s.CurrentCluster, l.model.GetPath(), l.logs.GetText(true)); err != nil {
|
|
l.app.Flash().Err(err)
|
|
} else {
|
|
l.app.Flash().Infof("Log %s saved successfully!", path)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ensureDir(dir string) error {
|
|
return os.MkdirAll(dir, 0744)
|
|
}
|
|
|
|
func saveData(cluster, name, data string) (string, error) {
|
|
dir := filepath.Join(config.K9sDumpDir, cluster)
|
|
if err := ensureDir(dir); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
now := time.Now().UnixNano()
|
|
fName := fmt.Sprintf("%s-%d.log", strings.Replace(name, "/", "-", -1), now)
|
|
|
|
path := filepath.Join(dir, fName)
|
|
mod := os.O_CREATE | os.O_WRONLY
|
|
file, err := os.OpenFile(path, mod, 0600)
|
|
if err != nil {
|
|
log.Error().Err(err).Msgf("LogFile create %s", path)
|
|
return "", nil
|
|
|
|
}
|
|
defer func() {
|
|
if err := file.Close(); err != nil {
|
|
log.Error().Err(err).Msg("Closing Log file")
|
|
}
|
|
}()
|
|
if _, err := file.Write([]byte(data)); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return path, nil
|
|
}
|
|
|
|
func (l *Log) clearCmd(*tcell.EventKey) *tcell.EventKey {
|
|
l.model.Clear()
|
|
return nil
|
|
}
|
|
|
|
func (l *Log) markCmd(*tcell.EventKey) *tcell.EventKey {
|
|
fmt.Fprintln(l.ansiWriter, fmt.Sprintf("[white::b]%s[::]", strings.Repeat("─", 80)))
|
|
return nil
|
|
}
|
|
|
|
func (l *Log) toggleTimestampCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
if l.app.InCmdMode() {
|
|
return evt
|
|
}
|
|
|
|
l.indicator.ToggleTimestamp()
|
|
l.model.Refresh()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *Log) toggleTextWrapCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
if l.app.InCmdMode() {
|
|
return evt
|
|
}
|
|
|
|
l.indicator.ToggleTextWrap()
|
|
l.logs.SetWrap(l.indicator.textWrap)
|
|
return nil
|
|
}
|
|
|
|
// ToggleAutoScrollCmd toggles autoscroll status.
|
|
func (l *Log) toggleAutoScrollCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
if l.app.InCmdMode() {
|
|
return evt
|
|
}
|
|
|
|
l.indicator.ToggleAutoScroll()
|
|
if l.indicator.AutoScroll() {
|
|
l.model.Start()
|
|
} else {
|
|
l.model.Stop()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (l *Log) toggleFullScreenCmd(evt *tcell.EventKey) *tcell.EventKey {
|
|
if l.app.InCmdMode() {
|
|
return evt
|
|
}
|
|
l.indicator.ToggleFullScreen()
|
|
l.goFullScreen()
|
|
return nil
|
|
}
|
|
|
|
func (l *Log) goFullScreen() {
|
|
l.SetFullScreen(l.indicator.FullScreen())
|
|
l.Box.SetBorder(!l.indicator.FullScreen())
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// Helpers...
|
|
|
|
func buildLogOpts(path, co string, prevLogs, showTime bool, tailLineCount int) dao.LogOptions {
|
|
return dao.LogOptions{
|
|
Path: path,
|
|
Container: co,
|
|
Lines: int64(tailLineCount),
|
|
Previous: prevLogs,
|
|
ShowTimestamp: showTime,
|
|
}
|
|
}
|