···

       
5
       
 
       
} from "@/lib/sessions";

     

       
6
       
 
       
import { shardMessageSchema } from "@/lib/types/messages";

     

       
7
       
 
       
import type { PreHandler, WsRouteHandler } from "@/lib/types/routes";

     

       
0
       
 
       

     

       
8
       
 
       
import { storeMessageInDb } from "@/lib/utils/gmstn";

     

       
9
       
 
       
import {

     

       
10
       
 
       
    rawDataToString,

     
···

       
71
       
 
       
        return;

     

       
72
       
 
       
    }

     

       
73
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
74
       
 
       
    socket.on("message", (rawData) => {

     

       
75
       
 
       
        const event = rawDataToString(rawData);

     

       
76
       
 
       


     
···

       
96
       
 
       
                    console.error(z.treeifyError(error));

     

       
97
       
 
       
                    return;

     

       
98
       
 
       
                }

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
99
       
 
       
                storeMessageInDb(shardMessage)

     

       
100
       
 
       
                    .then(() => {

     

       
101
       
 
       
                        console.log("stored", shardMessage);

     

···

       
5
       
 
       
} from "@/lib/sessions";

     

       
6
       
 
       
import { shardMessageSchema } from "@/lib/types/messages";

     

       
7
       
 
       
import type { PreHandler, WsRouteHandler } from "@/lib/types/routes";

     

       
8
       
+
       
import { atUriToString } from "@/lib/utils/atproto";

     

       
9
       
 
       
import { storeMessageInDb } from "@/lib/utils/gmstn";

     

       
10
       
 
       
import {

     

       
11
       
 
       
    rawDataToString,

     
···

       
72
       
 
       
        return;

     

       
73
       
 
       
    }

     

       
74
       
 
       


     

       
75
       
+
       
    // convert at uri objects array to set of at uri strings for easier lookup.

     

       
76
       
+
       
    const socketAllowedChannels = new Set(sessionInfo.allowedChannels.map(channel => atUriToString(channel)));

     

       
77
       
+
       


     

       
78
       
 
       
    socket.on("message", (rawData) => {

     

       
79
       
 
       
        const event = rawDataToString(rawData);

     

       
80
       
 
       


     
···

       
100
       
 
       
                    console.error(z.treeifyError(error));

     

       
101
       
 
       
                    return;

     

       
102
       
 
       
                }

     

       
103
       
+
       


     

       
104
       
+
       
                const { channel } = shardMessage

     

       
105
       
+
       
                if(!socketAllowedChannels.has(channel)) return;

     

       
106
       
+
       


     

       
107
       
 
       
                storeMessageInDb(shardMessage)

     

       
108
       
 
       
                    .then(() => {

     

       
109
       
 
       
                        console.log("stored", shardMessage);