airshipctl/pkg/config/options.go

/*
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package config

import (
	"fmt"
	"io"
	"sort"
	"strings"

	"k8s.io/cli-runtime/pkg/printers"
	"sigs.k8s.io/yaml"
)

// ContextOptions holds all configurable options for context
type ContextOptions struct {
	Name                    string
	CurrentContext          bool
	Manifest                string
	Current                 bool
	ManagementConfiguration string
	Format                  string
}

// ManifestOptions holds all configurable options for manifest configuration
type ManifestOptions struct {
	Name         string
	RepoName     string
	URL          string
	Branch       string
	CommitHash   string
	Tag          string
	Force        bool
	IsPhase      bool
	TargetPath   string
	MetadataPath string
}

// TODO(howell): The following functions are tightly coupled with flags passed
// on the command line. We should find a way to remove this coupling, since it
// is possible to create (and validate) these objects without using the command
// line.

// Validate checks for the possible context option values and returns
// Error when invalid value or incompatible choice of values given
func (o *ContextOptions) Validate() error {
	if !o.Current && o.Name == "" {
		return ErrEmptyContextName{}
	}

	if o.Current && o.Name != "" {
		return ErrConflictingContextOptions{}
	}

	// If the user simply wants to change the current context, no further validation is needed
	if o.CurrentContext {
		return nil
	}

	// TODO Manifest, Cluster could be validated against the existing config maps
	return nil
}

// Print prints the config contexts using one of formats `yaml` or `table` to a given output
func (o *ContextOptions) Print(cfg *Config, w io.Writer) error {
	if o.CurrentContext {
		o.Name = cfg.CurrentContext
	}

	switch o.Format {
	case "yaml":
		type reducedConfig struct {
			Contexts       map[string]*Context `json:"contexts"`
			CurrentContext string              `json:"currentContext,omitempty"`
		}
		contexts := &reducedConfig{
			Contexts:       cfg.Contexts,
			CurrentContext: cfg.CurrentContext,
		}
		if o.Name != "" {
			c, err := cfg.GetContext(o.Name)
			if err != nil {
				return err
			}
			contexts = &reducedConfig{
				Contexts: map[string]*Context{o.Name: c},
			}
		}
		data, err := yaml.Marshal(contexts)
		if err != nil {
			return err
		}
		fmt.Fprintf(w, string(data))
	case "table":
		out := printers.GetNewTabWriter(w)
		defer out.Flush()

		toPrint := []string{}
		if o.Name != "" {
			toPrint = append(toPrint, o.Name)
		} else {
			for name := range cfg.Contexts {
				toPrint = append(toPrint, name)
			}
		}

		columnNames := []string{"CURRENT", "NAME", "MANIFEST", "MANAGEMENTCONFIGURATION"}
		_, err := fmt.Fprintf(out, "%s\n", strings.Join(columnNames, "\t"))
		if err != nil {
			return err
		}

		sort.Strings(toPrint)
		for _, name := range toPrint {
			prefix := " "
			if cfg.CurrentContext == name {
				prefix = "*"
			}
			context, err := cfg.GetContext(name)
			if err != nil {
				return err
			}
			_, err = fmt.Fprintf(out, "%s\t%s\t%s\t%s\n", prefix, name, context.Manifest, context.ManagementConfiguration)
			if err != nil {
				return err
			}
		}
	default:
		return ErrWrongOutputFormat{Wrong: o.Format, Possible: []string{"yaml", "table"}}
	}
	return nil
}

// Validate checks for the possible manifest option values and returns
// Error when invalid value or incompatible choice of values given
func (o *ManifestOptions) Validate() error {
	if o.Name == "" {
		return ErrMissingManifestName{}
	}
	if o.IsPhase && o.RepoName == "" {
		return ErrMissingRepositoryName{}
	}
	possibleValues := [3]string{o.CommitHash, o.Branch, o.Tag}
	var count int
	for _, val := range possibleValues {
		if val != "" {
			count++
		}
	}
	if count > 1 {
		return ErrMutuallyExclusiveCheckout{}
	}
	return nil
}