版本封存

cmii-uav-watchdog-center/services/project_service.go

@@ -0,0 +1,101 @@
+package services
+
+import (
+	"cmii-uav-watchdog-common/models"
+	"cmii-uav-watchdog-common/utils"
+	"cmii-uav-watchdog-common/wdd_log"
+	"encoding/json"
+	"fmt"
+	"os"
+)
+
+const (
+	ProjectFilePath = "C:\\Users\\wddsh\\Documents\\IdeaProjects\\cmii-uav-watchdog-project\\cmii-uav-watchdog-center\\project_file\\"
+)
+
+func CreateProject(project *models.Project) (string, error) {
+
+	// 生成TierOneSecret
+	secret, err := GenerateTierOneTOTPSecret()
+	if err != nil {
+		wdd_log.Error("生成TierOneSecret失败: %v", err)
+		return "", err
+	}
+
+	project.TierOneSecret = secret
+	project.CreateTime = utils.CurentTimeString()
+
+	// 存储 project到file中
+	project.AuthFilePath = fmt.Sprintf("%s%s-%s.json", ProjectFilePath, project.Namespace, project.Name)
+
+	// 存储到file中
+	jsonData, err := json.Marshal(project)
+	if err != nil {
+		wdd_log.Error("序列化失败: %v", err)
+		return "", err
+	}
+
+	// 写入文件
+	err = os.WriteFile(project.AuthFilePath, jsonData, 0644)
+	if err != nil {
+		wdd_log.Error("写入文件失败: %v", err)
+		return "", err
+	}
+
+	return project.AuthFilePath, nil
+}
+
+// GetProjectList 获取项目列表
+func GetProjectList() ([]models.Project, error) {
+
+	// 获取项目列表
+	projectFileList, err := os.ReadDir(ProjectFilePath)
+	if err != nil {
+		wdd_log.Error("读取项目列表失败: %v", err)
+		return nil, err
+	}
+
+	projectList := make([]models.Project, 0)
+
+	// 遍历项目列表
+	for _, projectFilePath := range projectFileList {
+		// 读取项目文件
+		projectFile, err := os.ReadFile(fmt.Sprintf("%s%s", ProjectFilePath, projectFilePath.Name()))
+		if err != nil {
+			wdd_log.Error("读取项目文件失败: %v", err)
+			return nil, err
+		}
+
+		// 反序列化
+		var project models.Project
+		err = json.Unmarshal(projectFile, &project)
+		if err != nil {
+			wdd_log.Error("反序列化失败: %v", err)
+			return nil, err
+		}
+
+		// 添加到项目列表
+		projectList = append(projectList, project)
+	}
+
+	return projectList, nil
+}
+
+// GetProjectInfo 获取项目信息
+func GetProjectInfo(projectNamespace string) (models.Project, error) {
+	// 获取项目列表
+	projectList, err := GetProjectList()
+	if err != nil {
+		wdd_log.Error("获取项目列表失败: %v", err)
+		return models.Project{}, err
+	}
+
+	// 遍历项目列表
+	for _, project := range projectList {
+		if project.Namespace == projectNamespace {
+			return project, nil
+		}
+	}
+
+	return models.Project{}, fmt.Errorf("项目不存在")
+}