cmd/config.go
author Mikael Berthe <mikael@lilotux.net>
Tue, 09 May 2017 20:14:28 +0200
changeset 110 57843255fd1a
parent 91 02312ccc1fd3
child 113 2e411da68fd3
permissions -rw-r--r--
Refactor printers
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     1
// Copyright © 2017 Mikael Berthe <mikael@lilotux.net>
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     2
//
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     3
// Licensed under the MIT license.
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     4
// Please see the LICENSE file is this directory.
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     5
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     6
package cmd
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     7
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     8
import (
10
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
     9
	"os"
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    10
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    11
	"github.com/spf13/cobra"
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    12
	"github.com/spf13/viper"
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    13
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    14
	"github.com/McKael/madonctl/printer"
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    15
)
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    16
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    17
var configCmd = &cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    18
	Use:   "config",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    19
	Short: "Display configuration",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    20
}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    21
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    22
func init() {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    23
	RootCmd.AddCommand(configCmd)
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    24
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    25
	// Subcommands
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    26
	configCmd.AddCommand(configSubcommands...)
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    27
}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    28
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    29
var configSubcommands = []*cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    30
	&cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    31
		Use:   "dump",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    32
		Short: "Dump the configuration",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    33
		RunE: func(cmd *cobra.Command, args []string) error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    34
			return configDump()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    35
		},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    36
	},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    37
	&cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    38
		Use:     "whoami",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    39
		Aliases: []string{"token"},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    40
		Short:   "Display user token",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    41
		RunE: func(cmd *cobra.Command, args []string) error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    42
			return configDisplayToken()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    43
		},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    44
	},
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    45
	&cobra.Command{
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    46
		Use: "themes",
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    47
		//Aliases: []string{},
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    48
		Short: "Display available themes",
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    49
		RunE: func(cmd *cobra.Command, args []string) error {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    50
			return configDisplayThemes()
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    51
		},
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    52
	},
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    53
}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    54
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    55
const configurationTemplate = `---
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    56
instance: '{{.InstanceURL}}'
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    57
app_id: '{{.ID}}'
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    58
app_secret: '{{.Secret}}'
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    59
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    60
{{if .UserToken}}token: {{.UserToken.access_token}}{{else}}#token: ''{{end}}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    61
#login: ''
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    62
#password: ''
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    63
safe_mode: true
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    64
...
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    65
`
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    66
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    67
func configDump() error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    68
	if viper.GetBool("safe_mode") {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    69
		errPrint("Cannot dump: disabled by configuration (safe_mode)")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    70
		return nil
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    71
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    72
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    73
	if err := madonInitClient(); err != nil {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    74
		return err
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    75
	}
10
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    76
	// Try to sign in if a login was provided
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    77
	if viper.GetString("token") != "" || viper.GetString("login") != "" {
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    78
		if err := madonLogin(); err != nil {
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    79
			errPrint("Error: could not log in: %v", err)
10
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    80
			os.Exit(-1)
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    81
		}
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    82
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    83
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    84
	var p printer.ResourcePrinter
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    85
	var err error
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    86
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    87
	if getOutputFormat() == "plain" {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    88
		cfile := viper.ConfigFileUsed()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    89
		if cfile == "" {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    90
			cfile = defaultConfigFile
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    91
		}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    92
		errPrint("You can copy the following lines into a configuration file.")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    93
		errPrint("E.g. %s -i INSTANCE -L USERNAME -P PASS config dump > %s\n", AppName, cfile)
83
57afac822019 Use a map for ResourcePrinter options
Mikael Berthe <mikael@lilotux.net>
parents: 81
diff changeset
    94
		pOptions := printer.Options{"template": configurationTemplate}
57afac822019 Use a map for ResourcePrinter options
Mikael Berthe <mikael@lilotux.net>
parents: 81
diff changeset
    95
		p, err = printer.NewPrinterTemplate(pOptions)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    96
	} else {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    97
		p, err = getPrinter()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    98
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    99
	if err != nil {
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   100
		errPrint("Error: %v", err)
81
b1671f83e91b Do not display usage when GetPrinter fails
Mikael Berthe <mikael@lilotux.net>
parents: 10
diff changeset
   101
		os.Exit(1)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   102
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   103
	return p.PrintObj(gClient, nil, "")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   104
}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   105
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   106
func configDisplayToken() error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   107
	if viper.GetBool("safe_mode") {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   108
		errPrint("Cannot dump: disabled by configuration (safe_mode)")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   109
		return nil
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   110
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   111
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   112
	if err := madonInit(true); err != nil {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   113
		return err
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   114
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   115
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   116
	p, err := getPrinter()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   117
	if err != nil {
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   118
		errPrint("Error: %v", err)
81
b1671f83e91b Do not display usage when GetPrinter fails
Mikael Berthe <mikael@lilotux.net>
parents: 10
diff changeset
   119
		os.Exit(1)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   120
	}
110
57843255fd1a Refactor printers
Mikael Berthe <mikael@lilotux.net>
parents: 91
diff changeset
   121
	return p.printObj(gClient.UserToken)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   122
}
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   123
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   124
// configDisplayThemes lists the available themes
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   125
// It is intended for shell completion.
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   126
func configDisplayThemes() error {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   127
	var p printer.ResourcePrinter
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   128
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   129
	themes, err := getThemes()
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   130
	if err != nil {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   131
		errPrint("Error: %v", err)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   132
		os.Exit(1)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   133
	}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   134
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   135
	if getOutputFormat() == "plain" {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   136
		pOptions := printer.Options{"template": `{{printf "%s\n" .}}`}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   137
		p, err = printer.NewPrinterTemplate(pOptions)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   138
	} else {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   139
		p, err = getPrinter()
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   140
	}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   141
	if err != nil {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   142
		errPrint("Error: %v", err)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   143
		os.Exit(1)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   144
	}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   145
	return p.PrintObj(themes, nil, "")
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   146
}