k9s/internal/render/context.go

103 lines
2.2 KiB
Go

package render
import (
"fmt"
"strings"
"github.com/gdamore/tcell"
"github.com/rs/zerolog/log"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/runtime/schema"
"k8s.io/client-go/tools/clientcmd/api"
)
// Context renders a K8s ConfigMap to screen.
type Context struct{}
// ColorerFunc colors a resource row.
func (Context) ColorerFunc() ColorerFunc {
return func(ns string, r RowEvent) tcell.Color {
c := DefaultColorer(ns, r)
if r.Kind == EventAdd || r.Kind == EventUpdate {
return c
}
if strings.Contains(strings.TrimSpace(r.Row.Fields[0]), "*") {
c = HighlightColor
}
return c
}
}
// Header returns a header row.
func (Context) Header(ns string) HeaderRow {
return HeaderRow{
Header{Name: "NAME"},
Header{Name: "CLUSTER"},
Header{Name: "AUTHINFO"},
Header{Name: "NAMESPACE"},
}
}
// Render renders a K8s resource to screen.
func (c Context) Render(o interface{}, _ string, r *Row) error {
ctx, ok := o.(*NamedContext)
if !ok {
return fmt.Errorf("expected *NamedContext, but got %T", o)
}
name := ctx.Name
if ctx.IsCurrentContext(ctx.Name) {
name += "(*)"
}
r.ID = ctx.Name
r.Fields = Fields{
name,
ctx.Context.Cluster,
ctx.Context.AuthInfo,
ctx.Context.Namespace,
}
return nil
}
// Helpers...
// NamedContext represents a named cluster context.
type NamedContext struct {
Name string
Context *api.Context
Config ContextNamer
}
// ContextNamer represents a named context.
type ContextNamer interface {
CurrentContextName() (string, error)
}
// NewNamedContext returns a new named context.
func NewNamedContext(c ContextNamer, n string, ctx *api.Context) *NamedContext {
return &NamedContext{Name: n, Context: ctx, Config: c}
}
// IsCurrentContext return the active context name.
func (c *NamedContext) IsCurrentContext(n string) bool {
cl, err := c.Config.CurrentContextName()
if err != nil {
log.Fatal().Err(err).Msg("Fetching current context")
return false
}
return cl == n
}
// GetObjectKind returns a schema object.
func (c *NamedContext) GetObjectKind() schema.ObjectKind {
return nil
}
// DeepCopyObject returns a container copy.
func (c *NamedContext) DeepCopyObject() runtime.Object {
return c
}