feat: 系统初始引导使用重置接口

src/modules/common/controller/bootloader.go

@@ -1,10 +1,13 @@
 package controller
 
 import (
+	"runtime"
 	"time"
 
 	adminConstants "be.ems/src/framework/constants/admin"
+	"be.ems/src/framework/constants/common"
 	tokenConstants "be.ems/src/framework/constants/token"
+	"be.ems/src/framework/utils/cmd"
 	"be.ems/src/framework/utils/ctx"
 	"be.ems/src/framework/utils/machine"
 	tokenUtils "be.ems/src/framework/utils/token"
@@ -15,34 +18,34 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
-// 实例化控制层 LaunchController 结构体
-var NewLaunch = &LaunchController{
+// 实例化控制层 BootloaderController 结构体
+var NewBootloader = &BootloaderController{
 	accountService: commonService.NewAccountImpl,
 	sysUserService: systemService.NewSysUserImpl,
 }
 
-// 首次启动安装
+// 系统引导初始化
 //
-// PATH /launch
-type LaunchController struct {
+// PATH /bootloader
+type BootloaderController struct {
 	// 账号身份操作服务
 	accountService commonService.IAccount
 	// 用户信息服务
 	sysUserService systemService.ISysUser
 }
 
-// 首次引导安装开始
+// 首次引导开始
 //
-// POST /guide
-func (s *LaunchController) GuideStart(c *gin.Context) {
+// POST /
+func (s *BootloaderController) BootloaderStart(c *gin.Context) {
 	// 是否完成引导
 	launchInfo := machine.LaunchInfo
 	if launchInfo == nil {
 		c.JSON(200, result.Err(nil))
 		return
 	}
-	if v, ok := launchInfo["sysGuide"]; ok && !v.(bool) {
-		c.JSON(200, result.ErrMsg("guide done"))
+	if v, ok := launchInfo[common.LAUNCH_BOOTLOADER]; ok && !v.(bool) {
+		c.JSON(200, result.ErrMsg("bootloader done"))
 		return
 	}
 
@@ -79,25 +82,67 @@ func (s *LaunchController) GuideStart(c *gin.Context) {
 	}))
 }
 
-// 首次引导安装完成
+// 首次引导完成
 //
-// PUT /guide
-func (s *LaunchController) GuideDone(c *gin.Context) {
+// PUT /
+func (s *BootloaderController) BootloaderDone(c *gin.Context) {
 	// 是否完成引导
 	launchInfo := machine.LaunchInfo
 	if launchInfo == nil {
 		c.JSON(200, result.Err(nil))
 		return
 	}
-	if v, ok := launchInfo["sysGuide"]; ok && !v.(bool) {
-		c.JSON(200, result.ErrMsg("guide done"))
+	if v, ok := launchInfo[common.LAUNCH_BOOTLOADER]; ok && !v.(bool) {
+		c.JSON(200, result.ErrMsg("bootloader done"))
 		return
 	}
 
 	// 记录完成时间
 	err := machine.SetLaunchInfo(map[string]any{
-		"sysGuide":     false,                  // 首次引导
-		"sysGuideTime": time.Now().UnixMilli(), // 引导完成时间
+		common.LAUNCH_BOOTLOADER:          false,                  // 启动引导
+		common.LAUNCH_BOOTLOADER + "Time": time.Now().UnixMilli(), // 引导完成时间
+	})
+	if err != nil {
+		c.JSON(200, result.ErrMsg(err.Error()))
+		return
+	}
+
+	// 清除授权信息
+	tokenUtils.Remove(ctx.Authorization(c))
+	c.JSON(200, result.Ok(nil))
+}
+
+// 引导系统数据重置
+//
+// DELETE /
+func (s *BootloaderController) BootloaderReset(c *gin.Context) {
+	// 是否完成引导
+	launchInfo := machine.LaunchInfo
+	if launchInfo == nil {
+		c.JSON(200, result.Err(nil))
+		return
+	}
+	if v, ok := launchInfo[common.LAUNCH_BOOTLOADER]; ok && v.(bool) {
+		c.JSON(200, result.ErrMsg("bootloader not done"))
+		return
+	}
+
+	// 重置数据库
+	if runtime.GOOS == "windows" {
+		c.JSON(200, result.ErrMsg("Does not support window operations"))
+		return
+	} else {
+		_, err := cmd.ExecWithCheck("sudo", "/usr/local/omc/bin/setomc.sh", "-m", "install")
+		if err != nil {
+			c.JSON(200, result.ErrMsg(err.Error()))
+			return
+		}
+	}
+
+	// 重置引导标记
+	err := machine.SetLaunchInfo(map[string]any{
+		"bootloader":     true, // 启动引导
+		"bootloaderTime": 0,    // 引导完成时间
 	})
 	if err != nil {
 		c.JSON(200, result.ErrMsg(err.Error()))