package main import ( "fmt" "github.com/ente-io/cli/cmd" "github.com/ente-io/cli/internal" "github.com/ente-io/cli/internal/api" "github.com/ente-io/cli/pkg" "github.com/ente-io/cli/pkg/secrets" "github.com/ente-io/cli/utils/constants" "github.com/spf13/viper" "log" "os" "path/filepath" "strings" ) var AppVersion = "0.1.13" func main() { cliDBPath, err := GetCLIConfigPath() if secrets.IsRunningInContainer() { cliDBPath = constants.CliDataPath _, err := internal.ValidateDirForWrite(cliDBPath) if err != nil { log.Fatalf("Please mount a volume to %s to persist cli data\n%v\n", cliDBPath, err) } } if err != nil { log.Fatalf("Could not create cli config path\n%v\n", err) } initConfig(cliDBPath) newCliPath := fmt.Sprintf("%s/ente-cli.db", cliDBPath) if !strings.HasPrefix(cliDBPath, "/") { oldCliPath := fmt.Sprintf("%sente-cli.db", cliDBPath) if _, err := os.Stat(oldCliPath); err == nil { log.Printf("migrating old cli db from %s to %s\n", oldCliPath, newCliPath) if err := os.Rename(oldCliPath, newCliPath); err != nil { log.Fatalf("Could not rename old cli db\n%v\n", err) } } } db, err := pkg.GetDB(newCliPath) if err != nil { if strings.Contains(err.Error(), "timeout") { log.Fatalf("Please close all other instances of the cli and try again\n%v\n", err) } else { panic(err) } } ctrl := pkg.ClICtrl{ Client: api.NewClient(api.Params{ Debug: viper.GetBool("log.http"), Host: viper.GetString("endpoint.api"), }), DB: db, KeyHolder: secrets.NewKeyHolder(secrets.GetOrCreateClISecret()), } err = ctrl.Init() if err != nil { panic(err) } defer func() { if err := db.Close(); err != nil { panic(err) } }() if len(os.Args) == 2 && os.Args[1] == "docs" { log.Println("Generating docs") err = cmd.GenerateDocs() if err != nil { log.Fatal(err) } return } cmd.Execute(&ctrl, AppVersion) } func initConfig(cliConfigPath string) { viper.SetConfigName("config") // name of config file (without extension) viper.SetConfigType("yaml") // REQUIRED if the config file does not have the extension in the name viper.AddConfigPath(cliConfigPath + "/") // path to look for the config file in viper.AddConfigPath(".") // optionally look for config in the working directory viper.SetDefault("endpoint.api", constants.EnteApiUrl) viper.SetDefault("log.http", false) if err := viper.ReadInConfig(); err != nil { if _, ok := err.(viper.ConfigFileNotFoundError); ok { } else { // Config file was found but another error was produced } } } // GetCLIConfigPath returns the path to the .ente-cli folder and creates it if it doesn't exist. func GetCLIConfigPath() (string, error) { if os.Getenv("ENTE_CLI_CONFIG_PATH") != "" { return os.Getenv("ENTE_CLI_CONFIG_PATH"), nil } // Get the user's home directory homeDir, err := os.UserHomeDir() if err != nil { return "", err } cliDBPath := filepath.Join(homeDir, ".ente") // Check if the folder already exists, if not, create it if _, err := os.Stat(cliDBPath); os.IsNotExist(err) { err := os.MkdirAll(cliDBPath, 0755) if err != nil { return "", err } } return cliDBPath, nil }