x

internal/wechatdb/wechatdb.go

@@ -0,0 +1,117 @@
+package wechatdb
+
+import (
+	"time"
+
+	"github.com/sjzar/chatlog/pkg/model"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+type DB struct {
+	BasePath string
+	Version  int
+
+	contact *Contact
+	message *Message
+}
+
+func New(path string, version int) (*DB, error) {
+	w := &DB{
+		BasePath: path,
+		Version:  version,
+	}
+
+	// 初始化，加载数据库文件信息
+	if err := w.Initialize(); err != nil {
+		return nil, err
+	}
+
+	return w, nil
+}
+
+func (w *DB) Close() error {
+	return nil
+}
+
+func (w *DB) Initialize() error {
+
+	var err error
+	w.message, err = NewMessage(w.BasePath, w.Version)
+	if err != nil {
+		return err
+	}
+
+	w.contact, err = NewContact(w.BasePath, w.Version)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (w *DB) GetMessages(start, end time.Time, talker string, limit, offset int) ([]*model.Message, error) {
+
+	if talker != "" {
+		if contact := w.contact.GetContact(talker); contact != nil {
+			talker = contact.UserName
+		}
+	}
+
+	messages, err := w.message.GetMessages(start, end, talker, limit, offset)
+	if err != nil {
+		return nil, err
+	}
+	for i := range messages {
+		w.contact.MessageFillInfo(messages[i])
+	}
+
+	return messages, nil
+}
+
+type ListContactResp struct {
+	Items []*model.Contact `json:"items"`
+}
+
+func (w *DB) ListContact() (*ListContactResp, error) {
+	list, err := w.contact.ListContact()
+	if err != nil {
+		return nil, err
+	}
+	return &ListContactResp{
+		Items: list,
+	}, nil
+}
+
+func (w *DB) GetContact(userName string) *model.Contact {
+	return w.contact.GetContact(userName)
+}
+
+type ListChatRoomResp struct {
+	Items []*model.ChatRoom `json:"items"`
+}
+
+func (w *DB) ListChatRoom() (*ListChatRoomResp, error) {
+	list, err := w.contact.ListChatRoom()
+	if err != nil {
+		return nil, err
+	}
+	return &ListChatRoomResp{
+		Items: list,
+	}, nil
+}
+
+func (w *DB) GetChatRoom(userName string) *model.ChatRoom {
+	return w.contact.GetChatRoom(userName)
+}
+
+type GetSessionResp struct {
+	Items []*model.Session `json:"items"`
+}
+
+func (w *DB) GetSession(limit int) (*GetSessionResp, error) {
+	sessions := w.contact.GetSession(limit)
+	return &GetSessionResp{
+		Items: sessions,
+	}, nil
+}