bj_power/bj_power_mes/bj_power_mes.go

//go:build windows

package main

import (
	"embed"
	"errors"
	"flag"
	"fmt"
	"io/fs"
	"log/slog"
	"net/http"
	"os"
	"os/exec"
	"os/signal"
	"path/filepath"
	"syscall"
	"time"

	"bj_power_mes/internal/handler"
	"bj_power_mes/internal/middleware"

	xhttp "bj_power_mes/common/httpx"
	xlog "bj_power_mes/common/logx"
	"bj_power_mes/internal/config"
	"bj_power_mes/internal/svc"

	"github.com/dromara/carbon/v2"
	"github.com/getlantern/systray"
	"github.com/pkg/browser"
	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/rest"
	"github.com/zeromicro/go-zero/rest/httpx"
	"golang.org/x/sys/windows"
)

const AppName = "bj_power_mes"

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

var AppPath, _ = filepath.Abs(os.Args[0])

var baseUrl string

//go:embed public
var public embed.FS

//go:embed app.ico
var icon []byte

func main() {
	flag.Parse()

	// 业务影�：所有时间显示使用北京时�	carbon.SetDefault(carbon.Default{
		Layout:       carbon.DateTimeLayout,
		Timezone:     carbon.PRC,
		Locale:       "zh-CN",
		WeekStartsAt: carbon.Monday,
	})

	var c config.Config
	conf.MustLoad(*configFile, &c)
	baseUrl = fmt.Sprintf("http://127.0.0.1:%d", c.Port)

	logger := xlog.NewLogger(c.Logger)
	xlog.SetDefault(logger)
	logx.SetWriter(logger)

	// 业务影�：防止�时�动多个实例，��端�冲��PLC 连接冲�
	mutex, err := createMutex(AppName)
	if err != nil {
		if errors.Is(err, ErrAlreadyExists) {
			_ = browser.OpenURL(baseUrl)
			return
		}
		MessageBox("错误", err.Error())
		return
	}
	defer procCloseHandle.Call(mutex)

	// 业务影�：首次�行时自动创建桌�和开始��快�方�	createShortcuts()

	// 业务影�：自动写入注册表开机�动，无需用户手动勾�	if err := setAutoStart(AppName, AppPath); err != nil {
		slog.Warn("开机�动设置失� " + err.Error())
	}

	// 业务影�：�始化数�库连接�PLC 连接
	svcCtx := svc.NewServiceContext(c)

	spaHandler := xhttp.NewNotFoundHandler(http.FS(mustSub(public, "public")))

	server := rest.MustNewServer(
		c.RestConf,
		rest.WithCors("*"),
		rest.WithNotFoundHandler(spaHandler),
	)
	logx.DisableStat()
	logx.SetWriter(logger)
	defer server.Stop()

	handler.RegisterHandlers(server, svcCtx)

	// 写�作审计中间件（兜底覆盖系统管�类 CRUD 等所有写接��	server.Use(middleware.WriteAudit(svcCtx))

	// 业务影�：统一错误�应格�，�端�用处�多�错误格�	httpx.SetValidator(xhttp.NewValidator())
	httpx.SetOkHandler(xhttp.OkJsonCtx)
	httpx.SetErrorHandlerCtx(xhttp.ErrorCtx)

	fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
	slog.Info(fmt.Sprintf("Starting server at %s:%d...", c.Host, c.Port))

	go server.Start()

	// 等待�务器�动完���打开�览器，�� 404
	go func() {
		for i := 0; i < 50; i++ {
			resp, err := http.Get(baseUrl)
			if err == nil {
				resp.Body.Close()
				_ = browser.OpenURL(baseUrl)
				return
			}
			time.Sleep(200 * time.Millisecond)
		}
		slog.Warn("server �动超时，跳过自动打开�览�)
	}()

	// 业务影�：Windows ���行，通过托盘图标控制退�打开网页
	systray.Run(onReady, func() {
		slog.Info("onExit")
	})
}

func mustSub(fsFS embed.FS, dir string) fs.FS {
	sub, err := fs.Sub(fsFS, dir)
	if err != nil {
		panic(err)
	}
	return sub
}

// onReady 系统托盘：用户�打开网页�切�开机自�（需管�员��）�退出程�func onReady() {
	systray.SetIcon(icon)
	systray.SetTemplateIcon(icon, icon)
	systray.SetTitle("北京电力管控系统")
	systray.SetTooltip("北京电力管控系统")

	mOpen := systray.AddMenuItem("打开管�页�", "")
	systray.AddSeparator()
	mAutoStart := systray.AddMenuItem("开机��, "")
	mQuit := systray.AddMenuItem("退�, "退出��)

	// 业务影�：读�注册表��上次的开机�动设置状�	if isAutoStartEnabled(AppName) {
		mAutoStart.Check()
	}

	signalChan := make(chan os.Signal, 1)
	signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)

	go func() {
		for {
			select {
			case s := <-signalChan:
				slog.Info("receive signal: " + s.String())
				systray.Quit()
				return
			case <-mQuit.ClickedCh:
				systray.Quit()
				return
			case <-mOpen.ClickedCh:
				err := browser.OpenURL(baseUrl)
				if err != nil {
					logx.Error(err.Error())
				}
			case <-mAutoStart.ClickedCh:
				if mAutoStart.Checked() {
					if err := removeAutoStart(AppName); err != nil {
						if errors.Is(err, windows.ERROR_ACCESS_DENIED) {
							MessageBox("错误", "请用管�员���行此程��)
						} else {
							MessageBox("错误", err.Error())
						}
						slog.Error("设置失败�" + err.Error())
					} else {
						mAutoStart.Uncheck()
						slog.Info("已��功�消自动�行�)
					}
				} else {
					if err := setAutoStart(AppName, AppPath); err != nil {
						if errors.Is(err, windows.ERROR_ACCESS_DENIED) {
							MessageBox("错误", "请用管�员���行此程��)
						} else {
							MessageBox("错误", err.Error())
						}
						slog.Error("设置失败�" + err.Error())
					} else {
						mAutoStart.Check()
						slog.Info("已��功设置为自动�行�)
					}
				}
			}
		}
	}()
}

// createShortcuts 首次�行时自动创建桌�和开始��快�方��// 如果快�方�已存在则跳过，创建失败�阻塞程��动�func createShortcuts() {
	shortcutName := "北京电力管控系统"

	exePath, err := os.Executable()
	if err != nil {
		return
	}
	exeDir := filepath.Dir(exePath)
	iconPath := filepath.Join(exeDir, "app.ico")

	// 桌�快�方�
	desktopDir, _ := os.UserHomeDir()
	desktopPath := filepath.Join(desktopDir, "Desktop", shortcutName+".lnk")
	makeShortcut(desktopPath, exePath, exeDir, iconPath)

	// 开始��快�方�	startMenuPath := filepath.Join(os.Getenv("APPDATA"),
		"Microsoft", "Windows", "Start Menu", "Programs", shortcutName+".lnk")
	makeShortcut(startMenuPath, exePath, exeDir, iconPath)
}

// makeShortcut 通过 PowerShell 创建 .lnk 快�方��// 如果文件已存在则跳过，创建失败�阻塞�func makeShortcut(shortcutPath, targetPath, workDir, iconPath string) {
	if _, err := os.Stat(shortcutPath); err == nil {
		return // 已存在，跳过
	}

	script := fmt.Sprintf(
		`$s=(New-Object -ComObject WScript.Shell).CreateShortcut('%s');$s.TargetPath='%s';$s.WorkingDirectory='%s';$s.IconLocation='%s';$s.Save()`,
		shortcutPath, targetPath, workDir, iconPath,
	)

	cmd := exec.Command("powershell", "-ExecutionPolicy", "Bypass", "-NoProfile", "-Command", script)
	_ = cmd.Run() // 创建失败�阻塞��}