fix: "update" keyword treat as mutation

dalechyn · Jan 29, 2025 · 3b0d86a · 3b0d86a
1 parent b4fc7ab
commit 3b0d86a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/cli/commands/Generate.ts b/src/cli/commands/Generate.ts
@@ -106,7 +106,9 @@ export async function generate(options: Generate = {}) {
 
         await Promise.all(
           functionNames.map(async (functionName) => {
-            const hookType = functionName === 'create' ? 'Mutation' : 'Query'
+            const hookType = ['create', 'update'].includes(functionName)
+              ? 'Mutation'
+              : 'Query'
             const outputHook = `use${namespaceName}${functionName.slice(0, 1).toUpperCase()}${functionName.slice(1)}${hookType}`
             await fs.ensureDir(resolve(config.out, outputHook))