85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package modules
|
|
|
|
import (
|
|
"os"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/commonflags"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/config"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/storagecfg"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/misc"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/autocomplete"
|
|
utilConfig "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/config"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/gendoc"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
rootCmd = &cobra.Command{
|
|
Use: "frostfs-adm",
|
|
Short: "FrostFS Administrative Tool",
|
|
Long: `FrostFS Administrative Tool provides functions to setup and
|
|
manage FrostFS network deployment.`,
|
|
RunE: entryPoint,
|
|
SilenceUsage: true,
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
cobra.OnInitialize(func() { initConfig(rootCmd) })
|
|
// we need to init viper config to bind viper and cobra configurations for
|
|
// rpc endpoint, alphabet wallet dir, key credentials, etc.
|
|
|
|
// use stdout as default output for cmd.Print()
|
|
rootCmd.SetOut(os.Stdout)
|
|
|
|
rootCmd.PersistentFlags().StringP(commonflags.ConfigFlag, commonflags.ConfigFlagShorthand, "", commonflags.ConfigFlagUsage)
|
|
rootCmd.PersistentFlags().String(commonflags.ConfigDirFlag, "", commonflags.ConfigDirFlagUsage)
|
|
rootCmd.PersistentFlags().BoolP(commonflags.Verbose, commonflags.VerboseShorthand, false, commonflags.VerboseUsage)
|
|
_ = viper.BindPFlag(commonflags.Verbose, rootCmd.PersistentFlags().Lookup(commonflags.Verbose))
|
|
rootCmd.Flags().Bool("version", false, "Application version")
|
|
|
|
rootCmd.AddCommand(config.RootCmd)
|
|
rootCmd.AddCommand(morph.RootCmd)
|
|
rootCmd.AddCommand(storagecfg.RootCmd)
|
|
|
|
rootCmd.AddCommand(autocomplete.Command("frostfs-adm"))
|
|
rootCmd.AddCommand(gendoc.Command(rootCmd, gendoc.Options{}))
|
|
}
|
|
|
|
func Execute() error {
|
|
return rootCmd.Execute()
|
|
}
|
|
|
|
func entryPoint(cmd *cobra.Command, _ []string) error {
|
|
printVersion, _ := cmd.Flags().GetBool("version")
|
|
if printVersion {
|
|
cmd.Print(misc.BuildInfo("FrostFS Adm"))
|
|
return nil
|
|
}
|
|
|
|
return cmd.Usage()
|
|
}
|
|
|
|
func initConfig(cmd *cobra.Command) {
|
|
configFile, err := cmd.Flags().GetString(commonflags.ConfigFlag)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if configFile != "" {
|
|
viper.SetConfigType("yml")
|
|
viper.SetConfigFile(configFile)
|
|
_ = viper.ReadInConfig() // if config file is set but unavailable, ignore it
|
|
}
|
|
|
|
configDir, err := cmd.Flags().GetString(commonflags.ConfigDirFlag)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if configDir != "" {
|
|
_ = utilConfig.ReadConfigDir(viper.GetViper(), configDir) // if config files cannot be read, ignore it
|
|
}
|
|
}
|