From f80cb4a0fe26f5ebbb9658da671771dc79d416d6 Mon Sep 17 00:00:00 2001 From: Kyren223 Date: Sun, 6 Jul 2025 14:40:32 +0300 Subject: Implemented server-side authentication in accordance to the protocol --- internal/server/api/helpers.go | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'internal/server/api/helpers.go') diff --git a/internal/server/api/helpers.go b/internal/server/api/helpers.go index 4327a49..2ed0694 100644 --- a/internal/server/api/helpers.go +++ b/internal/server/api/helpers.go @@ -79,8 +79,7 @@ func NetworkPropagateWithFilter( context, cancel := context.WithTimeout(context.Background(), timeout) go func() { defer cancel() - pkt := packet.NewPacket(packet.NewMsgPackEncoder(payload)) - if ok := session.Write(context, pkt); !ok { + if ok := session.Write(context, WrapPayload(payload)); !ok { log.Println(sess.Addr(), "propagation to", session.Addr(), "failed") } }() @@ -122,8 +121,7 @@ func UserPropagate( context, cancel := context.WithTimeout(context.Background(), timeout) go func() { defer cancel() - pkt := packet.NewPacket(packet.NewMsgPackEncoder(payload)) - if ok := session.Write(context, pkt); !ok { + if ok := session.Write(context, WrapPayload(payload)); !ok { log.Println(sess.Addr(), "propagation to", session.Addr(), "failed") } }() @@ -191,3 +189,7 @@ func getNotifications(ctx context.Context, userId snowflake.ID) (packet.Notifica } return items, nil } + +func WrapPayload(payload packet.Payload) packet.Packet { + return packet.NewPacket(packet.NewMsgPackEncoder(payload)) +} -- cgit v1.3.1