drizzle-auto 1.0.24 → 1.1.1

package/index.js

@@ -6,124 +6,176 @@ const chokidar = require("chokidar");
 const { spawn } = require("child_process");
 
 /* =========================
-   🧭 PROJECT ROOT
+🎨 COLORS (clean & pro)
 ========================= */
-const ROOT = process.cwd();
-const PKG = path.join(ROOT, "package.json");
-
-/* =========================
-   🎨 COLORS
-========================= */
-const colors = {
+const C = {
   reset: "\x1b[0m",
-  red: "\x1b[31m",
+  dim: "\x1b[2m",
+  bold: "\x1b[1m",
+  cyan: "\x1b[36m",
   green: "\x1b[32m",
   yellow: "\x1b[33m",
-  cyan: "\x1b[36m",
+  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}`),
+};
+
 /* =========================
-   🧩 SCRIPT INJECTOR
+🧭 PROJECT ROOT
 ========================= */
-function injectTeaScript() {
+const ROOT = process.cwd();
+const PKG = path.join(ROOT, "package.json");
+
+/* =========================
+🧩 SCRIPT INJECTOR
+========================= */
+function injectScript() {
   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}`);
+    log.ok("Added script → npm run tea");
   }
 }
 
 /* =========================
-   🔍 DRIZZLE CONFIG DETECTION
+🔍 FIND DRIZZLE CONFIG
+(js | mjs | ts | mts)
 ========================= */
 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}`);
-    return null;
-  }
-
-  let schema = null;
-  try {
-    const content = fs.readFileSync(found, "utf8");
-    const match = content.match(/schema:\s*["'](.+?)["']/);
-    if (match) schema = path.join(ROOT, match[1]);
-  } catch {}
-  
-  if (schema && !fs.existsSync(schema)) {
-    console.log(`${colors.red}❌ Schema file missing: ${schema}${colors.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;
+}
 
-  return { config: found, schema };
+/* =========================
+🔎 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));
 }
 
 /* =========================
-   ⚙️ SAFE 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(`${colors.green}${d}${colors.reset}`));
-    p.stderr.on("data", d => {
-      failed = true;
-      process.stderr.write(`${colors.red}${d}${colors.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);
     });
-    p.on("close", code => resolve(code === 0 && !failed));
   });
 }
 
 /* =========================
-   🔁 PIPELINE
+🛡️ 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 busy = false;
+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 (busy) return;
-  busy = true;
-  console.log(`${colors.magenta}\n⚡ Trigger → ${trigger}${colors.reset}`);
+  if (running) return;
+  running = true;
+
+  log.trig(`Trigger → ${trigger}`);
 
-  const drizzle = findDrizzleConfig();
-  if (!drizzle) {
-    busy = false;
+  const config = findDrizzleConfig();
+  if (!config) {
+    log.warn("No drizzle.config found");
+    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;
+  log.info("Running drizzle generate…");
+  const gen = await exec("npx", ["drizzle-kit", "generate"]);
+  if (!gen) {
+    log.err("Generate failed — watching continues");
+    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;
+  const now = Date.now();
+  if (now - lastPush < PUSH_COOLDOWN) {
+    log.warn("Push skipped (cooldown)");
+    running = false;
     return;
   }
 
-  console.log(`${colors.cyan}✨ Drizzle fully synced${colors.reset}`);
-  busy = false;
+  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)
 ========================= */
-injectTeaScript();
+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(file));
+  interval: 300,
+}).on("all", (_, file) => {
+  if (isValidFile(file)) {
+    pipeline(path.relative(ROOT, file));
+  }
+});

package/package.json

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