mirror of
https://github.com/Kyren223/eko.git
synced 2025-09-07 05:58:21 +00:00
88 lines
2.6 KiB
Go
88 lines
2.6 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ed25519"
|
|
"database/sql"
|
|
"log"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/kyren223/eko/internal/data"
|
|
"github.com/kyren223/eko/internal/packet"
|
|
"github.com/kyren223/eko/internal/server/session"
|
|
"github.com/kyren223/eko/pkg/assert"
|
|
"github.com/kyren223/eko/pkg/snowflake"
|
|
)
|
|
|
|
func SendMessage(ctx context.Context, request *packet.SendMessage) packet.Payload {
|
|
sess, ok := session.FromContext(ctx)
|
|
assert.Assert(ok, "context in process packet should always have a session")
|
|
|
|
if (request.ReceiverID != nil) == (request.FrequencyID != nil) {
|
|
return &packet.Error{Error: "either receiver id or frequency id must exist"}
|
|
}
|
|
|
|
content := strings.TrimSpace(request.Content)
|
|
if content == "" {
|
|
return &packet.Error{Error: "message content must not be blank"}
|
|
}
|
|
|
|
node := sess.Manager().Node()
|
|
|
|
queries := data.New(db)
|
|
message, err := queries.CreateMessage(ctx, data.CreateMessageParams{
|
|
ID: node.Generate(),
|
|
SenderID: sess.ID(),
|
|
Content: content,
|
|
FrequencyID: request.FrequencyID,
|
|
ReceiverID: request.ReceiverID,
|
|
})
|
|
if err != nil {
|
|
log.Println(sess.Addr(), "SendMessage database error:", err)
|
|
return &packet.Error{Error: "internal server error"}
|
|
}
|
|
|
|
return &packet.MessagesInfo{Messages: []data.Message{message}}
|
|
}
|
|
|
|
func GetMessages(ctx context.Context, request *packet.RequestMessages) packet.Payload {
|
|
queries := data.New(db)
|
|
messages, err := queries.GetFrequencyMessages(ctx, request.FrequencyID)
|
|
if err != nil {
|
|
log.Println("database error when retrieving messages:", err)
|
|
return &packet.Error{Error: "internal server error"}
|
|
}
|
|
return &packet.MessagesInfo{Messages: messages}
|
|
}
|
|
|
|
// func GetUserById(ctx context.Context, request *packet.GetUserByID) packet.Payload {
|
|
// queries := data.New(db)
|
|
// user, err := queries.GetUserById(ctx, request.UserID)
|
|
// if err == sql.ErrNoRows {
|
|
// return &packet.Users{Users: []data.User{}}
|
|
// }
|
|
// if err != nil {
|
|
// log.Println("database error when retrieving user by id:", err)
|
|
// return &packet.ErrorMessage{Error: "internal server error"}
|
|
// }
|
|
// return &packet.Users{Users: []data.User{user}}
|
|
// }
|
|
|
|
func CreateOrGetUser(ctx context.Context, node *snowflake.Node, pubKey ed25519.PublicKey) (data.User, error) {
|
|
queries := data.New(db)
|
|
user, err := queries.GetUserByPublicKey(ctx, pubKey)
|
|
if err == sql.ErrNoRows {
|
|
id := node.Generate()
|
|
user, err = queries.CreateUser(ctx, data.CreateUserParams{
|
|
ID: id,
|
|
Name: "User" + strconv.FormatInt(id.Time()%1000, 10),
|
|
PublicKey: pubKey,
|
|
})
|
|
}
|
|
if err != nil {
|
|
return data.User{}, err
|
|
}
|
|
return user, nil
|
|
}
|