From 32ff0b76e5e2eb3c60b63bb107af17492290dfe3 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 10 Apr 2025 15:54:32 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=8E=BB=E9=99=A4=E5=BC=80=E7=AB=99?=
 =?UTF-8?q?=E5=BC=95=E5=AF=BC=E7=B3=BB=E7=BB=9F=E9=87=8D=E7=BD=AE=E5=88=A4?=
 =?UTF-8?q?=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/common/controller/bootloader.go | 24 ++++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/modules/common/controller/bootloader.go b/src/modules/common/controller/bootloader.go
index f7425bca..048ca18c 100644
--- a/src/modules/common/controller/bootloader.go
+++ b/src/modules/common/controller/bootloader.go
@@ -1,6 +1,7 @@
 package controller
 
 import (
+	"runtime"
 	"strings"
 
 	"be.ems/src/framework/constants"
@@ -8,6 +9,7 @@ import (
 	"be.ems/src/framework/reqctx"
 	"be.ems/src/framework/resp"
 	"be.ems/src/framework/token"
+	"be.ems/src/framework/utils/cmd"
 	"be.ems/src/framework/utils/machine"
 	"be.ems/src/framework/utils/regular"
 	"be.ems/src/modules/common/service"
@@ -122,9 +124,25 @@ func (s *BootloaderController) Reset(c *gin.Context) {
 	// 	return
 	// }
 
-	if err := machine.Reset(); err != nil {
-		c.JSON(200, resp.ErrMsg(err.Error()))
-		return
+	// if err := machine.Reset(); err != nil {
+	// 	c.JSON(200, resp.ErrMsg(err.Error()))
+	// 	return
+	// }
+
+	// 重置数据库
+	if runtime.GOOS == "windows" {
+		// return fmt.Errorf("not support window")
+	} else {
+		// 重置数据库
+		if _, err := cmd.Execf("sudo cp -rf /usr/local/omc/etc/db/omc_db.sqlite /usr/local/omc/database/omc_db.sqlite"); err != nil {
+			c.JSON(200, resp.ErrMsg(err.Error()))
+			return
+		}
+		// 重启服务
+		if _, err := cmd.Execf("nohup sh -c \"sleep 1s && %s\" > /dev/null 2>&1 &", "sudo systemctl restart restagent"); err != nil {
+			c.JSON(200, resp.ErrMsg(err.Error()))
+			return
+		}
 	}
 
 	// 清除授权信息