support setting for writing supabase migration files (#427)

src/ipc/processors/response_processor.ts

@@ -14,8 +14,10 @@ import {
   executeSupabaseSql,
 } from "../../supabase_admin/supabase_management_client";
 import { isServerFunction } from "../../supabase_admin/supabase_utils";
-import { SqlQuery } from "../../lib/schemas";
+import { SqlQuery, UserSettings } from "../../lib/schemas";
 import { gitCommit } from "../utils/git_utils";
+import { readSettings } from "@/main/settings";
+import { writeMigrationFile } from "../utils/file_utils";
 
 const readFile = fs.promises.readFile;
 const logger = log.scope("response_processor");
@@ -207,6 +209,7 @@ export async function processFullResponseActions(
     return {};
   }
 
+  const settings: UserSettings = readSettings();
   const appPath = getDyadAppPath(chatWithApp.app.path);
   const writtenFiles: string[] = [];
   const renamedFiles: string[] = [];
@@ -247,6 +250,22 @@ export async function processFullResponseActions(
             supabaseProjectId: chatWithApp.app.supabaseProjectId!,
             query: query.content,
           });
+
+          // Only write migration file if SQL execution succeeded
+          if (settings.enableSupabaseWriteSqlMigration) {
+            try {
+              await writeMigrationFile(
+                appPath,
+                query.content,
+                query.description,
+              );
+            } catch (error) {
+              errors.push({
+                message: `Failed to write SQL migration file for: ${query.description}`,
+                error: error,
+              });
+            }
+          }
         } catch (error) {
           errors.push({
             message: `Failed to execute SQL query: ${query.content}`,

src/ipc/utils/file_utils.ts

@@ -1,6 +1,8 @@
 import fs from "node:fs";
 import { promises as fsPromises } from "node:fs";
 import path from "node:path";
+import fsExtra from "fs-extra";
+import { generateCuteAppName } from "../../lib/utils";
 
 /**
  * Recursively gets all files in a directory, excluding node_modules and .git
@@ -57,3 +59,36 @@ export async function copyDirectoryRecursive(
     }
   }
 }
+
+export async function writeMigrationFile(
+  appPath: string,
+  queryContent: string,
+  queryDescription?: string,
+) {
+  const migrationsDir = path.join(appPath, "supabase", "migrations");
+  await fsExtra.ensureDir(migrationsDir);
+
+  const files = await fsExtra.readdir(migrationsDir);
+  const migrationNumbers = files
+    .map((file) => {
+      const match = file.match(/^(\d{4})_/);
+      return match ? parseInt(match[1], 10) : -1;
+    })
+    .filter((num) => num !== -1);
+
+  const nextMigrationNumber =
+    migrationNumbers.length > 0 ? Math.max(...migrationNumbers) + 1 : 0;
+  const paddedNumber = String(nextMigrationNumber).padStart(4, "0");
+
+  let description = "migration";
+  if (queryDescription) {
+    description = queryDescription.toLowerCase().replace(/[\s\W-]+/g, "_");
+  } else {
+    description = generateCuteAppName().replace(/-/g, "_");
+  }
+
+  const migrationFileName = `${paddedNumber}_${description}.sql`;
+  const migrationFilePath = path.join(migrationsDir, migrationFileName);
+
+  await fsExtra.writeFile(migrationFilePath, queryContent);
+}