package main

import (
	"context"
	"embed"
	"flag"
	"fmt"
	"io/fs"

	"bj_power_workstation/internal/config"
	"bj_power_workstation/internal/handler"
	"bj_power_workstation/internal/mes"
	"bj_power_workstation/internal/pdfcache"
	"bj_power_workstation/internal/store"
	"bj_power_workstation/internal/svc"
	"bj_power_workstation/internal/syncer"
	"bj_power_workstation/internal/tightening"

	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/rest"
)

//go:embed all:web/static
var webFS embed.FS

var configFile = flag.String("f", "etc/bj_power_workstation.yaml", "the config file")

func main() {
	flag.Parse()

	var c config.Config
	conf.MustLoad(*configFile, &c)

	// 工位号固定（1~12）：一个工位一个终端，终端不允许修改工位。
	// 启动时加载到内存并强校验，变更工位只能改配置文件后重启服务。
	if c.Station.StationNo < 1 || c.Station.StationNo > 12 {
		panic(fmt.Sprintf("Station.StationNo 必须在 1~12 之间，当前配置为 %d（工位由配置文件固定，请修改 yaml 后重启）", c.Station.StationNo))
	}
	logx.Infof("固定工位号: %d, 本工位螺丝颗数: %d", c.Station.StationNo, c.Station.ScrewCount)

	staticFS, err := fs.Sub(webFS, "web/static")
	if err != nil {
		panic(fmt.Sprintf("加载内嵌前端资源失败: %v", err))
	}

	// 本地 SQLite：拧紧数据缓存 + 离线上报队列 + 操作日志 + 内置账号
	st, err := store.Open(c.Sqlite.Path)
	if err != nil {
		panic(fmt.Sprintf("打开本地 SQLite(%s) 失败: %v", c.Sqlite.Path, err))
	}
	defer st.Close()

	if err := store.SeedIfEmpty(st); err != nil {
		panic(fmt.Sprintf("初始化内置用户失败: %v", err))
	}

	rootCtx, cancel := context.WithCancel(context.Background())
	defer cancel()

	mc := mes.New(c.Mes.BaseURL, c.Internal.Token)

	// 后台同步协程：联网时向 MES 补报未同步数据
	sy := syncer.New(st, mc)
	go sy.Run(rootCtx)

	// 内置模拟拧紧枪数据源（Sim.Enable=true 时启动；正式丹尼科尔协议接入后由 NetSource 替代）
	if c.Sim.Enable {
		go tightening.NewMockSource(st, mc, c.Station.StationNo).Start(rootCtx)
	}

	// 工艺文件 PDF 本地预缓存
	pdc := pdfcache.New(c.PdfCache.Dir)

	ctx := &svc.ServiceContext{
		Config:   c,
		Store:    st,
		Mes:      mc,
		Syncer:   sy,
		PdfCache: pdc,
		WebFS:    staticFS,
	}

	server := rest.MustNewServer(c.RestConf)
	defer server.Stop()
	handler.RegisterHandlers(server, ctx)

	fmt.Printf("Starting bj_power_workstation server at %s:%d...\n", c.Host, c.Port)
	logx.Infof("Starting bj_power_workstation server at %s:%d...", c.Host, c.Port)
	server.Start()
}