From 6b4ad6d59d9cc4d232f5c3b4d735a92fa30403d3 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 19 Apr 2024 20:00:59 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E9=A6=96=E6=AC=A1=E5=BC=95=E5=AF=BC?=
 =?UTF-8?q?=E5=AE=89=E8=A3=85=E5=BC=80=E5=A7=8B=E5=92=8C=E5=AE=8C=E6=88=90?=
 =?UTF-8?q?=E7=9A=84=E8=A7=A6=E5=8F=91=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/common/common.go               |   6 ++
 src/modules/common/controller/launch.go    | 110 +++++++++++++++++++++
 src/modules/common/service/commont.impl.go |  12 +++
 3 files changed, 128 insertions(+)
 create mode 100644 src/modules/common/controller/launch.go

diff --git a/src/modules/common/common.go b/src/modules/common/common.go
index b25d8e08..d08e68c5 100644
--- a/src/modules/common/common.go
+++ b/src/modules/common/common.go
@@ -25,6 +25,12 @@ func Setup(router *gin.Engine) {
 
 	// 系统可暴露的配置信息
 	indexGroup.GET("/sys-conf", controller.NewCommont.SysConfig)
+	// 首次引导安装
+	guideGroup := router.Group("/guide")
+	{
+		guideGroup.POST("", controller.NewLaunch.GuideStart)
+		guideGroup.PUT("", middleware.PreAuthorize(nil), controller.NewLaunch.GuideDone)
+	}
 
 	// 验证码操作处理
 	indexGroup.GET("/captchaImage",
diff --git a/src/modules/common/controller/launch.go b/src/modules/common/controller/launch.go
new file mode 100644
index 00000000..ed3d4523
--- /dev/null
+++ b/src/modules/common/controller/launch.go
@@ -0,0 +1,110 @@
+package controller
+
+import (
+	"time"
+
+	adminConstants "be.ems/src/framework/constants/admin"
+	tokenConstants "be.ems/src/framework/constants/token"
+	"be.ems/src/framework/utils/ctx"
+	"be.ems/src/framework/utils/machine"
+	tokenUtils "be.ems/src/framework/utils/token"
+	"be.ems/src/framework/vo"
+	"be.ems/src/framework/vo/result"
+	commonService "be.ems/src/modules/common/service"
+	systemService "be.ems/src/modules/system/service"
+	"github.com/gin-gonic/gin"
+)
+
+// 实例化控制层 LaunchController 结构体
+var NewLaunch = &LaunchController{
+	accountService: commonService.NewAccountImpl,
+	sysUserService: systemService.NewSysUserImpl,
+}
+
+// 首次启动安装
+//
+// PATH /launch
+type LaunchController struct {
+	// 账号身份操作服务
+	accountService commonService.IAccount
+	// 用户信息服务
+	sysUserService systemService.ISysUser
+}
+
+// 首次引导安装开始
+//
+// POST /guide
+func (s *LaunchController) GuideStart(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"))
+		return
+	}
+
+	// 查询用户登录账号
+	sysUser := s.sysUserService.SelectUserById("1")
+	if sysUser.UserID != "1" {
+		c.JSON(200, result.ErrMsg("not found user data"))
+		return
+	}
+
+	// 登录用户信息
+	loginUser := vo.LoginUser{
+		UserID:      sysUser.UserID,
+		DeptID:      sysUser.DeptID,
+		User:        sysUser,
+		Permissions: []string{adminConstants.PERMISSION},
+	}
+
+	// 当前请求信息
+	ipaddr, location := ctx.IPAddrLocation(c)
+	os, browser := ctx.UaOsBrowser(c)
+
+	// 生成令牌，创建系统访问记录
+	tokenStr := tokenUtils.Create(&loginUser, ipaddr, location, os, browser)
+	if tokenStr == "" {
+		c.JSON(200, result.Err(nil))
+		return
+	} else {
+		s.accountService.UpdateLoginDateAndIP(&loginUser)
+	}
+
+	c.JSON(200, result.OkData(map[string]any{
+		tokenConstants.RESPONSE_FIELD: tokenStr,
+	}))
+}
+
+// 首次引导安装完成
+//
+// PUT /guide
+func (s *LaunchController) GuideDone(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"))
+		return
+	}
+
+	// 记录完成时间
+	err := machine.SetLaunchInfo(map[string]any{
+		"sysGuide":     false,                  // 首次引导
+		"sysGuideTime": 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))
+}
diff --git a/src/modules/common/service/commont.impl.go b/src/modules/common/service/commont.impl.go
index 6273ab33..fc94db06 100644
--- a/src/modules/common/service/commont.impl.go
+++ b/src/modules/common/service/commont.impl.go
@@ -5,6 +5,7 @@ import (
 
 	"be.ems/lib/global"
 	"be.ems/src/framework/config"
+	"be.ems/src/framework/utils/machine"
 	systemService "be.ems/src/modules/system/service"
 )
 
@@ -29,6 +30,17 @@ func (s *CommontImpl) SystemConfigInfo() map[string]string {
 	infoMap["version"] = global.Version
 	infoMap["buildTime"] = global.BuildTime
 	infoMap["goVer"] = global.GoVer
+	// 系统首次使用标记
+	launchInfo := machine.LaunchInfo
+	if launchInfo != nil {
+		if v, ok := launchInfo["sysGuide"]; ok {
+			infoMap["sysGuide"] = fmt.Sprint(v)
+		} else {
+			infoMap["sysGuide"] = "true"
+		}
+	} else {
+		infoMap["sysGuide"] = "true"
+	}
 	// 序列号
 	infoMap["serialNum"] = fmt.Sprint(config.Get("omc.sn"))
 	// 获取LOGO类型