drizzle-auto 1.1.3 → 1.1.5

package/index.js

@@ -1,155 +1,168 @@
 #!/usr/bin/env node
 
-const fs = require("fs");
-const path = require("path");
 const chokidar = require("chokidar");
 const { spawn } = require("child_process");
+const path = require("path");
+const fs = require("fs");
 
-/* =========================
-🎨 COLORS (PRO ONLY)
-========================= */
+/* =============================================================
+   1. UI & COLOR CONFIGURATION (NeonPulse Style)
+   ============================================================= */
 const C = {
-  reset: "\x1b[0m",
-  bold: "\x1b[1m",
-  cyan: "\x1b[36m",
-  green: "\x1b[32m",
-  yellow: "\x1b[33m",
-  red: "\x1b[31m",
-  magenta: "\x1b[35m",
-  dim: "\x1b[2m",
+    reset: "\x1b[0m",
+    bold: "\x1b[1m",
+    red: "\x1b[31m",
+    green: "\x1b[32m",
+    yellow: "\x1b[33m",
+    blue: "\x1b[34m",
+    magenta: "\x1b[35m",
+    cyan: "\x1b[36m",
+    dim: "\x1b[90m"
 };
 
-/* =========================
-🧭 ROOT
-========================= */
-const ROOT = process.cwd();
-const PKG = path.join(ROOT, "package.json");
-
-/* =========================
-☕ ADD tea SCRIPT
-========================= */
-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(`${C.green}☕ Added:${C.reset} npm run tea`);
-  }
+const PALETTE = [C.red, C.green, C.yellow, C.blue, C.magenta, C.cyan];
+let colorIndex = 0;
+const nextColor = () => PALETTE[colorIndex++ % PALETTE.length];
+
+/* =============================================================
+   2. AUTOMATIC PACKAGE.JSON INJECTION (The "tea" Script)
+   ============================================================= */
+function injectTeaScript() {
+    const pkgPath = path.join(process.cwd(), "package.json");
+    if (fs.existsSync(pkgPath)) {
+        const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf8"));
+        if (!pkg.scripts) pkg.scripts = {};
+        if (!pkg.scripts.tea) {
+            pkg.scripts.tea = "drizzle-auto";
+            fs.writeFileSync(pkgPath, JSON.stringify(pkg, null, 2));
+            console.log(`${C.green}✔ Added "tea": "drizzle-auto" to package.json${C.reset}`);
+        }
+    }
 }
 
-/* =========================
-🔍 FIND drizzle.config.*
-========================= */
-function findConfig() {
-  const exts = ["js", "mjs", "cjs", "ts", "mts"];
-  let found = null;
-
-  (function walk(dir) {
-    if (found) return;
-    for (const f of fs.readdirSync(dir, { withFileTypes: true })) {
-      if (f.name === "node_modules" || f.name.startsWith(".")) continue;
-      const p = path.join(dir, f.name);
-      if (f.isDirectory()) walk(p);
-      else {
-        for (const e of exts) {
-          if (f.name === `drizzle.config.${e}`) {
-            found = p;
-            return;
-          }
+/* =============================================================
+   3. DEEP INFRASTRUCTURE JUSTIFICATION (Recursive Search)
+   ============================================================= */
+function findConfig(dir = process.cwd()) {
+    const files = fs.readdirSync(dir);
+    for (const file of files) {
+        const fullPath = path.join(dir, file);
+        if (IGNORED_REGEX.test(fullPath)) continue;
+
+        if (fs.statSync(fullPath).isDirectory()) {
+            const found = findConfig(fullPath);
+            if (found) return found;
+        } else if (/^drizzle\.config\.(js|mjs|cjs|ts|mts)$/.test(file)) {
+            return fullPath;
         }
-      }
     }
-  })(ROOT);
-
-  return found;
+    return null;
 }
 
-/* =========================
-⚙️ SAFE RUNNER (ERROR AWARE)
-========================= */
-function run(cmd, args) {
-  return new Promise((resolve) => {
-    let stderr = "";
+function justifyInfrastructure() {
+    const configPath = findConfig();
+    if (!configPath) return { ok: false, msg: "Missing drizzle.config.*" };
 
-    const p = spawn(cmd, args, { stdio: ["ignore", "pipe", "pipe"] });
-
-    p.stdout.on("data", d => process.stdout.write(d));
-    p.stderr.on("data", d => {
-      stderr += d.toString();
-      process.stderr.write(d);
-    });
+    try {
+        const content = fs.readFileSync(configPath, "utf8");
+        const schemaMatch = content.match(/schema:\s*["'](.+?)["']/);
+        if (schemaMatch && schemaMatch[1]) {
+            const schemaPath = path.resolve(path.dirname(configPath), schemaMatch[1]);
+            if (!fs.existsSync(schemaPath)) return { ok: false, msg: `Schema missing at ${schemaMatch[1]}` };
+        }
+    } catch (e) {
+        return { ok: false, msg: "Config unreadable" };
+    }
+    return { ok: true, path: configPath };
+}
 
-    p.on("close", (code) => {
-      resolve({ ok: code === 0, stderr });
+/* =============================================================
+   4. SAFE COMMAND EXECUTION (Step-by-Step Chain)
+   ============================================================= */
+function execute(args, name) {
+    return new Promise((resolve) => {
+        let hasError = false;
+        console.log(`${C.cyan}⏳ ${name}...${C.reset}`);
+
+        const child = spawn("npx", args, { shell: true, stdio: ["inherit", "pipe", "pipe"] });
+
+        child.stdout.on("data", (data) => {
+            const out = data.toString();
+            // Filter noise from Cloud DB pullers
+            if (!out.includes('Pulling schema')) process.stdout.write(`${C.dim}${out}${C.reset}`);
+            if (/error|failed|ENOTFOUND|ECONNREFUSED/i.test(out)) hasError = true;
+        });
+
+        child.stderr.on("data", (data) => {
+            process.stderr.write(`${C.red}${data}${C.reset}`);
+            hasError = true;
+        });
+
+        child.on("close", (code) => resolve(code === 0 && !hasError));
     });
-  });
 }
 
-/* =========================
-🔁 PIPELINE (NO LIES)
-========================= */
-let busy = false;
-
-async function pipeline(trigger) {
-  if (busy) return;
-  busy = true;
-
-  console.log(`\n${C.cyan}⚡ Trigger →${C.reset} ${trigger}`);
-
-  const config = findConfig();
-  if (!config) {
-    console.log(`${C.yellow}⚠ No drizzle.config found${C.reset}`);
-    busy = false;
-    return;
-  }
-
-  console.log(`${C.magenta}▶ drizzle-kit generate${C.reset}`);
-  const gen = await run("npx", ["drizzle-kit", "generate"]);
-  if (!gen.ok) {
-    console.log(`${C.red}🛑 Generate failed — watcher alive${C.reset}`);
-    busy = false;
-    return;
-  }
-
-  console.log(`${C.magenta}▶ drizzle-kit push${C.reset}`);
-  const push = await run("npx", ["drizzle-kit", "push"]);
-
-  if (!push.ok) {
-    console.log(`\n${C.red}${C.bold}🚨 PUSH FAILED${C.reset}`);
-
-    if (
-      push.stderr.includes("ENOTFOUND") ||
-      push.stderr.includes("websocket")
-    ) {
-      console.log(
-        `${C.yellow}⚠ Neon / network issue detected${C.reset}\n` +
-        `${C.dim}• Check internet\n` +
-        `• Check DATABASE_URL\n` +
-        `• Neon endpoint may be sleeping${C.reset}`
-      );
+/* =============================================================
+   5. DYNAMIC WORKFLOW MANAGER
+   ============================================================= */
+let isRunning = false;
+async function startWorkflow(trigger) {
+    if (isRunning) return;
+    isRunning = true;
+    
+    console.log(`\n${nextColor()}⚡ Triggered: ${trigger}${C.reset}`);
+
+    // Step 2: Justify Files
+    const infra = justifyInfrastructure();
+    if (!infra.ok) {
+        console.log(`${C.red}🛑 Infrastructure Failure: ${infra.msg}${C.reset}`);
+        isRunning = false;
+        return;
     }
 
-    console.log(`${C.dim}Watcher still running. Server NOT stopped.${C.reset}`);
-    busy = false;
-    return;
-  }
+    // Step 3 & 4: Sequence (Generate -> Push)
+    const steps = [
+        { name: "Generating Schema", args: ["drizzle-kit", "generate"] },
+        { name: "Pushing to Database", args: ["drizzle-kit", "push"] }
+    ];
+
+    for (const step of steps) {
+        const ok = await execute(step.args, step.name);
+        if (!ok) {
+            console.log(`${C.red}${C.bold}✖ Pipeline Halted. Check logs above.${C.reset}`);
+            isRunning = false;
+            return;
+        }
+    }
 
-  console.log(`${C.green}✨ Drizzle fully synced${C.reset}`);
-  busy = false;
+    console.log(`${C.green}${C.bold}✨ Success: All steps completed.${C.reset}`);
+    isRunning = false;
 }
 
-/* =========================
-👀 WATCH EVERYTHING
-========================= */
-injectTea();
-pipeline("Initial");
-
-chokidar.watch(ROOT, {
-  ignored: [/node_modules/, /\.git/, /\.next/, /dist/],
-  ignoreInitial: true,
-}).on("all", (_, file) => {
-  pipeline(path.relative(ROOT, file));
-});
+/* =============================================================
+   6. UNIVERSAL WATCHER & INITIALIZATION
+   ============================================================= */
+const IGNORED_REGEX = /(node_modules|\.git|\.next|dist)/;
+
+injectTeaScript();
+
+const watcher = chokidar.watch(".", {
+    ignored: IGNORED_REGEX,
+    persistent: true,
+    ignoreInitial: true,
+    usePolling: true,
+    interval: 300
+});
+
+watcher.on("all", (event, file) => {
+    startWorkflow(`${event.toUpperCase()} -> ${path.basename(file)}`);
+});
+
+// Initial startup audit
+startWorkflow("Initial System Audit");
+
+// Crash Protection
+process.on("uncaughtException", (err) => {
+    console.log(`${C.red}🛡️ Shield: ${err.message}${C.reset}`);
+    isRunning = false;
+});

package/package.json

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