keel/bot/formatter/deployments.go

package formatter

import (
	"fmt"
	"strings"
	"time"
)

// Deployment - internal deployment, used to better represent keel related info
type Deployment struct {
	Namespace         string `json:"namespace,omitempty"`
	Name              string `json:"name,omitempty"`
	CreatedAt         time.Time
	Replicas          int32
	AvailableReplicas int32
	Images            []string `json:"images,omitempty"` // image:tag list
}

// Formatter headers
const (
	defaultDeploymentQuietFormat = "{{.Name}}"
	defaultDeploymentTableFormat = "table {{.Namespace}}\t{{.Name}}\t{{.Ready}}\t{{.Images}}"

	DeploymentNamespaceHeader = "NAMESPACE"
	DeploymentNameHeader      = "NAME"
	DeploymentReadyHeader     = "READY"
	DeploymentImagesHeader    = "IMAGES"
)

// NewDeploymentsFormat returns a format for use with a deployment Context
func NewDeploymentsFormat(source string, quiet bool) Format {
	switch source {
	case TableFormatKey:
		if quiet {
			return defaultDeploymentQuietFormat
		}
		return defaultDeploymentTableFormat
	case RawFormatKey:
		if quiet {
			return `name: {{.Name}}`
		}
		return `name: {{.Name}}\n`
	}
	return Format(source)
}

// DeploymentWrite writes formatted deployments using the Context
func DeploymentWrite(ctx Context, Deployments []Deployment) error {
	render := func(format func(subContext subContext) error) error {
		for _, deployment := range Deployments {
			if err := format(&DeploymentContext{v: deployment}); err != nil {
				return err
			}
		}
		return nil
	}
	return ctx.Write(&DeploymentContext{}, render)
}

// DeploymentContext - deployment context is a container for each line
type DeploymentContext struct {
	HeaderContext
	v Deployment
}

// MarshalJSON - marshal to json (inspect)
func (c *DeploymentContext) MarshalJSON() ([]byte, error) {
	return marshalJSON(c)
}

// Namespace - print namespace
func (c *DeploymentContext) Namespace() string {
	c.AddHeader(DeploymentNamespaceHeader)
	return c.v.Namespace
}

// Name - print name
func (c *DeploymentContext) Name() string {
	c.AddHeader(DeploymentNameHeader)
	return c.v.Name
}

// Ready - print readiness
func (c *DeploymentContext) Ready() string {
	c.AddHeader(DeploymentReadyHeader)
	return fmt.Sprintf("%d/%d", c.v.AvailableReplicas, c.v.Replicas)
}

// Images - print used images
func (c *DeploymentContext) Images() string {
	c.AddHeader(DeploymentImagesHeader)
	return strings.Join(c.v.Images, ", ")
}