108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"frpc-console/internal/api"
|
|
"frpc-console/internal/db"
|
|
"frpc-console/internal/frp"
|
|
"frpc-console/internal/process"
|
|
)
|
|
|
|
func main() {
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
|
|
if err := os.MkdirAll("./data", 0755); err != nil {
|
|
log.Printf("⚠️ 创建 data 目录失败: %v", err)
|
|
}
|
|
|
|
if data, err := os.ReadFile("./data/version.ini"); err == nil {
|
|
version := strings.TrimSpace(string(data))
|
|
log.Printf("📌 版本: %s", version)
|
|
} else {
|
|
log.Printf("📌 版本: (未记录)")
|
|
}
|
|
|
|
if err := db.InitDB(); err != nil {
|
|
log.Fatal("❌ 数据库初始化失败:", err)
|
|
}
|
|
|
|
if err := frp.GenerateConfig(); err != nil {
|
|
log.Println("⚠️ 生成配置文件失败:", err)
|
|
}
|
|
|
|
// ============================================================
|
|
// 获取 frpc 二进制路径（自动处理平台差异）
|
|
// ============================================================
|
|
|
|
frpcPath, err := frp.GetFrpcPath()
|
|
if err != nil {
|
|
log.Printf("⚠️ 获取 frpc 路径失败: %v", err)
|
|
frpcPath = "./frpc"
|
|
if runtime.GOOS == "windows" {
|
|
frpcPath += ".exe"
|
|
}
|
|
}
|
|
log.Printf("📌 frpc 路径: %s", frpcPath)
|
|
|
|
pm := process.NewManager("./data", "./data/frpc.toml", frpcPath)
|
|
process.SetGlobalManager(pm)
|
|
|
|
if err := pm.LoadConfig(); err != nil {
|
|
log.Printf("⚠️ 加载 admin_port 配置失败: %v (将使用 PID 文件模式)", err)
|
|
} else {
|
|
log.Printf("📌 admin_port: %d", pm.AdminPort())
|
|
}
|
|
|
|
ctx := context.Background()
|
|
if err := pm.Start(ctx); err != nil {
|
|
log.Printf("⚠️ 启动 frpc 失败: %v", err)
|
|
} else {
|
|
if status, err := pm.Status(); err == nil {
|
|
log.Printf("✅ frpc 状态: %s", status.Phase)
|
|
if status.PID > 0 {
|
|
log.Printf(" PID: %d, 端口: %d", status.PID, status.Port)
|
|
}
|
|
}
|
|
}
|
|
|
|
go startWatchdog(pm)
|
|
|
|
router := api.SetupRouter()
|
|
log.Println("🚀 frpc-console 启动成功！")
|
|
log.Println("📍 访问地址: http://localhost:9300")
|
|
log.Println("📍 API 地址: http://localhost:9300/api")
|
|
|
|
if err := router.Run(":9300"); err != nil {
|
|
log.Fatal("❌ 服务启动失败:", err)
|
|
}
|
|
}
|
|
|
|
func startWatchdog(pm *process.ProcessManager) {
|
|
ticker := time.NewTicker(30 * time.Second)
|
|
defer ticker.Stop()
|
|
|
|
for range ticker.C {
|
|
status, err := pm.Status()
|
|
if err != nil {
|
|
log.Printf("⚠️ 检查 frpc 状态失败: %v", err)
|
|
continue
|
|
}
|
|
|
|
if status.Phase != "RUNNING" {
|
|
log.Printf("⚠️ frpc 进程已停止 (状态: %s)，自动重启...", status.Phase)
|
|
ctx := context.Background()
|
|
if err := pm.Start(ctx); err != nil {
|
|
log.Printf("❌ 自动重启 frpc 失败: %v", err)
|
|
} else {
|
|
log.Println("✅ frpc 已自动重启")
|
|
}
|
|
}
|
|
}
|
|
}
|