drizzle-auto 1.0.7 → 1.0.9

package/index.js

@@ -3,85 +3,173 @@
 const fs = require("fs");
 const path = require("path");
 const chokidar = require("chokidar");
-const spawn = require("cross-spawn");
-
-/* =========================
-   🧠 Detect project root
-========================= */
-function getProjectRoot() {
-  const init = process.env.INIT_CWD;
-  if (init && fs.existsSync(path.join(init, "package.json"))) return init;
-  let cwd = process.cwd();
-  if (cwd.includes("node_modules")) cwd = cwd.split(`node_modules${path.sep}`)[0];
-  return cwd;
+const { spawn } = require("child_process");
+
+/* =======================
+   🎨 NeonPulse Colors
+======================= */
+const COLORS = ["\x1b[31m","\x1b[32m","\x1b[33m","\x1b[34m","\x1b[35m","\x1b[36m"];
+const RESET = "\x1b[0m";
+const BOLD = "\x1b[1m";
+const DIM = "\x1b[90m";
+let colorIndex = 0;
+const nextColor = () => COLORS[colorIndex++ % COLORS.length];
+const rainbow = text => text.split("").map(c => `${nextColor()}${c}`).join("") + RESET;
+
+/* =======================
+   ⏳ Spinner
+======================= */
+const FRAMES = ["⠋","⠙","⠹","⠸","⠼","⠴","⠦","⠧","⠇","⠏"];
+let spinIndex = 0;
+let spinnerInterval = null;
+function startSpinner(msg) {
+  stopSpinner();
+  spinnerInterval = setInterval(() => {
+    process.stdout.write(`\r${nextColor()}${FRAMES[spinIndex++ % FRAMES.length]} ${msg}${RESET}`);
+  }, 80);
+}
+function stopSpinner() {
+  if (spinnerInterval) {
+    clearInterval(spinnerInterval);
+    spinnerInterval = null;
+    process.stdout.write("\r\x1b[K");
+  }
 }
 
-const projectRoot = getProjectRoot();
-const pkgPath = path.join(projectRoot, "package.json");
+/* =======================
+   🧠 Project Root + Package.json
+======================= */
+const PROJECT_ROOT = process.env.INIT_CWD || process.cwd();
+const PKG_PATH = path.join(PROJECT_ROOT, "package.json");
 
-/* =========================
-   ✍️ Inject "tea" script automatically
-========================= */
-function injectTeaScript() {
-  if (!fs.existsSync(pkgPath)) return;
-  const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf-8"));
+function addTeaScript() {
+  if (!fs.existsSync(PKG_PATH)) return;
+  const pkg = JSON.parse(fs.readFileSync(PKG_PATH, "utf8"));
   pkg.scripts ||= {};
   if (!pkg.scripts.tea) {
     pkg.scripts.tea = "drizzle-auto";
-    fs.writeFileSync(pkgPath, JSON.stringify(pkg, null, 2));
-    console.log("⚡ Script injected → npm run tea");
+    fs.writeFileSync(PKG_PATH, JSON.stringify(pkg, null, 2));
+    console.log(`⚡ tea script added → npm run tea`);
+  }
+}
+
+/* =======================
+   🧪 Infra Audit
+======================= */
+function auditInfra() {
+  const results = [];
+
+  // Drizzle config
+  const configFile = ["ts","js","mjs","mts"].map(ext => `drizzle.config.${ext}`)
+    .find(f => fs.existsSync(path.join(PROJECT_ROOT, f)));
+  results.push({ name: "Drizzle Config", ok: !!configFile });
+
+  // Schema path
+  if (configFile) {
+    try {
+      const cfg = fs.readFileSync(path.join(PROJECT_ROOT, configFile), "utf8");
+      const match = cfg.match(/schema:\s*["'](.+?)["']/);
+      if (match) {
+        results.push({
+          name: `Schema (${match[1]})`,
+          ok: fs.existsSync(path.join(PROJECT_ROOT, match[1]))
+        });
+      } else results.push({ name: "Schema Path", ok: false });
+    } catch {
+      results.push({ name: "Config Readable", ok: false });
+    }
   }
+
+  // Essentials
+  results.push({ name: ".env", ok: fs.existsSync(path.join(PROJECT_ROOT, ".env")) });
+  results.push({ name: "node_modules", ok: fs.existsSync(path.join(PROJECT_ROOT, "node_modules")) });
+
+  const missing = results.filter(r => !r.ok).map(r => r.name);
+  return { ok: missing.length === 0, missing };
 }
 
-/* =========================
-   🚀 Run Drizzle pipeline
-========================= */
+/* =======================
+   ⚡ Run npx command
+======================= */
+function runCmd(cmd) {
+  return new Promise(resolve => {
+    let failed = false;
+    startSpinner(`npx ${cmd.join(" ")}`);
+    const p = spawn("npx", cmd, { shell: true });
+    p.stdout.on("data", d => {
+      stopSpinner();
+      process.stdout.write(DIM + d.toString() + RESET);
+      if (/error|failed/i.test(d)) failed = true;
+    });
+    p.stderr.on("data", d => {
+      stopSpinner();
+      process.stderr.write("\x1b[31m" + d + RESET);
+      failed = true;
+    });
+    p.on("close", code => resolve(code === 0 && !failed));
+  });
+}
+
+/* =======================
+   🏁 Workflow
+======================= */
 let running = false;
+let hasError = false;
 
-function runPipeline() {
+async function workflow(trigger) {
   if (running) return;
   running = true;
-  console.clear();
-  console.log("🚀 Drizzle Pipeline starting...");
 
-  const p = spawn("npx", ["drizzle-kit", "check", "generate", "push"], {
-    stdio: "inherit",
-    shell: true,
-    cwd: projectRoot,
-  });
+  if (!hasError) console.clear();
+  console.log(`\n${nextColor()}⚡ Trigger → ${trigger}${RESET}`);
 
-  p.on("close", (code) => {
+  const audit = auditInfra();
+  if (!audit.ok) {
+    console.log("\n\x1b[31m❌ Infra not ready\x1b[0m");
+    audit.missing.forEach(m => console.log(DIM + "• " + m + RESET));
+    hasError = true;
     running = false;
-    if (code === 0) console.log("✅ Drizzle synced!");
-    else console.log("❌ Drizzle pipeline failed. Fix errors & save files.");
-  });
-}
+    return;
+  }
 
-/* =========================
-   👀 Postinstall hook
-========================= */
-if (process.argv.includes("--postinstall") || process.env.npm_config_argv) {
-  injectTeaScript();
-  runPipeline();
+  const steps = [
+    ["drizzle-kit", "check"],
+    ["drizzle-kit", "generate"],
+    ["drizzle-kit", "push"]
+  ];
+
+  for (const step of steps) {
+    if (!(await runCmd(step))) {
+      console.log("\n\x1b[31m💀 Pipeline crashed\x1b[0m");
+      hasError = true;
+      running = false;
+      return;
+    }
+  }
+
+  hasError = false;
+  console.log("\n\x1b[32m✨ Drizzle synced\x1b[0m");
+  setTimeout(() => running = false, 1000);
 }
 
-/* =========================
-   👀 Watcher for any changes
-========================= */
-injectTeaScript();
-runPipeline(); // run immediately on npx drizzle-auto
-
-chokidar
-  .watch(projectRoot, {
-    ignored: /node_modules|\.git|\.next/,
-    ignoreInitial: true,
-  })
-  .on("all", (event, filePath) => {
-    console.log(`⚡ File changed → ${filePath}`);
-    runPipeline();
-  });
+/* =======================
+   👀 Watcher
+======================= */
+chokidar.watch(PROJECT_ROOT, {
+  ignored: [/node_modules/, /\.git/, /\.next/, /dist/],
+  ignoreInitial: true,
+  usePolling: true,
+  interval: 300
+}).on("all", (e, f) => workflow(`${e.toUpperCase()} → ${path.basename(f)}`));
+
+/* =======================
+   🚀 Init
+======================= */
+addTeaScript();      // inject tea script automatically
+workflow("Initial Audit");
 
-process.on("uncaughtException", (err) => {
+process.on("uncaughtException", err => {
+  stopSpinner();
   console.log("\x1b[31m🔥 Crash:\x1b[0m", err.message);
   running = false;
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "drizzle-auto",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "bin": {
     "drizzle-auto": "index.js"
   },