feat(data): update message queries

1 files changed, 22 insertions, 8 deletions

diff --git a/internal/packet/messages.go b/internal/packet/messages.go
index c561079..04929cd 100644
--- a/internal/packet/messages.go
+++ b/internal/packet/messages.go
@@ -2,25 +2,20 @@ package packet
 
 import (
 	"github.com/kyren223/eko/internal/data"
+	"github.com/kyren223/eko/pkg/snowflake"
 )
 
 type ErrorMessage struct {
 	Error string `msgpack:"error"`
 }
 
-func NewOkMessage() *ErrorMessage {
-	return &ErrorMessage{}
-}
-
 func (m *ErrorMessage) Type() PacketType {
 	return PacketError
 }
 
-func (m *ErrorMessage) IsOk() bool {
-	return m.Error == ""
-}
-
 type SendMessage struct {
+	ReceiverID *snowflake.ID
+	FrequencyID *snowflake.ID
 	Content string
 }
 
@@ -28,6 +23,14 @@ func (m *SendMessage) Type() PacketType {
 	return PacketSendMessage
 }
 
+type PushedMessages struct {
+	Messages []data.Message
+}
+
+func (m *PushedMessages) Type() PacketType {
+	return PacketPushedMessages
+}
+
 type Messages struct {
 	Messages []data.Message
 }
@@ -35,3 +38,14 @@ type Messages struct {
 func (m *Messages) Type() PacketType {
 	return PacketMessages
 }
+
+type GetMessagesRange struct {
+	FrequencyID *snowflake.ID
+	ReceiverID *snowflake.ID
+	From *int64
+	To *int64
+}
+
+func (m *GetMessagesRange) Type() PacketType {
+	return PacketGetMessageRange
+}