context_configs.go

package qliksense

import (
	"errors"
	"fmt"
	"io"

	"github.com/qlik-oss/k-apis/pkg/config"
	"github.com/robfig/cron/v3"

	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"reflect"
	"strings"
	"text/tabwriter"

	b64 "encoding/base64"

	. "github.com/logrusorgru/aurora"
	ansi "github.com/mattn/go-colorable"
	"github.com/qlik-oss/sense-installer/pkg/api"
	_ "gopkg.in/yaml.v2"
	v1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

const (
	// Below are some constants to support qliksense context setup
	QliksenseConfigFile     = "config.yaml"
	QliksenseContextsDir    = "contexts"
	DefaultQliksenseContext = "qlik-default"
	MaxContextNameLength    = 17
	QliksenseSecretsDir     = "secrets"

	imageRegistryConfigKey     = "imageRegistry"
	pullSecretName             = "artifactory-docker-secret"
	qliksenseOperatorImageRepo = "qlik-docker-oss.bintray.io"
	qliksenseOperatorImageName = "qliksense-operator"
)

func (q *Qliksense) SetSecretsFromReader(arg string, reader io.Reader, createSecret, base64Encoded bool) error {
	//take only name from the arguments, value should be from reader
	argName := strings.SplitN(arg, "=", 1)
	if len(argName) != 1 {
		return errors.New("can only have one argument from pipe")
	}
	valueBytes, err := ioutil.ReadAll(reader)
	if err != nil {
		return err
	}
	return q.SetSecrets([]string{argName[0] + "=" + string(valueBytes)}, createSecret, base64Encoded)
}

// SetSecrets - set-secrets <key>=<value> commands
func (q *Qliksense) SetSecrets(args []string, isSecretSet bool, base64Encoded bool) error {
	qConfig := api.NewQConfig(q.QliksenseHome)
	qliksenseCR, err := qConfig.GetCurrentCR()
	if err != nil {
		return err
	}

	// Metadata name in qliksense CR is the name of the current context
	api.LogDebugMessage("Current context: %s\n", qliksenseCR.GetName())
	encryptionKey, err := qConfig.GetEncryptionKeyForCurrent()
	if err != nil {
		return err
	}
	resultArgs, err := api.ProcessConfigArgs(args, base64Encoded)
	if err != nil {
		return err
	}
	for _, ra := range resultArgs {
		api.LogDebugMessage("value args to be encrypted: %s\n", ra.Value)
		if err := q.processSecret(ra, encryptionKey, qliksenseCR, isSecretSet); err != nil {
			return err
		}
	}
	// write modified content into context-yaml
	return qConfig.WriteCR(qliksenseCR)
}

func (q *Qliksense) processSecret(ra *api.ServiceKeyValue, encryptionKey string, qliksenseCR *api.QliksenseCR, isSecretSet bool) error {
	cipherText, e2 := api.EncryptData([]byte(ra.Value), encryptionKey)
	if e2 != nil {
		return e2
	}
	secretName := ""
	if isSecretSet {
		secretFolder := qliksenseCR.GetK8sSecretsFolder(q.QliksenseHome)
		secretFileName := filepath.Join(secretFolder, ra.SvcName+".yaml")

		secretName = fmt.Sprintf("%s-%s-%s", qliksenseCR.GetName(), ra.SvcName, "senseinstaller")
		api.LogDebugMessage("Constructed secret name: %s", secretName)

		k8sSecret := v1.Secret{
			TypeMeta: metav1.TypeMeta{
				APIVersion: "v1",
				Kind:       "Secret",
			},
			ObjectMeta: metav1.ObjectMeta{
				Name: secretName,
			},
			Type: v1.SecretTypeOpaque,
		}

		if !api.DirExists(secretFolder) {
			if err := os.MkdirAll(secretFolder, os.ModePerm); err != nil {
				err = fmt.Errorf("Not able to create %s dir: %v", secretFolder, err)
				log.Println(err)
				return err
			}
		}
		// if read from file errors out, we can ignore it here
		_ = api.ReadFromFile(&k8sSecret, secretFileName)
		if k8sSecret.Data == nil {
			k8sSecret.Data = map[string][]byte{}
		}
		// v1.Secret does enconding, so no need to encode again
		k8sSecret.Data[ra.Key] = []byte(cipherText)

		// Write secret to file
		k8sSecretBytes, err := api.K8sSecretToYaml(k8sSecret)
		if err != nil {
			api.LogDebugMessage("Error while converting K8s secret to yaml")
			return err
		}
		if err = ioutil.WriteFile(secretFileName, k8sSecretBytes, os.ModePerm); err != nil {
			api.LogDebugMessage("Error while writing K8s secret to file")
			return err
		}
		api.LogDebugMessage("Created a Kubernetes secret")
	}
	base64EncodedSecret := b64.StdEncoding.EncodeToString([]byte(cipherText))
	// write into CR the keyref of the secret
	qliksenseCR.Spec.AddToSecrets(ra.SvcName, ra.Key, base64EncodedSecret, secretName)
	return nil
}

func (q *Qliksense) SetConfigFromReader(arg string, reader io.Reader, base64Encoded bool) error {
	//take only name from the arguments, value should be from reader
	argName := strings.SplitN(arg, "=", 1)
	if len(argName) != 1 {
		return errors.New("can only have one argument from pipe")
	}
	valueBytes, err := ioutil.ReadAll(reader)
	if err != nil {
		return err
	}
	return q.SetConfigs([]string{argName[0] + "=" + string(valueBytes)}, base64Encoded)
}

// SetConfigs - set-configs <key>=<value> commands
func (q *Qliksense) SetConfigs(args []string, base64Encoded bool) error {
	// retieve current context from config.yaml
	qConfig := api.NewQConfig(q.QliksenseHome)
	qliksenseCR, err := qConfig.GetCurrentCR()
	if err != nil {
		return err
	}

	resultArgs, err := api.ProcessConfigArgs(args, base64Encoded)
	if err != nil {
		return err
	}
	for _, ra := range resultArgs {
		qliksenseCR.Spec.AddToConfigs(ra.SvcName, ra.Key, ra.Value)
	}
	// write modified content into context.yaml
	return qConfig.WriteCR(qliksenseCR)
}

func caseInsenstiveFieldByName(v reflect.Value, name string) reflect.Value {
	name = strings.ToLower(name)
	return v.FieldByNameFunc(func(n string) bool { return strings.ToLower(n) == name })
}

// SetOtherConfigs - set profile/storageclassname/git.repository/manifestRoot commands
func (q *Qliksense) SetOtherConfigs(args []string) error {
	// retieve current context from config.yaml
	qConfig := api.NewQConfig(q.QliksenseHome)
	qliksenseCR, err := qConfig.GetCurrentCR()
	if err != nil {
		return err
	}

	// modify appropriate fields
	if len(args) == 0 {
		err := fmt.Errorf("No args were provided. Please provide args to configure the current context")
		log.Println(err)
		return err
	}

	for _, arg := range args {
		if strings.HasPrefix(arg, "git.") {
			if err := q.processSetGit(arg, qliksenseCR); err != nil {
				return err
			}
		} else if strings.HasPrefix(arg, "opsRunner.") {
			if err := q.processSetOpsRunner(arg, qliksenseCR); err != nil {
				return err
			}
		} else {
			if err := processSetSingleArg(arg, qliksenseCR); err != nil {
				return err
			}
		}
		fmt.Println(Green("Successfully added to Custom Resource Spec"))
	}

	// write modified content into context.yaml
	return qConfig.WriteCR(qliksenseCR)
}

func processSetSingleArg(arg string, cr *api.QliksenseCR) error {
	nv := strings.Split(arg, "=")
	switch nv[0] {
	case "manifestsRoot":
		cr.Spec.ManifestsRoot = nv[1]
	case "profile":
		cr.Spec.Profile = nv[1]
	case "storageClassName":
		cr.Spec.StorageClassName = nv[1]
	default:
		return errors.New("Please enter one of: profile, storageClassName, manifestRoot, git to configure the current context")
	}
	return nil
}

func (q *Qliksense) processSetGit(arg string, cr *api.QliksenseCR) error {
	s := strings.Split(arg, "=")
	tArg0 := strings.TrimSpace(s[0])
	tArg1 := strings.TrimSpace(s[1])
	subs := strings.Split(tArg0, ".")
	if cr.Spec.Git == nil {
		cr.Spec.Git = &config.Repo{}
	}
	switch subs[1] {
	case "repository":
		cr.Spec.Git.Repository = tArg1
	case "accessToken":
		qConfig := api.NewQConfig(q.QliksenseHome)
		key, err := qConfig.GetEncryptionKeyFor(cr.GetName())
		if err != nil {
			return err
		}
		return cr.SetFetchAccessToken(tArg1, key)
	case "secretName":
		cr.Spec.Git.SecretName = tArg1
	case "userName":
		cr.Spec.Git.UserName = tArg1
	default:
		return errors.New(arg + " does not match any cr spec")
	}
	return nil
}

func (q *Qliksense) processSetOpsRunner(arg string, cr *api.QliksenseCR) error {
	args := strings.Split(arg, "=")
	subs := strings.Split(args[0], ".")
	if cr.Spec.OpsRunner == nil {
		cr.Spec.OpsRunner = &config.OpsRunner{}
	}
	switch subs[1] {
	case "enabled":
		if args[1] != "yes" && args[1] != "no" {
			return errors.New("Please use yes or no for key enabled")
		}
		cr.Spec.OpsRunner.Enabled = args[1]
	case "schedule":
		if _, err := cron.ParseStandard(args[1]); err != nil {
			return errors.New("Please enter string with standard cron scheduling syntax ")
		}
		cr.Spec.OpsRunner.Schedule = args[1]
	case "watchBranch":
		cr.Spec.OpsRunner.WatchBranch = args[1]
	case "image":
		cr.Spec.OpsRunner.Image = args[1]
	default:
		return errors.New(arg + " does not match any cr spec")
	}
	return nil
}

// SetContextConfig - set the context for qliksense kubernetes resources to live in
func (q *Qliksense) SetContextConfig(args []string) error {
	if len(args) == 1 {
		err := q.SetUpQliksenseContext(args[0])
		if err != nil {
			return err
		}
	} else {
		err := fmt.Errorf("Please provide a name to configure the context with")
		log.Println(err)
		return err
	}
	return nil
}

func (q *Qliksense) ListContextConfigs() error {
	qliksenseConfigFile := filepath.Join(q.QliksenseHome, QliksenseConfigFile)
	var qliksenseConfig api.QliksenseConfig
	if err := api.ReadFromFile(&qliksenseConfig, qliksenseConfigFile); err != nil {
		log.Println(err)
		return err
	}
	out := ansi.NewColorableStdout()
	w := tabwriter.NewWriter(out, 5, 8, 0, '\t', 0)
	fmt.Fprintln(w, Underline("Context Name"), "\t", Underline("CR File Location"))
	w.Flush()
	if len(qliksenseConfig.Spec.Contexts) > 0 {
		for _, cont := range qliksenseConfig.Spec.Contexts {
			fmt.Fprintln(w, cont.Name, "\t", qliksenseConfig.GetCRFilePath(cont.Name), "\t")
		}
		w.Flush()
		fmt.Fprintln(out, "")
		fmt.Fprintln(out, Bold("Current Context : "), qliksenseConfig.Spec.CurrentContext)
	} else {
		fmt.Fprintln(out, "No Contexts Available")
	}
	return nil
}

func (q *Qliksense) DeleteContextConfig(args []string, flag bool) error {
	if len(args) == 1 {
		qliksenseConfigFile := filepath.Join(q.QliksenseHome, QliksenseConfigFile)
		var qliksenseConfig api.QliksenseConfig
		api.ReadFromFile(&qliksenseConfig, qliksenseConfigFile)
		out := ansi.NewColorableStdout()
		switch args[0] {
		case qliksenseConfig.Spec.CurrentContext:
			fmt.Fprintln(out, Yellow("Please switch contexts to be able to delete this context."))
			err := fmt.Errorf(Red("Cannot delete current context - %s").String(), White(Bold(qliksenseConfig.Spec.CurrentContext)))
			return err
		case DefaultQliksenseContext:
			err := fmt.Errorf(Red("Cannot delete default qliksense context").String())
			return err
		default:
			qliksenseContextsDir1 := filepath.Join(q.QliksenseHome, QliksenseContextsDir)
			qliksenseContextFile := filepath.Join(qliksenseContextsDir1, args[0])
			qliksenseSecretsDir1 := filepath.Join(q.QliksenseHome, QliksenseSecretsDir, QliksenseContextsDir)
			qliksenseSecretsFile := filepath.Join(qliksenseSecretsDir1, args[0])
			if err := os.RemoveAll(qliksenseContextFile); err != nil {
				err = fmt.Errorf("Not able to delete %s dir: %v", qliksenseContextsDir1, err)
				log.Println(err)
				return err
			} else if err := os.RemoveAll(qliksenseSecretsFile); err != nil {
				err = fmt.Errorf("No Secrets Folder Detected")
				log.Println(err)
				return err
			} else {
				currentLength := len(qliksenseConfig.Spec.Contexts)
				if currentLength > 0 {
					temp := qliksenseConfig.Spec.Contexts
					qliksenseConfig.Spec.Contexts = nil
					for _, ctx := range temp {
						if ctx.Name != args[0] {
							qliksenseConfig.Spec.Contexts = append(qliksenseConfig.Spec.Contexts, api.Context{
								Name:   ctx.Name,
								CrFile: ctx.CrFile,
							})
						}
					}
					newLength := len(qliksenseConfig.Spec.Contexts)
					if currentLength != newLength {
						ans := flag
						if ans == false {
							ans = AskForConfirmation("Are You Sure? ")
						}
						if ans == true {
							api.WriteToFile(&qliksenseConfig, qliksenseConfigFile)
							fmt.Fprintln(out, Yellow(Underline("Warning: Active resources may still be running in-cluster")))
							fmt.Fprintln(out, Green("Successfully deleted context: "), Bold(args[0]))
						} else {
							return nil
						}
					} else {
						err := fmt.Errorf(Red("Context not found").String())
						return err
					}
				}
			}
		}
	} else {
		err := fmt.Errorf("Please provide a context as an argument to delete")
		log.Println(err)
		return err
	}
	return nil
}

// SetUpQliksenseDefaultContext - to setup dir structure for default qliksense context
func (q *Qliksense) SetUpQliksenseDefaultContext() error {
	if api.FileExists(filepath.Join(q.QliksenseHome, "config.yaml")) {
		qliksenseConfig := api.NewQConfig(q.QliksenseHome)
		if qliksenseConfig.IsContextExist(DefaultQliksenseContext) {
			return nil
		}
	}
	return q.SetUpQliksenseContext(DefaultQliksenseContext)
}

// SetUpQliksenseContext - to setup qliksense context
func (q *Qliksense) SetUpQliksenseContext(contextName string) error {
	if contextName == "" {
		err := fmt.Errorf("Please enter a non-empty context-name")
		log.Println(err)
		return err
	}
	// check the length of the context name entered by the user, it should not exceed 17 chars
	if len(contextName) > MaxContextNameLength {
		err := fmt.Errorf("Please enter a context-name with utmost 17 characters")
		log.Println(err)
		return err
	}

	qliksenseConfigFile := filepath.Join(q.QliksenseHome, QliksenseConfigFile)
	qliksenseConfig := api.NewQConfigEmpty(q.QliksenseHome)

	if !api.FileExists(qliksenseConfigFile) {
		qliksenseConfig.AddBaseQliksenseConfigs(contextName)
	} else {
		if err := api.ReadFromFile(qliksenseConfig, qliksenseConfigFile); err != nil {
			log.Println(err)
			return err
		}
	}

	if qliksenseConfig.IsContextExist(contextName) {
		qliksenseConfig.Spec.CurrentContext = contextName
		return qliksenseConfig.Write()
	}
	qliksenseCR := &api.QliksenseCR{}
	qliksenseCR.AddCommonConfig(contextName)
	qliksenseConfig.Spec.CurrentContext = contextName
	if err := qliksenseConfig.CreateOrWriteCrAndContext(qliksenseCR); err != nil {
		return err
	}

	// set the encrypted default mongo for the context in current CR
	return q.SetSecrets([]string{fmt.Sprintf("qliksense.mongodbUri=mongodb://%s-mongodb:27017/qliksense?ssl=false", contextName)}, false, false)
}

func validateInput(input string) (string, error) {
	var err error
	validInputs := []string{"yes", "no", "None"}
	isValid := false
	for _, elem := range validInputs {
		if input == elem {
			isValid = true
			break
		}
	}
	if !isValid {
		err = fmt.Errorf("Please enter one of: yes, no or None")
		log.Println(err)

	}
	return input, err
}

// PrepareK8sSecret targetFile contains base64 encoded value of encrypted value.
// this method decodes and decrypts the secret value in the secret.yaml file and returns a B64encoded string
func (q *Qliksense) PrepareK8sSecret(targetFile string) (string, error) {
	// check if targetFile exists
	if !api.FileExists(targetFile) {
		err := fmt.Errorf("Target file does not exist in the path provided")
		log.Println(err)
		return "", err
	}
	qConfig := api.NewQConfig(q.QliksenseHome)
	encryptionKey, err := qConfig.GetEncryptionKeyForCurrent()
	if err != nil {
		return "", err
	}

	// read the target file
	k8sSecret, err := readTargetfile(targetFile)
	if err != nil {
		return "", err
	}
	// retrieve value from data section
	k8sSecret1, err := api.K8sSecretFromYaml(k8sSecret)
	if err != nil {
		return "", err
	}
	dataMap := k8sSecret1.Data
	var resultMap = make(map[string][]byte)
	for k, v := range dataMap {
		//k8s secrets has already base64 decoed value
		decryptedString, err := api.DecryptData(v, encryptionKey)
		if err != nil {
			err := fmt.Errorf("Not able to decrypt message: %v", err)
			return "", err
		}
		resultMap[k] = []byte(decryptedString)
	}

	// putting the above map back into the k8sSecret struct
	k8sSecret1.Data = resultMap
	k8sSecretBytes, err := api.K8sSecretToYaml(k8sSecret1)
	if err != nil {
		return "", err
	}
	return string(k8sSecretBytes), nil
}

func readTargetfile(targetFile string) ([]byte, error) {
	k8sSecret, err := ioutil.ReadFile(targetFile)
	if err != nil {
		err := fmt.Errorf("Unable to read the targetFile")
		log.Println(err)
		return nil, err
	}
	return k8sSecret, nil
}

func (q *Qliksense) SetImageRegistry(registry, pushUsername, pushPassword, pullUsername, pullPassword string) error {
	qConfig := api.NewQConfig(q.QliksenseHome)
	qliksenseCR, err := qConfig.GetCurrentCR()
	if err != nil {
		return err
	}
	if pushUsername != "" {
		if err := qConfig.SetPushDockerConfigJsonSecret(&api.DockerConfigJsonSecret{
			Uri:      registry,
			Username: pushUsername,
			Password: pushPassword,
		}); err != nil {
			return err
		} else if err := qConfig.SetPullDockerConfigJsonSecret(&api.DockerConfigJsonSecret{
			Name:     pullSecretName,
			Uri:      registry,
			Username: pullUsername,
			Password: pullPassword,
			Email:    pullUsername,
		}); err != nil {
			return err
		}
	} else if err := qConfig.DeletePushDockerConfigJsonSecret(); err != nil && !os.IsNotExist(err) {
		return err
	} else if err := qConfig.DeletePullDockerConfigJsonSecret(); err != nil && !os.IsNotExist(err) {
		return err
	}

	qliksenseCR.Spec.AddToConfigs("qliksense", imageRegistryConfigKey, registry)
	return qConfig.WriteCR(qliksenseCR)
}

func (q *Qliksense) SetEulaAccepted() error {
	qConfig := api.NewQConfig(q.QliksenseHome)
	qcr, err := qConfig.GetCurrentCR()
	if err != nil {
		return err
	}
	if !qcr.IsEULA() {
		qcr.SetEULA("yes")
		return qConfig.WriteCurrentContextCR(qcr)
	}
	return nil
}