Reorder sections of useHttpServer function

apteryxxyz · Nov 5, 2023 · 3571c27 · 3571c27
1 parent ece070a
commit 3571c27
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/packages/core/package.json b/packages/core/package.json
@@ -1,6 +1,6 @@
 {
   "name": "next-ws",
-  "version": "1.0.1-canary.1",
+  "version": "1.0.1-experimental.1",
   "description": "Add support for WebSockets in Next.js 13 app directory",
   "keywords": [
     "next",

diff --git a/packages/core/src/server/utilities/server.ts b/packages/core/src/server/utilities/server.ts
@@ -13,6 +13,12 @@ export const CustomHttpServer = Symbol('NextWS::CustomHttpServer');
  * @returns The HTTP Server instance.
  */
 export function useHttpServer(nextServer: NextNodeServer) {
+  // NextNodeServer is always required, so check for it before attempting to use custom server
+  if (!nextServer || !(nextServer instanceof NextNodeServer)) {
+    Log.error('[next-ws] could not find the NextNodeServer instance');
+    process.exit(1);
+  }
+
   const existing = Reflect.get(globalThis, CustomHttpServer) as Server;
   if (existing) {
     Log.warnOnce(
@@ -21,11 +27,6 @@ export function useHttpServer(nextServer: NextNodeServer) {
     return existing;
   }
 
-  if (!nextServer || !(nextServer instanceof NextNodeServer)) {
-    Log.error('[next-ws] could not find the NextNodeServer instance');
-    process.exit(1);
-  }
-
   // @ts-expect-error - serverOptions is protected
   const httpServer = nextServer.serverOptions?.httpServer;
   if (!httpServer || !(httpServer instanceof Server)) {