🎨 优化扩展模块，完成ai接入和对话功能

2026-02-12 23:12:28 +08:00
parent 4e611d3a5e
commit 572f3aa15b
779 changed files with 194400 additions and 3136 deletions
--- a/server/service/app/chat.go
+++ b/server/service/app/chat.go
@@ -0,0 +1,408 @@
+package app
+
+import (
+	"errors"
+	"time"
+
+	"git.echol.cn/loser/st/server/global"
+	"git.echol.cn/loser/st/server/model/app"
+	"git.echol.cn/loser/st/server/model/app/request"
+	"git.echol.cn/loser/st/server/model/app/response"
+	"gorm.io/gorm"
+)
+
+type ChatService struct{}
+
+// ==================== 对话 CRUD ====================
+
+// CreateChat 创建对话
+func (cs *ChatService) CreateChat(req request.CreateChatRequest, userID uint) (response.ChatResponse, error) {
+	// 获取角色卡信息
+	var character app.AICharacter
+	err := global.GVA_DB.First(&character, req.CharacterID).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return response.ChatResponse{}, errors.New("角色卡不存在")
+		}
+		return response.ChatResponse{}, err
+	}
+
+	// 对话标题
+	title := req.Title
+	if title == "" {
+		title = character.Name
+	}
+
+	now := time.Now()
+	chat := app.AIChat{
+		Title:         title,
+		UserID:        userID,
+		CharacterID:   &req.CharacterID,
+		ChatType:      "single",
+		LastMessageAt: &now,
+	}
+
+	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
+		// 创建对话
+		if err := tx.Create(&chat).Error; err != nil {
+			return err
+		}
+
+		// 如果角色卡有 FirstMessage，自动创建第一条系统消息
+		if character.FirstMessage != "" {
+			firstMsg := app.AIMessage{
+				ChatID:         chat.ID,
+				Content:        character.FirstMessage,
+				Role:           "assistant",
+				CharacterID:    &req.CharacterID,
+				SequenceNumber: 1,
+			}
+			if err := tx.Create(&firstMsg).Error; err != nil {
+				return err
+			}
+			chat.MessageCount = 1
+			tx.Model(&chat).Update("message_count", 1)
+		}
+
+		// 更新角色卡使用次数
+		tx.Model(&character).Update("total_chats", gorm.Expr("total_chats + 1"))
+
+		return nil
+	})
+
+	if err != nil {
+		return response.ChatResponse{}, err
+	}
+
+	return toChatResponse(&chat, &character, nil), nil
+}
+
+// GetChatList 获取用户的对话列表
+func (cs *ChatService) GetChatList(req request.ChatListRequest, userID uint) (response.ChatListResponse, error) {
+	db := global.GVA_DB.Model(&app.AIChat{}).Where("user_id = ?", userID)
+
+	var total int64
+	db.Count(&total)
+
+	var chats []app.AIChat
+	offset := (req.Page - 1) * req.PageSize
+	err := db.Preload("Character").
+		Order("is_pinned DESC, last_message_at DESC").
+		Offset(offset).Limit(req.PageSize).
+		Find(&chats).Error
+	if err != nil {
+		return response.ChatListResponse{}, err
+	}
+
+	// 获取每个对话的最后一条消息
+	chatIDs := make([]uint, len(chats))
+	for i, c := range chats {
+		chatIDs[i] = c.ID
+	}
+
+	lastMessages := make(map[uint]*response.MessageBrief)
+	if len(chatIDs) > 0 {
+		var messages []app.AIMessage
+		// 获取每个对话的最后一条消息（通过子查询）
+		global.GVA_DB.Where("chat_id IN ? AND is_deleted = ?", chatIDs, false).
+			Order("sequence_number DESC").
+			Find(&messages)
+
+		// 只保留每个对话的最后一条
+		for _, msg := range messages {
+			if _, exists := lastMessages[msg.ChatID]; !exists {
+				content := msg.Content
+				if len(content) > 100 {
+					content = content[:100] + "..."
+				}
+				lastMessages[msg.ChatID] = &response.MessageBrief{
+					Content: content,
+					Role:    msg.Role,
+				}
+			}
+		}
+	}
+
+	list := make([]response.ChatResponse, len(chats))
+	for i, c := range chats {
+		list[i] = toChatResponse(&c, c.Character, lastMessages[c.ID])
+	}
+
+	return response.ChatListResponse{
+		List:     list,
+		Total:    total,
+		Page:     req.Page,
+		PageSize: req.PageSize,
+	}, nil
+}
+
+// GetChatDetail 获取对话详情（包含消息列表）
+func (cs *ChatService) GetChatDetail(chatID uint, userID uint) (response.ChatDetailResponse, error) {
+	var chat app.AIChat
+	err := global.GVA_DB.Preload("Character").
+		Where("id = ? AND user_id = ?", chatID, userID).
+		First(&chat).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return response.ChatDetailResponse{}, errors.New("对话不存在")
+		}
+		return response.ChatDetailResponse{}, err
+	}
+
+	// 获取消息列表（最近的50条）
+	var messages []app.AIMessage
+	global.GVA_DB.Where("chat_id = ? AND is_deleted = ?", chatID, false).
+		Order("sequence_number ASC").
+		Limit(50).
+		Find(&messages)
+
+	msgList := make([]response.MessageResponse, len(messages))
+	for i, msg := range messages {
+		msgList[i] = toMessageResponse(&msg, chat.Character)
+	}
+
+	return response.ChatDetailResponse{
+		Chat:     toChatResponse(&chat, chat.Character, nil),
+		Messages: msgList,
+	}, nil
+}
+
+// GetChatMessages 分页获取对话消息
+func (cs *ChatService) GetChatMessages(chatID uint, req request.ChatMessagesRequest, userID uint) (response.MessageListResponse, error) {
+	// 验证对话归属
+	var chat app.AIChat
+	err := global.GVA_DB.Where("id = ? AND user_id = ?", chatID, userID).First(&chat).Error
+	if err != nil {
+		return response.MessageListResponse{}, errors.New("对话不存在")
+	}
+
+	db := global.GVA_DB.Model(&app.AIMessage{}).
+		Where("chat_id = ? AND is_deleted = ?", chatID, false)
+
+	var total int64
+	db.Count(&total)
+
+	var messages []app.AIMessage
+	offset := (req.Page - 1) * req.PageSize
+	err = db.Order("sequence_number ASC").
+		Offset(offset).Limit(req.PageSize).
+		Find(&messages).Error
+	if err != nil {
+		return response.MessageListResponse{}, err
+	}
+
+	// 预加载角色信息
+	var character *app.AICharacter
+	if chat.CharacterID != nil {
+		var c app.AICharacter
+		global.GVA_DB.First(&c, *chat.CharacterID)
+		character = &c
+	}
+
+	list := make([]response.MessageResponse, len(messages))
+	for i, msg := range messages {
+		list[i] = toMessageResponse(&msg, character)
+	}
+
+	return response.MessageListResponse{
+		List:     list,
+		Total:    total,
+		Page:     req.Page,
+		PageSize: req.PageSize,
+	}, nil
+}
+
+// DeleteChat 删除对话
+func (cs *ChatService) DeleteChat(chatID uint, userID uint) error {
+	var chat app.AIChat
+	err := global.GVA_DB.Where("id = ? AND user_id = ?", chatID, userID).First(&chat).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return errors.New("对话不存在")
+		}
+		return err
+	}
+
+	return global.GVA_DB.Transaction(func(tx *gorm.DB) error {
+		// 删除消息
+		if err := tx.Where("chat_id = ?", chatID).Delete(&app.AIMessage{}).Error; err != nil {
+			return err
+		}
+		// 删除消息变体
+		if err := tx.Where("message_id IN (?)",
+			tx.Model(&app.AIMessage{}).Select("id").Where("chat_id = ?", chatID),
+		).Delete(&app.AIMessageSwipe{}).Error; err != nil {
+			// 忽略错误，可能没有变体
+		}
+		// 删除对话
+		return tx.Delete(&chat).Error
+	})
+}
+
+// ==================== 消息操作 ====================
+
+// SaveUserMessage 保存用户消息（内部方法）
+func (cs *ChatService) SaveUserMessage(chatID uint, userID uint, content string) (*app.AIMessage, error) {
+	// 获取当前最大序号
+	var maxSeq int
+	global.GVA_DB.Model(&app.AIMessage{}).
+		Where("chat_id = ?", chatID).
+		Select("COALESCE(MAX(sequence_number), 0)").
+		Scan(&maxSeq)
+
+	msg := &app.AIMessage{
+		ChatID:         chatID,
+		Content:        content,
+		Role:           "user",
+		SenderID:       &userID,
+		SequenceNumber: maxSeq + 1,
+	}
+
+	if err := global.GVA_DB.Create(msg).Error; err != nil {
+		return nil, err
+	}
+
+	// 更新对话的消息数和最后消息时间
+	now := time.Now()
+	global.GVA_DB.Model(&app.AIChat{}).Where("id = ?", chatID).Updates(map[string]interface{}{
+		"message_count":   gorm.Expr("message_count + 1"),
+		"last_message_at": now,
+	})
+
+	return msg, nil
+}
+
+// SaveAssistantMessage 保存AI回复消息（内部方法）
+func (cs *ChatService) SaveAssistantMessage(chatID uint, characterID *uint, content string, model string, promptTokens, completionTokens int) (*app.AIMessage, error) {
+	var maxSeq int
+	global.GVA_DB.Model(&app.AIMessage{}).
+		Where("chat_id = ?", chatID).
+		Select("COALESCE(MAX(sequence_number), 0)").
+		Scan(&maxSeq)
+
+	msg := &app.AIMessage{
+		ChatID:           chatID,
+		Content:          content,
+		Role:             "assistant",
+		CharacterID:      characterID,
+		SequenceNumber:   maxSeq + 1,
+		Model:            model,
+		PromptTokens:     promptTokens,
+		CompletionTokens: completionTokens,
+		TotalTokens:      promptTokens + completionTokens,
+	}
+
+	if err := global.GVA_DB.Create(msg).Error; err != nil {
+		return nil, err
+	}
+
+	now := time.Now()
+	global.GVA_DB.Model(&app.AIChat{}).Where("id = ?", chatID).Updates(map[string]interface{}{
+		"message_count":   gorm.Expr("message_count + 1"),
+		"last_message_at": now,
+	})
+
+	return msg, nil
+}
+
+// EditMessage 编辑消息
+func (cs *ChatService) EditMessage(req request.EditMessageRequest, userID uint) (response.MessageResponse, error) {
+	var msg app.AIMessage
+	err := global.GVA_DB.Joins("JOIN ai_chats ON ai_chats.id = ai_messages.chat_id").
+		Where("ai_messages.id = ? AND ai_chats.user_id = ?", req.MessageID, userID).
+		First(&msg).Error
+	if err != nil {
+		return response.MessageResponse{}, errors.New("消息不存在")
+	}
+
+	msg.Content = req.Content
+	if err := global.GVA_DB.Save(&msg).Error; err != nil {
+		return response.MessageResponse{}, err
+	}
+
+	return toMessageResponse(&msg, nil), nil
+}
+
+// DeleteMessage 删除消息（软删除）
+func (cs *ChatService) DeleteMessage(messageID uint, userID uint) error {
+	var msg app.AIMessage
+	err := global.GVA_DB.Joins("JOIN ai_chats ON ai_chats.id = ai_messages.chat_id").
+		Where("ai_messages.id = ? AND ai_chats.user_id = ?", messageID, userID).
+		First(&msg).Error
+	if err != nil {
+		return errors.New("消息不存在")
+	}
+
+	return global.GVA_DB.Model(&msg).Update("is_deleted", true).Error
+}
+
+// GetChatForAI 获取对话用于AI调用的完整上下文（内部方法）
+func (cs *ChatService) GetChatForAI(chatID uint, userID uint) (*app.AIChat, *app.AICharacter, []app.AIMessage, error) {
+	var chat app.AIChat
+	err := global.GVA_DB.Where("id = ? AND user_id = ?", chatID, userID).First(&chat).Error
+	if err != nil {
+		return nil, nil, nil, errors.New("对话不存在")
+	}
+
+	var character *app.AICharacter
+	if chat.CharacterID != nil {
+		var c app.AICharacter
+		if err := global.GVA_DB.First(&c, *chat.CharacterID).Error; err == nil {
+			character = &c
+		}
+	}
+
+	// 获取历史消息（最近30条）
+	var messages []app.AIMessage
+	global.GVA_DB.Where("chat_id = ? AND is_deleted = ?", chatID, false).
+		Order("sequence_number ASC").
+		Limit(30).
+		Find(&messages)
+
+	return &chat, character, messages, nil
+}
+
+// ==================== 辅助函数 ====================
+
+func toChatResponse(chat *app.AIChat, character *app.AICharacter, lastMsg *response.MessageBrief) response.ChatResponse {
+	resp := response.ChatResponse{
+		ID:            chat.ID,
+		Title:         chat.Title,
+		CharacterID:   chat.CharacterID,
+		ChatType:      chat.ChatType,
+		LastMessageAt: chat.LastMessageAt,
+		MessageCount:  chat.MessageCount,
+		IsPinned:      chat.IsPinned,
+		LastMessage:   lastMsg,
+		CreatedAt:     chat.CreatedAt,
+	}
+
+	if character != nil {
+		resp.CharacterName = character.Name
+		resp.CharacterAvatar = character.Avatar
+	}
+
+	return resp
+}
+
+func toMessageResponse(msg *app.AIMessage, character *app.AICharacter) response.MessageResponse {
+	resp := response.MessageResponse{
+		ID:               msg.ID,
+		ChatID:           msg.ChatID,
+		Content:          msg.Content,
+		Role:             msg.Role,
+		CharacterID:      msg.CharacterID,
+		Model:            msg.Model,
+		PromptTokens:     msg.PromptTokens,
+		CompletionTokens: msg.CompletionTokens,
+		TotalTokens:      msg.TotalTokens,
+		SequenceNumber:   msg.SequenceNumber,
+		CreatedAt:        msg.CreatedAt,
+	}
+
+	if msg.Role == "assistant" && character != nil {
+		resp.CharacterName = character.Name
+	}
+
+	return resp
+}