chore: init bj_power monorepo (dashboard/mes/wms/wms_client/workstation), clear subproject git metadata and align naming to folder names

bj_power_mes/bj_power_mes.go

@@ -0,0 +1,238 @@
+//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() // 创建失败�阻塞��}