From 59be303670d464431eee6588dd220c9f36ce3394 Mon Sep 17 00:00:00 2001
From: Carson Full <carsonfull@gmail.com>
Date: Wed, 26 Jul 2023 15:10:55 -0500
Subject: [PATCH] Allow custom suffix for generated query files

---
 packages/generate/src/queries.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/generate/src/queries.ts b/packages/generate/src/queries.ts
index 538727a92..2f3bec4ea 100644
--- a/packages/generate/src/queries.ts
+++ b/packages/generate/src/queries.ts
@@ -8,7 +8,7 @@ import type { Target } from "./genutil";
 // generate per-file queries, then listen for changes and update
 export async function generateQueryFiles(params: {
   root: string | null;
-  options: CommandOptions;
+  options: CommandOptions & { suffix?: string };
   client: Client;
 }) {
   if (params.options.file && params.options.watch) {
@@ -64,6 +64,7 @@ currently supported.`);
             target: params.options.target!,
             path,
             types,
+            suffix: params.options.suffix,
           });
           for (const f of files) {
             if (!filesByExtension[f.extension]) {
@@ -122,6 +123,7 @@ currently supported.`);
         target: params.options.target!,
         path,
         types,
+        suffix: params.options.suffix,
       });
       for (const f of files) {
         const prettyPath = "./" + adapter.path.posix.relative(root, f.path);
@@ -181,6 +183,7 @@ function generateFiles(params: {
   target: Target;
   path: string;
   types: QueryType;
+  suffix?: string;
 }): {
   path: string;
   contents: string;
@@ -192,7 +195,7 @@ function generateFiles(params: {
   const outputDirname = adapter.path.dirname(params.path);
   const outputBaseFileName = adapter.path.join(
     outputDirname,
-    `${baseFileName}.query`
+    `${baseFileName}${params.suffix ?? '.query'}`
   );
 
   const method =