drift-ml 0.1.4 → 0.1.6

package/bin/drift.js

@@ -4,51 +4,68 @@ const { spawnSync } = require("child_process");
 const path = require("path");
 const fs = require("fs");
 
-// Known pipx install locations
+const isWindows = process.platform === "win32";
+
+// Known pipx install locations (Unix and Windows)
 function getPipxBinPaths() {
   const home = process.env.HOME || process.env.USERPROFILE || "";
+  if (!home) return [];
+  const localBin = path.join(home, ".local", "bin");
+  if (isWindows) {
+    return [
+      path.join(localBin, "drift.exe"),
+      path.join(localBin, "drift"),
+    ];
+  }
   return [
-    path.join(home, ".local", "bin", "drift"),
+    path.join(localBin, "drift"),
     "/usr/local/bin/drift",
   ];
 }
 
-// Find Python-based drift (not this Node script)
+function isLikelyUsOrNpmWrapper(filePath, content) {
+  if (!content || typeof content !== "string") return false;
+  const s = content.slice(0, 200);
+  // Node launcher
+  if (s.includes("#!/usr/bin/env node")) return true;
+  // Windows npm .cmd wrapper (invokes node)
+  if (s.includes("node") && (s.startsWith("@") || s.includes("cmd.exe"))) return true;
+  return false;
+}
+
+// Find Python-based drift (not this Node script or npm's drift.cmd)
 function findPythonDrift() {
-  // First check pipx standard locations
   for (const p of getPipxBinPaths()) {
     if (fs.existsSync(p)) {
-      // Verify it's not a Node script (i.e., not us)
       try {
-        const content = fs.readFileSync(p, "utf8").slice(0, 100);
-        if (!content.includes("#!/usr/bin/env node")) {
-          return p;
-        }
-      } catch (_) {}
+        const content = fs.readFileSync(p, "utf8").slice(0, 200);
+        if (!isLikelyUsOrNpmWrapper(p, content)) return p;
+      } catch (_) {
+        return p;
+      }
     }
   }
-  
-  // Fallback: search PATH but skip Node scripts
+
   const pathEnv = process.env.PATH || "";
   const dirs = pathEnv.split(path.delimiter);
-  
+
   for (const dir of dirs) {
-    const candidate = path.join(dir, "drift");
-    if (fs.existsSync(candidate)) {
+    const base = path.join(dir, "drift");
+    const candidates = isWindows ? [base + ".exe", base + ".cmd", base] : [base];
+    for (const candidate of candidates) {
+      if (!fs.existsSync(candidate)) continue;
+      const ext = path.extname(candidate).toLowerCase();
+      if (isWindows && (ext === ".cmd" || ext === ".bat" || ext === ".ps1")) continue;
       try {
-        const content = fs.readFileSync(candidate, "utf8").slice(0, 100);
-        // Skip if it's a Node script (that's us or another Node launcher)
-        if (content.includes("#!/usr/bin/env node")) {
-          continue;
-        }
+        const content = fs.readFileSync(candidate, "utf8").slice(0, 200);
+        if (isLikelyUsOrNpmWrapper(candidate, content)) continue;
         return candidate;
       } catch (_) {
-        // Binary file or unreadable - might be the real one
         return candidate;
       }
     }
   }
-  
+
   return null;
 }
 
@@ -59,8 +76,10 @@ function main() {
     console.error(`
 drift is not installed.
 
-Install it with:
+Install the Python CLI first:
 
+  pip install --user pipx
+  pipx ensurepath
   pipx install drift-ml
 
 Then run:
@@ -70,10 +89,10 @@ Then run:
     process.exit(1);
   }
 
-  // Run the Python drift and forward stdin/stdout
   const result = spawnSync(driftPath, process.argv.slice(2), {
     stdio: "inherit",
     env: process.env,
+    shell: isWindows,
   });
 
   process.exit(result.status === null ? (result.signal ? 128 + 9 : 1) : result.status);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "drift-ml",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "Drift — terminal-first, chat-based AutoML. Same engine as the web app. On first run: downloads and starts the engine locally (never exposes engine source).",
   "bin": {
     "drift": "./bin/drift.js"