blitzwing 0.1.3 → 0.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blitzwing",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "Interactive setup wizard to join a Blitzwing Petals mother swarm as a compute contributor",
   "bin": {
     "blitzwing": "bin/blitzwing.js"

package/src/install.js

@@ -18,41 +18,85 @@ function venvPip() {
   return path.join(VENV_PATH, "bin", "pip");
 }
 
-export function ensurePython() {
-  // Prefer 3.10/3.11 — Petals/hivemind break on 3.12+ / 3.13+
-  const candidates = ["python3.11", "python3.10", "python3", "python"];
-  let py = null;
-  let ver = "";
-  for (const cmd of candidates) {
-    const found = which(cmd);
-    if (!found) continue;
-    try {
-      ver = execFileSync(found, ["--version"], { encoding: "utf8" }).trim();
-      const m = ver.match(/Python (\d+)\.(\d+)/);
-      if (!m) continue;
-      const major = Number(m[1]);
-      const minor = Number(m[2]);
-      if (major === 3 && minor >= 10 && minor <= 11) {
-        py = found;
-        break;
-      }
-      if (!py) {
-        py = found; // fallback, may fail later
-      }
-    } catch {
-      /* try next */
-    }
+function pythonVersion(bin) {
+  try {
+    const ver = execFileSync(bin, ["--version"], { encoding: "utf8" }).trim();
+    const m = ver.match(/Python (\d+)\.(\d+)/);
+    if (!m) return null;
+    return { ver, major: Number(m[1]), minor: Number(m[2]) };
+  } catch {
+    return null;
   }
-  if (!py) {
-    throw new Error("Python 3.10 or 3.11 is required. Install via conda/miniconda and re-run.");
+}
+
+function isPetalsCompatible(info) {
+  return info && info.major === 3 && info.minor >= 10 && info.minor <= 11;
+}
+
+function pinnedPythonPath() {
+  try {
+    const p = path.join(HOME_DIR, "python");
+    if (!fs.existsSync(p)) return null;
+    const bin = fs.readFileSync(p, "utf8").trim();
+    return bin || null;
+  } catch {
+    return null;
   }
-  const m = ver.match(/Python (\d+)\.(\d+)/);
-  if (m && (Number(m[1]) !== 3 || Number(m[2]) > 11)) {
-    throw new Error(
-      `${ver} is too new for Petals. Use Python 3.11 (e.g. conda create -n blitzwing python=3.11).`
+}
+
+export function ensurePython() {
+  // Petals/hivemind break on 3.12+. Never fall back to a too-new interpreter.
+  const home = process.env.HOME || process.env.USERPROFILE || "";
+  const candidates = [];
+  const envBin = process.env.BLITZWING_PYTHON || process.env.PETALS_PYTHON;
+  if (envBin) candidates.push(envBin);
+  const pinned = pinnedPythonPath();
+  if (pinned) candidates.push(pinned);
+  if (process.env.CONDA_PREFIX) {
+    candidates.push(
+      path.join(process.env.CONDA_PREFIX, "bin", "python3.11"),
+      path.join(process.env.CONDA_PREFIX, "bin", "python3.10"),
+      path.join(process.env.CONDA_PREFIX, "bin", "python")
     );
   }
-  return { py, ver };
+  for (const root of ["miniforge3", "mambaforge", "miniconda", "miniconda3", "anaconda3"]) {
+    candidates.push(
+      path.join(home, root, "envs", "blitzwing", "bin", "python3.11"),
+      path.join(home, root, "envs", "blitzwing", "bin", "python")
+    );
+  }
+  candidates.push("python3.11", "python3.10");
+
+  const tried = [];
+  const seen = new Set();
+  for (const cmd of candidates) {
+    if (!cmd || seen.has(cmd)) continue;
+    seen.add(cmd);
+    const found = cmd.includes("/") || cmd.includes("\\") ? (fs.existsSync(cmd) ? cmd : null) : which(cmd);
+    if (!found) {
+      tried.push(`${cmd} (missing)`);
+      continue;
+    }
+    const info = pythonVersion(found);
+    if (isPetalsCompatible(info)) {
+      fs.mkdirSync(HOME_DIR, { recursive: true });
+      fs.writeFileSync(path.join(HOME_DIR, "python"), found + "\n");
+      return { py: found, ver: info.ver };
+    }
+    tried.push(`${found} (${info ? info.ver : "unreadable"})`);
+  }
+
+  throw new Error(
+    "Python 3.10 or 3.11 is required (3.12+ will not work).\n" +
+      "Checked:\n  - " +
+      tried.slice(0, 12).join("\n  - ") +
+      "\nFix:\n" +
+      "  curl -LsSf https://astral.sh/uv/install.sh | sh && source $HOME/.local/bin/env\n" +
+      "  uv python install 3.11\n" +
+      "  export BLITZWING_PYTHON=\"$(uv python find 3.11)\"\n" +
+      "  mkdir -p ~/.blitzwing && echo \"$BLITZWING_PYTHON\" > ~/.blitzwing/python\n" +
+      "  blitzwing"
+  );
 }
 
 function petalsImportable(python) {