fix: allow client only middleware

TanStack · Jan 25, 2025 · 3514996 · 3514996
1 parent b9f44bc
commit 3514996
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/packages/start-plugin/src/compilers.ts b/packages/start-plugin/src/compilers.ts
@@ -413,17 +413,16 @@ function handleCreateMiddlewareCallExpression(
     }
   }
 
-  const useFnPath = callExpressionPaths.server?.get(
+  const serverFnPath = callExpressionPaths.server?.get(
     'arguments.0',
   ) as babel.NodePath<any>
 
-  if (!callExpressionPaths.server || !useFnPath.node) {
-    throw new Error('createMiddleware must be called with a "use" property!')
-  }
-
-  // If we're on the client, remove the server call expression
-
-  if (opts.env === 'client' || opts.env === 'ssr') {
+  if (
+    callExpressionPaths.server &&
+    serverFnPath.node &&
+    (opts.env === 'client' || opts.env === 'ssr')
+  ) {
+    // If we're on the client, remove the server call expression
     if (t.isMemberExpression(callExpressionPaths.server.node.callee)) {
       callExpressionPaths.server.replaceWith(
         callExpressionPaths.server.node.callee.object,