commit d351ba48f8cefc035a68cea49ca957681eafe04f · gmstn.systems/shard

+32 -5
src/index.ts
···

       1
       1
        
       import { setupDbWithMigrations } from "@/db";

     

       2
       2
        
       import { DB_URL, SERVER_PORT } from "@/lib/env";

     

       3
       3
       +
       import { newErrorResponse } from "@/lib/utils/http/responses";

     

       4
       4
       +
       import {

     

       5
       5
       +
           wrapHttpRegistrationCheck,

     

       6
       6
       +
           wrapWsRegistrationCheck,

     

       7
       7
       +
       } from "@/lib/utils/registration";

     

       3
       8
        
       import { routes } from "@/routes";

     

       4
       9
        
       import { setupServer } from "@/server";

     

       5
       10
        
       

     
···

       9
       14
        
           const server = await setupServer();

     

       10
       15
        
           for (const [url, route] of Object.entries(routes)) {

     

       11
       16
        
               if (!route.wsHandler) {

     

       12
       12
       -
                   const { handler, method } = route;

     

       17
       17
       +
                   const { handler, method, skipRegistrationCheck } = route;

     

       13
       18
        
                   server.route({

     

       14
       19
        
                       url,

     

       15
       20
        
                       method,

     

       16
       16
       -
                       handler,

     

       21
       21
       +
                       handler: skipRegistrationCheck

     

       22
       22
       +
                           ? handler

     

       23
       23
       +
                           : wrapHttpRegistrationCheck(handler),

     

       17
       24
        
                   });

     

       18
       25
        
               } else {

     

       19
       19
       -
                   const { wsHandler, method, handler, preHandler } = route;

     

       26
       26
       +
                   const {

     

       27
       27
       +
                       wsHandler,

     

       28
       28
       +
                       method,

     

       29
       29
       +
                       handler: httpHandler,

     

       30
       30
       +
                       preHandler,

     

       31
       31
       +
                       skipRegistrationCheckHttp,

     

       32
       32
       +
                       skipRegistrationCheckWs,

     

       33
       33
       +
                   } = route;

     

       34
       34
       +
       

     

       35
       35
       +
                   const handler =

     

       36
       36
       +
                       httpHandler ??

     

       37
       37
       +
                       (() =>

     

       38
       38
       +
                           newErrorResponse(404, {

     

       39
       39
       +
                               message:

     

       40
       40
       +
                                   "This is a websocket only route. Did you mean to initiate a websocket connection here?",

     

       41
       41
       +
                           }));

     

       42
       42
       +
       

     

       20
       43
        
                   server.register(() => {

     

       21
       44
        
                       server.route({

     

       22
       45
        
                           url,

     

       23
       46
        
                           method: method ?? "GET",

     

       24
       24
       -
                           handler: handler ?? (() => new Response()),

     

       25
       25
       -
                           wsHandler,

     

       47
       47
       +
                           handler: skipRegistrationCheckHttp

     

       48
       48
       +
                               ? handler

     

       49
       49
       +
                               : wrapHttpRegistrationCheck(handler),

     

       50
       50
       +
                           wsHandler: skipRegistrationCheckWs

     

       51
       51
       +
                               ? wsHandler

     

       52
       52
       +
                               : wrapWsRegistrationCheck(wsHandler),

     

       26
       53
        
                           preHandler,

     

       27
       54
        
                       });

     

       28
       55
        
                   });