ente/cli/main.go

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
}