Skip to content

Commit 092858f

Browse files
committed
isStartFile XOR isRouteFile
1 parent f7f4c22 commit 092858f

File tree

1 file changed

+19
-12
lines changed

1 file changed

+19
-12
lines changed

packages/tanstackstart-react/src/vite/autoInstrumentMiddleware.ts

Lines changed: 19 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -92,18 +92,25 @@ export function makeAutoInstrumentMiddlewarePlugin(options: AutoInstrumentMiddle
9292
let needsImport = false;
9393
const skippedMiddlewares: string[] = [];
9494

95-
if (isStartFile) {
96-
const result = wrapGlobalMiddleware(transformed, id, debug);
97-
transformed = result.code;
98-
needsImport = needsImport || result.didWrap;
99-
skippedMiddlewares.push(...result.skipped);
100-
}
101-
102-
if (isRouteFile) {
103-
const result = wrapRouteMiddleware(transformed, id, debug);
104-
transformed = result.code;
105-
needsImport = needsImport || result.didWrap;
106-
skippedMiddlewares.push(...result.skipped);
95+
switch (true) {
96+
// global middleware
97+
case isStartFile: {
98+
const result = wrapGlobalMiddleware(transformed, id, debug);
99+
transformed = result.code;
100+
needsImport = needsImport || result.didWrap;
101+
skippedMiddlewares.push(...result.skipped);
102+
break;
103+
}
104+
// route middleware
105+
case isRouteFile: {
106+
const result = wrapRouteMiddleware(transformed, id, debug);
107+
transformed = result.code;
108+
needsImport = needsImport || result.didWrap;
109+
skippedMiddlewares.push(...result.skipped);
110+
break;
111+
}
112+
default:
113+
break;
107114
}
108115

109116
// Warn about middlewares that couldn't be auto-wrapped

0 commit comments

Comments
 (0)