package env

import (
	"log"

	"gitee.com/carlmax_my/console-core-go/pkg/file"

	"github.com/joho/godotenv"
)

const (
	DEF_ENV_FILE = ".env"
)

func LoadEnvFile(envFile string) error {
	return godotenv.Load(envFile)
}

// default from exe folder
func LoadEnv() error {
	envFile := file.JoinPath(file.GetExecPath(), DEF_ENV_FILE)
	return godotenv.Load(envFile)
}

// for dev mode, exe bin in a tmp random folder, so wd is fine
func LoadFromWd() error {
	dir, err := file.GetRootDir()
	if err != nil {
		log.Printf("LoadFromWd[%s] error: %s\n", dir, err.Error())
		return err
	}

	envFile := file.JoinPath(dir, DEF_ENV_FILE)
	log.Println("LoadFromWd", envFile)
	return godotenv.Load(envFile)
}

func LoadFromExeDir() error {
	dir := file.GetExecPath()
	envFile := file.JoinPath(dir, DEF_ENV_FILE)
	log.Println("LoadFromExeDir", envFile)
	return godotenv.Load(envFile)
}

func LoadFromBoth() error {
	dir, err := file.GetRootDir()
	if err != nil {
		log.Printf("LoadFromBoth[%s] error: %s\n", dir, err.Error())
		return LoadFromExeDir()
	}

	dir2 := file.GetExecPath()
	if dir == dir2 {
		return LoadFromExeDir()
	}

	LoadFromExeDir()
	envFile := file.JoinPath(dir, DEF_ENV_FILE)
	return godotenv.Load(envFile)
}