github.com/olli-ai/jx/v2@v2.0.400-0.20210921045218-14731b4dd448/pkg/cmd/step/step_credential.go

package step

import (
	"fmt"
	"io/ioutil"
	"sort"

	"github.com/olli-ai/jx/v2/pkg/cmd/opts/step"

	"github.com/olli-ai/jx/v2/pkg/cmd/helper"

	"github.com/jenkins-x/jx-logging/pkg/log"
	"github.com/olli-ai/jx/v2/pkg/cmd/opts"
	"github.com/olli-ai/jx/v2/pkg/cmd/templates"
	"github.com/olli-ai/jx/v2/pkg/kube"
	"github.com/olli-ai/jx/v2/pkg/util"
	"github.com/pkg/errors"
	"github.com/spf13/cobra"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

// StepCredentialOptions contains the command line arguments for this command
type StepCredentialOptions struct {
	step.StepOptions

	Namespace string
	Secret    string
	Key       string
	File      string
	Optional  bool
}

var (
	stepCredentialLong = templates.LongDesc(`
		Returns a credential from a Secret for easy scripting in pipeline steps.

		Supports the [Jenkins Credentials Provider labels on the Secrets](https://jenkinsci.github.io/kubernetes-credentials-provider-plugin/examples/)

		If you specify --optional then if the key or secret doesn't exist then the command will only print a warning and will not error.
`)

	stepCredentialExample = templates.Examples(`
		# get the password of a secret 'foo' which uses the Jenkins Credentials Provider labels
		export MY_PWD="$(jx step credential -s foo)"

		# get the password entry of a secret 'foo' as an environment variable
		export MY_PWD="$(jx step credential -s foo -k passwordj)"

		# create a local file from a file based secret using the Jenkins Credentials Provider labels
        export MY_KEY_FILE="$(jx step credential -s foo)"
         
		# create a local file called cheese from a given key
        export MY_KEY_FILE="$(jx step credential -s foo -f cheese -k data)"

		# create a local file called cheese from a given key, if the key exists'
        export MY_KEY_FILE="$(jx step credential -s foo -f cheese -k data --optional)"
         
`)
)

// NewCmdStepCredential creates the command
func NewCmdStepCredential(commonOpts *opts.CommonOptions) *cobra.Command {
	options := StepCredentialOptions{
		StepOptions: step.StepOptions{
			CommonOptions: commonOpts,
		},
	}
	cmd := &cobra.Command{
		Use:     "credential",
		Short:   "Returns a secret entry for easy scripting in pipeline steps",
		Long:    stepCredentialLong,
		Example: stepCredentialExample,
		Aliases: []string{"secret", "cred"},
		Run: func(cmd *cobra.Command, args []string) {
			options.Cmd = cmd
			options.Args = args
			err := options.Run()
			helper.CheckErr(err)
		},
	}

	cmd.Flags().StringVarP(&options.Namespace, "namespace", "n", "", "the namespace to look for a Secret")
	cmd.Flags().StringVarP(&options.Secret, "name", "s", "", "the name of the Secret")
	cmd.Flags().StringVarP(&options.Key, "key", "k", "", "the key in the Secret to output")
	cmd.Flags().StringVarP(&options.File, "file", "f", "", "the key for the filename to use if this is a file based Secret")
	cmd.Flags().BoolVarP(&options.Optional, "optional", "", false, "if true, then the command will only warn (not error) if the secret or the key doesn't exist")

	return cmd
}

// Run runs the command
func (o *StepCredentialOptions) Run() error {
	kubeClient, devNs, err := o.KubeClientAndDevNamespace()
	if err != nil {
		return err
	}

	ns := o.Namespace
	if ns == "" {
		ns = devNs
	}

	name := o.Secret
	if name == "" {
		return util.MissingOption("name")
	}
	secret, err := kubeClient.CoreV1().Secrets(ns).Get(name, metav1.GetOptions{})
	if err != nil {
		if o.Optional {
			log.Logger().Warnf("failed to find Secret %s in namespace %s", name, ns)
			return nil
		}
		return errors.Wrapf(err, "failed to find Secret %s in namespace %s", name, ns)
	}
	data := secret.Data
	if data == nil {
		if o.Optional {
			log.Logger().Warnf("Secret %s in namespace %s has no data", name, ns)
			return nil
		}
		return errors.Wrapf(err, "Secret %s in namespace %s has no data", name, ns)
	}
	keys := []string{}
	for k := range data {
		keys = append(keys, k)
	}
	sort.Strings(keys)

	filename := o.File
	key := o.Key

	labels := secret.Labels
	if labels != nil {
		kind := labels[kube.LabelCredentialsType]
		if filename == "" && kind == kube.ValueCredentialTypeSecretFile {
			filenameData, ok := data["filename"]
			if ok {
				filename = string(filenameData)
			} else {
				return fmt.Errorf("the Secret %s in namespace %s has label %s with value %s but has no filename key", name, ns, kube.LabelCredentialsType, kind)
			}
			if key == "" {
				key = "data"
			}
		}

		if key == "" && kind == kube.ValueCredentialTypeUsernamePassword {
			key = "password"
		}
	}

	if key == "" {
		return util.MissingOptionWithOptions("key", keys)
	}

	value, ok := data[key]
	if !ok {
		log.Logger().Warnf("Secret %s in namespace %s does not have key %s", name, ns, key)
		if o.Optional {
			return nil
		}
		return util.InvalidOption("key", key, keys)
	}
	if filename != "" {
		err = ioutil.WriteFile(filename, value, util.DefaultWritePermissions)
		if err != nil {
			return errors.Wrapf(err, "failed to store file %s", filename)
		}
		fmt.Fprintf(o.Out, "%s\n", filename)
		return nil
	}
	fmt.Fprintf(o.Out, "%s\n", value)
	return nil
}