Initial commit: Эфир мессенджер

internal/websocket/message.go

@@ -0,0 +1,136 @@
+package websocket
+
+import (
+	"encoding/json"
+	"time"
+)
+
+type MessageType string
+
+const (
+	MsgTypeNewMessage    MessageType = "new_message"
+	MsgTypeTyping        MessageType = "typing"
+	MsgTypeReadReceipt   MessageType = "read_receipt"
+	MsgTypeEditMessage   MessageType = "edit_message"
+	MsgTypeDeleteMessage MessageType = "delete_message"
+	MsgTypePing          MessageType = "ping"
+	
+	MsgTypeNewMessageResp MessageType = "new_message"
+	MsgTypeUserTyping     MessageType = "user_typing"
+	MsgTypeMessageRead    MessageType = "message_read"
+	MsgTypeMessageEdited  MessageType = "message_edited"
+	MsgTypeMessageDeleted MessageType = "message_deleted"
+	MsgTypeUserOnline     MessageType = "user_online"
+	MsgTypeUserOffline    MessageType = "user_offline"
+	MsgTypePong           MessageType = "pong"
+	MsgTypeError          MessageType = "error"
+)
+
+type WSMessage struct {
+	Type      MessageType     `json:"type"`
+	Data      json.RawMessage `json:"data,omitempty"`
+	Timestamp time.Time       `json:"timestamp"`
+}
+
+type NewMessageRequest struct {
+	ChatID       int64  `json:"chat_id"`
+	Plaintext    string `json:"plaintext"`
+	AttachmentID *int64 `json:"attachment_id,omitempty"`
+	TempID       string `json:"temp_id,omitempty"`
+}
+
+type NewMessageResponse struct {
+	ID         int64          `json:"id"`
+	ChatID     int64          `json:"chat_id"`
+	SenderID   int64          `json:"sender_id"`
+	Plaintext  string         `json:"plaintext"`
+	Attachment *AttachmentInfo `json:"attachment,omitempty"`
+	CreatedAt  time.Time      `json:"created_at"`
+	TempID     string         `json:"temp_id,omitempty"`
+}
+
+type AttachmentInfo struct {
+	ID       int64  `json:"id"`
+	FileName string `json:"file_name"`
+	FileSize int64  `json:"file_size"`
+	MimeType string `json:"mime_type"`
+}
+
+type TypingRequest struct {
+	ChatID   int64 `json:"chat_id"`
+	IsTyping bool  `json:"is_typing"`
+}
+
+type TypingResponse struct {
+	ChatID   int64 `json:"chat_id"`
+	UserID   int64 `json:"user_id"`
+	IsTyping bool  `json:"is_typing"`
+}
+
+type ReadReceiptRequest struct {
+	MessageID int64 `json:"message_id"`
+}
+
+type ReadReceiptResponse struct {
+	MessageID int64     `json:"message_id"`
+	UserID    int64     `json:"user_id"`
+	ReadAt    time.Time `json:"read_at"`
+}
+
+type EditMessageRequest struct {
+	MessageID int64  `json:"message_id"`
+	Plaintext string `json:"plaintext"`
+}
+
+type EditMessageResponse struct {
+	MessageID    int64     `json:"message_id"`
+	NewPlaintext string    `json:"new_plaintext"`
+	EditedAt     time.Time `json:"edited_at"`
+}
+
+type DeleteMessageRequest struct {
+	MessageID int64 `json:"message_id"`
+}
+
+type DeleteMessageResponse struct {
+	MessageID int64     `json:"message_id"`
+	DeletedAt time.Time `json:"deleted_at"`
+}
+
+type UserOnlineResponse struct {
+	UserID   int64      `json:"user_id"`
+	IsOnline bool       `json:"is_online"`
+	LastSeen *time.Time `json:"last_seen,omitempty"`
+}
+
+type ErrorResponse struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+}
+
+func ParseMessage(raw []byte) (*WSMessage, error) {
+	var msg WSMessage
+	if err := json.Unmarshal(raw, &msg); err != nil {
+		return nil, err
+	}
+	return &msg, nil
+}
+
+func CreateMessage(msgType MessageType, data interface{}) ([]byte, error) {
+	var dataBytes json.RawMessage
+	if data != nil {
+		bytes, err := json.Marshal(data)
+		if err != nil {
+			return nil, err
+		}
+		dataBytes = bytes
+	}
+	
+	msg := WSMessage{
+		Type:      msgType,
+		Data:      dataBytes,
+		Timestamp: time.Now(),
+	}
+	
+	return json.Marshal(msg)
+}