campaigner/cmd/token.go

package cmd

import (
	"git.ctrlz.es/mgdelacroix/campaigner/config"

	"github.com/spf13/cobra"
)

func TokenSetJiraCmd() *cobra.Command {
	return &cobra.Command{
		Use:     "jira USERNAME TOKEN",
		Short:   "Sets the value of the jira token",
		Args:    cobra.ExactArgs(2),
		RunE:    tokenSetJiraCmdF,
	}
}

func TokenSetGithubCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "github TOKEN",
		Short: "Sets the value of the github token",
		Args:  cobra.ExactArgs(1),
		RunE:  tokenSetGithubCmdF,
	}
}

func TokenSetCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "set",
		Short: "Sets the value of the platform tokens",
	}

	cmd.AddCommand(
		TokenSetJiraCmd(),
		TokenSetGithubCmd(),
	)

	return cmd
}

func TokenCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "token",
		Short: "Subcommands related to tokens",
	}

	cmd.AddCommand(
		TokenSetCmd(),
	)

	return cmd
}

func tokenSetJiraCmdF(cmd *cobra.Command, args []string) error {
	cfg, err := config.ReadConfig()
	if err != nil {
		ErrorAndExit(cmd, err)
	}

	cfg.JiraUsername = args[0]
	cfg.JiraToken = args[1]
	if err := config.SaveConfig(cfg); err != nil {
		ErrorAndExit(cmd, err)
	}
	return nil
}

func tokenSetGithubCmdF(cmd *cobra.Command, args []string) error {
	cfg, err := config.ReadConfig()
	if err != nil {
		ErrorAndExit(cmd, err)
	}

	cfg.GithubToken = args[0]
	if err := config.SaveConfig(cfg); err != nil {
		ErrorAndExit(cmd, err)
	}
	return nil
}