blitzwing 0.1.4 → 0.1.5

package/package.json

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

@@ -33,11 +33,25 @@ 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;
+  }
+}
+
 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"),
@@ -45,31 +59,43 @@ export function ensurePython() {
       path.join(process.env.CONDA_PREFIX, "bin", "python")
     );
   }
-  candidates.push(
-    path.join(process.env.HOME || "", "miniconda", "envs", "blitzwing", "bin", "python3.11"),
-    path.join(process.env.HOME || "", "miniconda", "envs", "blitzwing", "bin", "python"),
-    path.join(process.env.HOME || "", "miniconda3", "envs", "blitzwing", "bin", "python3.11"),
-    path.join(process.env.HOME || "", "miniconda3", "envs", "blitzwing", "bin", "python"),
-    "python3.11",
-    "python3.10"
-  );
+  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) continue;
+    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). " +
-      "Install with: conda create -n blitzwing python=3.11 && conda activate blitzwing  " +
-      "or set BLITZWING_PYTHON=/path/to/python3.11"
+    "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"
   );
 }