84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
package dao
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"math"
|
|
"regexp"
|
|
|
|
"github.com/derailed/tview"
|
|
runewidth "github.com/mattn/go-runewidth"
|
|
"github.com/rs/zerolog/log"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/cli-runtime/pkg/printers"
|
|
)
|
|
|
|
var (
|
|
inverseRx = regexp.MustCompile(`\A\!`)
|
|
fuzzyRx = regexp.MustCompile(`\A\-f`)
|
|
)
|
|
|
|
func inList(ll []string, s string) bool {
|
|
for _, l := range ll {
|
|
if l == s {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// IsInverseSelector checks if inverse char has been provided.
|
|
func IsInverseSelector(s string) bool {
|
|
if s == "" {
|
|
return false
|
|
}
|
|
return inverseRx.MatchString(s)
|
|
}
|
|
|
|
// IsFuzzySelector checks if filter is fuzzy or not.
|
|
func IsFuzzySelector(s string) bool {
|
|
if s == "" {
|
|
return false
|
|
}
|
|
return fuzzyRx.MatchString(s)
|
|
}
|
|
|
|
func toPerc(v1, v2 float64) float64 {
|
|
if v2 == 0 {
|
|
return 0
|
|
}
|
|
return math.Round((v1 / v2) * 100)
|
|
}
|
|
|
|
// Truncate a string to the given l and suffix ellipsis if needed.
|
|
func Truncate(str string, width int) string {
|
|
return runewidth.Truncate(str, width, string(tview.SemigraphicsHorizontalEllipsis))
|
|
}
|
|
|
|
// ToYAML converts a resource to its YAML representation.
|
|
func ToYAML(o runtime.Object, showManaged bool) (string, error) {
|
|
if o == nil {
|
|
return "", errors.New("no object to yamlize")
|
|
}
|
|
|
|
var (
|
|
buff bytes.Buffer
|
|
p printers.YAMLPrinter
|
|
)
|
|
if !showManaged {
|
|
o = o.DeepCopyObject()
|
|
uo := o.(*unstructured.Unstructured).Object
|
|
if meta, ok := uo["metadata"].(map[string]interface{}); ok {
|
|
delete(meta, "managedFields")
|
|
}
|
|
}
|
|
err := p.PrintObj(o, &buff)
|
|
if err != nil {
|
|
log.Error().Msgf("Marshal Error %v", err)
|
|
return "", err
|
|
}
|
|
|
|
return buff.String(), nil
|
|
}
|