drizzle-auto 1.1.0 → 1.1.1

package/index.js

@@ -6,28 +6,37 @@ const chokidar = require("chokidar");
 const { spawn } = require("child_process");
 
 /* =========================
-   🎨 COLORS (PRO)
+🎨 COLORS (clean & pro)
 ========================= */
 const C = {
   reset: "\x1b[0m",
-  blue: "\x1b[34m",
+  dim: "\x1b[2m",
+  bold: "\x1b[1m",
   cyan: "\x1b[36m",
   green: "\x1b[32m",
   yellow: "\x1b[33m",
   red: "\x1b[31m",
-  dim: "\x1b[90m",
+  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}`),
 };
 
 /* =========================
-   🧭 ROOT + PKG
+🧭 PROJECT ROOT
 ========================= */
 const ROOT = process.cwd();
 const PKG = path.join(ROOT, "package.json");
 
 /* =========================
-   ☕ SCRIPT INJECTOR
+🧩 SCRIPT INJECTOR
 ========================= */
-function injectTea() {
+function injectScript() {
   if (!fs.existsSync(PKG)) return;
 
   const pkg = JSON.parse(fs.readFileSync(PKG, "utf8"));
@@ -36,115 +45,137 @@ function injectTea() {
   if (!pkg.scripts.tea) {
     pkg.scripts.tea = "drizzle-auto";
     fs.writeFileSync(PKG, JSON.stringify(pkg, null, 2));
-    console.log(`${C.cyan}☕ Added script: npm run tea${C.reset}`);
+    log.ok("Added script → npm run tea");
   }
 }
 
 /* =========================
-   🔍 RECURSIVE SEARCH
+🔍 FIND DRIZZLE CONFIG
+(js | mjs | ts | mts)
 ========================= */
-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 = 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 schemaPath = null;
-  try {
-    const content = fs.readFileSync(config, "utf8");
-    const match = content.match(/schema:\s*["'](.+?)["']/);
-    if (match) {
-      schemaPath = path.resolve(path.dirname(config), match[1]);
-    }
-  } catch {}
-
-  if (schemaPath && !fs.existsSync(schemaPath)) {
-    console.log(`${C.red}❌ Schema missing: ${schemaPath}${C.reset}`);
-    return null;
+  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;
   }
+  return null;
+}
 
-  console.log(`${C.green}✔ Drizzle config found${C.reset}`);
-  return true;
+/* =========================
+🔎 FILE FILTER (ALL TYPES)
+========================= */
+const WATCH_EXTS = [
+  ".js", ".mjs", ".cjs",
+  ".ts", ".mts",
+  ".jsx", ".tsx",
+  ".sql"
+];
+
+function isValidFile(file) {
+  return WATCH_EXTS.includes(path.extname(file));
 }
 
 /* =========================
-   ⚙️ COMMAND RUNNER
+⚙️ SAFE SPAWN (NO SHELL)
 ========================= */
-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(C.dim + d.toString() + C.reset)
-    );
-    p.stderr.on("data", d => {
-      failed = true;
-      process.stderr.write(C.red + d.toString() + C.reset);
+function exec(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);
     });
 
-    p.on("close", code => resolve(code === 0 && !failed));
+    p.on("close", (code) => {
+      resolve(code === 0);
+    });
   });
 }
 
 /* =========================
-   🔁 PIPELINE (SAFE)
+🛡️ CRASH PROTECTION
+========================= */
+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
 
 async function pipeline(trigger) {
   if (running) return;
   running = true;
 
-  console.log(`${C.blue}\n⚡ Trigger → ${trigger}${C.reset}`);
+  log.trig(`Trigger → ${trigger}`);
 
-  if (!findDrizzleConfig()) {
+  const config = findDrizzleConfig();
+  if (!config) {
+    log.warn("No drizzle.config found");
     running = false;
     return;
   }
 
-  if (!(await run(["drizzle-kit", "generate"]))) {
-    console.log(`${C.red}🛑 generate failed (watcher alive)${C.reset}`);
+  log.info("Running drizzle generate…");
+  const gen = await exec("npx", ["drizzle-kit", "generate"]);
+  if (!gen) {
+    log.err("Generate failed — watching continues");
     running = false;
     return;
   }
 
-  if (!(await run(["drizzle-kit", "push"]))) {
-    console.log(`${C.red}🛑 push failed (watcher alive)${C.reset}`);
+  const now = Date.now();
+  if (now - lastPush < PUSH_COOLDOWN) {
+    log.warn("Push skipped (cooldown)");
     running = false;
     return;
   }
 
-  console.log(`${C.green}✨ Drizzle synced${C.reset}`);
+  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;
+    return;
+  }
+
+  log.ok("Drizzle fully synced");
   running = false;
 }
 
 /* =========================
-   👀 WATCHER
+👀 WATCHER (ALL FILES)
 ========================= */
-injectTea();
-pipeline("initial");
+injectScript();
+pipeline("Initial");
+
+log.info(`Watching files (${DEV_MODE ? "DEV MODE" : "SAFE MODE"})…`);
 
 chokidar.watch(ROOT, {
-  ignored: [/node_modules/, /\.git/, /\.next/, /dist/],
+  ignored: [
+    /node_modules/,
+    /\.git/,
+    /\.next/,
+    /dist/,
+    /build/
+  ],
   ignoreInitial: true,
   usePolling: true,
   interval: 300,
-}).on("all", (_, file) => pipeline(path.basename(file)));
+}).on("all", (_, file) => {
+  if (isValidFile(file)) {
+    pipeline(path.relative(ROOT, file));
+  }
+});

package/package.json

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