drift-ml 0.1.3 → 0.1.4

package/bin/drift.js

@@ -1,17 +1,59 @@
 #!/usr/bin/env node
 
 const { spawnSync } = require("child_process");
+const path = require("path");
+const fs = require("fs");
 
-function which(cmd) {
-  const isWindows = process.platform === "win32";
-  const whichCmd = isWindows ? "where" : "which";
-  const { status, stdout } = spawnSync(whichCmd, [cmd], { encoding: "utf8" });
-  return status === 0 ? (stdout || "").trim().split("\n")[0] : null;
+// Known pipx install locations
+function getPipxBinPaths() {
+  const home = process.env.HOME || process.env.USERPROFILE || "";
+  return [
+    path.join(home, ".local", "bin", "drift"),
+    "/usr/local/bin/drift",
+  ];
+}
+
+// Find Python-based drift (not this Node script)
+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 (_) {}
+    }
+  }
+  
+  // 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)) {
+      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;
+        }
+        return candidate;
+      } catch (_) {
+        // Binary file or unreadable - might be the real one
+        return candidate;
+      }
+    }
+  }
+  
+  return null;
 }
 
 function main() {
-  // Check if drift binary exists on PATH
-  const driftPath = which("drift");
+  const driftPath = findPythonDrift();
   
   if (!driftPath) {
     console.error(`
@@ -28,7 +70,7 @@ Then run:
     process.exit(1);
   }
 
-  // Run drift and forward stdin/stdout
+  // Run the Python drift and forward stdin/stdout
   const result = spawnSync(driftPath, process.argv.slice(2), {
     stdio: "inherit",
     env: process.env,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "drift-ml",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "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"