drizzle-auto 1.0.24 → 1.1.0

package/index.js

@@ -6,124 +6,145 @@ const chokidar = require("chokidar");
 const { spawn } = require("child_process");
 
 /* =========================
-   🧭 PROJECT ROOT
+   🎨 COLORS (PRO)
 ========================= */
-const ROOT = process.cwd();
-const PKG = path.join(ROOT, "package.json");
-
-/* =========================
-   🎨 COLORS
-========================= */
-const colors = {
+const C = {
   reset: "\x1b[0m",
-  red: "\x1b[31m",
+  blue: "\x1b[34m",
+  cyan: "\x1b[36m",
   green: "\x1b[32m",
   yellow: "\x1b[33m",
-  cyan: "\x1b[36m",
-  magenta: "\x1b[35m",
+  red: "\x1b[31m",
+  dim: "\x1b[90m",
 };
 
 /* =========================
-   🧩 SCRIPT INJECTOR
+   🧭 ROOT + PKG
+========================= */
+const ROOT = process.cwd();
+const PKG = path.join(ROOT, "package.json");
+
+/* =========================
+   ☕ SCRIPT INJECTOR
 ========================= */
-function injectTeaScript() {
+function injectTea() {
   if (!fs.existsSync(PKG)) return;
+
   const pkg = JSON.parse(fs.readFileSync(PKG, "utf8"));
   pkg.scripts ||= {};
+
   if (!pkg.scripts.tea) {
     pkg.scripts.tea = "drizzle-auto";
     fs.writeFileSync(PKG, JSON.stringify(pkg, null, 2));
-    console.log(`${colors.cyan}☕ Added npm run tea${colors.reset}`);
+    console.log(`${C.cyan}☕ Added script: npm run tea${C.reset}`);
   }
 }
 
 /* =========================
-   🔍 DRIZZLE CONFIG DETECTION
+   🔍 RECURSIVE SEARCH
 ========================= */
+function walk(dir, results = []) {
+  for (const file of fs.readdirSync(dir)) {
+    const p = path.join(dir, file);
+    if (p.includes("node_modules") || p.includes(".git")) continue;
+
+    const stat = fs.statSync(p);
+    if (stat.isDirectory()) walk(p, results);
+    else results.push(p);
+  }
+  return results;
+}
+
 function findDrizzleConfig() {
-  const files = ["drizzle.config.js","drizzle.config.mjs","drizzle.config.ts","drizzle.config.mts"];
-  const found = files.map(f => path.join(ROOT, f)).find(fs.existsSync);
-  if (!found) {
-    console.log(`${colors.yellow}⚠️ No drizzle.config found${colors.reset}`);
+  const files = walk(ROOT);
+  const config = files.find(f =>
+    /drizzle\.config\.(js|mjs|ts|mts)$/.test(f)
+  );
+
+  if (!config) {
+    console.log(`${C.yellow}⚠️ drizzle.config not found (anywhere)${C.reset}`);
     return null;
   }
 
-  let schema = null;
+  let schemaPath = null;
   try {
-    const content = fs.readFileSync(found, "utf8");
+    const content = fs.readFileSync(config, "utf8");
     const match = content.match(/schema:\s*["'](.+?)["']/);
-    if (match) schema = path.join(ROOT, match[1]);
+    if (match) {
+      schemaPath = path.resolve(path.dirname(config), match[1]);
+    }
   } catch {}
-  
-  if (schema && !fs.existsSync(schema)) {
-    console.log(`${colors.red}❌ Schema file missing: ${schema}${colors.reset}`);
+
+  if (schemaPath && !fs.existsSync(schemaPath)) {
+    console.log(`${C.red}❌ Schema missing: ${schemaPath}${C.reset}`);
     return null;
   }
 
-  return { config: found, schema };
+  console.log(`${C.green}✔ Drizzle config found${C.reset}`);
+  return true;
 }
 
 /* =========================
-   ⚙️ SAFE RUNNER
+   ⚙️ COMMAND RUNNER
 ========================= */
 function run(cmd) {
   return new Promise(resolve => {
     let failed = false;
     const p = spawn("npx", cmd, { shell: true });
-    p.stdout.on("data", d => process.stdout.write(`${colors.green}${d}${colors.reset}`));
+
+    p.stdout.on("data", d =>
+      process.stdout.write(C.dim + d.toString() + C.reset)
+    );
     p.stderr.on("data", d => {
       failed = true;
-      process.stderr.write(`${colors.red}${d}${colors.reset}`);
+      process.stderr.write(C.red + d.toString() + C.reset);
     });
+
     p.on("close", code => resolve(code === 0 && !failed));
   });
 }
 
 /* =========================
-   🔁 PIPELINE
+   🔁 PIPELINE (SAFE)
 ========================= */
-let busy = false;
+let running = false;
 
 async function pipeline(trigger) {
-  if (busy) return;
-  busy = true;
-  console.log(`${colors.magenta}\n⚡ Trigger → ${trigger}${colors.reset}`);
+  if (running) return;
+  running = true;
+
+  console.log(`${C.blue}\n⚡ Trigger → ${trigger}${C.reset}`);
 
-  const drizzle = findDrizzleConfig();
-  if (!drizzle) {
-    busy = false;
+  if (!findDrizzleConfig()) {
+    running = false;
     return;
   }
 
-  // 1️⃣ Generate
-  const genOK = await run(["drizzle-kit", "generate"]);
-  if (!genOK) {
-    console.log(`${colors.red}🛑 Generate failed — pipeline paused${colors.reset}`);
-    busy = false;
+  if (!(await run(["drizzle-kit", "generate"]))) {
+    console.log(`${C.red}🛑 generate failed (watcher alive)${C.reset}`);
+    running = false;
     return;
   }
 
-  // 2️⃣ Push
-  const pushOK = await run(["drizzle-kit", "push"]);
-  if (!pushOK) {
-    console.log(`${colors.red}🛑 Push failed — server still running${colors.reset}`);
-    busy = false;
+  if (!(await run(["drizzle-kit", "push"]))) {
+    console.log(`${C.red}🛑 push failed (watcher alive)${C.reset}`);
+    running = false;
     return;
   }
 
-  console.log(`${colors.cyan}✨ Drizzle fully synced${colors.reset}`);
-  busy = false;
+  console.log(`${C.green}✨ Drizzle synced${C.reset}`);
+  running = false;
 }
 
 /* =========================
    👀 WATCHER
 ========================= */
-injectTeaScript();
-pipeline("Initial");
+injectTea();
+pipeline("initial");
 
 chokidar.watch(ROOT, {
   ignored: [/node_modules/, /\.git/, /\.next/, /dist/],
   ignoreInitial: true,
   usePolling: true,
-  interval: 300
-}).on("all", (_, file) => pipeline(file));
+  interval: 300,
+}).on("all", (_, file) => pipeline(path.basename(file)));

package/package.json

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