refactor(backend): isolate workspace aggregation

2026-07-21 14:58:25 +08:00
parent 14a140985c
commit a53f047efe
8 changed files with 818 additions and 734 deletions
--- a/backend/internal/logic/projects/object_crud.go
+++ b/backend/internal/logic/projects/object_crud.go
@@ -0,0 +1,198 @@
+package projects
+
+import (
+	"errors"
+	"strings"
+
+	"gorm.io/gorm"
+	"senlinai-agent/backend/internal/models"
+)
+
+func (s *Service) CreateTask(ownerID uint, projectID uint, input CreateTaskInput) (*models.SenlinAgentTask, error) {
+	if err := ensureProjectOwner(ownerID, projectID); err != nil {
+		return nil, err
+	}
+	title := strings.TrimSpace(input.Title)
+	if title == "" {
+		return nil, errors.New("task title is required")
+	}
+	status := strings.TrimSpace(input.Status)
+	if status == "" {
+		status = "open"
+	}
+	tagID, err := s.findOrCreateTagID(projectID, input.Tag)
+	if err != nil {
+		return nil, err
+	}
+	task := &models.SenlinAgentTask{
+		ProjectID: projectID, CreatedBy: ownerID, TagID: tagID, Title: title,
+		Description: strings.TrimSpace(input.Description), Status: status, DueAt: input.DueAt,
+	}
+	return task, models.DBService.Create(task).Error
+}
+
+func (s *Service) UpdateTask(ownerID uint, projectID uint, taskID uint, input UpdateTaskInput) (*models.SenlinAgentTask, error) {
+	if err := ensureProjectOwner(ownerID, projectID); err != nil {
+		return nil, err
+	}
+	var task models.SenlinAgentTask
+	if err := models.DBService.Where("id = ? AND project_id = ?", taskID, projectID).First(&task).Error; err != nil {
+		return nil, err
+	}
+	nextProjectID := input.NextProjectID
+	if nextProjectID == 0 {
+		nextProjectID = task.ProjectID
+	}
+	if err := ensureProjectOwner(ownerID, nextProjectID); err != nil {
+		return nil, err
+	}
+	title := strings.TrimSpace(input.Title)
+	if title == "" {
+		return nil, errors.New("task title is required")
+	}
+	tagID, err := s.findOrCreateTagID(nextProjectID, input.Tag)
+	if err != nil {
+		return nil, err
+	}
+	projectIdentity, err := projectIdentity(nextProjectID)
+	if err != nil {
+		return nil, err
+	}
+	tagIdentity, err := optionalTagIdentity(tagID)
+	if err != nil {
+		return nil, err
+	}
+	status := strings.TrimSpace(input.Status)
+	if status == "" {
+		if input.Completed {
+			status = "done"
+		} else {
+			status = "open"
+		}
+	}
+	task.ProjectID = nextProjectID
+	task.ProjectIdentity = projectIdentity
+	task.Title = title
+	task.Description = strings.TrimSpace(input.Description)
+	task.Status = status
+	task.TagID = tagID
+	task.TagIdentity = tagIdentity
+	if err := models.DBService.Save(&task).Error; err != nil {
+		return nil, err
+	}
+	return &task, nil
+}
+
+func (s *Service) CreateFileSource(ownerID uint, projectID uint, input CreateFileSourceInput) (*models.SenlinAgentSource, error) {
+	if err := ensureProjectOwner(ownerID, projectID); err != nil {
+		return nil, err
+	}
+	title := strings.TrimSpace(input.Title)
+	if title == "" {
+		return nil, errors.New("source title is required")
+	}
+	filePath := strings.TrimSpace(input.FilePath)
+	if filePath == "" {
+		return nil, errors.New("source file path is required")
+	}
+	source := &models.SenlinAgentSource{ProjectID: projectID, CreatedBy: ownerID, Kind: "file", Title: title, FilePath: filePath}
+	return source, models.DBService.Create(source).Error
+}
+
+func (s *Service) CreateCronPlan(ownerID uint, projectID uint, input CreateCronPlanInput) (*models.SenlinAgentCronPlan, error) {
+	if err := ensureProjectOwner(ownerID, projectID); err != nil {
+		return nil, err
+	}
+	title := strings.TrimSpace(input.Title)
+	if title == "" {
+		return nil, errors.New("cron plan title is required")
+	}
+	schedule := strings.TrimSpace(input.Schedule)
+	if schedule == "" {
+		return nil, errors.New("cron schedule is required")
+	}
+	plan := &models.SenlinAgentCronPlan{
+		ProjectID: projectID, CreatedBy: ownerID, Title: title, Schedule: schedule,
+		Enabled: input.Enabled, NextRunAt: input.NextRunAt, LastResult: "Not run yet",
+	}
+	return plan, models.DBService.Create(plan).Error
+}
+
+func (s *Service) CreateProjectTag(ownerID uint, projectID uint, name string) (*models.SenlinAgentTag, error) {
+	if err := ensureProjectOwner(ownerID, projectID); err != nil {
+		return nil, err
+	}
+	return s.CreateTag(projectID, name)
+}
+
+func (s *Service) CreateTag(projectID uint, name string) (*models.SenlinAgentTag, error) {
+	name = strings.TrimSpace(name)
+	if name == "" {
+		return nil, errors.New("tag name is required")
+	}
+	var existing models.SenlinAgentTag
+	err := models.DBService.Where("project_id = ? AND name = ?", projectID, name).First(&existing).Error
+	if err == nil {
+		return &existing, nil
+	}
+	if !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+	tag := &models.SenlinAgentTag{ProjectID: projectID, Name: name}
+	return tag, models.DBService.Create(tag).Error
+}
+
+func (s *Service) ListProjectTags(ownerID uint, projectID uint) ([]models.SenlinAgentTag, error) {
+	if err := ensureProjectOwner(ownerID, projectID); err != nil {
+		return nil, err
+	}
+	return s.ListTags(projectID)
+}
+
+func (s *Service) ListTags(projectID uint) ([]models.SenlinAgentTag, error) {
+	var tags []models.SenlinAgentTag
+	err := models.DBService.Where("project_id = ?", projectID).Order("name asc").Find(&tags).Error
+	return tags, err
+}
+
+func (s *Service) findOrCreateTagID(projectID uint, name string) (*uint, error) {
+	name = strings.TrimSpace(name)
+	if name == "" {
+		return nil, nil
+	}
+	tag, err := s.CreateTag(projectID, name)
+	if err != nil {
+		return nil, err
+	}
+	return &tag.ID, nil
+}
+
+func ensureProjectOwner(ownerID uint, projectID uint) error {
+	var count int64
+	if err := models.DBService.Model(&models.SenlinAgentProject{}).Where("id = ? AND owner_id = ?", projectID, ownerID).Count(&count).Error; err != nil {
+		return err
+	}
+	if count == 0 {
+		return errors.New("project not found")
+	}
+	return nil
+}
+
+func projectIdentity(projectID uint) (string, error) {
+	var project models.SenlinAgentProject
+	if err := models.DBService.Select("identity").First(&project, projectID).Error; err != nil {
+		return "", err
+	}
+	return project.Identity, nil
+}
+
+func optionalTagIdentity(tagID *uint) (*string, error) {
+	if tagID == nil {
+		return nil, nil
+	}
+	var tag models.SenlinAgentTag
+	if err := models.DBService.Select("identity").First(&tag, *tagID).Error; err != nil {
+		return nil, err
+	}
+	return &tag.Identity, nil
+}