Await default property of route module

.changeset/clever-ends-wink.md

@@ -0,0 +1,5 @@
+---
+"next-ws": patch
+---
+
+Await default property of route module

src/server/helpers/route.ts

@@ -123,7 +123,10 @@ async function importModule<T>(modulePath: string): Promise<T> {
   }
 }
 
-export function getSocketHandler(routeModule: RouteModule) {
+export async function getSocketHandler(routeModule: RouteModule) {
+  if (routeModule.default instanceof Promise)
+    return getSocketHandler(await routeModule.default);
+
   return (
     routeModule?.default?.routeModule?.userland?.SOCKET ??
     routeModule?.routeModule?.userland?.SOCKET ??
@@ -133,7 +136,7 @@ export function getSocketHandler(routeModule: RouteModule) {
 }
 
 export interface RouteModule {
-  default?: RouteModule;
+  default?: Promise<RouteModule> | RouteModule;
   routeModule?: { userland?: RouteModule['handlers'] };
   handlers?: { SOCKET?: SocketHandler };
 }

src/server/setup.ts

@@ -41,7 +41,7 @@ export function setupWebSocketServer(nextServer: NextNodeServer) {
       return socket.destroy();
     }
 
-    const socketHandler = getSocketHandler(routeModule);
+    const socketHandler = await getSocketHandler(routeModule);
     if (!socketHandler || typeof socketHandler !== 'function') {
       logger.error(`[next-ws] ${pathname} does not export a SOCKET handler`);
       return socket.destroy();