hotmilk 0.1.9 → 0.1.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hotmilk",
-  "version": "0.1.9",
+  "version": "0.1.10",
   "description": "hotmilk — Pi package bundling gentle-pi, context-mode, graphify, subagents, and bundled extension toggles via hotmilk.json",
   "keywords": [
     "hotmilk",

package/src/bootstrap/extensions.ts

@@ -1,5 +1,6 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { CONTEXT_STACK_EXTENSION_IDS } from "./context-stack.ts";
+import { bundledImportUrl } from "./resolve-bundled.ts";
 import { BUNDLED_EXTENSION_IDS, type BundledExtensionId } from "../config/hotmilk.ts";
 
 type ExtensionFactory = (pi: ExtensionAPI) => void | Promise<void>;
@@ -7,35 +8,40 @@ type ExtensionFactory = (pi: ExtensionAPI) => void | Promise<void>;
 /** Bundled deps may type against @mariozechner/pi-coding-agent; hotmilk uses @earendil-works. */
 type ExtensionModule = { default: unknown };
 
+function loadBundled(relativePath: string): () => Promise<ExtensionModule> {
+  return () => import(bundledImportUrl(relativePath));
+}
+
 /**
  * Load bundled extensions on demand so disabled toggles do not pay import cost.
  * Context stack registers sequentially; other enabled extensions load in parallel.
  */
 const BUNDLED_EXTENSION_LOADERS: Record<BundledExtensionId, () => Promise<ExtensionModule>> = {
-  "skill-registry": () => import("../../node_modules/gentle-pi/extensions/skill-registry.ts"),
-  "sdd-init": () => import("../../node_modules/gentle-pi/extensions/sdd-init.ts"),
-  "gentle-ai": () => import("../../node_modules/gentle-pi/extensions/gentle-ai.ts"),
-  // Same entry as upstream .pi/extensions/context-mode → build/adapters/pi/extension.js
-  "context-mode": () => import("../../node_modules/context-mode/build/adapters/pi/extension.js"),
-  "ask-user": () => import("pi-ask-user"),
-  graphify: () => import("../../node_modules/graphify-pi/extensions/graphify.ts"),
-  subagents: () => import("../../node_modules/pi-subagents/src/extension/index.ts"),
-  goal: () => import("../../node_modules/pi-goal/.pi/extensions/pi-goal/index.ts"),
-  docparser: () => import("../../node_modules/pi-docparser/extensions/docparser/index.ts"),
-  obsidian: () => import("../../node_modules/@haispeed/pi-obsidian/extensions/obsidian-cli.ts"),
-  "cursor-provider": () => import("../../node_modules/@netandreus/pi-cursor-provider/index.ts"),
-  btw: () => import("../../node_modules/pi-btw/extensions/btw.ts"),
-  simplify: () => import("../../node_modules/pi-simplify/dist/index.js"),
-  "rtk-optimizer": () => import("../../node_modules/pi-rtk-optimizer/index.ts"),
-  "mcp-adapter": () => import("../../node_modules/pi-mcp-adapter/index.ts"),
-  "planning-with-files": () =>
-    import("../../node_modules/@tomxprime/planning-with-files/extensions/planning-with-files/index.ts"),
-  caveman: () => import("../../node_modules/pi-caveman/extensions/caveman.ts"),
-  "red-green": () => import("../../node_modules/pi-red-green/dist/index.js"),
-  "agent-dashboard": () =>
-    import("../../node_modules/@blackbelt-technology/pi-agent-dashboard/packages/extension/src/bridge.ts"),
-  "web-access": () => import("../../node_modules/pi-web-access/index.ts"),
-  "pi-flows": () => import("../../node_modules/@blackbelt-technology/pi-flows/extensions/index.ts"),
+  "skill-registry": loadBundled("gentle-pi/extensions/skill-registry.ts"),
+  "sdd-init": loadBundled("gentle-pi/extensions/sdd-init.ts"),
+  "gentle-ai": loadBundled("gentle-pi/extensions/gentle-ai.ts"),
+  "context-mode": loadBundled("context-mode/build/adapters/pi/extension.js"),
+  "ask-user": loadBundled("pi-ask-user/index.ts"),
+  graphify: loadBundled("graphify-pi/extensions/graphify.ts"),
+  subagents: loadBundled("pi-subagents/src/extension/index.ts"),
+  goal: loadBundled("pi-goal/.pi/extensions/pi-goal/index.ts"),
+  docparser: loadBundled("pi-docparser/extensions/docparser/index.ts"),
+  obsidian: loadBundled("@haispeed/pi-obsidian/extensions/obsidian-cli.ts"),
+  "cursor-provider": loadBundled("@netandreus/pi-cursor-provider/index.ts"),
+  btw: loadBundled("pi-btw/extensions/btw.ts"),
+  simplify: loadBundled("pi-simplify/dist/index.js"),
+  "rtk-optimizer": loadBundled("pi-rtk-optimizer/index.ts"),
+  "mcp-adapter": loadBundled("pi-mcp-adapter/index.ts"),
+  "planning-with-files": loadBundled(
+    "@tomxprime/planning-with-files/extensions/planning-with-files/index.ts",
+  ),
+  caveman: loadBundled("pi-caveman/extensions/caveman.ts"),
+  "red-green": loadBundled("pi-red-green/dist/index.js"),
+  "agent-dashboard": loadBundled(
+    "@blackbelt-technology/pi-agent-dashboard/packages/extension/src/bridge.ts",
+  ),
+  "web-access": loadBundled("pi-web-access/index.ts"),
+  "pi-flows": loadBundled("@blackbelt-technology/pi-flows/extensions/index.ts"),
 };
 
 async function registerOne(pi: ExtensionAPI, id: BundledExtensionId): Promise<void> {

package/src/bootstrap/resolve-bundled.ts

@@ -0,0 +1,55 @@
+import { existsSync } from "node:fs";
+import { basename, dirname, join } from "node:path";
+import { fileURLToPath, pathToFileURL } from "node:url";
+
+/** Split `pkg/subpath` — supports scoped packages (`@scope/name/...`). */
+export function parseBundledModulePath(relativePath: string): { pkgName: string; subpath: string } {
+  if (relativePath.startsWith("@")) {
+    const parts = relativePath.split("/");
+    return { pkgName: `${parts[0]}/${parts[1]}`, subpath: parts.slice(2).join("/") };
+  }
+  const slash = relativePath.indexOf("/");
+  if (slash === -1) {
+    return { pkgName: relativePath, subpath: "index.ts" };
+  }
+  return { pkgName: relativePath.slice(0, slash), subpath: relativePath.slice(slash + 1) };
+}
+
+/**
+ * Resolve a bundled dependency file whether npm nested it under hotmilk or hoisted
+ * it next to hotmilk (e.g. `~/.pi/npm/node_modules/context-mode`).
+ */
+export function resolveBundledModule(
+  relativePath: string,
+  fromModuleUrl = import.meta.url,
+): string {
+  const { pkgName, subpath } = parseBundledModulePath(relativePath);
+  let dir = dirname(fileURLToPath(fromModuleUrl));
+
+  while (true) {
+    const nested = join(dir, "node_modules", pkgName, subpath);
+    if (existsSync(nested)) {
+      return nested;
+    }
+
+    if (basename(dir) === "node_modules") {
+      const sibling = join(dir, pkgName, subpath);
+      if (existsSync(sibling)) {
+        return sibling;
+      }
+    }
+
+    const parent = dirname(dir);
+    if (parent === dir) {
+      break;
+    }
+    dir = parent;
+  }
+
+  throw new Error(`Cannot resolve bundled module "${relativePath}" from hotmilk`);
+}
+
+/** Dynamic import URL for a bundled extension entry (works with nested and hoisted installs). */
+export function bundledImportUrl(relativePath: string): string {
+  return pathToFileURL(resolveBundledModule(relativePath)).href;
+}

package/src/index.ts

@@ -15,6 +15,9 @@ import { registerHotmilkSessionLogo } from "./ui/session-logo.ts";
 export default async function registerHotmilk(pi: ExtensionAPI): Promise<void> {
   const runtime = createHotmilkRuntime();
 
+  // Register before bundled imports so session_start handlers exist when bindExtensions emits.
+  registerHotmilkSessionLogo(pi);
+
   if (runtime.extensionToggles["agent-dashboard"]) {
     const { scheduleDashboardWarmStart } = await import("./bootstrap/dashboard.ts");
     scheduleDashboardWarmStart();
@@ -27,6 +30,5 @@ export default async function registerHotmilk(pi: ExtensionAPI): Promise<void> {
   registerDefaultsHandlers(pi, runtime.defaults);
   registerSessionHandlers(pi, runtime);
   registerInputCommands(pi);
-  registerHotmilkSessionLogo(pi);
   pi.on("input", (event, ctx) => routeInputCommand(event.text, pi, ctx));
 }

package/src/ui/session-logo.ts

@@ -56,7 +56,6 @@ function showHotmilkSessionLogo(ctx: ExtensionContext): void {
         },
       };
     });
-    activeTui?.requestRender();
   }, LOGO_SHOW_DELAY_MS);
 }