drizzle-auto 1.0.8 → 1.0.9

package/index.js

@@ -5,32 +5,29 @@ const path = require("path");
 const chokidar = require("chokidar");
 const { spawn } = require("child_process");
 
-/* =========================
-   🎨 Neon Colors
-========================= */
+/* =======================
+   🎨 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;
+const rainbow = text => text.split("").map(c => `${nextColor()}${c}`).join("") + RESET;
 
-/* =========================
-   ⏳ Neon Spinner
-========================= */
-const spinnerFrames = ["⠋","⠙","⠹","⠸","⠼","⠴","⠦","⠧","⠇","⠏"];
+/* =======================
+   ⏳ Spinner
+======================= */
+const FRAMES = ["⠋","⠙","⠹","⠸","⠼","⠴","⠦","⠧","⠇","⠏"];
 let spinIndex = 0;
-let spinnerInterval;
-
-function startSpinner(text) {
+let spinnerInterval = null;
+function startSpinner(msg) {
   stopSpinner();
   spinnerInterval = setInterval(() => {
-    process.stdout.write(`\r${nextColor()}${spinnerFrames[spinIndex++ % spinnerFrames.length]} ${text}${RESET}`);
+    process.stdout.write(`\r${nextColor()}${FRAMES[spinIndex++ % FRAMES.length]} ${msg}${RESET}`);
   }, 80);
 }
-
 function stopSpinner() {
   if (spinnerInterval) {
     clearInterval(spinnerInterval);
@@ -39,109 +36,99 @@ function stopSpinner() {
   }
 }
 
-/* =========================
-   🧭 Project Root & Package.json
-========================= */
-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 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");
 
-/* =========================
-   ✍️ Auto Inject Script "tea"
-========================= */
-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("⚡ tea script injected → npm run tea");
+    fs.writeFileSync(PKG_PATH, JSON.stringify(pkg, null, 2));
+    console.log(`⚡ tea script added → npm run tea`);
   }
 }
 
-/* =========================
-   🧪 Infrastructure Audit
-========================= */
+/* =======================
+   🧪 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(projectRoot, f)));
+    .find(f => fs.existsSync(path.join(PROJECT_ROOT, f)));
   results.push({ name: "Drizzle Config", ok: !!configFile });
 
+  // Schema path
   if (configFile) {
     try {
-      const content = fs.readFileSync(path.join(projectRoot, configFile), "utf-8");
-      const match = content.match(/schema:\s*["'](.+?)["']/);
+      const cfg = fs.readFileSync(path.join(PROJECT_ROOT, configFile), "utf8");
+      const match = cfg.match(/schema:\s*["'](.+?)["']/);
       if (match) {
-        const schemaPath = path.join(projectRoot, match[1]);
-        results.push({ name: `Schema (${match[1]})`, ok: fs.existsSync(schemaPath) });
-      } else {
-        results.push({ name: "Schema Path", ok: false });
-      }
+        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 Read", ok: false });
+      results.push({ name: "Config Readable", ok: false });
     }
   }
 
-  results.push({ name: ".env", ok: fs.existsSync(path.join(projectRoot, ".env")) });
-  results.push({ name: "node_modules", ok: fs.existsSync(path.join(projectRoot, "node_modules")) });
+  // 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 NPX Command
-========================= */
-function runStep(cmd) {
-  return new Promise((resolve) => {
+/* =======================
+   ⚡ 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));
   });
 }
 
-/* =========================
-   🚀 Main Workflow
-========================= */
-let isRunning = false;
+/* =======================
+   🏁 Workflow
+======================= */
+let running = false;
 let hasError = false;
 
 async function workflow(trigger) {
-  if (isRunning) return;
-  isRunning = true;
+  if (running) return;
+  running = true;
 
   if (!hasError) console.clear();
-  console.log(`\n${nextColor()}⚡ ${trigger}${RESET}`);
+  console.log(`\n${nextColor()}⚡ Trigger → ${trigger}${RESET}`);
 
   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;
-    isRunning = false;
+    running = false;
     return;
   }
 
@@ -151,40 +138,38 @@ async function workflow(trigger) {
     ["drizzle-kit", "push"]
   ];
 
-  for (const s of steps) {
-    if (!(await runStep(s))) {
+  for (const step of steps) {
+    if (!(await runCmd(step))) {
       console.log("\n\x1b[31m💀 Pipeline crashed\x1b[0m");
       hasError = true;
-      isRunning = false;
+      running = false;
       return;
     }
   }
 
   hasError = false;
   console.log("\n\x1b[32m✨ Drizzle synced\x1b[0m");
-  setTimeout(() => (isRunning = false), 1200);
+  setTimeout(() => running = false, 1000);
 }
 
-/* =========================
+/* =======================
    👀 Watcher
-========================= */
-chokidar.watch(projectRoot, {
+======================= */
+chokidar.watch(PROJECT_ROOT, {
   ignored: [/node_modules/, /\.git/, /\.next/, /dist/],
   ignoreInitial: true,
   usePolling: true,
   interval: 300
-}).on("all", (e, f) => workflow(`${e.toUpperCase()} → ${f ? path.basename(f) : ""}`));
-
-/* =========================
-   🛠 Postinstall Mode
-========================= */
-if (process.argv.includes("--postinstall") || process.env.npm_config_argv) {
-  injectTeaScript();
-  process.exit(0);
-}
+}).on("all", (e, f) => workflow(`${e.toUpperCase()} → ${path.basename(f)}`));
+
+/* =======================
+   🚀 Init
+======================= */
+addTeaScript();      // inject tea script automatically
+workflow("Initial Audit");
 
-/* =========================
-   🌟 Init
-========================= */
-injectTeaScript();
-workflow("Initial Audit");
+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.8",
+  "version": "1.0.9",
   "bin": {
     "drizzle-auto": "index.js"
   },