db

type Database interface {
	// Open 初始化数据库
	Open() error

	// GetMessageByGlobalID 通过 GlobalID 来获取消息
	GetMessageByGlobalID(int32) (StoredMessage, error)
	// GetGroupMessageByGlobalID 通过 GlobalID 来获取群消息
	GetGroupMessageByGlobalID(int32) (*StoredGroupMessage, error)
	// GetPrivateMessageByGlobalID 通过 GlobalID 来获取私聊消息
	GetPrivateMessageByGlobalID(int32) (*StoredPrivateMessage, error)
	// GetGuildChannelMessageByID 通过 ID 来获取频道消息
	GetGuildChannelMessageByID(string) (*StoredGuildChannelMessage, error)

	// InsertGroupMessage 向数据库写入新的群消息
	InsertGroupMessage(*StoredGroupMessage) error
	// InsertPrivateMessage 向数据库写入新的私聊消息
	InsertPrivateMessage(*StoredPrivateMessage) error
	// InsertGuildChannelMessage 向数据库写入新的频道消息
	InsertGuildChannelMessage(*StoredGuildChannelMessage) error
}

type QuotedInfo struct {
	PrevID        string       `bson:"prevId" yaml:"-"`
	PrevGlobalID  int32        `bson:"prevGlobalId" yaml:"-"`
	QuotedContent []global.MSG `bson:"quotedContent" yaml:"quoted_content"`
}

type StoredGroupMessage struct {
	ID          string                  `bson:"_id" yaml:"-"`
	GlobalID    int32                   `bson:"globalId" yaml:"-"`
	Attribute   *StoredMessageAttribute `bson:"attribute" yaml:"-"`
	SubType     string                  `bson:"subType" yaml:"-"`
	QuotedInfo  *QuotedInfo             `bson:"quotedInfo" yaml:"-"`
	GroupCode   int64                   `bson:"groupCode" yaml:"-"`
	AnonymousID string                  `bson:"anonymousId" yaml:"-"`
	Content     []global.MSG            `bson:"content" yaml:"content"`
}

type StoredGuildChannelMessage struct {
	ID         string                       `bson:"_id" yaml:"-"`
	Attribute  *StoredGuildMessageAttribute `bson:"attribute" yaml:"-"`
	GuildID    uint64                       `bson:"guildId" yaml:"-"`
	ChannelID  uint64                       `bson:"channelId" yaml:"-"`
	QuotedInfo *QuotedInfo                  `bson:"quotedInfo" yaml:"-"`
	Content    []global.MSG                 `bson:"content" yaml:"content"`
}

type StoredGuildMessageAttribute struct {
	MessageSeq   uint64 `bson:"messageSeq" yaml:"-"`
	InternalID   uint64 `bson:"internalId" yaml:"-"`
	SenderTinyID uint64 `bson:"senderTinyId" yaml:"-"`
	SenderName   string `bson:"senderName" yaml:"-"`
	Timestamp    int64  `bson:"timestamp" yaml:"-"`
}

type StoredMessage interface {
	GetID() string
	GetType() string
	GetGlobalID() int32
	GetAttribute() *StoredMessageAttribute
	GetContent() []global.MSG
}

type StoredMessageAttribute struct {
	MessageSeq int32  `bson:"messageSeq" yaml:"-"`
	InternalID int32  `bson:"internalId" yaml:"-"`
	SenderUin  int64  `bson:"senderUin" yaml:"-"`
	SenderName string `bson:"senderName" yaml:"-"`
	Timestamp  int64  `bson:"timestamp" yaml:"-"`
}

type StoredPrivateMessage struct {
	ID         string                  `bson:"_id" yaml:"-"`
	GlobalID   int32                   `bson:"globalId" yaml:"-"`
	Attribute  *StoredMessageAttribute `bson:"attribute" yaml:"-"`
	SubType    string                  `bson:"subType" yaml:"-"`
	QuotedInfo *QuotedInfo             `bson:"quotedInfo" yaml:"-"`
	SessionUin int64                   `bson:"sessionUin" yaml:"-"`
	TargetUin  int64                   `bson:"targetUin" yaml:"-"`
	Content    []global.MSG            `bson:"content" yaml:"content"`
}