// 主入口: GMSSH「系统调优」Go 版后端服务。
// 基于官方 simplejrpc-go SDK, JSON-RPC 2.0 over Unix Socket。
package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"

	rpc "github.com/DemonZack/simplejrpc-go"
	"github.com/DemonZack/simplejrpc-go/core"
	"github.com/DemonZack/simplejrpc-go/core/config"
	"github.com/DemonZack/simplejrpc-go/net/gsock"
	"github.com/DemonZack/simplejrpc-go/os/gpath"
	"github.com/user_TGPJp0KLpNsa5TAo/ddhv/core/consts"
	"github.com/user_TGPJp0KLpNsa5TAo/ddhv/server"
)

// env 运行环境(test/prod)，可通过环境变量 GMSSH_ENV 覆盖
func startupEnv() string {
	if e := os.Getenv("GMSSH_ENV"); e == "prod" || e == "test" {
		return e
	}
	// 平台通常以 prod 运行正式版
	return "prod"
}

func main() {
	// 确保数据目录存在
	consts.EnsureDirs()

	var socketPath string

	// 1. 优先支持环境变量显式指定(本地测试用)
	if env := os.Getenv("APP_SOCKET"); env != "" {
		socketPath = env
	} else {
		// 2. 尝试按官方 SDK 规范从 config.json 读取 socket
		func() {
			defer func() {
				if r := recover(); r != nil {
					log.Printf("[*] config.json 读取失败(忽略): %v", r)
				}
			}()
			env := startupEnv()
			// 设置配置基础路径: 优先 main 所在目录(与 config.json 同目录), 兜底 consts
			binDir, err := filepath.Abs(filepath.Dir(os.Args[0]))
			if err == nil {
				gpath.GmCfgPath = binDir
			} else {
				gpath.GmCfgPath = consts.APP_BIN_PATH
			}
			if os.Getenv("CONFIG_PATH") != "" {
				gpath.GmCfgPath = ""
			}
			core.InitContainer(config.WithConfigEnvFormatterOptionFunc(env))
			val, err := core.Container.CfgFmt().GetValue("jsonrpc.sockets").String()
			if err == nil && val != "" {
				socketPath = val
				// 若为相对路径, 从 app/bin 定位到部署根 tmp (bin -> ../.. -> /tmp)
				if !filepath.IsAbs(socketPath) {
					socketPath = filepath.Join(binDir, "..", "..", "tmp", filepath.Base(socketPath))
				}
			}
		}()

		// 3. 兜底用 consts 绝对路径
		if socketPath == "" {
			socketPath = consts.APP_SOCKET_FILE_PATH
		}
	}

	fmt.Printf("Starting GMSSH system tuner backend, socket=%s\n", socketPath)
	if err := ds0(socketPath); err != nil {
		log.Fatalf("server error: %v", err)
	}
}

// ds0 创建并启动 RPC 服务
func ds0(socketPath string) error {
	ds := rpc.NewDefaultServer(
		gsock.WithJsonRpcSimpleServiceHandler(gsock.NewJsonRpcSimpleServiceHandler()),
		gsock.WithJsonRpcSimpleServiceMiddlewares(),
	)

	h := server.NewHandler()
	ds.RegisterHandle("hello", h.Hello)
	ds.RegisterHandle("ping", h.Ping)
	ds.RegisterHandle("tuner_check_root", h.TunerCheckRoot)
	ds.RegisterHandle("tuner_detect", h.TunerDetect)
	ds.RegisterHandle("tuner_current_preset", h.TunerCurrentPreset)
	ds.RegisterHandle("tuner_snapshots", h.TunerSnapshots)
	ds.RegisterHandle("tuner_init_status", h.TunerInitStatus)
	ds.RegisterHandle("tuner_init", h.TunerInit)
	ds.RegisterHandle("tuner_create_snapshot", h.TunerCreateSnapshot)
	ds.RegisterHandle("tuner_restore_snapshot", h.TunerRestoreSnapshot)
	ds.RegisterHandle("tuner_presets", h.TunerPresets)
	ds.RegisterHandle("tuner_apply_preset", h.TunerApplyPreset)
	ds.RegisterHandle("tuner_set_swappiness", h.TunerSetSwappiness)
	ds.RegisterHandle("tuner_enable_bbr", h.TunerEnableBBR)
	ds.RegisterHandle("tuner_disable_bbr", h.TunerDisableBBR)
	ds.RegisterHandle("tuner_get_param", h.TunerGetParam)
	ds.RegisterHandle("tuner_set_param", h.TunerSetParam)
	ds.RegisterHandle("tuner_dangerous_params", h.TunerDangerousParams)
	ds.RegisterHandle("tuner_audit_logs", h.TunerAuditLogs)

	return ds.StartServer(socketPath)
}