drizzle-auto 1.1.1 → 1.1.2

package/index.js

@@ -6,25 +6,17 @@ const chokidar = require("chokidar");
 const { spawn } = require("child_process");
 
 /* =========================
-🎨 COLORS (clean & pro)
+🎨 COLORS (CLEAN + PRO)
 ========================= */
 const C = {
   reset: "\x1b[0m",
   dim: "\x1b[2m",
-  bold: "\x1b[1m",
   cyan: "\x1b[36m",
   green: "\x1b[32m",
   yellow: "\x1b[33m",
   red: "\x1b[31m",
   magenta: "\x1b[35m",
-};
-
-const log = {
-  info: (m) => console.log(`${C.cyan}ℹ ${m}${C.reset}`),
-  ok: (m) => console.log(`${C.green}✓ ${m}${C.reset}`),
-  warn: (m) => console.log(`${C.yellow}⚠ ${m}${C.reset}`),
-  err: (m) => console.log(`${C.red}✗ ${m}${C.reset}`),
-  trig: (m) => console.log(`${C.magenta}⚡ ${m}${C.reset}`),
+  bold: "\x1b[1m",
 };
 
 /* =========================
@@ -34,9 +26,9 @@ const ROOT = process.cwd();
 const PKG = path.join(ROOT, "package.json");
 
 /* =========================
-🧩 SCRIPT INJECTOR
+🧩 SCRIPT INJECTOR (REAL FIX)
 ========================= */
-function injectScript() {
+function injectTeaScript() {
   if (!fs.existsSync(PKG)) return;
 
   const pkg = JSON.parse(fs.readFileSync(PKG, "utf8"));
@@ -45,137 +37,126 @@ function injectScript() {
   if (!pkg.scripts.tea) {
     pkg.scripts.tea = "drizzle-auto";
     fs.writeFileSync(PKG, JSON.stringify(pkg, null, 2));
-    log.ok("Added script → npm run tea");
+    console.log(
+      `${C.green}☕ Added script:${C.reset} ${C.bold}npm run tea${C.reset}`
+    );
   }
 }
 
 /* =========================
-🔍 FIND DRIZZLE CONFIG
-(js | mjs | ts | mts)
+🔍 FIND DRIZZLE CONFIG (ANYWHERE)
 ========================= */
 function findDrizzleConfig() {
-  const exts = ["js", "mjs", "ts", "mts"];
-  for (const e of exts) {
-    const p = path.join(ROOT, `drizzle.config.${e}`);
-    if (fs.existsSync(p)) return p;
+  const exts = ["js", "mjs", "cjs", "ts", "mts"];
+  let found = null;
+
+  function walk(dir) {
+    if (found) return;
+    for (const file of fs.readdirSync(dir, { withFileTypes: true })) {
+      if (file.name === "node_modules" || file.name.startsWith(".")) continue;
+      const full = path.join(dir, file.name);
+
+      if (file.isDirectory()) walk(full);
+      else {
+        for (const e of exts) {
+          if (file.name === `drizzle.config.${e}`) {
+            found = full;
+            return;
+          }
+        }
+      }
+    }
   }
-  return null;
+
+  walk(ROOT);
+  return found;
 }
 
 /* =========================
-🔎 FILE FILTER (ALL TYPES)
+🧪 VALIDATE SCHEMA EXISTS
 ========================= */
-const WATCH_EXTS = [
-  ".js", ".mjs", ".cjs",
-  ".ts", ".mts",
-  ".jsx", ".tsx",
-  ".sql"
-];
-
-function isValidFile(file) {
-  return WATCH_EXTS.includes(path.extname(file));
+function validateConfig(configPath) {
+  try {
+    const txt = fs.readFileSync(configPath, "utf8");
+    const m = txt.match(/schema:\s*["'](.+?)["']/);
+    if (!m) return true;
+
+    const schemaPath = path.resolve(path.dirname(configPath), m[1]);
+    return fs.existsSync(schemaPath);
+  } catch {
+    return false;
+  }
 }
 
 /* =========================
-⚙️ SAFE SPAWN (NO SHELL)
+⚙️ SAFE RUNNER (NO SERVER KILL)
 ========================= */
-function exec(cmd, args = []) {
+function run(cmd, args) {
   return new Promise((resolve) => {
-    const p = spawn(cmd, args, { stdio: "inherit" });
-
-    p.on("error", (e) => {
-      log.err(`Process error: ${e.message}`);
-      resolve(false);
+    const p = spawn(cmd, args, {
+      stdio: "inherit",
+      shell: false,
     });
 
-    p.on("close", (code) => {
-      resolve(code === 0);
-    });
+    p.on("close", (code) => resolve(code === 0));
   });
 }
 
 /* =========================
-🛡️ CRASH PROTECTION
+🔁 PIPELINE (SERVER SAFE)
 ========================= */
-process.on("uncaughtException", (e) => {
-  log.err(`Crash prevented: ${e.message}`);
-});
-process.on("unhandledRejection", (e) => {
-  log.err(`Promise rejected: ${e}`);
-});
-
-/* =========================
-🔁 PIPELINE (DEV SAFE)
-========================= */
-let running = false;
-let lastPush = 0;
-
-const DEV_MODE = process.argv.includes("--dev");
-const PUSH_COOLDOWN = DEV_MODE ? 120_000 : 30_000; // safer in dev
+let busy = false;
 
 async function pipeline(trigger) {
-  if (running) return;
-  running = true;
+  if (busy) return;
+  busy = true;
 
-  log.trig(`Trigger → ${trigger}`);
+  console.log(
+    `\n${C.cyan}⚡ Trigger →${C.reset} ${C.bold}${trigger}${C.reset}`
+  );
 
   const config = findDrizzleConfig();
   if (!config) {
-    log.warn("No drizzle.config found");
-    running = false;
+    console.log(`${C.yellow}⚠ No drizzle.config found${C.reset}`);
+    busy = false;
     return;
   }
 
-  log.info("Running drizzle generate…");
-  const gen = await exec("npx", ["drizzle-kit", "generate"]);
-  if (!gen) {
-    log.err("Generate failed — watching continues");
-    running = false;
+  if (!validateConfig(config)) {
+    console.log(`${C.red}❌ Schema path invalid${C.reset}`);
+    busy = false;
     return;
   }
 
-  const now = Date.now();
-  if (now - lastPush < PUSH_COOLDOWN) {
-    log.warn("Push skipped (cooldown)");
-    running = false;
+  console.log(`${C.magenta}▶ drizzle-kit generate${C.reset}`);
+  const genOK = await run("npx", ["drizzle-kit", "generate"]);
+  if (!genOK) {
+    console.log(`${C.red}🛑 Generate failed — watcher alive${C.reset}`);
+    busy = false;
     return;
   }
 
-  log.info("Running drizzle push…");
-  lastPush = now;
-
-  const push = await exec("npx", ["drizzle-kit", "push"]);
-  if (!push) {
-    log.err("Push failed — DB untouched, server alive");
-    running = false;
+  console.log(`${C.magenta}▶ drizzle-kit push${C.reset}`);
+  const pushOK = await run("npx", ["drizzle-kit", "push"]);
+  if (!pushOK) {
+    console.log(`${C.red}🛑 Push failed — watcher alive${C.reset}`);
+    busy = false;
     return;
   }
 
-  log.ok("Drizzle fully synced");
-  running = false;
+  console.log(`${C.green}✨ Drizzle fully synced${C.reset}`);
+  busy = false;
 }
 
 /* =========================
-👀 WATCHER (ALL FILES)
+👀 WATCH EVERYTHING
 ========================= */
-injectScript();
+injectTeaScript();
 pipeline("Initial");
 
-log.info(`Watching files (${DEV_MODE ? "DEV MODE" : "SAFE MODE"})…`);
-
 chokidar.watch(ROOT, {
-  ignored: [
-    /node_modules/,
-    /\.git/,
-    /\.next/,
-    /dist/,
-    /build/
-  ],
+  ignored: [/node_modules/, /\.git/, /\.next/, /dist/],
   ignoreInitial: true,
-  usePolling: true,
-  interval: 300,
 }).on("all", (_, file) => {
-  if (isValidFile(file)) {
-    pipeline(path.relative(ROOT, file));
-  }
+  pipeline(path.relative(ROOT, file));
 });

package/package.json

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