package env

import (
	"os"
	"path/filepath"
	"runtime"
	"strings"

	"gitee.com/gitwcx/jenv/sdk/constants"
)

// oldPath 保存原始的 PATH 环境变量值
var oldPath string

// SetEnv 临时设置环境变量
func SetEnv(key string, value string) error {
	if runtime.GOOS == "windows" {
		return doSetEnv(key, value)
	} else {
		return doSetEnv(key, value)
		//return os.Setenv(key, value)
	}
}

func getDefaultJavaHome(sdkType string) string {
	return constants.GetDefaultEnvHome(sdkType)
}

// CleanJDKPath 清理 PATH 中的 JDK 相关设置
func CleanJDKPath() error {
	// 获取当前的 PATH
	path := os.Getenv("PATH")

	// 将 PATH 分割成数组
	paths := strings.Split(path, string(filepath.ListSeparator))

	// 过滤掉包含 jdk、jre 的路径
	var cleanedPaths []string
	for _, p := range paths {
		if !strings.Contains(strings.ToLower(p), "jdk") && !strings.Contains(strings.ToLower(p), "jre") {
			cleanedPaths = append(cleanedPaths, p)
		}
	}

	// 重新组合 PATH
	newPath := strings.Join(cleanedPaths, string(filepath.ListSeparator))

	// 设置新的 PATH
	return os.Setenv("PATH", newPath)
}

// RestoreOldPath 恢复原始的 PATH 环境变量
func RestoreOldPath() error {
	if oldPath != "" {
		return os.Setenv("PATH", oldPath)
	}
	return nil
}

// IsInPath checks if a directory is in the system PATH
func IsInPath(dir string) bool {
	if runtime.GOOS == "windows" {
		path, err := QuerySystemEnvironmentVariable(`PATH`)
		if err != nil {
			return false
		}
		// Split path by semicolon on Windows
		paths := strings.Split(path, ";")
		for _, p := range paths {
			if p == dir {
				return true
			}
		}
		return false
	} else {
		// For non-Windows systems, use os.Getenv
		path := os.Getenv("PATH")
		// Split path by colon on Unix-like systems
		paths := strings.Split(path, ":")
		for _, p := range paths {
			if p == dir {
				return true
			}
		}
		return false
	}
}

// AddToPath adds a directory to the system PATH
func AddToPath(binDir string) error {
	if runtime.GOOS == "windows" {
		oldPath, err := QuerySystemEnvironmentVariable("PATH")
		if err != nil {
			return err
		}
		return UpdateSystemEnvironmentVariable("PATH", binDir+";"+oldPath)
	} else {
		// For non-Windows systems
		currentPath := os.Getenv("PATH")
		newPath := binDir + ":" + currentPath
		return os.Setenv("PATH", newPath)
	}
}