Add files via upload

2026-07-06 04:27:55 +02:00 · 2025-12-25 01:44:19 +08:00
parent 27a37346c1
commit 26f131bb77
9 changed files with 2705 additions and 12 deletions
@@ -14,6 +14,7 @@ import (
 type Conversation struct {
 	ID        string    `json:"id"`
 	Title     string    `json:"title"`
+	Pinned    bool      `json:"pinned"`
 	CreatedAt time.Time `json:"createdAt"`
 	UpdatedAt time.Time `json:"updatedAt"`
 	Messages  []Message `json:"messages,omitempty"`
@@ -55,11 +56,12 @@ func (db *DB) CreateConversation(title string) (*Conversation, error) {
 func (db *DB) GetConversation(id string) (*Conversation, error) {
 	var conv Conversation
 	var createdAt, updatedAt string
+	var pinned int

 	err := db.QueryRow(
-		"SELECT id, title, created_at, updated_at FROM conversations WHERE id = ?",
+		"SELECT id, title, pinned, created_at, updated_at FROM conversations WHERE id = ?",
 		id,
-	).Scan(&conv.ID, &conv.Title, &createdAt, &updatedAt)
+	).Scan(&conv.ID, &conv.Title, &pinned, &createdAt, &updatedAt)
 	if err != nil {
 		if err == sql.ErrNoRows {
 			return nil, fmt.Errorf("对话不存在")
@@ -85,6 +87,8 @@ func (db *DB) GetConversation(id string) (*Conversation, error) {
 		conv.UpdatedAt, _ = time.Parse(time.RFC3339, updatedAt)
 	}

+	conv.Pinned = pinned != 0
+
 	// 加载消息
 	messages, err := db.GetMessages(id)
 	if err != nil {
@@ -138,7 +142,7 @@ func (db *DB) ListConversations(limit, offset int, search string) ([]*Conversati
 		searchPattern := "%" + search + "%"
 		// 使用DISTINCT避免重复，因为一个对话可能有多条消息匹配
 		rows, err = db.Query(
-			`SELECT DISTINCT c.id, c.title, c.created_at, c.updated_at 
+			`SELECT DISTINCT c.id, c.title, COALESCE(c.pinned, 0), c.created_at, c.updated_at 
 			 FROM conversations c
 			 LEFT JOIN messages m ON c.id = m.conversation_id
 			 WHERE c.title LIKE ? OR m.content LIKE ?
@@ -148,7 +152,7 @@ func (db *DB) ListConversations(limit, offset int, search string) ([]*Conversati
 		)
 	} else {
 		rows, err = db.Query(
-			"SELECT id, title, created_at, updated_at FROM conversations ORDER BY updated_at DESC LIMIT ? OFFSET ?",
+			"SELECT id, title, COALESCE(pinned, 0), created_at, updated_at FROM conversations ORDER BY updated_at DESC LIMIT ? OFFSET ?",
 			limit, offset,
 		)
 	}
@@ -162,8 +166,9 @@ func (db *DB) ListConversations(limit, offset int, search string) ([]*Conversati
 	for rows.Next() {
 		var conv Conversation
 		var createdAt, updatedAt string
+		var pinned int

-		if err := rows.Scan(&conv.ID, &conv.Title, &createdAt, &updatedAt); err != nil {
+		if err := rows.Scan(&conv.ID, &conv.Title, &pinned, &createdAt, &updatedAt); err != nil {
 			return nil, fmt.Errorf("扫描对话失败: %w", err)
 		}

@@ -185,6 +190,8 @@ func (db *DB) ListConversations(limit, offset int, search string) ([]*Conversati
 			conv.UpdatedAt, _ = time.Parse(time.RFC3339, updatedAt)
 		}

+		conv.Pinned = pinned != 0
+
 		conversations = append(conversations, &conv)
 	}

@@ -148,6 +148,28 @@ func (db *DB) initTables() error {
 		FOREIGN KEY (message_id) REFERENCES messages(id) ON DELETE SET NULL
 	);`

+	// 创建对话分组表
+	createConversationGroupsTable := `
+	CREATE TABLE IF NOT EXISTS conversation_groups (
+		id TEXT PRIMARY KEY,
+		name TEXT NOT NULL,
+		icon TEXT,
+		created_at DATETIME NOT NULL,
+		updated_at DATETIME NOT NULL
+	);`
+
+	// 创建对话分组映射表
+	createConversationGroupMappingsTable := `
+	CREATE TABLE IF NOT EXISTS conversation_group_mappings (
+		id TEXT PRIMARY KEY,
+		conversation_id TEXT NOT NULL,
+		group_id TEXT NOT NULL,
+		created_at DATETIME NOT NULL,
+		FOREIGN KEY (conversation_id) REFERENCES conversations(id) ON DELETE CASCADE,
+		FOREIGN KEY (group_id) REFERENCES conversation_groups(id) ON DELETE CASCADE,
+		UNIQUE(conversation_id, group_id)
+	);`
+
 	// 创建索引
 	createIndexes := `
 	CREATE INDEX IF NOT EXISTS idx_messages_conversation_id ON messages(conversation_id);
@@ -164,6 +186,9 @@ func (db *DB) initTables() error {
 	CREATE INDEX IF NOT EXISTS idx_knowledge_retrieval_logs_conversation ON knowledge_retrieval_logs(conversation_id);
 	CREATE INDEX IF NOT EXISTS idx_knowledge_retrieval_logs_message ON knowledge_retrieval_logs(message_id);
 	CREATE INDEX IF NOT EXISTS idx_knowledge_retrieval_logs_created_at ON knowledge_retrieval_logs(created_at);
+	CREATE INDEX IF NOT EXISTS idx_conversation_group_mappings_conversation ON conversation_group_mappings(conversation_id);
+	CREATE INDEX IF NOT EXISTS idx_conversation_group_mappings_group ON conversation_group_mappings(group_id);
+	CREATE INDEX IF NOT EXISTS idx_conversations_pinned ON conversations(pinned);
 	`

 	if _, err := db.Exec(createConversationsTable); err != nil {
@@ -198,16 +223,29 @@ func (db *DB) initTables() error {
 		return fmt.Errorf("创建knowledge_retrieval_logs表失败: %w", err)
 	}

-	if _, err := db.Exec(createIndexes); err != nil {
-		return fmt.Errorf("创建索引失败: %w", err)
+	if _, err := db.Exec(createConversationGroupsTable); err != nil {
+		return fmt.Errorf("创建conversation_groups表失败: %w", err)
 	}

-	// 为已有表添加新字段（如果不存在）
+	if _, err := db.Exec(createConversationGroupMappingsTable); err != nil {
+		return fmt.Errorf("创建conversation_group_mappings表失败: %w", err)
+	}
+
+	// 为已有表添加新字段（如果不存在）- 必须在创建索引之前
 	if err := db.migrateConversationsTable(); err != nil {
 		db.logger.Warn("迁移conversations表失败", zap.Error(err))
 		// 不返回错误，允许继续运行
 	}

+	if err := db.migrateConversationGroupsTable(); err != nil {
+		db.logger.Warn("迁移conversation_groups表失败", zap.Error(err))
+		// 不返回错误，允许继续运行
+	}
+
+	if _, err := db.Exec(createIndexes); err != nil {
+		return fmt.Errorf("创建索引失败: %w", err)
+	}
+
 	db.logger.Info("数据库表初始化完成")
 	return nil
 }
@@ -251,6 +289,48 @@ func (db *DB) migrateConversationsTable() error {
 		}
 	}

+	// 检查pinned字段是否存在
+	err = db.QueryRow("SELECT COUNT(*) FROM pragma_table_info('conversations') WHERE name='pinned'").Scan(&count)
+	if err != nil {
+		// 如果查询失败，尝试添加字段
+		if _, addErr := db.Exec("ALTER TABLE conversations ADD COLUMN pinned INTEGER DEFAULT 0"); addErr != nil {
+			// 如果字段已存在，忽略错误
+			errMsg := strings.ToLower(addErr.Error())
+			if !strings.Contains(errMsg, "duplicate column") && !strings.Contains(errMsg, "already exists") {
+				db.logger.Warn("添加pinned字段失败", zap.Error(addErr))
+			}
+		}
+	} else if count == 0 {
+		// 字段不存在，添加它
+		if _, err := db.Exec("ALTER TABLE conversations ADD COLUMN pinned INTEGER DEFAULT 0"); err != nil {
+			db.logger.Warn("添加pinned字段失败", zap.Error(err))
+		}
+	}
+
+	return nil
+}
+
+// migrateConversationGroupsTable 迁移conversation_groups表，添加新字段
+func (db *DB) migrateConversationGroupsTable() error {
+	// 检查pinned字段是否存在
+	var count int
+	err := db.QueryRow("SELECT COUNT(*) FROM pragma_table_info('conversation_groups') WHERE name='pinned'").Scan(&count)
+	if err != nil {
+		// 如果查询失败，尝试添加字段
+		if _, addErr := db.Exec("ALTER TABLE conversation_groups ADD COLUMN pinned INTEGER DEFAULT 0"); addErr != nil {
+			// 如果字段已存在，忽略错误
+			errMsg := strings.ToLower(addErr.Error())
+			if !strings.Contains(errMsg, "duplicate column") && !strings.Contains(errMsg, "already exists") {
+				db.logger.Warn("添加pinned字段失败", zap.Error(addErr))
+			}
+		}
+	} else if count == 0 {
+		// 字段不存在，添加它
+		if _, err := db.Exec("ALTER TABLE conversation_groups ADD COLUMN pinned INTEGER DEFAULT 0"); err != nil {
+			db.logger.Warn("添加pinned字段失败", zap.Error(err))
+		}
+	}
+
 	return nil
 }

@@ -0,0 +1,319 @@
+package database
+
+import (
+	"database/sql"
+	"fmt"
+	"time"
+
+	"github.com/google/uuid"
+)
+
+// ConversationGroup 对话分组
+type ConversationGroup struct {
+	ID        string    `json:"id"`
+	Name      string    `json:"name"`
+	Icon      string    `json:"icon"`
+	Pinned    bool      `json:"pinned"`
+	CreatedAt time.Time `json:"createdAt"`
+	UpdatedAt time.Time `json:"updatedAt"`
+}
+
+// GroupExistsByName 检查分组名称是否已存在
+func (db *DB) GroupExistsByName(name string, excludeID string) (bool, error) {
+	var count int
+	var err error
+
+	if excludeID != "" {
+		err = db.QueryRow(
+			"SELECT COUNT(*) FROM conversation_groups WHERE name = ? AND id != ?",
+			name, excludeID,
+		).Scan(&count)
+	} else {
+		err = db.QueryRow(
+			"SELECT COUNT(*) FROM conversation_groups WHERE name = ?",
+			name,
+		).Scan(&count)
+	}
+
+	if err != nil {
+		return false, fmt.Errorf("检查分组名称失败: %w", err)
+	}
+
+	return count > 0, nil
+}
+
+// CreateGroup 创建分组
+func (db *DB) CreateGroup(name, icon string) (*ConversationGroup, error) {
+	// 检查名称是否已存在
+	exists, err := db.GroupExistsByName(name, "")
+	if err != nil {
+		return nil, err
+	}
+	if exists {
+		return nil, fmt.Errorf("分组名称已存在")
+	}
+
+	id := uuid.New().String()
+	now := time.Now()
+
+	if icon == "" {
+		icon = "📁"
+	}
+
+	_, err = db.Exec(
+		"INSERT INTO conversation_groups (id, name, icon, pinned, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?)",
+		id, name, icon, 0, now, now,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("创建分组失败: %w", err)
+	}
+
+	return &ConversationGroup{
+		ID:        id,
+		Name:      name,
+		Icon:      icon,
+		Pinned:    false,
+		CreatedAt: now,
+		UpdatedAt: now,
+	}, nil
+}
+
+// ListGroups 列出所有分组
+func (db *DB) ListGroups() ([]*ConversationGroup, error) {
+	rows, err := db.Query(
+		"SELECT id, name, icon, COALESCE(pinned, 0), created_at, updated_at FROM conversation_groups ORDER BY COALESCE(pinned, 0) DESC, created_at ASC",
+	)
+	if err != nil {
+		return nil, fmt.Errorf("查询分组列表失败: %w", err)
+	}
+	defer rows.Close()
+
+	var groups []*ConversationGroup
+	for rows.Next() {
+		var group ConversationGroup
+		var createdAt, updatedAt string
+		var pinned int
+
+		if err := rows.Scan(&group.ID, &group.Name, &group.Icon, &pinned, &createdAt, &updatedAt); err != nil {
+			return nil, fmt.Errorf("扫描分组失败: %w", err)
+		}
+
+		group.Pinned = pinned != 0
+
+		// 尝试多种时间格式解析
+		var err1, err2 error
+		group.CreatedAt, err1 = time.Parse("2006-01-02 15:04:05.999999999-07:00", createdAt)
+		if err1 != nil {
+			group.CreatedAt, err1 = time.Parse("2006-01-02 15:04:05", createdAt)
+		}
+		if err1 != nil {
+			group.CreatedAt, _ = time.Parse(time.RFC3339, createdAt)
+		}
+
+		group.UpdatedAt, err2 = time.Parse("2006-01-02 15:04:05.999999999-07:00", updatedAt)
+		if err2 != nil {
+			group.UpdatedAt, err2 = time.Parse("2006-01-02 15:04:05", updatedAt)
+		}
+		if err2 != nil {
+			group.UpdatedAt, _ = time.Parse(time.RFC3339, updatedAt)
+		}
+
+		groups = append(groups, &group)
+	}
+
+	return groups, nil
+}
+
+// GetGroup 获取分组
+func (db *DB) GetGroup(id string) (*ConversationGroup, error) {
+	var group ConversationGroup
+	var createdAt, updatedAt string
+	var pinned int
+
+	err := db.QueryRow(
+		"SELECT id, name, icon, COALESCE(pinned, 0), created_at, updated_at FROM conversation_groups WHERE id = ?",
+		id,
+	).Scan(&group.ID, &group.Name, &group.Icon, &pinned, &createdAt, &updatedAt)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			return nil, fmt.Errorf("分组不存在")
+		}
+		return nil, fmt.Errorf("查询分组失败: %w", err)
+	}
+
+	// 尝试多种时间格式解析
+	var err1, err2 error
+	group.CreatedAt, err1 = time.Parse("2006-01-02 15:04:05.999999999-07:00", createdAt)
+	if err1 != nil {
+		group.CreatedAt, err1 = time.Parse("2006-01-02 15:04:05", createdAt)
+	}
+	if err1 != nil {
+		group.CreatedAt, _ = time.Parse(time.RFC3339, createdAt)
+	}
+
+	group.UpdatedAt, err2 = time.Parse("2006-01-02 15:04:05.999999999-07:00", updatedAt)
+	if err2 != nil {
+		group.UpdatedAt, err2 = time.Parse("2006-01-02 15:04:05", updatedAt)
+	}
+	if err2 != nil {
+		group.UpdatedAt, _ = time.Parse(time.RFC3339, updatedAt)
+	}
+
+	group.Pinned = pinned != 0
+
+	return &group, nil
+}
+
+// UpdateGroup 更新分组
+func (db *DB) UpdateGroup(id, name, icon string) error {
+	// 检查名称是否已存在（排除当前分组）
+	exists, err := db.GroupExistsByName(name, id)
+	if err != nil {
+		return err
+	}
+	if exists {
+		return fmt.Errorf("分组名称已存在")
+	}
+
+	_, err = db.Exec(
+		"UPDATE conversation_groups SET name = ?, icon = ?, updated_at = ? WHERE id = ?",
+		name, icon, time.Now(), id,
+	)
+	if err != nil {
+		return fmt.Errorf("更新分组失败: %w", err)
+	}
+	return nil
+}
+
+// DeleteGroup 删除分组
+func (db *DB) DeleteGroup(id string) error {
+	_, err := db.Exec("DELETE FROM conversation_groups WHERE id = ?", id)
+	if err != nil {
+		return fmt.Errorf("删除分组失败: %w", err)
+	}
+	return nil
+}
+
+// AddConversationToGroup 将对话添加到分组
+func (db *DB) AddConversationToGroup(conversationID, groupID string) error {
+	id := uuid.New().String()
+	_, err := db.Exec(
+		"INSERT OR REPLACE INTO conversation_group_mappings (id, conversation_id, group_id, created_at) VALUES (?, ?, ?, ?)",
+		id, conversationID, groupID, time.Now(),
+	)
+	if err != nil {
+		return fmt.Errorf("添加对话到分组失败: %w", err)
+	}
+	return nil
+}
+
+// RemoveConversationFromGroup 从分组中移除对话
+func (db *DB) RemoveConversationFromGroup(conversationID, groupID string) error {
+	_, err := db.Exec(
+		"DELETE FROM conversation_group_mappings WHERE conversation_id = ? AND group_id = ?",
+		conversationID, groupID,
+	)
+	if err != nil {
+		return fmt.Errorf("从分组中移除对话失败: %w", err)
+	}
+	return nil
+}
+
+// GetConversationsByGroup 获取分组中的所有对话
+func (db *DB) GetConversationsByGroup(groupID string) ([]*Conversation, error) {
+	rows, err := db.Query(
+		`SELECT c.id, c.title, COALESCE(c.pinned, 0), c.created_at, c.updated_at 
+		 FROM conversations c
+		 INNER JOIN conversation_group_mappings cgm ON c.id = cgm.conversation_id
+		 WHERE cgm.group_id = ?
+		 ORDER BY c.updated_at DESC`,
+		groupID,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("查询分组对话失败: %w", err)
+	}
+	defer rows.Close()
+
+	var conversations []*Conversation
+	for rows.Next() {
+		var conv Conversation
+		var createdAt, updatedAt string
+		var pinned int
+
+		if err := rows.Scan(&conv.ID, &conv.Title, &pinned, &createdAt, &updatedAt); err != nil {
+			return nil, fmt.Errorf("扫描对话失败: %w", err)
+		}
+
+		// 尝试多种时间格式解析
+		var err1, err2 error
+		conv.CreatedAt, err1 = time.Parse("2006-01-02 15:04:05.999999999-07:00", createdAt)
+		if err1 != nil {
+			conv.CreatedAt, err1 = time.Parse("2006-01-02 15:04:05", createdAt)
+		}
+		if err1 != nil {
+			conv.CreatedAt, _ = time.Parse(time.RFC3339, createdAt)
+		}
+
+		conv.UpdatedAt, err2 = time.Parse("2006-01-02 15:04:05.999999999-07:00", updatedAt)
+		if err2 != nil {
+			conv.UpdatedAt, err2 = time.Parse("2006-01-02 15:04:05", updatedAt)
+		}
+		if err2 != nil {
+			conv.UpdatedAt, _ = time.Parse(time.RFC3339, updatedAt)
+		}
+
+		conv.Pinned = pinned != 0
+
+		conversations = append(conversations, &conv)
+	}
+
+	return conversations, nil
+}
+
+// GetGroupByConversation 获取对话所属的分组
+func (db *DB) GetGroupByConversation(conversationID string) (string, error) {
+	var groupID string
+	err := db.QueryRow(
+		"SELECT group_id FROM conversation_group_mappings WHERE conversation_id = ? LIMIT 1",
+		conversationID,
+	).Scan(&groupID)
+	if err != nil {
+		if err == sql.ErrNoRows {
+			return "", nil // 没有分组
+		}
+		return "", fmt.Errorf("查询对话分组失败: %w", err)
+	}
+	return groupID, nil
+}
+
+// UpdateConversationPinned 更新对话置顶状态
+func (db *DB) UpdateConversationPinned(id string, pinned bool) error {
+	pinnedValue := 0
+	if pinned {
+		pinnedValue = 1
+	}
+	_, err := db.Exec(
+		"UPDATE conversations SET pinned = ?, updated_at = ? WHERE id = ?",
+		pinnedValue, time.Now(), id,
+	)
+	if err != nil {
+		return fmt.Errorf("更新对话置顶状态失败: %w", err)
+	}
+	return nil
+}
+
+// UpdateGroupPinned 更新分组置顶状态
+func (db *DB) UpdateGroupPinned(id string, pinned bool) error {
+	pinnedValue := 0
+	if pinned {
+		pinnedValue = 1
+	}
+	_, err := db.Exec(
+		"UPDATE conversation_groups SET pinned = ?, updated_at = ? WHERE id = ?",
+		pinnedValue, time.Now(), id,
+	)
+	if err != nil {
+		return fmt.Errorf("更新分组置顶状态失败: %w", err)
+	}
+	return nil
+}