Fixed Unknown Procedure Type - Next.js App Router

PotatoHD404 · Jan 15, 2024 · 51505ca · 51505ca · PotatoHD404 · Jan 15, 2024
1 parent d1eab87
commit 51505ca
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/utils/procedure.ts b/src/utils/procedure.ts
@@ -20,9 +20,10 @@ export const getInputOutputParsers = (procedure: OpenApiProcedure) => {
 };
 
 const getProcedureType = (procedure: OpenApiProcedure): ProcedureType => {
-  if (procedure._def.query) return 'query';
-  if (procedure._def.mutation) return 'mutation';
-  if (procedure._def.subscription) return 'subscription';
+  const allowedProcedureTypes = ['query', 'mutation', 'subscription']
+  if (allowedProcedureTypes.includes(procedure._def.type)) {
+    return procedure._def.type;
+  }
   throw new Error('Unknown procedure type');
 };