x

internal/wechatdb/datasource/darwinv3/datasource.go

@@ -0,0 +1,510 @@
+package darwinv3
+
+import (
+	"context"
+	"crypto/md5"
+	"database/sql"
+	"encoding/hex"
+	"fmt"
+	"log"
+	"strings"
+	"time"
+
+	"github.com/sjzar/chatlog/internal/model"
+	"github.com/sjzar/chatlog/pkg/util"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+const (
+	MessageFilePattern  = "^msg_([0-9]?[0-9])?\\.db$"
+	ContactFilePattern  = "^wccontact_new2\\.db$"
+	ChatRoomFilePattern = "^group_new\\.db$"
+	SessionFilePattern  = "^session_new\\.db$"
+)
+
+type DataSource struct {
+	path       string
+	messageDbs []*sql.DB
+	contactDb  *sql.DB
+	chatRoomDb *sql.DB
+	sessionDb  *sql.DB
+
+	talkerDBMap      map[string]*sql.DB
+	user2DisplayName map[string]string
+}
+
+func New(path string) (*DataSource, error) {
+	ds := &DataSource{
+		path:             path,
+		messageDbs:       make([]*sql.DB, 0),
+		talkerDBMap:      make(map[string]*sql.DB),
+		user2DisplayName: make(map[string]string),
+	}
+
+	if err := ds.initMessageDbs(path); err != nil {
+		return nil, fmt.Errorf("初始化消息数据库失败: %w", err)
+	}
+	if err := ds.initContactDb(path); err != nil {
+		return nil, fmt.Errorf("初始化联系人数据库失败: %w", err)
+	}
+	if err := ds.initChatRoomDb(path); err != nil {
+		return nil, fmt.Errorf("初始化群聊数据库失败: %w", err)
+	}
+	if err := ds.initSessionDb(path); err != nil {
+		return nil, fmt.Errorf("初始化会话数据库失败: %w", err)
+	}
+
+	return ds, nil
+}
+
+func (ds *DataSource) initMessageDbs(path string) error {
+
+	files, err := util.FindFilesWithPatterns(path, MessageFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找消息数据库文件失败: %w", err)
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("未找到任何消息数据库文件: %s", path)
+	}
+
+	// 处理每个数据库文件
+	for _, filePath := range files {
+		// 连接数据库
+		db, err := sql.Open("sqlite3", filePath)
+		if err != nil {
+			log.Printf("警告: 连接数据库 %s 失败: %v", filePath, err)
+			continue
+		}
+		ds.messageDbs = append(ds.messageDbs, db)
+
+		// 获取所有表名
+		rows, err := db.Query("SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'Chat_%'")
+		if err != nil {
+			log.Printf("警告: 获取表名失败: %v", err)
+			continue
+		}
+
+		for rows.Next() {
+			var tableName string
+			if err := rows.Scan(&tableName); err != nil {
+				log.Printf("警告: 扫描表名失败: %v", err)
+				continue
+			}
+
+			// 从表名中提取可能的talker信息
+			talkerMd5 := extractTalkerFromTableName(tableName)
+			if talkerMd5 == "" {
+				continue
+			}
+			ds.talkerDBMap[talkerMd5] = db
+		}
+		rows.Close()
+
+	}
+	return nil
+}
+
+func (ds *DataSource) initContactDb(path string) error {
+
+	files, err := util.FindFilesWithPatterns(path, ContactFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找联系人数据库文件失败: %w", err)
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("未找到联系人数据库文件: %s", path)
+	}
+
+	ds.contactDb, err = sql.Open("sqlite3", files[0])
+	if err != nil {
+		return fmt.Errorf("连接联系人数据库失败: %w", err)
+	}
+
+	return nil
+}
+
+func (ds *DataSource) initChatRoomDb(path string) error {
+	files, err := util.FindFilesWithPatterns(path, ChatRoomFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找群聊数据库文件失败: %w", err)
+	}
+	if len(files) == 0 {
+		return fmt.Errorf("未找到群聊数据库文件: %s", path)
+	}
+	ds.chatRoomDb, err = sql.Open("sqlite3", files[0])
+	if err != nil {
+		return fmt.Errorf("连接群聊数据库失败: %w", err)
+	}
+
+	rows, err := ds.chatRoomDb.Query("SELECT m_nsUsrName, nickname FROM GroupMember")
+	if err != nil {
+		log.Printf("警告: 获取群聊成员失败: %v", err)
+		return nil
+	}
+
+	for rows.Next() {
+		var user string
+		var nickName string
+		if err := rows.Scan(&user, &nickName); err != nil {
+			log.Printf("警告: 扫描表名失败: %v", err)
+			continue
+		}
+		ds.user2DisplayName[user] = nickName
+	}
+	rows.Close()
+
+	return nil
+}
+
+func (ds *DataSource) initSessionDb(path string) error {
+	files, err := util.FindFilesWithPatterns(path, SessionFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找最近会话数据库文件失败: %w", err)
+	}
+	if len(files) == 0 {
+		return fmt.Errorf("未找到最近会话数据库文件: %s", path)
+	}
+	ds.sessionDb, err = sql.Open("sqlite3", files[0])
+	if err != nil {
+		return fmt.Errorf("连接最近会话数据库失败: %w", err)
+	}
+	return nil
+}
+
+// GetMessages 实现获取消息的方法
+func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+	// 在 darwinv3 中，每个联系人/群聊的消息存储在单独的表中，表名为 Chat_md5(talker)
+	// 首先需要找到对应的表名
+	if talker == "" {
+		return nil, fmt.Errorf("talker 不能为空")
+	}
+
+	_talkerMd5Bytes := md5.Sum([]byte(talker))
+	talkerMd5 := hex.EncodeToString(_talkerMd5Bytes[:])
+	db, ok := ds.talkerDBMap[talkerMd5]
+	if !ok {
+		return nil, fmt.Errorf("未找到 talker %s 的消息数据库", talker)
+	}
+	tableName := fmt.Sprintf("Chat_%s", talkerMd5)
+
+	// 构建查询条件
+	query := fmt.Sprintf(`
+		SELECT msgCreateTime, msgContent, messageType, mesDes, msgSource, CompressContent, ConBlob
+		FROM %s 
+		WHERE msgCreateTime >= ? AND msgCreateTime <= ? 
+		ORDER BY msgCreateTime ASC
+	`, tableName)
+
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := db.QueryContext(ctx, query, startTime.Unix(), endTime.Unix())
+	if err != nil {
+		return nil, fmt.Errorf("查询表 %s 失败: %w", tableName, err)
+	}
+	defer rows.Close()
+
+	// 处理查询结果
+	messages := []*model.Message{}
+	for rows.Next() {
+		var msg model.MessageDarwinV3
+		var compressContent, conBlob []byte
+		err := rows.Scan(
+			&msg.MesCreateTime,
+			&msg.MesContent,
+			&msg.MesType,
+			&msg.MesDes,
+			&msg.MesSource,
+			&compressContent,
+			&conBlob,
+		)
+		if err != nil {
+			log.Printf("警告: 扫描消息行失败: %v", err)
+			continue
+		}
+
+		// 将消息包装为通用模型
+		message := msg.Wrap(talker)
+		messages = append(messages, message)
+	}
+
+	return messages, nil
+}
+
+// 从表名中提取 talker
+func extractTalkerFromTableName(tableName string) string {
+
+	if !strings.HasPrefix(tableName, "Chat_") {
+		return ""
+	}
+
+	if strings.HasSuffix(tableName, "_dels") {
+		return ""
+	}
+
+	return strings.TrimPrefix(tableName, "Chat_")
+}
+
+// GetContacts 实现获取联系人信息的方法
+func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset int) ([]*model.Contact, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT m_nsUsrName, nickname, IFNULL(m_nsRemark,""), m_uiSex, IFNULL(m_nsAliasName,"") 
+				FROM WCContact 
+				WHERE m_nsUsrName = ? OR nickname = ? OR m_nsRemark = ? OR m_nsAliasName = ?`
+		args = []interface{}{key, key, key, key}
+	} else {
+		// 查询所有联系人
+		query = `SELECT m_nsUsrName, nickname, IFNULL(m_nsRemark,""), m_uiSex, IFNULL(m_nsAliasName,"") 
+				FROM WCContact`
+	}
+
+	// 添加排序、分页
+	query += ` ORDER BY m_nsUsrName`
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询联系人失败: %w", err)
+	}
+	defer rows.Close()
+
+	contacts := []*model.Contact{}
+	for rows.Next() {
+		var contactDarwinV3 model.ContactDarwinV3
+		err := rows.Scan(
+			&contactDarwinV3.M_nsUsrName,
+			&contactDarwinV3.Nickname,
+			&contactDarwinV3.M_nsRemark,
+			&contactDarwinV3.M_uiSex,
+			&contactDarwinV3.M_nsAliasName,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描联系人行失败: %w", err)
+		}
+
+		contacts = append(contacts, contactDarwinV3.Wrap())
+	}
+
+	return contacts, nil
+}
+
+// GetChatRooms 实现获取群聊信息的方法
+func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offset int) ([]*model.ChatRoom, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT m_nsUsrName, nickname, IFNULL(m_nsRemark,""), IFNULL(m_nsChatRoomMemList,""), IFNULL(m_nsChatRoomAdminList,"") 
+				FROM GroupContact 
+				WHERE m_nsUsrName = ? OR nickname = ? OR m_nsRemark = ?`
+		args = []interface{}{key, key, key}
+	} else {
+		// 查询所有群聊
+		query = `SELECT m_nsUsrName, nickname, IFNULL(m_nsRemark,""), IFNULL(m_nsChatRoomMemList,""), IFNULL(m_nsChatRoomAdminList,"") 
+				FROM GroupContact`
+	}
+
+	// 添加排序、分页
+	query += ` ORDER BY m_nsUsrName`
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.chatRoomDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询群聊失败: %w", err)
+	}
+	defer rows.Close()
+
+	chatRooms := []*model.ChatRoom{}
+	for rows.Next() {
+		var chatRoomDarwinV3 model.ChatRoomDarwinV3
+		err := rows.Scan(
+			&chatRoomDarwinV3.M_nsUsrName,
+			&chatRoomDarwinV3.Nickname,
+			&chatRoomDarwinV3.M_nsRemark,
+			&chatRoomDarwinV3.M_nsChatRoomMemList,
+			&chatRoomDarwinV3.M_nsChatRoomAdminList,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+		}
+
+		chatRooms = append(chatRooms, chatRoomDarwinV3.Wrap(ds.user2DisplayName))
+	}
+
+	// 如果没有找到群聊，尝试通过联系人查找
+	if len(chatRooms) == 0 && key != "" {
+		contacts, err := ds.GetContacts(ctx, key, 1, 0)
+		if err == nil && len(contacts) > 0 && strings.HasSuffix(contacts[0].UserName, "@chatroom") {
+			// 再次尝试通过用户名查找群聊
+			rows, err := ds.chatRoomDb.QueryContext(ctx,
+				`SELECT m_nsUsrName, nickname, m_nsRemark, m_nsChatRoomMemList, m_nsChatRoomAdminList 
+				FROM GroupContact 
+				WHERE m_nsUsrName = ?`,
+				contacts[0].UserName)
+
+			if err != nil {
+				return nil, fmt.Errorf("查询群聊失败: %w", err)
+			}
+			defer rows.Close()
+
+			for rows.Next() {
+				var chatRoomDarwinV3 model.ChatRoomDarwinV3
+				err := rows.Scan(
+					&chatRoomDarwinV3.M_nsUsrName,
+					&chatRoomDarwinV3.Nickname,
+					&chatRoomDarwinV3.M_nsRemark,
+					&chatRoomDarwinV3.M_nsChatRoomMemList,
+					&chatRoomDarwinV3.M_nsChatRoomAdminList,
+				)
+
+				if err != nil {
+					return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+				}
+
+				chatRooms = append(chatRooms, chatRoomDarwinV3.Wrap(ds.user2DisplayName))
+			}
+
+			// 如果群聊记录不存在，但联系人记录存在，创建一个模拟的群聊对象
+			if len(chatRooms) == 0 {
+				chatRooms = append(chatRooms, &model.ChatRoom{
+					Name:  contacts[0].UserName,
+					Users: make([]model.ChatRoomUser, 0),
+				})
+			}
+		}
+	}
+
+	return chatRooms, nil
+}
+
+// GetSessions 实现获取会话信息的方法
+func (ds *DataSource) GetSessions(ctx context.Context, key string, limit, offset int) ([]*model.Session, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT m_nsUserName, m_uLastTime 
+				FROM SessionAbstract 
+				WHERE m_nsUserName = ?`
+		args = []interface{}{key}
+	} else {
+		// 查询所有会话
+		query = `SELECT m_nsUserName, m_uLastTime 
+				FROM SessionAbstract`
+	}
+
+	// 添加排序、分页
+	query += ` ORDER BY m_uLastTime DESC`
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.sessionDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询会话失败: %w", err)
+	}
+	defer rows.Close()
+
+	sessions := []*model.Session{}
+	for rows.Next() {
+		var sessionDarwinV3 model.SessionDarwinV3
+		err := rows.Scan(
+			&sessionDarwinV3.M_nsUserName,
+			&sessionDarwinV3.M_uLastTime,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描会话行失败: %w", err)
+		}
+
+		// 包装成通用模型
+		session := sessionDarwinV3.Wrap()
+
+		// 尝试获取联系人信息以补充会话信息
+		contacts, err := ds.GetContacts(ctx, session.UserName, 1, 0)
+		if err == nil && len(contacts) > 0 {
+			session.NickName = contacts[0].DisplayName()
+		} else {
+			// 尝试获取群聊信息
+			chatRooms, err := ds.GetChatRooms(ctx, session.UserName, 1, 0)
+			if err == nil && len(chatRooms) > 0 {
+				session.NickName = chatRooms[0].DisplayName()
+			}
+		}
+
+		sessions = append(sessions, session)
+	}
+
+	return sessions, nil
+}
+
+// Close 实现关闭数据库连接的方法
+func (ds *DataSource) Close() error {
+	var errs []error
+
+	// 关闭消息数据库连接
+	for i, db := range ds.messageDbs {
+		if err := db.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭消息数据库 %d 失败: %w", i, err))
+		}
+	}
+
+	// 关闭联系人数据库连接
+	if ds.contactDb != nil {
+		if err := ds.contactDb.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭联系人数据库失败: %w", err))
+		}
+	}
+
+	// 关闭群聊数据库连接
+	if ds.chatRoomDb != nil {
+		if err := ds.chatRoomDb.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭群聊数据库失败: %w", err))
+		}
+	}
+
+	// 关闭会话数据库连接
+	if ds.sessionDb != nil {
+		if err := ds.sessionDb.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭会话数据库失败: %w", err))
+		}
+	}
+
+	if len(errs) > 0 {
+		return fmt.Errorf("关闭数据库连接时发生错误: %v", errs)
+	}
+
+	return nil
+}

internal/wechatdb/datasource/datasource.go

@@ -0,0 +1,49 @@
+package datasource
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/sjzar/chatlog/internal/model"
+	"github.com/sjzar/chatlog/internal/wechatdb/datasource/darwinv3"
+	v4 "github.com/sjzar/chatlog/internal/wechatdb/datasource/v4"
+	"github.com/sjzar/chatlog/internal/wechatdb/datasource/windowsv3"
+)
+
+// 错误定义
+var (
+	ErrUnsupportedPlatform = fmt.Errorf("unsupported platform")
+)
+
+type DataSource interface {
+
+	// 消息
+	GetMessages(ctx context.Context, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error)
+
+	// 联系人
+	GetContacts(ctx context.Context, key string, limit, offset int) ([]*model.Contact, error)
+
+	// 群聊
+	GetChatRooms(ctx context.Context, key string, limit, offset int) ([]*model.ChatRoom, error)
+
+	// 最近会话
+	GetSessions(ctx context.Context, key string, limit, offset int) ([]*model.Session, error)
+
+	Close() error
+}
+
+func NewDataSource(path string, platform string, version int) (DataSource, error) {
+	switch {
+	case platform == "windows" && version == 3:
+		return windowsv3.New(path)
+	case platform == "windows" && version == 4:
+		return v4.New(path)
+	case platform == "darwin" && version == 3:
+		return darwinv3.New(path)
+	case platform == "darwin" && version == 4:
+		return v4.New(path)
+	default:
+		return nil, fmt.Errorf("%w: %s v%d", ErrUnsupportedPlatform, platform, version)
+	}
+}

internal/wechatdb/datasource/v4/datasource.go

@@ -0,0 +1,634 @@
+package v4
+
+import (
+	"context"
+	"crypto/md5"
+	"database/sql"
+	"encoding/hex"
+	"fmt"
+	"log"
+	"sort"
+	"strings"
+	"time"
+
+	"github.com/sjzar/chatlog/internal/model"
+	"github.com/sjzar/chatlog/pkg/util"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+const (
+	MessageFilePattern = "^message_([0-9]?[0-9])?\\.db$"
+	ContactFilePattern = "^contact\\.db$"
+	SessionFilePattern = "^session\\.db$"
+)
+
+// MessageDBInfo 存储消息数据库的信息
+type MessageDBInfo struct {
+	FilePath  string
+	StartTime time.Time
+	EndTime   time.Time
+	ID2Name   map[int]string
+}
+
+type DataSource struct {
+	path       string
+	messageDbs map[string]*sql.DB
+	contactDb  *sql.DB
+	sessionDb  *sql.DB
+
+	// 消息数据库信息
+	messageFiles []MessageDBInfo
+}
+
+func New(path string) (*DataSource, error) {
+	ds := &DataSource{
+		path:         path,
+		messageDbs:   make(map[string]*sql.DB),
+		messageFiles: make([]MessageDBInfo, 0),
+	}
+
+	if err := ds.initMessageDbs(path); err != nil {
+		return nil, fmt.Errorf("初始化消息数据库失败: %w", err)
+	}
+	if err := ds.initContactDb(path); err != nil {
+		return nil, fmt.Errorf("初始化联系人数据库失败: %w", err)
+	}
+	if err := ds.initSessionDb(path); err != nil {
+		return nil, fmt.Errorf("初始化会话数据库失败: %w", err)
+	}
+
+	return ds, nil
+}
+
+func (ds *DataSource) initMessageDbs(path string) error {
+	// 查找所有消息数据库文件
+	files, err := util.FindFilesWithPatterns(path, MessageFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找消息数据库文件失败: %w", err)
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("未找到任何消息数据库文件: %s", path)
+	}
+
+	// 处理每个数据库文件
+	for _, filePath := range files {
+		// 连接数据库
+		db, err := sql.Open("sqlite3", filePath)
+		if err != nil {
+			log.Printf("警告: 连接数据库 %s 失败: %v", filePath, err)
+			continue
+		}
+
+		// 获取 Timestamp 表中的开始时间
+		var startTime time.Time
+		var timestamp int64
+
+		row := db.QueryRow("SELECT timestamp FROM Timestamp LIMIT 1")
+		if err := row.Scan(&timestamp); err != nil {
+			log.Printf("警告: 获取数据库 %s 的时间戳失败: %v", filePath, err)
+			db.Close()
+			continue
+		}
+		startTime = time.Unix(timestamp, 0)
+
+		// 获取 ID2Name 映射
+		id2Name := make(map[int]string)
+		rows, err := db.Query("SELECT user_name FROM Name2Id")
+		if err != nil {
+			log.Printf("警告: 获取数据库 %s 的 Name2Id 表失败: %v", filePath, err)
+			db.Close()
+			continue
+		}
+
+		i := 1
+		for rows.Next() {
+			var name string
+			if err := rows.Scan(&name); err != nil {
+				log.Printf("警告: 扫描 Name2Id 行失败: %v", err)
+				continue
+			}
+			id2Name[i] = name
+			i++
+		}
+		rows.Close()
+
+		// 保存数据库信息
+		ds.messageFiles = append(ds.messageFiles, MessageDBInfo{
+			FilePath:  filePath,
+			StartTime: startTime,
+			ID2Name:   id2Name,
+		})
+
+		// 保存数据库连接
+		ds.messageDbs[filePath] = db
+	}
+
+	// 按照 StartTime 排序数据库文件
+	sort.Slice(ds.messageFiles, func(i, j int) bool {
+		return ds.messageFiles[i].StartTime.Before(ds.messageFiles[j].StartTime)
+	})
+
+	// 设置结束时间
+	for i := range ds.messageFiles {
+		if i == len(ds.messageFiles)-1 {
+			ds.messageFiles[i].EndTime = time.Now()
+		} else {
+			ds.messageFiles[i].EndTime = ds.messageFiles[i+1].StartTime
+		}
+	}
+
+	return nil
+}
+
+func (ds *DataSource) initContactDb(path string) error {
+	files, err := util.FindFilesWithPatterns(path, ContactFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找联系人数据库文件失败: %w", err)
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("未找到联系人数据库文件: %s", path)
+	}
+
+	ds.contactDb, err = sql.Open("sqlite3", files[0])
+	if err != nil {
+		return fmt.Errorf("连接联系人数据库失败: %w", err)
+	}
+
+	return nil
+}
+
+func (ds *DataSource) initSessionDb(path string) error {
+	files, err := util.FindFilesWithPatterns(path, SessionFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找最近会话数据库文件失败: %w", err)
+	}
+	if len(files) == 0 {
+		return fmt.Errorf("未找到最近会话数据库文件: %s", path)
+	}
+	ds.sessionDb, err = sql.Open("sqlite3", files[0])
+	if err != nil {
+		return fmt.Errorf("连接最近会话数据库失败: %w", err)
+	}
+	return nil
+}
+
+// getDBInfosForTimeRange 获取时间范围内的数据库信息
+func (ds *DataSource) getDBInfosForTimeRange(startTime, endTime time.Time) []MessageDBInfo {
+	var dbs []MessageDBInfo
+	for _, info := range ds.messageFiles {
+		if info.StartTime.Before(endTime) && info.EndTime.After(startTime) {
+			dbs = append(dbs, info)
+		}
+	}
+	return dbs
+}
+
+func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+	if talker == "" {
+		return nil, fmt.Errorf("必须指定 talker 参数")
+	}
+
+	// 找到时间范围内的数据库文件
+	dbInfos := ds.getDBInfosForTimeRange(startTime, endTime)
+	if len(dbInfos) == 0 {
+		return nil, fmt.Errorf("未找到时间范围 %v 到 %v 内的数据库文件", startTime, endTime)
+	}
+
+	if len(dbInfos) == 1 {
+		// LIMIT 和 OFFSET 逻辑在单文件情况下可以直接在 SQL 里处理
+		return ds.getMessagesSingleFile(ctx, dbInfos[0], startTime, endTime, talker, limit, offset)
+	}
+
+	// 从每个相关数据库中查询消息
+	totalMessages := []*model.Message{}
+
+	for _, dbInfo := range dbInfos {
+		// 检查上下文是否已取消
+		if err := ctx.Err(); err != nil {
+			return nil, err
+		}
+
+		db, ok := ds.messageDbs[dbInfo.FilePath]
+		if !ok {
+			log.Printf("警告: 数据库 %s 未打开", dbInfo.FilePath)
+			continue
+		}
+
+		messages, err := ds.getMessagesFromDB(ctx, db, dbInfo, startTime, endTime, talker)
+		if err != nil {
+			log.Printf("警告: 从数据库 %s 获取消息失败: %v", dbInfo.FilePath, err)
+			continue
+		}
+
+		totalMessages = append(totalMessages, messages...)
+
+		if limit+offset > 0 && len(totalMessages) >= limit+offset {
+			break
+		}
+	}
+
+	// 对所有消息按时间排序
+	sort.Slice(totalMessages, func(i, j int) bool {
+		return totalMessages[i].Sequence < totalMessages[j].Sequence
+	})
+
+	// 处理分页
+	if limit > 0 {
+		if offset >= len(totalMessages) {
+			return []*model.Message{}, nil
+		}
+		end := offset + limit
+		if end > len(totalMessages) {
+			end = len(totalMessages)
+		}
+		return totalMessages[offset:end], nil
+	}
+
+	return totalMessages, nil
+}
+
+// getMessagesSingleFile 从单个数据库文件获取消息
+func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+	db, ok := ds.messageDbs[dbInfo.FilePath]
+	if !ok {
+		return nil, fmt.Errorf("数据库 %s 未打开", dbInfo.FilePath)
+	}
+
+	// 构建表名
+	_talkerMd5Bytes := md5.Sum([]byte(talker))
+	talkerMd5 := hex.EncodeToString(_talkerMd5Bytes[:])
+	tableName := "Msg_" + talkerMd5
+
+	// 构建查询条件
+	conditions := []string{"create_time >= ? AND create_time <= ?"}
+	args := []interface{}{startTime.Unix(), endTime.Unix()}
+
+	query := fmt.Sprintf(`
+		SELECT sort_seq, local_type, real_sender_id, create_time, message_content, packed_info_data, status
+		FROM %s 
+		WHERE %s 
+		ORDER BY sort_seq ASC
+	`, tableName, strings.Join(conditions, " AND "))
+
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := db.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询数据库 %s 失败: %w", dbInfo.FilePath, err)
+	}
+	defer rows.Close()
+
+	// 处理查询结果
+	messages := []*model.Message{}
+	isChatRoom := strings.HasSuffix(talker, "@chatroom")
+
+	for rows.Next() {
+		var msg model.MessageV4
+		err := rows.Scan(
+			&msg.SortSeq,
+			&msg.LocalType,
+			&msg.RealSenderID,
+			&msg.CreateTime,
+			&msg.MessageContent,
+			&msg.PackedInfoData,
+			&msg.Status,
+		)
+		if err != nil {
+			return nil, fmt.Errorf("扫描消息行失败: %w", err)
+		}
+
+		messages = append(messages, msg.Wrap(dbInfo.ID2Name, isChatRoom))
+	}
+
+	return messages, nil
+}
+
+// getMessagesFromDB 从数据库获取消息
+func (ds *DataSource) getMessagesFromDB(ctx context.Context, db *sql.DB, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string) ([]*model.Message, error) {
+	// 构建表名
+	_talkerMd5Bytes := md5.Sum([]byte(talker))
+	talkerMd5 := hex.EncodeToString(_talkerMd5Bytes[:])
+	tableName := "Msg_" + talkerMd5
+
+	// 检查表是否存在
+	var exists bool
+	err := db.QueryRowContext(ctx,
+		"SELECT 1 FROM sqlite_master WHERE type='table' AND name=?",
+		tableName).Scan(&exists)
+
+	if err != nil {
+		if err == sql.ErrNoRows {
+			// 表不存在，返回空结果
+			return []*model.Message{}, nil
+		}
+		return nil, fmt.Errorf("检查表 %s 是否存在失败: %w", tableName, err)
+	}
+
+	// 构建查询条件
+	conditions := []string{"create_time >= ? AND create_time <= ?"}
+	args := []interface{}{startTime.Unix(), endTime.Unix()}
+
+	query := fmt.Sprintf(`
+		SELECT sort_seq, local_type, real_sender_id, create_time, message_content, packed_info_data, status
+		FROM %s 
+		WHERE %s 
+		ORDER BY sort_seq ASC
+	`, tableName, strings.Join(conditions, " AND "))
+
+	// 执行查询
+	rows, err := db.QueryContext(ctx, query, args...)
+	if err != nil {
+		// 如果表不存在，SQLite 会返回错误
+		if strings.Contains(err.Error(), "no such table") {
+			return []*model.Message{}, nil
+		}
+		return nil, fmt.Errorf("查询数据库失败: %w", err)
+	}
+	defer rows.Close()
+
+	// 处理查询结果
+	messages := []*model.Message{}
+	isChatRoom := strings.HasSuffix(talker, "@chatroom")
+
+	for rows.Next() {
+		var msg model.MessageV4
+		err := rows.Scan(
+			&msg.SortSeq,
+			&msg.LocalType,
+			&msg.RealSenderID,
+			&msg.CreateTime,
+			&msg.MessageContent,
+			&msg.PackedInfoData,
+			&msg.Status,
+		)
+		if err != nil {
+			return nil, fmt.Errorf("扫描消息行失败: %w", err)
+		}
+
+		messages = append(messages, msg.Wrap(dbInfo.ID2Name, isChatRoom))
+	}
+
+	return messages, nil
+}
+
+// 联系人
+func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset int) ([]*model.Contact, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT username, local_type, alias, remark, nick_name 
+				FROM contact 
+				WHERE username = ? OR alias = ? OR remark = ? OR nick_name = ?`
+		args = []interface{}{key, key, key, key}
+	} else {
+		// 查询所有联系人
+		query = `SELECT username, local_type, alias, remark, nick_name FROM contact`
+	}
+
+	// 添加排序、分页
+	query += ` ORDER BY username`
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询联系人失败: %w", err)
+	}
+	defer rows.Close()
+
+	contacts := []*model.Contact{}
+	for rows.Next() {
+		var contactV4 model.ContactV4
+		err := rows.Scan(
+			&contactV4.UserName,
+			&contactV4.LocalType,
+			&contactV4.Alias,
+			&contactV4.Remark,
+			&contactV4.NickName,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描联系人行失败: %w", err)
+		}
+
+		contacts = append(contacts, contactV4.Wrap())
+	}
+
+	return contacts, nil
+}
+
+// 群聊
+func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offset int) ([]*model.ChatRoom, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT username, owner, ext_buffer FROM chat_room WHERE username = ?`
+		args = []interface{}{key}
+
+		// 执行查询
+		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		if err != nil {
+			return nil, fmt.Errorf("查询群聊失败: %w", err)
+		}
+		defer rows.Close()
+
+		chatRooms := []*model.ChatRoom{}
+		for rows.Next() {
+			var chatRoomV4 model.ChatRoomV4
+			err := rows.Scan(
+				&chatRoomV4.UserName,
+				&chatRoomV4.Owner,
+				&chatRoomV4.ExtBuffer,
+			)
+
+			if err != nil {
+				return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+			}
+
+			chatRooms = append(chatRooms, chatRoomV4.Wrap())
+		}
+
+		// 如果没有找到群聊，尝试通过联系人查找
+		if len(chatRooms) == 0 {
+			contacts, err := ds.GetContacts(ctx, key, 1, 0)
+			if err == nil && len(contacts) > 0 && strings.HasSuffix(contacts[0].UserName, "@chatroom") {
+				// 再次尝试通过用户名查找群聊
+				rows, err := ds.contactDb.QueryContext(ctx,
+					`SELECT username, owner, ext_buffer FROM chat_room WHERE username = ?`,
+					contacts[0].UserName)
+
+				if err != nil {
+					return nil, fmt.Errorf("查询群聊失败: %w", err)
+				}
+				defer rows.Close()
+
+				for rows.Next() {
+					var chatRoomV4 model.ChatRoomV4
+					err := rows.Scan(
+						&chatRoomV4.UserName,
+						&chatRoomV4.Owner,
+						&chatRoomV4.ExtBuffer,
+					)
+
+					if err != nil {
+						return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+					}
+
+					chatRooms = append(chatRooms, chatRoomV4.Wrap())
+				}
+
+				// 如果群聊记录不存在，但联系人记录存在，创建一个模拟的群聊对象
+				if len(chatRooms) == 0 {
+					chatRooms = append(chatRooms, &model.ChatRoom{
+						Name:             contacts[0].UserName,
+						Users:            make([]model.ChatRoomUser, 0),
+						User2DisplayName: make(map[string]string),
+					})
+				}
+			}
+		}
+
+		return chatRooms, nil
+	} else {
+		// 查询所有群聊
+		query = `SELECT username, owner, ext_buffer FROM chat_room`
+
+		// 添加排序、分页
+		query += ` ORDER BY username`
+		if limit > 0 {
+			query += fmt.Sprintf(" LIMIT %d", limit)
+			if offset > 0 {
+				query += fmt.Sprintf(" OFFSET %d", offset)
+			}
+		}
+
+		// 执行查询
+		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		if err != nil {
+			return nil, fmt.Errorf("查询群聊失败: %w", err)
+		}
+		defer rows.Close()
+
+		chatRooms := []*model.ChatRoom{}
+		for rows.Next() {
+			var chatRoomV4 model.ChatRoomV4
+			err := rows.Scan(
+				&chatRoomV4.UserName,
+				&chatRoomV4.Owner,
+				&chatRoomV4.ExtBuffer,
+			)
+
+			if err != nil {
+				return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+			}
+
+			chatRooms = append(chatRooms, chatRoomV4.Wrap())
+		}
+
+		return chatRooms, nil
+	}
+}
+
+// 最近会话
+func (ds *DataSource) GetSessions(ctx context.Context, key string, limit, offset int) ([]*model.Session, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT username, summary, last_timestamp, last_msg_sender, last_sender_display_name 
+				FROM SessionTable 
+				WHERE username = ? OR last_sender_display_name = ?
+				ORDER BY sort_timestamp DESC`
+		args = []interface{}{key, key}
+	} else {
+		// 查询所有会话
+		query = `SELECT username, summary, last_timestamp, last_msg_sender, last_sender_display_name 
+				FROM SessionTable 
+				ORDER BY sort_timestamp DESC`
+	}
+
+	// 添加分页
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.sessionDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询会话失败: %w", err)
+	}
+	defer rows.Close()
+
+	sessions := []*model.Session{}
+	for rows.Next() {
+		var sessionV4 model.SessionV4
+		err := rows.Scan(
+			&sessionV4.Username,
+			&sessionV4.Summary,
+			&sessionV4.LastTimestamp,
+			&sessionV4.LastMsgSender,
+			&sessionV4.LastSenderDisplayName,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描会话行失败: %w", err)
+		}
+
+		sessions = append(sessions, sessionV4.Wrap())
+	}
+
+	return sessions, nil
+}
+
+func (ds *DataSource) Close() error {
+	var errs []error
+
+	// 关闭消息数据库连接
+	for path, db := range ds.messageDbs {
+		if err := db.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭消息数据库 %s 失败: %w", path, err))
+		}
+	}
+
+	// 关闭联系人数据库连接
+	if ds.contactDb != nil {
+		if err := ds.contactDb.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭联系人数据库失败: %w", err))
+		}
+	}
+
+	// 关闭会话数据库连接
+	if ds.sessionDb != nil {
+		if err := ds.sessionDb.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭会话数据库失败: %w", err))
+		}
+	}
+
+	if len(errs) > 0 {
+		return fmt.Errorf("关闭数据库连接时发生错误: %v", errs)
+	}
+
+	return nil
+}

internal/wechatdb/datasource/windowsv3/datasource.go

@@ -0,0 +1,615 @@
+package windowsv3
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"log"
+	"sort"
+	"strings"
+	"time"
+
+	"github.com/sjzar/chatlog/internal/model"
+	"github.com/sjzar/chatlog/pkg/util"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+const (
+	MessageFilePattern = "^MSG([0-9]?[0-9])?\\.db$"
+	ContactFilePattern = "^MicroMsg.db$"
+)
+
+// MessageDBInfo 保存消息数据库的信息
+type MessageDBInfo struct {
+	FilePath  string
+	StartTime time.Time
+	EndTime   time.Time
+	TalkerMap map[string]int
+}
+
+// DataSource 实现了 DataSource 接口
+type DataSource struct {
+	// 消息数据库
+	messageFiles []MessageDBInfo
+	messageDbs   map[string]*sql.DB
+
+	// 联系人数据库
+	contactDbFile string
+	contactDb     *sql.DB
+}
+
+// New 创建一个新的 WindowsV3DataSource
+func New(path string) (*DataSource, error) {
+	ds := &DataSource{
+		messageFiles: make([]MessageDBInfo, 0),
+		messageDbs:   make(map[string]*sql.DB),
+	}
+
+	// 初始化消息数据库
+	if err := ds.initMessageDbs(path); err != nil {
+		return nil, fmt.Errorf("初始化消息数据库失败: %w", err)
+	}
+
+	// 初始化联系人数据库
+	if err := ds.initContactDb(path); err != nil {
+		return nil, fmt.Errorf("初始化联系人数据库失败: %w", err)
+	}
+
+	return ds, nil
+}
+
+// initMessageDbs 初始化消息数据库
+func (ds *DataSource) initMessageDbs(path string) error {
+	// 查找所有消息数据库文件
+	files, err := util.FindFilesWithPatterns(path, MessageFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找消息数据库文件失败: %w", err)
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("未找到任何消息数据库文件: %s", path)
+	}
+
+	// 处理每个数据库文件
+	for _, filePath := range files {
+		// 连接数据库
+		db, err := sql.Open("sqlite3", filePath)
+		if err != nil {
+			log.Printf("警告: 连接数据库 %s 失败: %v", filePath, err)
+			continue
+		}
+
+		// 获取 DBInfo 表中的开始时间
+		var startTime time.Time
+
+		rows, err := db.Query("SELECT tableIndex, tableVersion, tableDesc FROM DBInfo")
+		if err != nil {
+			log.Printf("警告: 查询数据库 %s 的 DBInfo 表失败: %v", filePath, err)
+			db.Close()
+			continue
+		}
+
+		for rows.Next() {
+			var tableIndex int
+			var tableVersion int64
+			var tableDesc string
+
+			if err := rows.Scan(&tableIndex, &tableVersion, &tableDesc); err != nil {
+				log.Printf("警告: 扫描 DBInfo 行失败: %v", err)
+				continue
+			}
+
+			// 查找描述为 "Start Time" 的记录
+			if strings.Contains(tableDesc, "Start Time") {
+				startTime = time.Unix(tableVersion/1000, (tableVersion%1000)*1000000)
+				break
+			}
+		}
+		rows.Close()
+
+		// 组织 TalkerMap
+		talkerMap := make(map[string]int)
+		rows, err = db.Query("SELECT UsrName FROM Name2ID")
+		if err != nil {
+			log.Printf("警告: 查询数据库 %s 的 Name2ID 表失败: %v", filePath, err)
+			db.Close()
+			continue
+		}
+
+		i := 1
+		for rows.Next() {
+			var userName string
+			if err := rows.Scan(&userName); err != nil {
+				log.Printf("警告: 扫描 Name2ID 行失败: %v", err)
+				continue
+			}
+			talkerMap[userName] = i
+			i++
+		}
+		rows.Close()
+
+		// 保存数据库信息
+		ds.messageFiles = append(ds.messageFiles, MessageDBInfo{
+			FilePath:  filePath,
+			StartTime: startTime,
+			TalkerMap: talkerMap,
+		})
+
+		// 保存数据库连接
+		ds.messageDbs[filePath] = db
+	}
+
+	// 按照 StartTime 排序数据库文件
+	sort.Slice(ds.messageFiles, func(i, j int) bool {
+		return ds.messageFiles[i].StartTime.Before(ds.messageFiles[j].StartTime)
+	})
+
+	// 设置结束时间
+	for i := range ds.messageFiles {
+		if i == len(ds.messageFiles)-1 {
+			ds.messageFiles[i].EndTime = time.Now()
+		} else {
+			ds.messageFiles[i].EndTime = ds.messageFiles[i+1].StartTime
+		}
+	}
+
+	return nil
+}
+
+// initContactDb 初始化联系人数据库
+func (ds *DataSource) initContactDb(path string) error {
+	files, err := util.FindFilesWithPatterns(path, ContactFilePattern, true)
+	if err != nil {
+		return fmt.Errorf("查找联系人数据库文件失败: %w", err)
+	}
+
+	if len(files) == 0 {
+		return fmt.Errorf("未找到联系人数据库文件: %s", path)
+	}
+
+	ds.contactDbFile = files[0]
+
+	ds.contactDb, err = sql.Open("sqlite3", ds.contactDbFile)
+	if err != nil {
+		return fmt.Errorf("连接联系人数据库失败: %w", err)
+	}
+
+	return nil
+}
+
+// getDBInfosForTimeRange 获取时间范围内的数据库信息
+func (ds *DataSource) getDBInfosForTimeRange(startTime, endTime time.Time) []MessageDBInfo {
+	var dbs []MessageDBInfo
+	for _, info := range ds.messageFiles {
+		if info.StartTime.Before(endTime) && info.EndTime.After(startTime) {
+			dbs = append(dbs, info)
+		}
+	}
+	return dbs
+}
+
+// GetMessages 实现 DataSource 接口的 GetMessages 方法
+func (ds *DataSource) GetMessages(ctx context.Context, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+	// 找到时间范围内的数据库文件
+	dbInfos := ds.getDBInfosForTimeRange(startTime, endTime)
+	if len(dbInfos) == 0 {
+		return nil, fmt.Errorf("未找到时间范围 %v 到 %v 内的数据库文件", startTime, endTime)
+	}
+
+	if len(dbInfos) == 1 {
+		// LIMIT 和 OFFSET 逻辑在单文件情况下可以直接在 SQL 里处理
+		return ds.getMessagesSingleFile(ctx, dbInfos[0], startTime, endTime, talker, limit, offset)
+	}
+
+	// 从每个相关数据库中查询消息
+	totalMessages := []*model.Message{}
+
+	for _, dbInfo := range dbInfos {
+		// 检查上下文是否已取消
+		if err := ctx.Err(); err != nil {
+			return nil, err
+		}
+
+		db, ok := ds.messageDbs[dbInfo.FilePath]
+		if !ok {
+			log.Printf("警告: 数据库 %s 未打开", dbInfo.FilePath)
+			continue
+		}
+
+		// 构建查询条件
+		conditions := []string{"Sequence >= ? AND Sequence <= ?"}
+		args := []interface{}{startTime.Unix() * 1000, endTime.Unix() * 1000}
+
+		if len(talker) > 0 {
+			talkerID, ok := dbInfo.TalkerMap[talker]
+			if ok {
+				conditions = append(conditions, "TalkerId = ?")
+				args = append(args, talkerID)
+			} else {
+				conditions = append(conditions, "StrTalker = ?")
+				args = append(args, talker)
+			}
+		}
+
+		query := fmt.Sprintf(`
+            SELECT Sequence, CreateTime, TalkerId, StrTalker, IsSender, 
+                Type, SubType, StrContent, CompressContent, BytesExtra
+            FROM MSG 
+            WHERE %s 
+            ORDER BY Sequence ASC
+        `, strings.Join(conditions, " AND "))
+
+		// 执行查询
+		rows, err := db.QueryContext(ctx, query, args...)
+		if err != nil {
+			log.Printf("警告: 查询数据库 %s 失败: %v", dbInfo.FilePath, err)
+			continue
+		}
+
+		// 处理查询结果
+		for rows.Next() {
+			var msg model.MessageV3
+			var compressContent []byte
+			var bytesExtra []byte
+
+			err := rows.Scan(
+				&msg.Sequence,
+				&msg.CreateTime,
+				&msg.TalkerID,
+				&msg.StrTalker,
+				&msg.IsSender,
+				&msg.Type,
+				&msg.SubType,
+				&msg.StrContent,
+				&compressContent,
+				&bytesExtra,
+			)
+			if err != nil {
+				log.Printf("警告: 扫描消息行失败: %v", err)
+				continue
+			}
+			msg.CompressContent = compressContent
+			msg.BytesExtra = bytesExtra
+
+			totalMessages = append(totalMessages, msg.Wrap())
+		}
+		rows.Close()
+
+		if limit+offset > 0 && len(totalMessages) >= limit+offset {
+			break
+		}
+	}
+
+	// 对所有消息按时间排序
+	sort.Slice(totalMessages, func(i, j int) bool {
+		return totalMessages[i].Sequence < totalMessages[j].Sequence
+	})
+
+	// 处理分页
+	if limit > 0 {
+		if offset >= len(totalMessages) {
+			return []*model.Message{}, nil
+		}
+		end := offset + limit
+		if end > len(totalMessages) {
+			end = len(totalMessages)
+		}
+		return totalMessages[offset:end], nil
+	}
+
+	return totalMessages, nil
+}
+
+// getMessagesSingleFile 从单个数据库文件获取消息
+func (ds *DataSource) getMessagesSingleFile(ctx context.Context, dbInfo MessageDBInfo, startTime, endTime time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+	// 构建查询条件
+	conditions := []string{"Sequence >= ? AND Sequence <= ?"}
+	args := []interface{}{startTime.Unix() * 1000, endTime.Unix() * 1000}
+	if len(talker) > 0 {
+		// TalkerId 有索引，优先使用
+		talkerID, ok := dbInfo.TalkerMap[talker]
+		if ok {
+			conditions = append(conditions, "TalkerId = ?")
+			args = append(args, talkerID)
+		} else {
+			conditions = append(conditions, "StrTalker = ?")
+			args = append(args, talker)
+		}
+	}
+	query := fmt.Sprintf(`
+        SELECT Sequence, CreateTime, TalkerId, StrTalker, IsSender, 
+            Type, SubType, StrContent, CompressContent, BytesExtra
+        FROM MSG 
+        WHERE %s 
+        ORDER BY Sequence ASC
+    `, strings.Join(conditions, " AND "))
+
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.messageDbs[dbInfo.FilePath].QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询数据库 %s 失败: %w", dbInfo.FilePath, err)
+	}
+	defer rows.Close()
+
+	// 处理查询结果
+	totalMessages := []*model.Message{}
+	for rows.Next() {
+		var msg model.MessageV3
+		var compressContent []byte
+		var bytesExtra []byte
+		err := rows.Scan(
+			&msg.Sequence,
+			&msg.CreateTime,
+			&msg.TalkerID,
+			&msg.StrTalker,
+			&msg.IsSender,
+			&msg.Type,
+			&msg.SubType,
+			&msg.StrContent,
+			&compressContent,
+			&bytesExtra,
+		)
+		if err != nil {
+			return nil, fmt.Errorf("扫描消息行失败: %w", err)
+		}
+		msg.CompressContent = compressContent
+		msg.BytesExtra = bytesExtra
+		totalMessages = append(totalMessages, msg.Wrap())
+	}
+	return totalMessages, nil
+}
+
+// GetContacts 实现获取联系人信息的方法
+func (ds *DataSource) GetContacts(ctx context.Context, key string, limit, offset int) ([]*model.Contact, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT UserName, Alias, Remark, NickName, Reserved1 FROM Contact 
+                WHERE UserName = ? OR Alias = ? OR Remark = ? OR NickName = ?`
+		args = []interface{}{key, key, key, key}
+	} else {
+		// 查询所有联系人
+		query = `SELECT UserName, Alias, Remark, NickName, Reserved1 FROM Contact`
+	}
+
+	// 添加排序、分页
+	query += ` ORDER BY UserName`
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询联系人失败: %w", err)
+	}
+	defer rows.Close()
+
+	contacts := []*model.Contact{}
+	for rows.Next() {
+		var contactV3 model.ContactV3
+		err := rows.Scan(
+			&contactV3.UserName,
+			&contactV3.Alias,
+			&contactV3.Remark,
+			&contactV3.NickName,
+			&contactV3.Reserved1,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描联系人行失败: %w", err)
+		}
+
+		contacts = append(contacts, contactV3.Wrap())
+	}
+
+	return contacts, nil
+}
+
+// GetChatRooms 实现获取群聊信息的方法
+func (ds *DataSource) GetChatRooms(ctx context.Context, key string, limit, offset int) ([]*model.ChatRoom, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT ChatRoomName, Reserved2, RoomData FROM ChatRoom WHERE ChatRoomName = ?`
+		args = []interface{}{key}
+
+		// 执行查询
+		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		if err != nil {
+			return nil, fmt.Errorf("查询群聊失败: %w", err)
+		}
+		defer rows.Close()
+
+		chatRooms := []*model.ChatRoom{}
+		for rows.Next() {
+			var chatRoomV3 model.ChatRoomV3
+			err := rows.Scan(
+				&chatRoomV3.ChatRoomName,
+				&chatRoomV3.Reserved2,
+				&chatRoomV3.RoomData,
+			)
+
+			if err != nil {
+				return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+			}
+
+			chatRooms = append(chatRooms, chatRoomV3.Wrap())
+		}
+
+		// 如果没有找到群聊，尝试通过联系人查找
+		if len(chatRooms) == 0 {
+			contacts, err := ds.GetContacts(ctx, key, 1, 0)
+			if err == nil && len(contacts) > 0 && strings.HasSuffix(contacts[0].UserName, "@chatroom") {
+				// 再次尝试通过用户名查找群聊
+				rows, err := ds.contactDb.QueryContext(ctx,
+					`SELECT ChatRoomName, Reserved2, RoomData FROM ChatRoom WHERE ChatRoomName = ?`,
+					contacts[0].UserName)
+
+				if err != nil {
+					return nil, fmt.Errorf("查询群聊失败: %w", err)
+				}
+				defer rows.Close()
+
+				for rows.Next() {
+					var chatRoomV3 model.ChatRoomV3
+					err := rows.Scan(
+						&chatRoomV3.ChatRoomName,
+						&chatRoomV3.Reserved2,
+						&chatRoomV3.RoomData,
+					)
+
+					if err != nil {
+						return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+					}
+
+					chatRooms = append(chatRooms, chatRoomV3.Wrap())
+				}
+
+				// 如果群聊记录不存在，但联系人记录存在，创建一个模拟的群聊对象
+				if len(chatRooms) == 0 {
+					chatRooms = append(chatRooms, &model.ChatRoom{
+						Name:             contacts[0].UserName,
+						Users:            make([]model.ChatRoomUser, 0),
+						User2DisplayName: make(map[string]string),
+					})
+				}
+			}
+		}
+
+		return chatRooms, nil
+	} else {
+		// 查询所有群聊
+		query = `SELECT ChatRoomName, Reserved2, RoomData FROM ChatRoom`
+
+		// 添加排序、分页
+		query += ` ORDER BY ChatRoomName`
+		if limit > 0 {
+			query += fmt.Sprintf(" LIMIT %d", limit)
+			if offset > 0 {
+				query += fmt.Sprintf(" OFFSET %d", offset)
+			}
+		}
+
+		// 执行查询
+		rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+		if err != nil {
+			return nil, fmt.Errorf("查询群聊失败: %w", err)
+		}
+		defer rows.Close()
+
+		chatRooms := []*model.ChatRoom{}
+		for rows.Next() {
+			var chatRoomV3 model.ChatRoomV3
+			err := rows.Scan(
+				&chatRoomV3.ChatRoomName,
+				&chatRoomV3.Reserved2,
+				&chatRoomV3.RoomData,
+			)
+
+			if err != nil {
+				return nil, fmt.Errorf("扫描群聊行失败: %w", err)
+			}
+
+			chatRooms = append(chatRooms, chatRoomV3.Wrap())
+		}
+
+		return chatRooms, nil
+	}
+}
+
+// GetSessions 实现获取会话信息的方法
+func (ds *DataSource) GetSessions(ctx context.Context, key string, limit, offset int) ([]*model.Session, error) {
+	var query string
+	var args []interface{}
+
+	if key != "" {
+		// 按照关键字查询
+		query = `SELECT strUsrName, nOrder, strNickName, strContent, nTime 
+                FROM Session 
+                WHERE strUsrName = ? OR strNickName = ?
+                ORDER BY nOrder DESC`
+		args = []interface{}{key, key}
+	} else {
+		// 查询所有会话
+		query = `SELECT strUsrName, nOrder, strNickName, strContent, nTime 
+                FROM Session 
+                ORDER BY nOrder DESC`
+	}
+
+	// 添加分页
+	if limit > 0 {
+		query += fmt.Sprintf(" LIMIT %d", limit)
+		if offset > 0 {
+			query += fmt.Sprintf(" OFFSET %d", offset)
+		}
+	}
+
+	// 执行查询
+	rows, err := ds.contactDb.QueryContext(ctx, query, args...)
+	if err != nil {
+		return nil, fmt.Errorf("查询会话失败: %w", err)
+	}
+	defer rows.Close()
+
+	sessions := []*model.Session{}
+	for rows.Next() {
+		var sessionV3 model.SessionV3
+		err := rows.Scan(
+			&sessionV3.StrUsrName,
+			&sessionV3.NOrder,
+			&sessionV3.StrNickName,
+			&sessionV3.StrContent,
+			&sessionV3.NTime,
+		)
+
+		if err != nil {
+			return nil, fmt.Errorf("扫描会话行失败: %w", err)
+		}
+
+		sessions = append(sessions, sessionV3.Wrap())
+	}
+
+	return sessions, nil
+}
+
+// Close 实现 DataSource 接口的 Close 方法
+func (ds *DataSource) Close() error {
+	var errs []error
+
+	// 关闭消息数据库连接
+	for path, db := range ds.messageDbs {
+		if err := db.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭消息数据库 %s 失败: %w", path, err))
+		}
+	}
+
+	// 关闭联系人数据库连接
+	if ds.contactDb != nil {
+		if err := ds.contactDb.Close(); err != nil {
+			errs = append(errs, fmt.Errorf("关闭联系人数据库失败: %w", err))
+		}
+	}
+
+	if len(errs) > 0 {
+		return fmt.Errorf("关闭数据库连接时发生错误: %v", errs)
+	}
+
+	return nil
+}