package setup

import (
	"context"
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"runtime"
	"strings"

	"github.com/gogf/gf/v2/frame/g"
)

var (
	envConfigured bool

	// DetectedWhisperPath 自动检测到的 whisper 命令行路径（空则使用 python -m whisper）
	DetectedWhisperPath string
)

// EnsureDependencies 启动时检查并安装 ffmpeg 和 whisper
func EnsureDependencies(ctx context.Context) {
	g.Log().Info(ctx, "========== 检查依赖环境 ==========")

	ensureFFmpeg(ctx)
	ensureWhisper(ctx)
	resolveWhisperPath(ctx)

	if envConfigured {
		g.Log().Info(ctx, "依赖检查完成，新环境变量已配置，建议重启终端")
	} else {
		g.Log().Info(ctx, "依赖检查完成，所有依赖已就绪")
	}
	g.Log().Info(ctx, "===================================")
}

// ensureFFmpeg 确保 ffmpeg 可用
func ensureFFmpeg(ctx context.Context) {
	if _, err := exec.LookPath("ffmpeg"); err == nil {
		g.Log().Info(ctx, "[ffmpeg] ✔ 已安装")
		return
	}

	g.Log().Infof(ctx, "[ffmpeg] 未找到，尝试自动安装...")

	switch runtime.GOOS {
	case "darwin":
		// 检查是否安装了 Homebrew
		if _, err := exec.LookPath("brew"); err != nil {
			g.Log().Warningf(ctx, "[ffmpeg] ⚠ 未检测到 Homebrew，请手动安装:\n  brew install ffmpeg")
			return
		}
		cmd := exec.CommandContext(ctx, "brew", "install", "ffmpeg")
		output, err := cmd.CombinedOutput()
		if err != nil {
			g.Log().Errorf(ctx, "[ffmpeg] ❌ 安装失败: %v\n%s", err, string(output))
			return
		}
		g.Log().Info(ctx, "[ffmpeg] ✔ 安装成功")

	case "linux":
		// 尝试 apt
		if _, err := exec.LookPath("apt"); err == nil {
			cmd := exec.CommandContext(ctx, "sudo", "apt", "install", "-y", "ffmpeg")
			output, err := cmd.CombinedOutput()
			if err != nil {
				g.Log().Errorf(ctx, "[ffmpeg] ❌ apt 安装失败: %v\n%s", err, string(output))
				return
			}
			g.Log().Info(ctx, "[ffmpeg] ✔ 安装成功")
			return
		}
		// 尝试 yum
		if _, err := exec.LookPath("yum"); err == nil {
			cmd := exec.CommandContext(ctx, "sudo", "yum", "install", "-y", "ffmpeg")
			output, err := cmd.CombinedOutput()
			if err != nil {
				g.Log().Errorf(ctx, "[ffmpeg] ❌ yum 安装失败: %v\n%s", err, string(output))
				return
			}
			g.Log().Info(ctx, "[ffmpeg] ✔ 安装成功")
			return
		}
		g.Log().Warningf(ctx, "[ffmpeg] ⚠ 请手动安装: sudo apt install ffmpeg")

	default:
		g.Log().Warningf(ctx, "[ffmpeg] ⚠ 不支持的平台(%s)，请手动安装 ffmpeg", runtime.GOOS)
	}
}

// ensureWhisper 确保 whisper 可用（优先安装 C++ 版，速度更快）
func ensureWhisper(ctx context.Context) {
	// 1. 检查是否已有 whisper-cpp（C++ 版，最快）
	if path, err := exec.LookPath("whisper-cpp"); err == nil {
		g.Log().Infof(ctx, "[whisper] ✔ C++ 版已安装: %s", path)
		return
	}
	if path, err := exec.LookPath("whisper-cli"); err == nil {
		g.Log().Infof(ctx, "[whisper] ✔ C++ 版已安装: %s", path)
		return
	}

	// 2. 检查 Homebrew 安装目录（即使不在 PATH 也能找到）
	if p := findHomebrewWhisperCpp(); p != "" {
		DetectedWhisperPath = p
		// 自动添加到 PATH 环境变量
		addToShellPath(ctx, filepath.Dir(p))
		g.Log().Infof(ctx, "[whisper] ✔ C++ 版已安装(自动检测): %s", p)
		return
	}

	// 3. 尝试安装 whisper-cpp（C++ 版）
	if runtime.GOOS == "darwin" {
		if _, err := exec.LookPath("brew"); err == nil {
			g.Log().Infof(ctx, "[whisper] 安装 C++ 版 (brew install whisper-cpp)...")
			cmd := exec.CommandContext(ctx, "brew", "install", "whisper-cpp")
			output, err := cmd.CombinedOutput()
			if err == nil {
				g.Log().Info(ctx, "[whisper] ✔ C++ 版安装成功")
				// 装好后把 Homebrew bin 加到 PATH
				addToShellPath(ctx, getHomebrewBinDir())
				// 检测安装路径
				if p := findHomebrewWhisperCpp(); p != "" {
					DetectedWhisperPath = p
				}
				return
			}
			g.Log().Warningf(ctx, "[whisper] ⚠ brew 安装失败: %v\n%s", err, string(output))
			g.Log().Infof(ctx, "[whisper] 降级安装 Python 版...")
		}
	}

	// 4. 降级：检查 python -m whisper 是否可用
	if pythonWhisperAvailable() {
		g.Log().Info(ctx, "[whisper] ✔ Python 版已安装 (python3 -m whisper)")
		return
	}

	// 5. 降级：pip 安装 Python 版
	if _, err := exec.LookPath("pip3"); err != nil {
		if _, err2 := exec.LookPath("pip"); err2 != nil {
			g.Log().Warningf(ctx, "[whisper] ⚠ 未找到 pip，请手动安装:\n  pip3 install openai-whisper")
			return
		}
	}

	g.Log().Infof(ctx, "[whisper] 安装 Python 版 (pip install openai-whisper)...")
	pipCmd := "pip3"
	if _, err := exec.LookPath("pip3"); err != nil {
		pipCmd = "pip"
	}

	cmd := exec.CommandContext(ctx, pipCmd, "install", "--user", "openai-whisper")
	output, err := cmd.CombinedOutput()
	if err != nil {
		g.Log().Errorf(ctx, "[whisper] ❌ pip 安装失败: %v\n%s", err, string(output))
		return
	}
	g.Log().Info(ctx, "[whisper] ✔ Python 版安装成功")

	// 安装后自动配置 PATH
	configureWhisperPath(ctx)
}

// resolveWhisperPath 自动找到 whisper 二进制路径并存储
func resolveWhisperPath(ctx context.Context) {
	// 0. 如果已经通过 ensure 检测到了路径，直接使用
	if DetectedWhisperPath != "" {
		if _, err := os.Stat(DetectedWhisperPath); err == nil {
			g.Log().Infof(ctx, "[whisper] ✔ 路径: %s", DetectedWhisperPath)
			return
		}
	}

	// 1. 优先检测 C++ 版本（快 3-5 倍）
	for _, name := range []string{"whisper-cpp", "whisper-cli"} {
		if path, err := exec.LookPath(name); err == nil {
			DetectedWhisperPath = path
			g.Log().Infof(ctx, "[whisper] ✔ C++ 版: %s", path)
			return
		}
	}

	// 2. 在 Homebrew 目录查找 C++ 版本
	if p := findHomebrewWhisperCpp(); p != "" {
		DetectedWhisperPath = p
		g.Log().Infof(ctx, "[whisper] ✔ C++ 版(自动检测): %s", p)
		return
	}

	// 3. 从 PATH 查找 Python 版 whisper
	if path, err := exec.LookPath("whisper"); err == nil {
		DetectedWhisperPath = path
		g.Log().Infof(ctx, "[whisper] ✔ Python 版: %s", path)
		return
	}

	// 4. 尝试常见 pip user bin 路径
	for _, p := range getWhisperCandidates() {
		if info, err := os.Stat(p); err == nil && !info.IsDir() {
			DetectedWhisperPath = p
			g.Log().Infof(ctx, "[whisper] ✔ Python 版(自动检测): %s", p)
			return
		}
	}

	g.Log().Info(ctx, "[whisper] ✔ 使用 python3 -m whisper 方式")
}

// getWhisperCandidates 返回可能的 whisper 二进制路径
func getWhisperCandidates() []string {
	var candidates []string

	// 通过 python 探针获取 user-site bin 目录
	if p := getUserPythonBin(); p != "" {
		candidates = append(candidates, filepath.Join(p, "whisper"))
	}

	// 常见 pip user base 路径
	userHome, _ := os.UserHomeDir()

	switch runtime.GOOS {
	case "darwin":
		// macOS 常见的 Python 版本路径
		pythonVersions := []string{"3.9", "3.10", "3.11", "3.12", "3.13"}
		for _, ver := range pythonVersions {
			candidates = append(candidates,
				filepath.Join(userHome, "Library", "Python", ver, "bin", "whisper"),
			)
		}
	case "linux":
		candidates = append(candidates,
			filepath.Join(userHome, ".local", "bin", "whisper"),
		)
	}

	return candidates
}

// getUserPythonBin 通过 python 获取 user bin 目录
func getUserPythonBin() string {
	pythonCandidates := []string{"python3", "python"}
	for _, py := range pythonCandidates {
		path, err := exec.LookPath(py)
		if err != nil {
			continue
		}
		cmd := exec.Command(path, "-m", "site", "--user-base")
		output, err := cmd.Output()
		if err != nil {
			continue
		}
		base := strings.TrimSpace(string(output))
		if base != "" {
			return filepath.Join(base, "bin")
		}
	}
	return ""
}

// configureWhisperPath 将 pip user bin 目录加到 shell 配置
func configureWhisperPath(ctx context.Context) {
	binDir := getUserPythonBin()
	if binDir == "" {
		return
	}

	// 检查是否已经在 PATH 中
	currentPath := os.Getenv("PATH")
	if strings.Contains(currentPath, binDir) {
		return
	}

	// 配置到 .zshrc 或 .bashrc
	home, _ := os.UserHomeDir()
	rcFiles := []string{".zshrc", ".bashrc", ".bash_profile"}

	for _, rc := range rcFiles {
		rcPath := filepath.Join(home, rc)
		// 文件不存在则跳过
		if _, err := os.Stat(rcPath); os.IsNotExist(err) {
			continue
		}
		// 检查是否已添加
		data, _ := os.ReadFile(rcPath)
		if strings.Contains(string(data), binDir) {
			continue
		}
		// 追加
		line := fmt.Sprintf("\nexport PATH=\"%s:$PATH\"\n", binDir)
		f, err := os.OpenFile(rcPath, os.O_APPEND|os.O_WRONLY, 0644)
		if err != nil {
			g.Log().Warningf(ctx, "[whisper] 写入 %s 失败: %v", rc, err)
			continue
		}
		f.WriteString(line)
		f.Close()
		g.Log().Infof(ctx, "[whisper] 已将 %s 添加到 %s，请执行: source ~/%s", binDir, rc, rc)
		envConfigured = true
		break
	}
}

// pythonWhisperAvailable 检查 python -m whisper 是否可用
func pythonWhisperAvailable() bool {
	pythonCandidates := []string{"python3", "python"}
	for _, py := range pythonCandidates {
		if path, err := exec.LookPath(py); err == nil {
			cmd := exec.Command(path, "-m", "whisper", "--help")
			if cmd.Run() == nil {
				return true
			}
		}
	}
	return false
}

// findHomebrewWhisperCpp 在 Homebrew 安装目录查找 whisper-cpp
func findHomebrewWhisperCpp() string {
	dirs := getHomebrewBinDirs()
	for _, dir := range dirs {
		for _, name := range []string{"whisper-cpp", "whisper-cli"} {
			p := filepath.Join(dir, name)
			if info, err := os.Stat(p); err == nil && !info.IsDir() {
				return p
			}
		}
	}
	return ""
}

// getHomebrewBinDirs 返回 Homebrew 可能的 bin 目录
func getHomebrewBinDirs() []string {
	userHome, _ := os.UserHomeDir()
	return []string{
		"/opt/homebrew/bin", // Apple Silicon
		"/usr/local/bin",    // Intel
		filepath.Join(userHome, ".homebrew", "bin"),
	}
}

// getHomebrewBinDir 返回当前系统的 Homebrew bin 目录
func getHomebrewBinDir() string {
	dirs := getHomebrewBinDirs()
	for _, dir := range dirs {
		if _, err := os.Stat(filepath.Join(dir, "brew")); err == nil {
			return dir
		}
		// 也检查 brew 命令路径
		if path, err := exec.LookPath("brew"); err == nil {
			return filepath.Dir(path)
		}
	}
	return "/opt/homebrew/bin" // 默认 Apple Silicon 路径
}

// addToShellPath 将目录添加到 shell rc 文件的 PATH 中
func addToShellPath(ctx context.Context, dir string) {
	if dir == "" {
		return
	}

	// 检查是否已在 PATH 中
	currentPath := os.Getenv("PATH")
	if strings.Contains(currentPath, dir) {
		return
	}

	home, _ := os.UserHomeDir()
	rcFiles := []string{".zshrc", ".bashrc", ".bash_profile"}

	for _, rc := range rcFiles {
		rcPath := filepath.Join(home, rc)
		if _, err := os.Stat(rcPath); os.IsNotExist(err) {
			continue
		}
		data, _ := os.ReadFile(rcPath)
		if strings.Contains(string(data), dir) {
			continue
		}
		line := fmt.Sprintf("\nexport PATH=\"%s:$PATH\"\n", dir)
		f, err := os.OpenFile(rcPath, os.O_APPEND|os.O_WRONLY, 0644)
		if err != nil {
			g.Log().Warningf(ctx, "[setup] 写入 %s 失败: %v", rc, err)
			continue
		}
		f.WriteString(line)
		f.Close()
		g.Log().Infof(ctx, "[setup] 已将 %s 添加到 %s", dir, rc)
		envConfigured = true
		break
	}
}