@justestif/pk 0.4.0 → 0.5.1

package/README.md

@@ -42,22 +42,22 @@ Non-interactive:
 
 ```bash
 pk init my-project --harness claude
-pk init my-project --harness claude,codex   # multiple harnesses
+pk init my-project --harness claude,opencode   # multiple harnesses
 ```
 
-Available harnesses: `claude` (Claude Code), `codex` (Codex), `opencode` (OpenCode).
+Available harnesses: `claude` (Claude Code), `opencode` (OpenCode), `pi` (Pi).
 
 `pk init` does three things:
 
 1. Creates `~/.pk/<name>/` as the knowledge home for this project
-2. Installs a hook or plugin that calls `pk prime` at session start to inject context into your agent
+2. Installs a hook or plugin that injects `PK_KNOWLEDGE_DIR` into the shell and calls `pk prime` for context
 3. Installs the pk skill so your agent knows how to use the CLI
 
-| Harness    | Files written                                       | Mechanism                                 |
-| ---------- | --------------------------------------------------- | ----------------------------------------- |
-| `claude`   | `.claude/hooks/pk-eval.ts`, `.claude/settings.json` | Hook spawns `pk prime` on every prompt    |
-| `codex`    | `AGENTS.md`                                         | Codex reads AGENTS.md natively            |
-| `opencode` | `.opencode/plugins/pk-eval.ts`                      | Plugin spawns `pk prime` at session start |
+| Harness    | Files written                                                              | Env injection                        |
+| ---------- | -------------------------------------------------------------------------- | ------------------------------------ |
+| `claude`   | `.claude/hooks/pk-session-start.sh`, `pk-eval.ts`, `.claude/settings.json`| `SessionStart` → `$CLAUDE_ENV_FILE`  |
+| `opencode` | `.opencode/plugins/pk-eval.ts`                                             | `shell.env` plugin hook              |
+| `pi`       | `.pi/extensions/pk-eval.ts`                                                | `tool_call` mutation                 |
 
 ## Commands
 

package/dist/index.js

@@ -36019,12 +36019,19 @@ async function writeJson2(filePath, data) {
   await Bun.write(filePath, JSON.stringify(data, null, 2) + `
 `);
 }
-function claudeHookScript(knowledgeDir) {
+function sessionStartScript(knowledgeDir) {
+  return `#!/bin/bash
+# pk hook \u2014 auto-generated by pk init
+if [ -n "$CLAUDE_ENV_FILE" ]; then
+  echo 'export PK_KNOWLEDGE_DIR=${knowledgeDir}' >> "$CLAUDE_ENV_FILE"
+fi
+exit 0
+`;
+}
+function userPromptSubmitScript() {
   return `// pk hook \u2014 auto-generated by pk init
 async function handleUserPromptSubmit() {
-  const proc = Bun.spawnSync(['pk', 'prime'], {
-    env: { ...process.env, PK_KNOWLEDGE_DIR: '${knowledgeDir}' }
-  });
+  const proc = Bun.spawnSync(['pk', 'prime']);
   const output = proc.stdout?.toString().trim() ?? '';
   console.log(JSON.stringify({
     hookSpecificOutput: {
@@ -36038,116 +36045,107 @@ async function handleUserPromptSubmit() {
 handleUserPromptSubmit().catch(() => process.exit(0));
 `;
 }
+function addHookEntry(hooks, event, command) {
+  const entries = hooks[event] ?? [];
+  const wrappedHook = { matcher: "", hooks: [{ type: "command", command }] };
+  const alreadyPresent = (entry) => Array.isArray(entry.hooks) && entry.hooks.some((h2) => typeof h2 === "object" && h2 !== null && h2.command === command);
+  if (!entries.some(alreadyPresent)) {
+    entries.push(wrappedHook);
+  }
+  hooks[event] = entries;
+}
 async function writeClaudeHook(projectRoot, knowledgeDir) {
   const hookDir = path9.join(projectRoot, ".claude", "hooks");
-  const hookPath = path9.join(hookDir, "pk-eval.ts");
   mkdirSync4(hookDir, { recursive: true });
-  await Bun.write(hookPath, claudeHookScript(knowledgeDir));
+  const sessionStartPath = path9.join(hookDir, "pk-session-start.sh");
+  await Bun.write(sessionStartPath, sessionStartScript(knowledgeDir));
+  const { chmodSync } = await import("fs");
+  chmodSync(sessionStartPath, 493);
+  const evalPath = path9.join(hookDir, "pk-eval.ts");
+  await Bun.write(evalPath, userPromptSubmitScript());
   const settingsPath = path9.join(projectRoot, ".claude", "settings.json");
   const settings = await readJson(settingsPath);
   const hooks = settings.hooks ?? {};
-  const ups = hooks.UserPromptSubmit ?? [];
-  const hookCmd = `bun run ${hookPath}`;
-  const wrappedHook = {
-    matcher: "",
-    hooks: [
-      {
-        type: "command",
-        command: hookCmd
-      }
-    ]
-  };
-  const hasPkEval = (entry) => typeof entry === "object" && entry !== null && ("hooks" in entry) && Array.isArray(entry.hooks) && entry.hooks.some((hook) => typeof hook === "object" && hook !== null && ("type" in hook) && hook.type === "command" && ("command" in hook) && hook.command === hookCmd);
-  if (!ups.some(hasPkEval)) {
-    ups.push(wrappedHook);
-  }
-  hooks.UserPromptSubmit = ups;
+  addHookEntry(hooks, "SessionStart", sessionStartPath);
+  addHookEntry(hooks, "UserPromptSubmit", `bun run ${evalPath}`);
   settings.hooks = hooks;
   await writeJson2(settingsPath, settings);
 }
 
-// src/commands/harnesses/codex.ts
+// src/commands/harnesses/opencode.ts
 import { mkdirSync as mkdirSync5 } from "fs";
 import path10 from "path";
-
-// src/lib/instruction.ts
-function pkInstruction(knowledgeDir) {
-  return `## pk \u2014 project knowledge
-
-Use the pk skill and its CLI commands to manage project knowledge for this project. Run \`pk synthesize --session-start\` at the start of every session.
-
-Knowledge directory: \`${knowledgeDir}\`
-
-\`\`\`bash
-export PK_KNOWLEDGE_DIR="${knowledgeDir}"
-\`\`\``;
-}
-var FORCED_EVAL_PROMPT = "SKILL EVALUATION REQUIRED: Before responding, explicitly decide YES or NO" + " \u2014 does this request involve project work, investigation, decisions, or knowledge capture" + " where the pk skill applies? If YES, activate the pk skill, then proceed.";
-
-// src/commands/harnesses/codex.ts
-var PK_SECTION_START = "<!-- pk:start -->";
-var PK_SECTION_END = "<!-- pk:end -->";
-async function writeInstructionSection(filePath, content) {
-  const section = `${PK_SECTION_START}
-${content}
-${PK_SECTION_END}
+function openCodePluginScript(knowledgeDir) {
+  return `// pk plugin \u2014 auto-generated by pk init
+export const pkPlugin = async (_input: unknown) => ({
+  'shell.env': async (
+    _input: unknown,
+    output: { env: Record<string, string> },
+  ): Promise<void> => {
+    output.env.PK_KNOWLEDGE_DIR = '${knowledgeDir}';
+  },
+  'experimental.chat.system.transform': async (
+    _input: unknown,
+    output: { system: string[] },
+  ): Promise<void> => {
+    const proc = Bun.spawnSync(['pk', 'prime']);
+    const text = proc.stdout?.toString().trim() ?? '';
+    if (text) output.system.unshift(text);
+  },
+});
 `;
-  let existing = "";
-  try {
-    existing = await Bun.file(filePath).text();
-  } catch {}
-  const startIdx = existing.indexOf(PK_SECTION_START);
-  const endIdx = existing.indexOf(PK_SECTION_END);
-  let updated;
-  if (startIdx !== -1 && endIdx !== -1) {
-    updated = existing.slice(0, startIdx) + section + existing.slice(endIdx + PK_SECTION_END.length + 1);
-  } else {
-    updated = existing ? existing.trimEnd() + `
-
-` + section : section;
-  }
-  mkdirSync5(path10.dirname(filePath), { recursive: true });
-  await Bun.write(filePath, updated);
 }
-async function writeAgentsMd(projectRoot, knowledgeDir) {
-  await writeInstructionSection(path10.join(projectRoot, "AGENTS.md"), pkInstruction(knowledgeDir));
+async function writeOpenCodePlugin(projectRoot, knowledgeDir) {
+  const pluginPath = path10.join(projectRoot, ".opencode", "plugins", "pk-eval.ts");
+  mkdirSync5(path10.dirname(pluginPath), { recursive: true });
+  await Bun.write(pluginPath, openCodePluginScript(knowledgeDir));
 }
 
-// src/commands/harnesses/opencode.ts
+// src/commands/harnesses/pi.ts
 import { mkdirSync as mkdirSync6 } from "fs";
 import path11 from "path";
-function openCodePluginScript(knowledgeDir) {
-  return `// pk plugin \u2014 auto-generated by pk init
-export const experimental = {
-	async 'chat.system.transform'({ system }: { system: string[] }): Promise<void> {
-	   const proc = Bun.spawnSync(['pk', 'prime'], {
-	     env: { ...process.env, PK_KNOWLEDGE_DIR: '${knowledgeDir}' }
-	   });
-	   const output = proc.stdout?.toString().trim() ?? '';
-	   if (output) {
-	     system.unshift(output);
-	   }
-	},
-};
+function piPluginScript(knowledgeDir) {
+  return `// pk extension \u2014 auto-generated by pk init
+import { spawnSync } from 'node:child_process';
+import type { ExtensionAPI } from '@earendil-works/pi-coding-agent';
+
+export default function (pi: ExtensionAPI) {
+  pi.on('before_agent_start', async (event, _ctx) => {
+    const proc = spawnSync('pk', ['prime'], {
+      env: { ...process.env, PK_KNOWLEDGE_DIR: '${knowledgeDir}' },
+      encoding: 'utf-8',
+    });
+    const output = proc.stdout?.trim() ?? '';
+    return output
+      ? { systemPrompt: output + '\\n\\n' + event.systemPrompt }
+      : undefined;
+  });
+
+  pi.on('tool_call', async (event, _ctx) => {
+    if (event.toolName === 'bash' && event.input?.command) {
+      event.input.command = 'export PK_KNOWLEDGE_DIR=${knowledgeDir}\\n' + event.input.command;
+    }
+  });
+}
 `;
 }
-async function writeOpenCodePlugin(projectRoot, knowledgeDir) {
-  const pluginPath = path11.join(projectRoot, ".opencode", "plugins", "pk-eval.ts");
+async function writePiPlugin(projectRoot, knowledgeDir) {
+  const pluginPath = path11.join(projectRoot, ".pi", "extensions", "pk-eval.ts");
   mkdirSync6(path11.dirname(pluginPath), { recursive: true });
-  await Bun.write(pluginPath, openCodePluginScript(knowledgeDir));
+  await Bun.write(pluginPath, piPluginScript(knowledgeDir));
 }
 
 // src/lib/project.ts
 var HARNESSES = [
-  { hint: "forced-eval hook", label: "Claude Code", value: "claude" },
-  { hint: "AGENTS.md injection", label: "Codex", value: "codex" },
-  { hint: "forced-eval plugin", label: "OpenCode", value: "opencode" }
+  { hint: "SessionStart env + UserPromptSubmit context", label: "Claude Code", value: "claude" },
+  { hint: "shell.env + chat.system.transform plugin", label: "OpenCode", value: "opencode" },
+  { hint: "tool_call env + before_agent_start context", label: "Pi", value: "pi" }
 ];
 var HARNESS_VALUES = new Set(HARNESSES.map((h2) => h2.value));
 var HARNESS_ACTIVATION = {
   claude: "start a new Claude Code session in this project",
-  codex: "start a new Codex session in this project",
-  opencode: "reload OpenCode or restart the app"
+  opencode: "reload OpenCode or restart the app",
+  pi: "start a new Pi session in this project"
 };
 async function ensureProject(name21) {
   const kDir = projectDir(name21);
@@ -36206,14 +36204,12 @@ async function initializeProject(options2) {
 function skillTargetDir(harness, projectRoot) {
   switch (harness) {
     case "claude": {
-      return path12.join(os3.homedir(), ".claude", "skills", "pk");
+      return path12.join(projectRoot, ".claude", "skills", "pk");
     }
-    case "opencode": {
+    case "opencode":
+    case "pi": {
       return path12.join(projectRoot, ".agents", "skills", "pk");
     }
-    case "codex": {
-      return path12.join(os3.homedir(), ".codex", "skills", "pk");
-    }
   }
 }
 function skillSourceDir() {
@@ -36241,14 +36237,14 @@ async function applyHarness(harness, ctx) {
       await writeClaudeHook(projectRoot, knowledgeDir);
       break;
     }
-    case "codex": {
-      await writeAgentsMd(projectRoot, knowledgeDir);
-      break;
-    }
     case "opencode": {
       await writeOpenCodePlugin(projectRoot, knowledgeDir);
       break;
     }
+    case "pi": {
+      await writePiPlugin(projectRoot, knowledgeDir);
+      break;
+    }
   }
 }
 async function applyHarnesses(harnesses, ctx) {
@@ -36408,7 +36404,7 @@ Validates frontmatter, required sections, file location, ID uniqueness, tag form
   init: `pk init [<name>] [--harness <harness>]
 Interactive when args omitted. Creates ~/.pk/<name>/ knowledge base, configures hooks, installs skill.
 Detects existing projects and connects without recreating.
-Harnesses: claude, codex, opencode.`,
+Harnesses: claude, opencode, pi.`,
   config: `pk config [--embedding model] [--no-embedding]
 Shows or updates pk configuration. Embedding config shape: {embedding:{enabled,provider,model}}.`
 };
@@ -36442,6 +36438,20 @@ function registerVocab(program2) {
   }));
 }
 
+// src/lib/instruction.ts
+function pkInstruction(knowledgeDir) {
+  return `## pk \u2014 project knowledge
+
+Use the pk skill and its CLI commands to manage project knowledge for this project. Run \`pk synthesize --session-start\` at the start of every session.
+
+Knowledge directory: \`${knowledgeDir}\`
+
+\`\`\`bash
+export PK_KNOWLEDGE_DIR="${knowledgeDir}"
+\`\`\``;
+}
+var FORCED_EVAL_PROMPT = "SKILL EVALUATION REQUIRED: Before responding, explicitly decide YES or NO" + " \u2014 does this request involve project work, investigation, decisions, or knowledge capture" + " where the pk skill applies? If YES, activate the pk skill, then proceed.";
+
 // src/commands/prime.ts
 init_git();
 function registerPrime(program2) {

package/package.json

@@ -1,7 +1,7 @@
 {
    "name": "@justestif/pk",
    "type": "module",
-   "version": "0.4.0",
+   "version": "0.5.1",
    "description": "Project knowledge — structured intake, search, and recall",
    "bin": {
       "pk": "dist/index.js"