create project

2026-01-16 17:30:40 +08:00
commit effac6b017
157 changed files with 45997 additions and 0 deletions
--- a/application/services/character_library_service.go
+++ b/application/services/character_library_service.go
@@ -0,0 +1,473 @@
+package services
+
+import (
+	"errors"
+	"fmt"
+	"time"
+
+	models "github.com/drama-generator/backend/domain/models"
+	"github.com/drama-generator/backend/pkg/logger"
+	"gorm.io/gorm"
+)
+
+type CharacterLibraryService struct {
+	db  *gorm.DB
+	log *logger.Logger
+}
+
+func NewCharacterLibraryService(db *gorm.DB, log *logger.Logger) *CharacterLibraryService {
+	return &CharacterLibraryService{
+		db:  db,
+		log: log,
+	}
+}
+
+type CreateLibraryItemRequest struct {
+	Name        string  `json:"name" binding:"required,min=1,max=100"`
+	Category    *string `json:"category"`
+	ImageURL    string  `json:"image_url" binding:"required"`
+	Description *string `json:"description"`
+	Tags        *string `json:"tags"`
+	SourceType  string  `json:"source_type"`
+}
+
+type CharacterLibraryQuery struct {
+	Page       int    `form:"page,default=1"`
+	PageSize   int    `form:"page_size,default=20"`
+	Category   string `form:"category"`
+	SourceType string `form:"source_type"`
+	Keyword    string `form:"keyword"`
+}
+
+// ListLibraryItems 获取用户角色库列表
+func (s *CharacterLibraryService) ListLibraryItems(query *CharacterLibraryQuery) ([]models.CharacterLibrary, int64, error) {
+	var items []models.CharacterLibrary
+	var total int64
+
+	db := s.db.Model(&models.CharacterLibrary{})
+
+	// 筛选条件
+	if query.Category != "" {
+		db = db.Where("category = ?", query.Category)
+	}
+
+	if query.SourceType != "" {
+		db = db.Where("source_type = ?", query.SourceType)
+	}
+
+	if query.Keyword != "" {
+		db = db.Where("name LIKE ? OR description LIKE ?", "%"+query.Keyword+"%", "%"+query.Keyword+"%")
+	}
+
+	// 获取总数
+	if err := db.Count(&total).Error; err != nil {
+		s.log.Errorw("Failed to count character library", "error", err)
+		return nil, 0, err
+	}
+
+	// 分页查询
+	offset := (query.Page - 1) * query.PageSize
+	err := db.Order("created_at DESC").
+		Offset(offset).
+		Limit(query.PageSize).
+		Find(&items).Error
+
+	if err != nil {
+		s.log.Errorw("Failed to list character library", "error", err)
+		return nil, 0, err
+	}
+
+	return items, total, nil
+}
+
+// CreateLibraryItem 添加到角色库
+func (s *CharacterLibraryService) CreateLibraryItem(req *CreateLibraryItemRequest) (*models.CharacterLibrary, error) {
+	sourceType := req.SourceType
+	if sourceType == "" {
+		sourceType = "generated"
+	}
+
+	item := &models.CharacterLibrary{
+		Name:        req.Name,
+		Category:    req.Category,
+		ImageURL:    req.ImageURL,
+		Description: req.Description,
+		Tags:        req.Tags,
+		SourceType:  sourceType,
+	}
+
+	if err := s.db.Create(item).Error; err != nil {
+		s.log.Errorw("Failed to create library item", "error", err)
+		return nil, err
+	}
+
+	s.log.Infow("Library item created", "item_id", item.ID)
+	return item, nil
+}
+
+// GetLibraryItem 获取角色库项
+func (s *CharacterLibraryService) GetLibraryItem(itemID string) (*models.CharacterLibrary, error) {
+	var item models.CharacterLibrary
+	err := s.db.Where("id = ? ", itemID).First(&item).Error
+
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.New("library item not found")
+		}
+		s.log.Errorw("Failed to get library item", "error", err)
+		return nil, err
+	}
+
+	return &item, nil
+}
+
+// DeleteLibraryItem 删除角色库项
+func (s *CharacterLibraryService) DeleteLibraryItem(itemID string) error {
+	result := s.db.Where("id = ? ", itemID).Delete(&models.CharacterLibrary{})
+
+	if result.Error != nil {
+		s.log.Errorw("Failed to delete library item", "error", result.Error)
+		return result.Error
+	}
+
+	if result.RowsAffected == 0 {
+		return errors.New("library item not found")
+	}
+
+	s.log.Infow("Library item deleted", "item_id", itemID)
+	return nil
+}
+
+// ApplyLibraryItemToCharacter 将角色库形象应用到角色
+func (s *CharacterLibraryService) ApplyLibraryItemToCharacter(characterID string, libraryItemID string) error {
+	// 验证角色库项存在且属于该用户
+	var libraryItem models.CharacterLibrary
+	if err := s.db.Where("id = ? ", libraryItemID).First(&libraryItem).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("library item not found")
+		}
+		return err
+	}
+
+	// 查找角色
+	var character models.Character
+	if err := s.db.Where("id = ?", characterID).First(&character).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("character not found")
+		}
+		return err
+	}
+
+	// 查询Drama验证权限
+	var drama models.Drama
+	if err := s.db.Where("id = ? ", character.DramaID).First(&drama).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("unauthorized")
+		}
+		return err
+	}
+
+	// 更新角色的image_url
+	if err := s.db.Model(&character).Update("image_url", libraryItem.ImageURL).Error; err != nil {
+		s.log.Errorw("Failed to update character image", "error", err)
+		return err
+	}
+
+	s.log.Infow("Library item applied to character", "character_id", characterID, "library_item_id", libraryItemID)
+	return nil
+}
+
+// UploadCharacterImage 上传角色图片
+func (s *CharacterLibraryService) UploadCharacterImage(characterID string, imageURL string) error {
+	// 查找角色
+	var character models.Character
+	if err := s.db.Where("id = ?", characterID).First(&character).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("character not found")
+		}
+		return err
+	}
+
+	// 查询Drama验证权限
+	var drama models.Drama
+	if err := s.db.Where("id = ? ", character.DramaID).First(&drama).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("unauthorized")
+		}
+		return err
+	}
+
+	// 更新图片URL
+	if err := s.db.Model(&character).Update("image_url", imageURL).Error; err != nil {
+		s.log.Errorw("Failed to update character image", "error", err)
+		return err
+	}
+
+	s.log.Infow("Character image uploaded", "character_id", characterID)
+	return nil
+}
+
+// AddCharacterToLibrary 将角色添加到角色库
+func (s *CharacterLibraryService) AddCharacterToLibrary(characterID string, category *string) (*models.CharacterLibrary, error) {
+	// 查找角色
+	var character models.Character
+	if err := s.db.Where("id = ?", characterID).First(&character).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.New("character not found")
+		}
+		return nil, err
+	}
+
+	// 查询Drama验证权限
+	var drama models.Drama
+	if err := s.db.Where("id = ? ", character.DramaID).First(&drama).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.New("unauthorized")
+		}
+		return nil, err
+	}
+
+	// 检查是否有图片
+	if character.ImageURL == nil || *character.ImageURL == "" {
+		return nil, fmt.Errorf("角色还没有形象图片")
+	}
+
+	// 创建角色库项
+	charLibrary := &models.CharacterLibrary{
+		Name:        character.Name,
+		ImageURL:    *character.ImageURL,
+		Description: character.Description,
+		SourceType:  "character",
+	}
+
+	if err := s.db.Create(charLibrary).Error; err != nil {
+		s.log.Errorw("Failed to add character to library", "error", err)
+		return nil, err
+	}
+
+	s.log.Infow("Character added to library", "character_id", characterID, "library_item_id", charLibrary.ID)
+	return charLibrary, nil
+}
+
+// DeleteCharacter 删除单个角色
+func (s *CharacterLibraryService) DeleteCharacter(characterID uint) error {
+	// 查找角色
+	var character models.Character
+	if err := s.db.Where("id = ?", characterID).First(&character).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("character not found")
+		}
+		return err
+	}
+
+	// 验证权限：检查角色所属的drama是否属于当前用户
+	var drama models.Drama
+	if err := s.db.Where("id = ? ", character.DramaID).First(&drama).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("unauthorized")
+		}
+		return err
+	}
+
+	// 删除角色
+	if err := s.db.Delete(&character).Error; err != nil {
+		s.log.Errorw("Failed to delete character", "error", err, "id", characterID)
+		return err
+	}
+
+	s.log.Infow("Character deleted", "id", characterID)
+	return nil
+}
+
+// GenerateCharacterImage AI生成角色形象
+func (s *CharacterLibraryService) GenerateCharacterImage(characterID string, imageService *ImageGenerationService, modelName string) (*models.ImageGeneration, error) {
+	// 查找角色
+	var character models.Character
+	if err := s.db.Where("id = ?", characterID).First(&character).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.New("character not found")
+		}
+		return nil, err
+	}
+
+	// 查询Drama验证权限
+	var drama models.Drama
+	if err := s.db.Where("id = ? ", character.DramaID).First(&drama).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, errors.New("unauthorized")
+		}
+		return nil, err
+	}
+
+	// 构建生成提示词 - 使用详细的外貌描述，添加干净背景要求
+	prompt := ""
+
+	// 优先使用appearance字段，它包含了最详细的外貌描述
+	if character.Appearance != nil && *character.Appearance != "" {
+		prompt = *character.Appearance
+	} else if character.Description != nil && *character.Description != "" {
+		prompt = *character.Description
+	} else {
+		prompt = character.Name
+	}
+
+	// 添加角色画像和风格要求
+	prompt += ", character portrait, full body or upper body shot"
+
+	// 添加干净背景要求 - 确保背景简洁不干扰主体
+	prompt += ", simple clean background, plain solid color background, white or light gray background"
+	prompt += ", studio lighting, professional photography"
+
+	// 添加质量和风格要求
+	prompt += ", high quality, detailed, anime style, character design"
+	prompt += ", no complex background, no scenery, focus on character"
+
+	// 调用图片生成服务
+	dramaIDStr := fmt.Sprintf("%d", character.DramaID)
+	imageType := "character"
+	req := &GenerateImageRequest{
+		DramaID:     dramaIDStr,
+		CharacterID: &character.ID,
+		ImageType:   imageType,
+		Prompt:      prompt,
+		Provider:    "openai",    // 或从配置读取
+		Model:       modelName,   // 使用用户指定的模型
+		Size:        "2560x1440", // 3,686,400像素，满足API最低要求（16:9比例）
+		Quality:     "standard",
+	}
+
+	imageGen, err := imageService.GenerateImage(req)
+	if err != nil {
+		s.log.Errorw("Failed to generate character image", "error", err)
+		return nil, fmt.Errorf("图片生成失败: %w", err)
+	}
+
+	// 异步处理：在后台监听图片生成完成，然后更新角色image_url
+	go s.waitAndUpdateCharacterImage(character.ID, imageGen.ID)
+
+	// 立即返回ImageGeneration对象，让前端可以轮询状态
+	s.log.Infow("Character image generation started", "character_id", characterID, "image_gen_id", imageGen.ID)
+	return imageGen, nil
+}
+
+// waitAndUpdateCharacterImage 后台异步等待图片生成完成并更新角色image_url
+func (s *CharacterLibraryService) waitAndUpdateCharacterImage(characterID uint, imageGenID uint) {
+	maxAttempts := 60
+	pollInterval := 5 * time.Second
+
+	for i := 0; i < maxAttempts; i++ {
+		time.Sleep(pollInterval)
+
+		// 查询图片生成状态
+		var imageGen models.ImageGeneration
+		if err := s.db.First(&imageGen, imageGenID).Error; err != nil {
+			s.log.Errorw("Failed to query image generation status", "error", err, "image_gen_id", imageGenID)
+			continue
+		}
+
+		// 检查是否完成
+		if imageGen.Status == models.ImageStatusCompleted && imageGen.ImageURL != nil && *imageGen.ImageURL != "" {
+			// 更新角色的image_url
+			if err := s.db.Model(&models.Character{}).Where("id = ?", characterID).Update("image_url", *imageGen.ImageURL).Error; err != nil {
+				s.log.Errorw("Failed to update character image_url", "error", err, "character_id", characterID)
+				return
+			}
+			s.log.Infow("Character image updated successfully", "character_id", characterID, "image_url", *imageGen.ImageURL)
+			return
+		}
+
+		// 检查是否失败
+		if imageGen.Status == models.ImageStatusFailed {
+			s.log.Errorw("Character image generation failed", "character_id", characterID, "image_gen_id", imageGenID, "error", imageGen.ErrorMsg)
+			return
+		}
+	}
+
+	s.log.Warnw("Character image generation timeout", "character_id", characterID, "image_gen_id", imageGenID)
+}
+
+// UpdateCharacter 更新角色信息
+func (s *CharacterLibraryService) UpdateCharacter(characterID string, req interface{}) error {
+	// 查找角色
+	var character models.Character
+	if err := s.db.Where("id = ?", characterID).First(&character).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("character not found")
+		}
+		return err
+	}
+
+	// 验证权限：查询角色所属的drama是否属于该用户
+	var drama models.Drama
+	if err := s.db.Where("id = ? ", character.DramaID).First(&drama).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("unauthorized")
+		}
+		return err
+	}
+
+	// 构建更新数据
+	updates := make(map[string]interface{})
+
+	// 使用类型断言获取请求数据
+	if reqMap, ok := req.(*struct {
+		Name        *string `json:"name"`
+		Appearance  *string `json:"appearance"`
+		Personality *string `json:"personality"`
+		Description *string `json:"description"`
+	}); ok {
+		if reqMap.Name != nil && *reqMap.Name != "" {
+			updates["name"] = *reqMap.Name
+		}
+		if reqMap.Appearance != nil {
+			updates["appearance"] = *reqMap.Appearance
+		}
+		if reqMap.Personality != nil {
+			updates["personality"] = *reqMap.Personality
+		}
+		if reqMap.Description != nil {
+			updates["description"] = *reqMap.Description
+		}
+	}
+
+	if len(updates) == 0 {
+		return errors.New("no fields to update")
+	}
+
+	// 更新角色信息
+	if err := s.db.Model(&character).Updates(updates).Error; err != nil {
+		s.log.Errorw("Failed to update character", "error", err, "character_id", characterID)
+		return err
+	}
+
+	s.log.Infow("Character updated", "character_id", characterID)
+	return nil
+}
+
+// BatchGenerateCharacterImages 批量生成角色图片（并发执行）
+func (s *CharacterLibraryService) BatchGenerateCharacterImages(characterIDs []string, imageService *ImageGenerationService, modelName string) {
+	s.log.Infow("Starting batch character image generation",
+		"count", len(characterIDs),
+		"model", modelName)
+
+	// 使用 goroutine 并发生成所有角色图片
+	for _, characterID := range characterIDs {
+		// 为每个角色启动单独的 goroutine
+		go func(charID string) {
+			imageGen, err := s.GenerateCharacterImage(charID, imageService, modelName)
+			if err != nil {
+				s.log.Errorw("Failed to generate character image in batch",
+					"character_id", charID,
+					"error", err)
+				return
+			}
+
+			s.log.Infow("Character image generated in batch",
+				"character_id", charID,
+				"image_gen_id", imageGen.ID)
+		}(characterID)
+	}
+
+	s.log.Infow("Batch character image generation tasks submitted",
+		"total", len(characterIDs))
+}