drizzle-auto 1.1.25 → 1.1.26

package/index.js

@@ -1,20 +1,20 @@
 #!/usr/bin/env node
 
 /**
- * 🍵 Drizzle-Auto v2.1 - Fail-Fast Edition
- * 🔹 Strictly catches WebSocket/Neon ENOTFOUND errors
- * 🔹 Stops immediately on any failure
- * 🔹 Debounced watcher to prevent race conditions
+ * 🍵 Drizzle-Auto v2.3 - Flexible Config + Fail-Fast
+ * 🔹 Detects any drizzle.config.{js,ts,mjs,cjs,mts} automatically
+ * 🔹 Stops immediately if config or schema missing
+ * 🔹 Beginner-friendly logs
  */
 
 const { spawn } = require("child_process");
 const path = require("path");
 const fs = require("fs");
 
-// Optional TS support for config parsing
-try { require("ts-node").register({ transpileOnly: true }); } catch (e) {}
+// TS support
+try { require("ts-node").register({ transpileOnly: true }); } catch(e){}
 
-// ---------------------- COLORS & UI ----------------------
+// ---------------- COLORS ----------------
 const T = {
   reset: "\x1b[0m",
   bold: "\x1b[1m",
@@ -25,152 +25,167 @@ const T = {
   gray: "\x1b[38;5;244m",
 };
 
-const frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
-let spinIdx = 0, spinInterval;
+let spinInterval;
+const frames = ["⠋","⠙","⠹","⠸","⠼","⠴","⠦","⠧","⠇","⠏"];
+let spinIdx = 0;
 
-function startLoading(msg) {
+function startLoading(msg){
   stopLoading();
-  spinInterval = setInterval(() => {
+  spinInterval = setInterval(()=>{
     process.stdout.write(`\r${T.cyan}${frames[spinIdx++ % frames.length]}${T.reset} ${T.bold}${msg}${T.reset} `);
-  }, 80);
+  },80);
 }
 
-function stopLoading() {
-  if (spinInterval) {
+function stopLoading(){
+  if(spinInterval){
     clearInterval(spinInterval);
-    spinInterval = null;
+    spinInterval=null;
     process.stdout.write("\r\x1b[K");
   }
 }
 
-// ---------------------- UTILS ----------------------
-const PKG_PATH = path.join(process.cwd(), "package.json");
-function injectScript() {
-  if (!fs.existsSync(PKG_PATH)) return;
-  try {
-    const pkg = JSON.parse(fs.readFileSync(PKG_PATH, "utf8"));
+// ---------------- ADD "tea" SCRIPT ----------------
+const PKG = path.join(process.cwd(),"package.json");
+function addTeaScript(){
+  if(!fs.existsSync(PKG)) return;
+  try{
+    const pkg = JSON.parse(fs.readFileSync(PKG,"utf8"));
     pkg.scripts ||= {};
-    if (!pkg.scripts.tea) {
+    if(!pkg.scripts.tea){
       pkg.scripts.tea = "drizzle-auto";
-      fs.writeFileSync(PKG_PATH, JSON.stringify(pkg, null, 2));
-      console.log(`${T.lime}🍵 Script added → npm run tea${T.reset}`);
+      fs.writeFileSync(PKG, JSON.stringify(pkg,null,2));
+      console.log(`${T.lime}🍵 Added script → npm run tea${T.reset}`);
     }
-  } catch (e) {}
+  } catch(e){}
 }
 
-// ---------------------- COMMAND RUNNER (STRICT) ----------------------
-function runCommand(cmd, label) {
-  return new Promise((resolve, reject) => {
+// ---------------- FIND ANY DRIZZLE CONFIG ----------------
+function findConfig(){
+  const allowedExts = ["js","ts","mjs","cjs","mts"];
+  const rootFiles = fs.readdirSync(process.cwd());
+  for(const file of rootFiles){
+    const match = file.match(/^drizzle\.config\.(.+)$/i);
+    if(match && allowedExts.includes(match[1])) return path.join(process.cwd(),file);
+  }
+  return null;
+}
+
+// ---------------- CHECK SCHEMA ----------------
+function checkSchema(configPath){
+  try{
+    const cfg = require(configPath);
+    const schemaPath = cfg.default?.schema || cfg.schema;
+    if(!schemaPath){
+      console.log(`${T.yellow}⚠ Warning: schema not defined in ${path.basename(configPath)}${T.reset}`);
+      return true; // optional
+    }
+    const fullSchema = path.resolve(path.dirname(configPath), schemaPath);
+    if(!fs.existsSync(fullSchema)){
+      console.log(`${T.red}❌ ERROR: Schema file missing → ${fullSchema}${T.reset}`);
+      process.exit(1);
+    }
+    return true;
+  } catch(e){
+    console.log(`${T.red}❌ ERROR: Cannot read config → ${e.message}${T.reset}`);
+    process.exit(1);
+  }
+}
+
+// ---------------- RUN SHELL COMMAND ----------------
+function runCommand(cmd,label){
+  return new Promise((resolve,reject)=>{
     startLoading(label);
-    
-    // We pipe stderr to catch hidden WebSocket/Neon errors that don't trigger exit codes
-    const child = spawn(cmd, { shell: true, stdio: ["inherit", "inherit", "pipe"] });
-    let errorOutput = "";
-
-    child.stderr.on("data", (data) => {
-      const str = data.toString();
-      errorOutput += str;
-      process.stderr.write(str); // Still show the error to the user
+    const child = spawn(cmd,{shell:true,stdio:["inherit","inherit","pipe"]});
+    let stderrData = "";
+    child.stderr.on("data", d=>{
+      stderrData += d.toString();
+      process.stderr.write(d);
     });
-
-    child.on("close", (code) => {
+    child.on("close", code=>{
       stopLoading();
-      
-      // Strict failure check: 
-      // 1. Non-zero exit code
-      // 2. Presence of "ErrorEvent" or "ENOTFOUND" in stderr (Neon driver specific)
-      const hasDriverError = /ErrorEvent|ENOTFOUND|failed to connect/i.test(errorOutput);
-
-      if (code !== 0 || hasDriverError) {
-        reject(new Error(`${label} failed execution.`));
-      } else {
-        resolve();
-      }
+      const hasError = code!==0 || /ErrorEvent|ENOTFOUND|failed to connect/i.test(stderrData);
+      if(hasError) reject(new Error(`${label} failed!`));
+      else resolve();
     });
-
-    child.on("error", (err) => {
+    child.on("error", err=>{
       stopLoading();
       reject(err);
     });
   });
 }
 
-// ---------------------- ENGINE ----------------------
-let isRunning = false;
-let isBroken = false;
+// ---------------- ENGINE ----------------
+let isRunning=false;
+async function triggerEngine(){
+  if(isRunning) return;
+  isRunning=true;
+
+  const configPath = findConfig();
+  if(!configPath){
+    console.log(`${T.red}❌ ERROR: No drizzle config found!${T.reset}`);
+    console.log(`${T.yellow}Expected one of: drizzle.config.js/ts/mjs/cjs/mts${T.reset}`);
+    process.exit(1);
+  }
 
-async function triggerEngine() {
-  if (isRunning) return;
-  isRunning = true;
+  checkSchema(configPath);
 
-  try {
-    const steps = [
-      { name: "Integrity Check", cmd: "npx drizzle-kit check" },
-      { name: "Generate Migration", cmd: "npx drizzle-kit generate" },
-      { name: "Push Database", cmd: "npx drizzle-kit push" }
-    ];
+  const steps = [
+    {name:"Integrity Check", cmd:"npx drizzle-kit check"},
+    {name:"Generate Migration", cmd:"npx drizzle-kit generate"},
+    {name:"Push Database", cmd:"npx drizzle-kit push"}
+  ];
 
-    for (const step of steps) {
-      await runCommand(step.cmd, step.name);
-    }
-
-    console.log(`\n${T.lime}${T.bold}🎉 SUCCESS: Database synced.${T.reset}\n`);
-    isBroken = false;
-  } catch (err) {
-    isBroken = true;
-    console.log(`\n${T.red}${T.bold}🚨 CRITICAL STOP:${T.reset} ${err.message}`);
-    console.log(`${T.yellow}Execution paused. Fix the error and save any file to retry.${T.reset}\n`);
-    
-    if (process.argv.includes('--postinstall')) {
-      process.exit(1);
+  try{
+    for(const step of steps){
+      await runCommand(step.cmd,step.name);
     }
-  } finally {
-    isRunning = false;
+    console.log(`\n${T.lime}🎉 SUCCESS: All steps finished.${T.reset}\n`);
+  } catch(e){
+    console.log(`\n${T.red}🚨 CRITICAL STOP: ${e.message}${T.reset}`);
+    process.exit(1);
+  } finally{
+    isRunning=false;
   }
 }
 
-// ---------------------- MAIN / WATCHER ----------------------
-injectScript();
+// ---------------- MAIN ----------------
+addTeaScript();
 
-const skipWatcher = process.argv.includes('--postinstall');
+const skipWatcher = process.argv.includes("--postinstall");
 
-if (skipWatcher) {
+if(skipWatcher){
   triggerEngine();
-} else {
+}else{
   triggerEngine();
 
-  try {
+  try{
     const chokidar = require("chokidar");
-    let debounceTimer;
-
-    const watcher = chokidar.watch(".", {
-      ignored: [/node_modules/, /\.git/, /\.next/, /dist/, /out/],
-      ignoreInitial: true,
-      usePolling: true,
-      interval: 300
-    });
-
-    watcher.on("all", (event, path) => {
-      clearTimeout(debounceTimer);
-      debounceTimer = setTimeout(() => {
-        if (isBroken) console.log(`${T.cyan}🔄 Change detected in ${path}. Retrying...${T.reset}`);
+    let timer;
+    chokidar.watch(".",{
+      ignored:[/node_modules/,/\.git/,/\.next/,/dist/,/out/],
+      ignoreInitial:true,
+      usePolling:true,
+      interval:300
+    }).on("all",(e,f)=>{
+      clearTimeout(timer);
+      timer=setTimeout(()=>{
+        console.log(`${T.gray}🔄 Change detected, re-running...${T.reset}`);
         triggerEngine();
-      }, 500);
+      },500);
     });
-
-    console.log(`${T.gray}👀 Watching for schema changes...${T.reset}`);
-  } catch (e) {
-    console.log(`${T.yellow}⚠ Chokidar not found. Watch mode disabled.${T.reset}`);
+    console.log(`${T.gray}👀 Watching for changes in config/schema...${T.reset}`);
+  } catch(e){
+    console.log(`${T.yellow}⚠ Chokidar not installed. Watch mode disabled.${T.reset}`);
   }
 }
 
-// ---------------------- GLOBAL ERROR CATCH ----------------------
-process.on("uncaughtException", (err) => {
-  console.log(`\n${T.red}${T.bold}🛡️ UNCAUGHT ERROR:${T.reset} ${err.message}`);
+// ---------------- GLOBAL ERROR CATCH ----------------
+process.on("uncaughtException",err=>{
+  console.log(`${T.red}🛡️ UNCAUGHT ERROR: ${err.message}${T.reset}`);
   process.exit(1);
 });
 
-process.on("SIGINT", () => {
-  console.log(`\n${T.yellow}Shutting down Drizzle-Auto...${T.reset}`);
+process.on("SIGINT",()=>{
+  console.log(`${T.yellow}Shutting down Drizzle-Auto...${T.reset}`);
   process.exit(0);
-});
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "drizzle-auto",
-  "version": "1.1.25",
+  "version": "1.1.26",
   "description": "",
   "license": "ISC",
   "author": "pavel ahmmed hridoy",