nexusql 0.9.6 → 0.9.8

package/dist/cli.js

@@ -3216,13 +3216,15 @@ var Database = class _Database {
   async installExtensions(extensions) {
     for (const ext of extensions) {
       try {
-        await this.exec(`CREATE EXTENSION IF NOT EXISTS ${ext};`);
+        const needsQuoting = /[^a-z0-9_]/i.test(ext);
+        const quotedExt = needsQuoting ? `"${ext}"` : ext;
+        await this.exec(`CREATE EXTENSION IF NOT EXISTS ${quotedExt};`);
         const result = await this.queryParams(
           `SELECT EXISTS(SELECT 1 FROM pg_extension WHERE extname = $1) as exists`,
           [ext]
         );
         if (!result.rows[0]?.exists) {
-          throw new Error(`Extension "${ext}" was not installed`);
+          throw new Error(`Extension ${quotedExt} was not installed`);
         }
       } catch (error) {
         const message = error instanceof Error ? error.message : String(error);
@@ -3412,8 +3414,16 @@ async function gen(options = {}) {
   const tempDb = db.withDatabase(tempDbName);
   let migrationSql = "";
   try {
+    const hasVectorType = targetSql.toLowerCase().includes("vector");
+    const extensions = [...config.extensions];
+    if (hasVectorType && !extensions.includes("vector")) {
+      spinner.info(
+        "Detected 'vector' type usage - auto-enabling vector extension"
+      );
+      extensions.push("vector");
+    }
     spinner.start("Installing extensions...");
-    await tempDb.installExtensions(config.extensions);
+    await tempDb.installExtensions(extensions);
     if (config.extensions.includes("vector")) {
       const vectorCheck = await tempDb.query(
         "SELECT typname FROM pg_type WHERE typname = 'vector'"
@@ -3428,9 +3438,7 @@ async function gen(options = {}) {
     spinner.succeed("Installed db extensions");
     spinner.start("Loading schema into temp database...");
     if (targetSql.toLowerCase().includes("vector")) {
-      const vectorLines = targetSql.split("\n").filter(
-        (line) => line.toLowerCase().includes("vector")
-      );
+      const vectorLines = targetSql.split("\n").filter((line) => line.toLowerCase().includes("vector"));
       console.log("\n[DEBUG] SQL lines with 'vector':");
       vectorLines.forEach((line) => console.log("  ", line.trim()));
       console.log("");