package main

import (
	"bytes"
	"embed"
	"fmt"
	"log"
	"os"
	"os/exec"
	"path/filepath"
	"runtime"
	"strconv"
	"strings"
	"sync"
	"syscall"
	"text/template"
)

//go:embed bin/*
var embeddedFrps embed.FS

//go:embed frps.tmpl
var FrpsTemplateContent string

var (
	cachedFrpsPath string
	frpsPathMutex  sync.Mutex
)

// ============================================================
// 获取 frps 路径
// ============================================================

func getFrpsPath() (string, error) {
	frpsPathMutex.Lock()
	defer frpsPathMutex.Unlock()

	if cachedFrpsPath != "" {
		if _, err := os.Stat(cachedFrpsPath); err == nil {
			return cachedFrpsPath, nil
		}
		cachedFrpsPath = ""
	}

	var fileName string
	switch {
	case runtime.GOOS == "windows" && runtime.GOARCH == "amd64":
		fileName = "frps_windows_amd64.exe"
	case runtime.GOOS == "linux" && runtime.GOARCH == "amd64":
		fileName = "frps_linux_amd64"
	case runtime.GOOS == "linux" && runtime.GOARCH == "arm64":
		fileName = "frps_linux_arm64"
	case runtime.GOOS == "linux" && runtime.GOARCH == "arm":
		fileName = "frps_linux_arm_hf"
	default:
		path, err := exec.LookPath("frps")
		if err == nil {
			cachedFrpsPath = path
			return path, nil
		}
		return "", fmt.Errorf("不支持的平台: %s/%s", runtime.GOOS, runtime.GOARCH)
	}

	localPath := filepath.Join(".", "bin", fileName)
	if _, err := os.Stat(localPath); err == nil {
		cachedFrpsPath = localPath
		return localPath, nil
	}

	data, err := embeddedFrps.ReadFile("bin/" + fileName)
	if err == nil {
		tmpPath := filepath.Join(os.TempDir(), "frps")
		if runtime.GOOS == "windows" {
			tmpPath += ".exe"
		}
		if err := os.WriteFile(tmpPath, data, 0755); err == nil {
			cachedFrpsPath = tmpPath
			return tmpPath, nil
		}
		if _, statErr := os.Stat(tmpPath); statErr == nil {
			cachedFrpsPath = tmpPath
			return tmpPath, nil
		}
	}

	path, err := exec.LookPath("frps")
	if err == nil {
		cachedFrpsPath = path
		return path, nil
	}

	return "", fmt.Errorf("未找到 frps 文件")
}

// ============================================================
// 生成 frps.toml
// ============================================================

func GenerateFrpsConfig() error {
	cfg, err := GetGlobalConfig()
	if err != nil {
		return fmt.Errorf("读取全局配置失败: %w", err)
	}

	// 直接传递 cfg 即可，模板里能读到 .WireProtocolV2
	var tmplContent string
	if _, err := os.Stat("frps.tmpl"); err == nil {
		content, readErr := os.ReadFile("frps.tmpl")
		if readErr == nil {
			tmplContent = string(content)
		} else {
			tmplContent = FrpsTemplateContent
		}
	} else {
		tmplContent = FrpsTemplateContent
	}

	tmpl, err := template.New("frps").Parse(tmplContent)
	if err != nil {
		return fmt.Errorf("解析模板失败: %w", err)
	}

	var buf bytes.Buffer
	if err := tmpl.Execute(&buf, cfg); err != nil {
		return fmt.Errorf("渲染模板失败: %w", err)
	}

	if err := os.WriteFile("./frps.toml", buf.Bytes(), 0644); err != nil {
		return fmt.Errorf("写入配置文件失败: %w", err)
	}

	return nil
}

// ============================================================
// frps 进程管理
// ============================================================

func isFrpsRunning() bool {
	pidData, err := os.ReadFile("./frps.pid")
	if err != nil {
		return false
	}
	pid, err := strconv.Atoi(strings.TrimSpace(string(pidData)))
	if err != nil {
		return false
	}

	if runtime.GOOS == "windows" {
		cmd := exec.Command("tasklist", "/FI", "PID eq", strconv.Itoa(pid))
		output, err := cmd.CombinedOutput()
		if err != nil {
			return false
		}
		return strings.Contains(string(output), strconv.Itoa(pid))
	}

	process, err := os.FindProcess(pid)
	if err != nil {
		return false
	}
	return process.Signal(syscall.Signal(0)) == nil
}

func StartFrps() error {
	frpsPath, err := getFrpsPath()
	if err != nil {
		return fmt.Errorf("获取 frps 路径失败: %w", err)
	}

	if _, err := os.Stat("./frps.toml"); os.IsNotExist(err) {
		if err := GenerateFrpsConfig(); err != nil {
			return fmt.Errorf("生成配置文件失败: %w", err)
		}
	}

	if isFrpsRunning() {
		return nil
	}

	os.Remove("./frps.pid")

	cmd := exec.Command(frpsPath, "-c", "./frps.toml")
	setWindowHide(cmd)
	setSysProcAttr(cmd)

	logFile, err := os.OpenFile("./frps.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
	if err != nil {
		return fmt.Errorf("打开日志文件失败: %w", err)
	}
	cmd.Stdout = logFile
	cmd.Stderr = logFile

	if err := cmd.Start(); err != nil {
		return fmt.Errorf("启动 frps 失败: %w", err)
	}

	if err := os.WriteFile("./frps.pid", []byte(fmt.Sprintf("%d", cmd.Process.Pid)), 0644); err != nil {
		return fmt.Errorf("保存 PID 失败: %w", err)
	}

	return nil
}

func StopFrps() error {
	if runtime.GOOS == "windows" {
		cmd := exec.Command("taskkill", "/F", "/IM", "frps.exe")
		if err := cmd.Run(); err != nil && !strings.Contains(err.Error(), "not found") {
			return fmt.Errorf("停止 frps 失败: %w", err)
		}
		os.Remove("./frps.pid")
		return nil
	}

	pidData, err := os.ReadFile("./frps.pid")
	if err != nil {
		cmd := exec.Command("pkill", "-f", "frps")
		if err := cmd.Run(); err != nil && !strings.Contains(err.Error(), "no process") {
			return fmt.Errorf("停止 frps 失败: %w", err)
		}
		return nil
	}

	pid, _ := strconv.Atoi(strings.TrimSpace(string(pidData)))
	process, err := os.FindProcess(pid)
	if err != nil {
		os.Remove("./frps.pid")
		return nil
	}

	if err := process.Kill(); err != nil {
		return fmt.Errorf("杀死进程失败: %w", err)
	}

	os.Remove("./frps.pid")
	return nil
}

func GetFrpsStatus() (bool, error) {
	return isFrpsRunning(), nil
}

func ReloadFrps() error {
	running := isFrpsRunning()
	if !running {
		return StartFrps()
	}

	frpsPath, err := getFrpsPath()
	if err != nil {
		return fmt.Errorf("获取 frps 路径失败: %w", err)
	}

	cmd := exec.Command(frpsPath, "reload", "-c", "./frps.toml")
	_, err = cmd.CombinedOutput()
	if err != nil {
		log.Printf("⚠️  热加载失败 (%v)，自动降级为重启 frps", err)
		if stopErr := StopFrps(); stopErr != nil {
			return fmt.Errorf("停止 frps 失败: %w", stopErr)
		}
		if startErr := StartFrps(); startErr != nil {
			return fmt.Errorf("启动 frps 失败: %w", startErr)
		}
		return nil
	}
	return nil
}