cmd/config.go
author Mikael Berthe <mikael@lilotux.net>
Wed, 10 May 2017 20:13:31 +0200
changeset 116 3dd35bdb63ac
parent 114 590c9dd7328e
child 127 2b4d0f198a94
permissions -rw-r--r--
Version 1.0.0
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
114
590c9dd7328e Improve config dump output
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    11
	"github.com/mattn/go-isatty"
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    12
	"github.com/spf13/cobra"
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    13
	"github.com/spf13/viper"
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    14
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    15
	"github.com/McKael/madonctl/printer"
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    18
var configCmd = &cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    19
	Use:   "config",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    20
	Short: "Display configuration",
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    23
func init() {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    24
	RootCmd.AddCommand(configCmd)
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    25
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    26
	// Subcommands
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    27
	configCmd.AddCommand(configSubcommands...)
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    30
var configSubcommands = []*cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    31
	&cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    32
		Use:   "dump",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    33
		Short: "Dump the configuration",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    34
		RunE: func(cmd *cobra.Command, args []string) error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    35
			return configDump()
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
	},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    38
	&cobra.Command{
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    39
		Use:     "whoami",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    40
		Aliases: []string{"token"},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    41
		Short:   "Display user token",
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    42
		RunE: func(cmd *cobra.Command, args []string) error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    43
			return configDisplayToken()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    44
		},
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    45
	},
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    46
	&cobra.Command{
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    47
		Use: "themes",
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    48
		//Aliases: []string{},
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    49
		Short: "Display available themes",
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    50
		RunE: func(cmd *cobra.Command, args []string) error {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    51
			return configDisplayThemes()
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    52
		},
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    53
	},
0
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    56
const configurationTemplate = `---
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    57
instance: '{{.InstanceURL}}'
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    58
app_id: '{{.ID}}'
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    59
app_secret: '{{.Secret}}'
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    60
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    61
{{if .UserToken}}token: {{.UserToken.access_token}}{{else}}#token: ''{{end}}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    62
#login: ''
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    63
#password: ''
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    64
safe_mode: true
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    68
func configDump() error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    69
	if viper.GetBool("safe_mode") {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    70
		errPrint("Cannot dump: disabled by configuration (safe_mode)")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    71
		return nil
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    74
	if err := madonInitClient(); err != nil {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    75
		return err
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    76
	}
10
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    77
	// 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
    78
	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
    79
		if err := madonLogin(); err != nil {
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
    80
			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
    81
			os.Exit(-1)
e4c490882126 Don't dump the configuration file if login failed
Mikael Berthe <mikael@lilotux.net>
parents: 0
diff changeset
    82
		}
0
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
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    85
	var p printer.ResourcePrinter
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    86
	var err error
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    87
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    88
	if getOutputFormat() == "plain" {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    89
		cfile := viper.ConfigFileUsed()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    90
		if cfile == "" {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    91
			cfile = defaultConfigFile
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    92
		}
114
590c9dd7328e Improve config dump output
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    93
		if isatty.IsTerminal(os.Stdout.Fd()) {
590c9dd7328e Improve config dump output
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    94
			errPrint("You can copy the following lines into a configuration file.")
590c9dd7328e Improve config dump output
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    95
			errPrint("E.g. %s -i INSTANCE -L USERNAME -P PASS config dump > %s", AppName, cfile)
590c9dd7328e Improve config dump output
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    96
			errPrint(" or  %s -i INSTANCE oauth2 > %s\n", AppName, cfile)
590c9dd7328e Improve config dump output
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    97
		}
83
57afac822019 Use a map for ResourcePrinter options
Mikael Berthe <mikael@lilotux.net>
parents: 81
diff changeset
    98
		pOptions := printer.Options{"template": configurationTemplate}
57afac822019 Use a map for ResourcePrinter options
Mikael Berthe <mikael@lilotux.net>
parents: 81
diff changeset
    99
		p, err = printer.NewPrinterTemplate(pOptions)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   100
	} else {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   101
		p, err = getPrinter()
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
	if err != nil {
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   104
		errPrint("Error: %v", err)
81
b1671f83e91b Do not display usage when GetPrinter fails
Mikael Berthe <mikael@lilotux.net>
parents: 10
diff changeset
   105
		os.Exit(1)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   106
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   107
	return p.PrintObj(gClient, nil, "")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   108
}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   109
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   110
func configDisplayToken() error {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   111
	if viper.GetBool("safe_mode") {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   112
		errPrint("Cannot dump: disabled by configuration (safe_mode)")
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   113
		return nil
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
	if err := madonInit(true); err != nil {
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   117
		return err
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   118
	}
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   119
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   120
	p, err := getPrinter()
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   121
	if err != nil {
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   122
		errPrint("Error: %v", err)
81
b1671f83e91b Do not display usage when GetPrinter fails
Mikael Berthe <mikael@lilotux.net>
parents: 10
diff changeset
   123
		os.Exit(1)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   124
	}
110
57843255fd1a Refactor printers
Mikael Berthe <mikael@lilotux.net>
parents: 91
diff changeset
   125
	return p.printObj(gClient.UserToken)
0
5abace724584 Initial public release
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
   126
}
91
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   127
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   128
// configDisplayThemes lists the available themes
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   129
// It is intended for shell completion.
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   130
func configDisplayThemes() error {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   131
	var p printer.ResourcePrinter
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   132
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   133
	themes, err := getThemes()
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   134
	if err != nil {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   135
		errPrint("Error: %v", err)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   136
		os.Exit(1)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   137
	}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   138
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   139
	if getOutputFormat() == "plain" {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   140
		pOptions := printer.Options{"template": `{{printf "%s\n" .}}`}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   141
		p, err = printer.NewPrinterTemplate(pOptions)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   142
	} else {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   143
		p, err = getPrinter()
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
	if err != nil {
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   146
		errPrint("Error: %v", err)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   147
		os.Exit(1)
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   148
	}
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   149
	return p.PrintObj(themes, nil, "")
02312ccc1fd3 Add shell completion for theme flag
Mikael Berthe <mikael@lilotux.net>
parents: 83
diff changeset
   150
}