package env

import (
	"fmt"
	"log"
	"os"
	"path/filepath"

	"github.com/joho/godotenv"
)

func getEnvDir() (string, error) {
	dir, err := os.Getwd()
	if err != nil {
		return "", err
	}
	for {
		if filepath.Base(dir) == "e2e" {
			return dir, nil
		}
		parrent := filepath.Dir(dir)
		if parrent == dir {
			return "", fmt.Errorf("未找到 e2e 目录")
		}
		dir = parrent
	}
}

func LoadEnv(envFile string) error {
	envDir, err := getEnvDir()
	if err != nil {
		return err
	}
	if err := godotenv.Load(filepath.Join(envDir, envFile)); err != nil {
		log.Printf("警告: 未找到 .env 文件：%v。将使用系统环境变量", err)
	}
	return nil
}