Replicate most of drone-helm's config

internal/helm/config.go

@@ -0,0 +1,58 @@
+package helm
+
+import (
+	"fmt"
+	"strings"
+)
+
+type Config struct {
+	// Configuration for drone-helm itself
+	Command            HelmCommand `envconfig:"HELM_COMMAND"`      // Helm command to run
+	DroneEvent         string      `envconfig:"DRONE_BUILD_EVENT"` // Drone event that invoked this plugin.
+	UpdateDependencies bool        `split_words:"true"`            // call `helm dependency update` before the main command
+	Repos              []string    `envconfig:"HELM_REPOS"`        // call `helm repo add` before the main command
+	Prefix             string      ``                              // Prefix to use when looking up secret env vars
+
+	// Global helm config
+	Debug          bool     ``                                                // global helm flag (also applies to drone-helm itself)
+	KubeConfig     string   `split_words:"true" default:"/root/.kube/config"` // path to the kube config file
+	Values         string   ``
+	StringValues   string   `split_words:"true"`
+	ValuesFiles    []string `split_words:"true"`
+	Namespace      string   ``
+	Token          string   `envconfig:"KUBERNETES_TOKEN"`
+	SkipTLSVerify  bool     `envconfig:"SKIP_TLS_VERIFY"`
+	Certificate    string   `envconfig:"KUBERNETES_CERTIFICATE"`
+	APIServer      string   `envconfig:"API_SERVER"`
+	ServiceAccount string   `envconfig:"SERVICE_ACCOUNT"` // Can't just use split_words; need envconfig to find the non-prefixed form
+
+	// Config specifically for `helm upgrade`
+	ChartVersion string `split_words:"true"` //
+	DryRun       bool   `split_words:"true"` // also available for `delete`
+	Wait         bool   ``                   //
+	ReuseValues  bool   `split_words:"true"` //
+	Timeout      string ``                   //
+	Chart        string ``                   // Also available for `lint`, in which case it must be a path to a chart directory
+	Release      string ``
+	Force        bool   `` //
+}
+
+type HelmCommand string
+
+// HelmCommand.Decode checks the given value against the list of known commands and generates a helpful error if the command is unknown.
+func (cmd *HelmCommand) Decode(value string) error {
+	known := []string{"upgrade", "delete", "lint", "help"}
+	for _, c := range known {
+		if value == c {
+			*cmd = HelmCommand(value)
+			return nil
+		}
+	}
+
+	if value == "" {
+		return nil
+	}
+	known[len(known)-1] = fmt.Sprintf("or %s", known[len(known)-1])
+	return fmt.Errorf("Unknown command '%s'. If specified, command must be %s.",
+		value, strings.Join(known, ", "))
+}

internal/helm/config_test.go

@@ -0,0 +1,28 @@
+package helm
+
+import (
+	"github.com/stretchr/testify/suite"
+	"testing"
+)
+
+type ConfigTestSuite struct {
+	suite.Suite
+}
+
+func TestConfigTestSuite(t *testing.T) {
+	suite.Run(t, new(ConfigTestSuite))
+}
+
+func (suite *ConfigTestSuite) TestHelmCommandDecodeSuccess() {
+	cmd := HelmCommand("")
+	err := cmd.Decode("upgrade")
+	suite.Require().Nil(err)
+
+	suite.EqualValues(cmd, "upgrade")
+}
+
+func (suite *ConfigTestSuite) TestHelmCommandDecodeFailure() {
+	cmd := HelmCommand("")
+	err := cmd.Decode("execute order 66")
+	suite.EqualError(err, "Unknown command 'execute order 66'. If specified, command must be upgrade, delete, lint, or help.")
+}

internal/helm/plan.go

@@ -0,0 +1,62 @@
+package helm
+
+import (
+	"errors"
+	"github.com/pelotech/drone-helm3/internal/run"
+)
+
+type Step interface {
+	Run() error
+}
+
+type Plan struct {
+	steps []Step
+}
+
+func NewPlan(cfg Config) (*Plan, error) {
+	p := Plan{}
+	switch cfg.Command {
+	case "upgrade":
+		steps, err := upgrade(cfg)
+		if err != nil {
+			return nil, err
+		}
+		p.steps = steps
+	case "delete":
+		return nil, errors.New("not implemented")
+	case "lint":
+		return nil, errors.New("not implemented")
+	case "help":
+		return nil, errors.New("not implemented")
+	default:
+		switch cfg.DroneEvent {
+		case "push", "tag", "deployment", "pull_request", "promote", "rollback":
+			steps, err := upgrade(cfg)
+			if err != nil {
+				return nil, err
+			}
+			p.steps = steps
+		default:
+			return nil, errors.New("not implemented")
+		}
+	}
+
+	return &p, nil
+}
+
+func (p *Plan) Execute() error {
+	for _, step := range p.steps {
+		if err := step.Run(); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func upgrade(cfg Config) ([]Step, error) {
+	steps := make([]Step, 0)
+	steps = append(steps, run.NewUpgrade(cfg.Release, cfg.Chart))
+
+	return steps, nil
+}

internal/helm/plan_test.go

@@ -0,0 +1,53 @@
+package helm
+
+import (
+	"fmt"
+	"github.com/stretchr/testify/suite"
+	"testing"
+
+	"github.com/pelotech/drone-helm3/internal/run"
+)
+
+type PlanTestSuite struct {
+	suite.Suite
+}
+
+func TestPlanTestSuite(t *testing.T) {
+	suite.Run(t, new(PlanTestSuite))
+}
+
+func (suite *PlanTestSuite) TestNewPlanUpgradeCommand() {
+	cfg := Config{
+		Command: "upgrade",
+		Chart:   "billboard_top_100",
+		Release: "post_malone_circles",
+	}
+
+	plan, err := NewPlan(cfg)
+	suite.Require().Nil(err)
+	suite.Equal(1, len(plan.steps))
+
+	switch step := plan.steps[0].(type) {
+	case *run.Upgrade:
+		suite.Equal("billboard_top_100", step.Chart)
+		suite.Equal("post_malone_circles", step.Release)
+	default:
+		suite.Failf("Wrong type for step 1", "Expected Upgrade, got %T", step)
+	}
+}
+
+func (suite *PlanTestSuite) TestNewPlanUpgradeFromDroneEvent() {
+	cfg := Config{
+		Chart:   "billboard_top_100",
+		Release: "lizzo_good_as_hell",
+	}
+
+	upgradeEvents := []string{"push", "tag", "deployment", "pull_request", "promote", "rollback"}
+	for _, event := range upgradeEvents {
+		cfg.DroneEvent = event
+		plan, err := NewPlan(cfg)
+		suite.Require().Nil(err)
+		suite.Require().Equal(1, len(plan.steps), fmt.Sprintf("for event type '%s'", event))
+		suite.IsType(&run.Upgrade{}, plan.steps[0], fmt.Sprintf("for event type '%s'", event))
+	}
+}