89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package model
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/derailed/k9s/internal/color"
|
|
)
|
|
|
|
type (
|
|
// Fqn uniquely describes a container
|
|
Fqn struct {
|
|
Namespace, Name, Container string
|
|
}
|
|
|
|
// LogOptions represent logger options.
|
|
LogOptions struct {
|
|
Fqn
|
|
|
|
Lines int64
|
|
Color color.Paint
|
|
Previous bool
|
|
SingleContainer bool
|
|
MultiPods bool
|
|
}
|
|
)
|
|
|
|
// HasContainer checks if a container is present.
|
|
func (o LogOptions) HasContainer() bool {
|
|
return o.Container != ""
|
|
}
|
|
|
|
// FQN returns resource fully qualified name.
|
|
func (o LogOptions) FQN() string {
|
|
return FQN(o.Namespace, o.Name)
|
|
}
|
|
|
|
// Path returns resource descriptor path.
|
|
func (o LogOptions) Path() string {
|
|
return o.FQN() + ":" + o.Container
|
|
}
|
|
|
|
// FixedSizeName returns a normalize fixed size pod name if possible.
|
|
func (o LogOptions) FixedSizeName() string {
|
|
tokens := strings.Split(o.Name, "-")
|
|
if len(tokens) < 3 {
|
|
return o.Name
|
|
}
|
|
var s []string
|
|
for i := 0; i < len(tokens)-1; i++ {
|
|
s = append(s, tokens[i])
|
|
}
|
|
return Truncate(strings.Join(s, "-"), 15) + "-" + tokens[len(tokens)-1]
|
|
}
|
|
|
|
// // NormalizeName colorizes a pod name.
|
|
// func (o LogOptions) NormalizeName() string {
|
|
// if o.Color == 0 {
|
|
// return ""
|
|
// }
|
|
|
|
// return color.Colorize(o.Name+":"+o.Container+" ", o.Color)
|
|
// // return o.Name + ":" + o.Container + " "
|
|
// }
|
|
|
|
func colorize(c color.Paint, txt string) string {
|
|
if c == 0 {
|
|
return ""
|
|
}
|
|
|
|
return color.Colorize(txt, c)
|
|
}
|
|
|
|
// DecorateLog add a log header to display po/co information along with the log message.
|
|
func (o LogOptions) DecorateLog(msg string) string {
|
|
if msg == "" {
|
|
return msg
|
|
}
|
|
|
|
if o.MultiPods {
|
|
return colorize(o.Color, o.Name+":"+o.Container+" ") + msg
|
|
}
|
|
|
|
if !o.SingleContainer {
|
|
return colorize(o.Color, o.Container+" ") + msg
|
|
}
|
|
|
|
return msg
|
|
}
|