merge: 合并OMC分支

src/framework/utils/machine/launch.go

@@ -0,0 +1,178 @@
+package machine
+
+import (
+	"encoding/json"
+	"fmt"
+	"hash/fnv"
+	"os"
+	"runtime"
+	"time"
+
+	"nms_cxy/src/framework/constants/common"
+	"nms_cxy/src/framework/logger"
+	"nms_cxy/src/framework/utils/cmd"
+	"nms_cxy/src/framework/utils/crypto"
+	"nms_cxy/src/framework/utils/parse"
+)
+
+// 机器的唯一标识符
+var Code string
+
+// 初始信息
+var LaunchInfo map[string]any
+
+// codeGenerate 生成机器的唯一标识符
+func codeGenerate() string {
+	var machineID string
+
+	// 获取主机名
+	hostname, err := os.Hostname()
+	if err != nil {
+		panic(err)
+	}
+	machineID += hostname
+
+	// 获取 CPU 信息
+	numCPU := runtime.NumCPU()
+	machineID += fmt.Sprintf("%d", numCPU)
+
+	// 获取操作系统信息
+	osInfo := runtime.GOOS
+	machineID += osInfo
+
+	// 使用哈希函数生成机器码
+	h := fnv.New32a()
+	h.Write([]byte(machineID))
+	machineCode := h.Sum32()
+
+	return fmt.Sprintf("%x", machineCode)
+}
+
+// 网管本地路径
+func filePath() string {
+	filePath := "/usr/local/etc/omc/machine.ini"
+	if runtime.GOOS == "windows" {
+		filePath = fmt.Sprintf("C:%s", filePath)
+	}
+	return filePath
+}
+
+// codeFileRead 读取机器保留的信息
+func codeFileRead() (map[string]any, error) {
+	var mapData map[string]any
+	// 读取文件内容
+	bytes, err := os.ReadFile(filePath())
+	if err != nil {
+		logger.Warnf("CodeFileRead ReadFile => %s", err.Error())
+		return mapData, fmt.Errorf("not file")
+	}
+	content := string(bytes)
+	// 解密
+	contentDe, err := crypto.StringDecryptByAES(content)
+	if err != nil {
+		logger.Errorf("CodeFileRead decrypt: %v", err.Error())
+		return mapData, fmt.Errorf("decrypt fail")
+	}
+	// 序列化Map
+	mapData, err = parse.ConvertConfigToMap("json", string(contentDe))
+	if err != nil {
+		logger.Warnf("CodeFileRead ConvertConfigToMap => %s", err.Error())
+		return mapData, fmt.Errorf("content error")
+	}
+	return mapData, nil
+}
+
+// codeFileWrite 写入机器保留的信息
+func codeFileWrite(data map[string]any) error {
+	jsonByte, _ := json.Marshal(data)
+	// 加密
+	contentEn, err := crypto.StringEncryptByAES(string(jsonByte))
+	if err != nil {
+		logger.Errorf("insert encrypt: %v", err.Error())
+		return fmt.Errorf("encrypt fail")
+	}
+	return parse.ConvertConfigToFile("txt", filePath(), contentEn)
+}
+
+// Launch 记录首次安装启动初始信息
+func Launch() {
+	Code = codeGenerate()
+	// 检查文件是否存在
+	if _, err := os.Stat(filePath()); err != nil {
+		LaunchInfo = map[string]any{
+			"code":    Code,                   // 机器码
+			"useTime": time.Now().UnixMilli(), // 首次使用时间
+
+			common.LAUNCH_BOOTLOADER:          true, // 启动引导
+			common.LAUNCH_BOOTLOADER + "Time": 0,    // 引导完成时间
+		}
+		codeFileWrite(LaunchInfo)
+	} else {
+		// 读取记录文件
+		data, err := codeFileRead()
+		if err != nil {
+			// 文件异常就重新生成
+			os.Remove(filePath())
+			Launch()
+			return
+		}
+		LaunchInfo = data
+	}
+}
+
+// SetLaunchInfo 新增额外的初始信息
+func SetLaunchInfo(info map[string]any) error {
+	if info == nil {
+		return fmt.Errorf("not info")
+	}
+
+	// 固定值禁止变更
+	constKeys := []string{"code", "useTime"}
+	for k, v := range info {
+		constKey := false
+		for _, ck := range constKeys {
+			if ck == k {
+				constKey = true
+				break
+			}
+		}
+		if constKey {
+			continue
+		} else {
+			LaunchInfo[k] = v
+		}
+	}
+	return codeFileWrite(LaunchInfo)
+}
+
+// Bootloader 启动引导标记
+func Bootloader(flag bool) error {
+	return SetLaunchInfo(map[string]any{
+		common.LAUNCH_BOOTLOADER:          flag,                   // 启动引导 true开 false关
+		common.LAUNCH_BOOTLOADER + "Time": time.Now().UnixMilli(), // 引导完成时间
+	})
+}
+
+// Reset 引导数据重置
+func Reset() error {
+	// 重置数据库
+	if runtime.GOOS == "windows" {
+		// return fmt.Errorf("not support window")
+	} else {
+		// 重置数据库
+		if _, err := cmd.Execf("sudo /usr/local/omc/bin/setomc.sh -m install"); err != nil {
+			return err
+		}
+		// 重启服务
+		if _, err := cmd.Execf("nohup sh -c \"sleep 1s && %s\" > /dev/null 2>&1 &", "sudo systemctl restart restagent"); err != nil {
+			return err
+		}
+	}
+
+	// 重置引导标记
+	if err := Bootloader(true); err != nil {
+		return err
+	}
+
+	return nil
+}