porter/cli/cmd/preview/env_group_driver.go

package preview

import (
	"context"
	"fmt"

	"github.com/fatih/color"
	"github.com/mitchellh/mapstructure"
	api "github.com/porter-dev/porter/api/client"
	"github.com/porter-dev/porter/api/types"
	"github.com/porter-dev/porter/cli/cmd/config"
	"github.com/porter-dev/porter/internal/integrations/preview"
	"github.com/porter-dev/switchboard/pkg/drivers"
	"github.com/porter-dev/switchboard/pkg/models"
)

type EnvGroupDriver struct {
	output      map[string]interface{}
	lookupTable *map[string]drivers.Driver
	target      *preview.Target
	config      *preview.EnvGroupDriverConfig
	apiClient   api.Client
	cliConfig   config.CLIConfig
}

// NewEnvGroupDriver extends switchboard with environment groups
func NewEnvGroupDriver(ctx context.Context, apiClient api.Client, cliConfig config.CLIConfig) func(resource *models.Resource, opts *drivers.SharedDriverOpts) (drivers.Driver, error) {
	return func(resource *models.Resource, opts *drivers.SharedDriverOpts) (drivers.Driver, error) {
		driver := &EnvGroupDriver{
			lookupTable: opts.DriverLookupTable,
			output:      make(map[string]interface{}),
			apiClient:   apiClient,
			cliConfig:   cliConfig,
		}

		target, err := GetTarget(ctx, resource.Name, resource.Target, apiClient, cliConfig)
		if err != nil {
			return nil, err
		}

		driver.target = target

		return driver, nil
	}
}

func (d *EnvGroupDriver) ShouldApply(resource *models.Resource) bool {
	return true
}

func (d *EnvGroupDriver) Apply(resource *models.Resource) (*models.Resource, error) {
	ctx := context.TODO() // switchboard blocks changing this for now

	driverConfig, err := d.getConfig(resource)
	if err != nil {
		return nil, err
	}

	d.config = driverConfig

	for _, group := range d.config.EnvGroups {
		if group.Name == "" {
			return nil, fmt.Errorf("env group name cannot be empty")
		}

		if group.Namespace == "" {
			color.New(color.FgYellow).Printf("env group %s has empty namespace so defaulting to target namespace %s\n",
				group.Name, d.target.Namespace)

			group.Namespace = d.target.Namespace
		}

		envGroupResp, err := d.apiClient.GetEnvGroup(
			ctx,
			d.target.Project,
			d.target.Cluster,
			group.Namespace,
			&types.GetEnvGroupRequest{
				Name: group.Name,
			},
		)

		if err != nil && err.Error() == "env group not found" {
			newEnvGroup, err := d.apiClient.CreateEnvGroup(
				ctx, d.target.Project, d.target.Cluster, group.Namespace,
				&types.CreateEnvGroupRequest{
					Name:      group.Name,
					Variables: group.Variables,
				},
			)
			if err != nil {
				return nil, err
			}

			envGroupResp = &types.GetEnvGroupResponse{
				EnvGroup: &types.EnvGroup{
					Name:      newEnvGroup.Name,
					Variables: newEnvGroup.Variables,
				},
			}
		} else if err != nil {
			return nil, err
		}

		d.output[envGroupResp.Name] = map[string]interface{}{
			"variables": envGroupResp.Variables,
		}
	}

	return resource, nil
}

func (d *EnvGroupDriver) Output() (map[string]interface{}, error) {
	return d.output, nil
}

func (d *EnvGroupDriver) getConfig(resource *models.Resource) (*preview.EnvGroupDriverConfig, error) {
	populatedConf, err := drivers.ConstructConfig(&drivers.ConstructConfigOpts{
		RawConf:      resource.Config,
		LookupTable:  *d.lookupTable,
		Dependencies: resource.Dependencies,
	})
	if err != nil {
		return nil, err
	}

	config := &preview.EnvGroupDriverConfig{}

	err = mapstructure.Decode(populatedConf, config)

	if err != nil {
		return nil, err
	}

	return config, nil
}