76 lines
1.5 KiB
Go
76 lines
1.5 KiB
Go
|
// Copyright (C) 2018 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package process
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
homedir "github.com/mitchellh/go-homedir"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"github.com/spf13/pflag"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
)
|
||
|
|
|
||
|
|
func defaultConfigPath(name string) string {
|
||
|
|
if name == "" {
|
||
|
|
name = filepath.Base(os.Args[0])
|
||
|
|
}
|
||
|
|
path := filepath.Join(".storj", fmt.Sprintf("%s.json", name))
|
||
|
|
home, err := homedir.Dir()
|
||
|
|
if err != nil {
|
||
|
|
log.Println(err)
|
||
|
|
return path
|
||
|
|
}
|
||
|
|
return filepath.Join(home, path)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Execute runs a *cobra.Command and sets up Storj-wide process configuration
|
||
|
|
// like a configuration file and logging.
|
||
|
|
func Execute(cmd *cobra.Command) {
|
||
|
|
cfgFile := flag.String("config", defaultConfigPath(cmd.Name()),
|
||
|
|
"config file")
|
||
|
|
|
||
|
|
pflag.CommandLine.AddGoFlagSet(flag.CommandLine)
|
||
|
|
|
||
|
|
cobra.OnInitialize(func() {
|
||
|
|
viper.BindPFlags(cmd.Flags())
|
||
|
|
viper.SetEnvPrefix("storj")
|
||
|
|
viper.AutomaticEnv()
|
||
|
|
if *cfgFile != "" {
|
||
|
|
viper.SetConfigFile(*cfgFile)
|
||
|
|
viper.ReadInConfig()
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
Must(cmd.Execute())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Main runs a Service
|
||
|
|
func Main(s Service) error {
|
||
|
|
cfgFile := flag.String("config", defaultConfigPath(""), "config file")
|
||
|
|
|
||
|
|
pflag.CommandLine.AddGoFlagSet(flag.CommandLine)
|
||
|
|
pflag.Parse()
|
||
|
|
viper.BindPFlags(pflag.CommandLine)
|
||
|
|
viper.SetEnvPrefix("storj")
|
||
|
|
viper.AutomaticEnv()
|
||
|
|
if *cfgFile != "" {
|
||
|
|
viper.SetConfigFile(*cfgFile)
|
||
|
|
viper.ReadInConfig()
|
||
|
|
}
|
||
|
|
|
||
|
|
return CtxService(s)(&cobra.Command{}, pflag.Args())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Must checks for errors
|
||
|
|
func Must(err error) {
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|