package helm

import (
	"os"
	"path"
)

var (
	// HelmV2 represents helm V2 specific settings
	HelmV2 = HelmVer{
		binaryName:           "helm2",
		templateNameArg:      "--name",
		kubeVersionSupported: true,
		showCommand:          "inspect",
		pullCommand:          "fetch",
		initSupported:        true,
	}
	// HelmV3 represents helm V3 specific settings
	HelmV3 = HelmVer{
		binaryName:                   "helm",
		templateNameArg:              "--name-template",
		kubeVersionSupported:         true,
		showCommand:                  "show",
		pullCommand:                  "pull",
		initSupported:                false,
		getPostTemplateCallback:      cleanupChartLockFile,
		additionalTemplateArgs:       []string{"--include-crds"},
		insecureSkipVerifySupported:  true,
		helmPassCredentialsSupported: true,
	}
)

// workaround for Helm3 bug. Remove after https://github.com/helm/helm/issues/6870 is fixed.
// The `helm template` command generates Chart.lock after which `helm dependency build` does not work
// As workaround removing lock file unless it exists before running helm template
func cleanupChartLockFile(chartPath string) (func(), error) {
	exists := true
	lockPath := path.Join(chartPath, "Chart.lock")
	if _, err := os.Stat(lockPath); err != nil {
		if os.IsNotExist(err) {
			exists = false
		} else {
			return nil, err
		}
	}
	return func() {
		if !exists {
			_ = os.Remove(lockPath)
		}
	}, nil
}

// HelmVer contains Helm version specific settings such as helm binary and command names
type HelmVer struct {
	binaryName                   string
	initSupported                bool
	templateNameArg              string
	showCommand                  string
	pullCommand                  string
	kubeVersionSupported         bool
	getPostTemplateCallback      func(chartPath string) (func(), error)
	additionalTemplateArgs       []string
	insecureSkipVerifySupported  bool
	helmPassCredentialsSupported bool
}
