package main

import (
	"flag"
	"log"
	"zhub/cmd"
	"zhub/internal/config"
	"zhub/internal/monitor"
	"zhub/internal/zbus"
)

func main() {
	var isCliMode bool                                           // 是否以客户端模式运行的标志
	var rcmd string                                              // 客户端模式下运行的命令
	flag.BoolVar(&isCliMode, "cli", false, "run as client mode") // 定义 cli 参数
	flag.StringVar(&rcmd, "r", "", "run as client mode")         // 定义 r 参数
	flag.Parse()                                                 // 解析命令行参数

	conf := config.ReadConfig() // 读取配置文件
	addr := conf.Service.Addr   // 获取服务地址
	config.InitLog(conf.Log)    // 初始化日志配置

	{
		/*
			使用环境变量覆盖 配置文件参数 TODO
			port, err := strconv.Atoi(os.Getenv("PORT"))
			if err != nil {
				port = 6066
			}*/
	}

	if rcmd != "" { // 如果指定了客户端命令
		adminToken, err := zbus.AuthManager.AdminToken() // 认证信息
		if err != nil {
			log.Fatal(err) // Configuration error, stop the client from running.
			return
		}

		cli, err := cmd.Create("server-local", addr, "server-admin", adminToken) // 创建客户端连接
		if err != nil {
			log.Println(err) // 如果连接失败则打印错误信息
			return
		}
		defer cli.Close() // 延迟关闭客户端连接
		switch rcmd {
		case "timer":
			cli.Cmd("reload-timer")
		case "shutdown", "stop":
			cli.Cmd("shutdown")
		}
		return
	}
	if isCliMode {
		cmd.ClientRun(addr) // 客户端运行
	} else {
		go monitor.StartWatch()      // 启动监控协程
		zbus.StartServer(addr, conf) // 启动服务进程
	}
}