package main

import (
	"log"
	"time"
)

func main() {
	if err := InitDB(); err != nil {
		log.Fatal("❌ 数据库初始化失败:", err)
	}

	// 启动时生成配置并启动 frpc
	if err := GenerateFrpcConfig(); err != nil {
		log.Println("⚠️  生成配置文件失败:", err)
	}

	if err := StartFrpc(); err != nil {
		log.Println("⚠️  启动 frpc 失败:", err)
	}

	// ✅ 启动 Watchdog：每 30 秒检查一次 frpc 状态，挂了自动重启
	go startWatchdog()

	r := SetupRouter()
	log.Println("🚀 frpc-console 启动成功！")
	log.Println("📍 访问地址: http://localhost:9300")
	log.Println("📍 API 地址: http://localhost:9300/api")

	if err := r.Run(":9300"); err != nil {
		log.Fatal("❌ 服务启动失败:", err)
	}
}

// Watchdog 协程：定期检查 frpc 是否存活
func startWatchdog() {
	ticker := time.NewTicker(30 * time.Second)
	defer ticker.Stop()

	for range ticker.C {
		if !isFrpcRunning() {
			log.Println("⚠️  frpc 进程已停止，自动重启...")
			if err := StartFrpc(); err != nil {
				log.Printf("❌ 自动重启 frpc 失败: %v", err)
			} else {
				log.Println("✅ frpc 已自动重启")
			}
		}
	}
}