salvacybersec/feynman
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: respect feynman agent dir in vendored pi-subagents

Browse Source
This commit is contained in:
Advait Paliwal
2026-03-28 21:44:50 -07:00
parent 62d63be1d8
commit ab8a284c74
5 changed files with 254 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/lib/pi-subagents-patch.d.mts Normal file
View File

@@ -0,0 +1,2 @@
export const PI_SUBAGENTS_PATCH_TARGETS: string[];
export function patchPiSubagentsSource(relativePath: string, source: string): string;

124
scripts/lib/pi-subagents-patch.mjs Normal file
View File

@@ -0,0 +1,124 @@
export const PI_SUBAGENTS_PATCH_TARGETS = [
"index.ts",
"agents.ts",
"artifacts.ts",
"run-history.ts",
"skills.ts",
"chain-clarify.ts",
];
const RESOLVE_PI_AGENT_DIR_HELPER = [
"function resolvePiAgentDir(): string {",
' const configured = process.env.PI_CODING_AGENT_DIR?.trim();',
' if (!configured) return path.join(os.homedir(), ".pi", "agent");',
' return configured.startsWith("~/") ? path.join(os.homedir(), configured.slice(2)) : configured;',
"}",
].join("\n");
function injectResolvePiAgentDirHelper(source) {
if (source.includes("function resolvePiAgentDir(): string {")) {
return source;
}
const lines = source.split("\n");
let insertAt = 0;
let importSeen = false;
let importOpen = false;
for (let index = 0; index < lines.length; index += 1) {
const trimmed = lines[index].trim();
if (!importSeen) {
if (trimmed === "" || trimmed.startsWith("/**") || trimmed.startsWith("*") || trimmed.startsWith("*/")) {
insertAt = index + 1;
continue;
}
if (trimmed.startsWith("import ")) {
importSeen = true;
importOpen = !trimmed.endsWith(";");
insertAt = index + 1;
continue;
}
break;
}
if (trimmed.startsWith("import ")) {
importOpen = !trimmed.endsWith(";");
insertAt = index + 1;
continue;
}
if (importOpen) {
if (trimmed.endsWith(";")) importOpen = false;
insertAt = index + 1;
continue;
}
if (trimmed === "") {
insertAt = index + 1;
continue;
}
insertAt = index;
break;
}
return [...lines.slice(0, insertAt), "", RESOLVE_PI_AGENT_DIR_HELPER, "", ...lines.slice(insertAt)].join("\n");
}
function replaceAll(source, from, to) {
return source.split(from).join(to);
}
export function patchPiSubagentsSource(relativePath, source) {
let patched = source;
switch (relativePath) {
case "index.ts":
patched = replaceAll(
patched,
'const configPath = path.join(os.homedir(), ".pi", "agent", "extensions", "subagent", "config.json");',
'const configPath = path.join(resolvePiAgentDir(), "extensions", "subagent", "config.json");',
);
break;
case "agents.ts":
patched = replaceAll(
patched,
'const userDir = path.join(os.homedir(), ".pi", "agent", "agents");',
'const userDir = path.join(resolvePiAgentDir(), "agents");',
);
break;
case "artifacts.ts":
patched = replaceAll(
patched,
'const sessionsBase = path.join(os.homedir(), ".pi", "agent", "sessions");',
'const sessionsBase = path.join(resolvePiAgentDir(), "sessions");',
);
break;
case "run-history.ts":
patched = replaceAll(
patched,
'const HISTORY_PATH = path.join(os.homedir(), ".pi", "agent", "run-history.jsonl");',
'const HISTORY_PATH = path.join(resolvePiAgentDir(), "run-history.jsonl");',
);
break;
case "skills.ts":
patched = replaceAll(
patched,
'const AGENT_DIR = path.join(os.homedir(), ".pi", "agent");',
"const AGENT_DIR = resolvePiAgentDir();",
);
break;
case "chain-clarify.ts":
patched = replaceAll(
patched,
'const dir = path.join(os.homedir(), ".pi", "agent", "agents");',
'const dir = path.join(resolvePiAgentDir(), "agents");',
);
break;
default:
return source;
}
if (patched === source) {
return source;
}
return injectResolvePiAgentDirHelper(patched);
}

15
scripts/patch-embedded-pi.mjs
View File

@@ -4,6 +4,7 @@ import { createRequire } from "node:module";
import { dirname, resolve } from "node:path";
import { fileURLToPath } from "node:url";
import { FEYNMAN_LOGO_HTML } from "../logo.mjs";
import { PI_SUBAGENTS_PATCH_TARGETS, patchPiSubagentsSource } from "./lib/pi-subagents-patch.mjs";
const here = dirname(fileURLToPath(import.meta.url));
const appRoot = resolve(here, "..");
@@ -54,6 +55,7 @@ const interactiveThemePath = piPackageRoot ? resolve(piPackageRoot, "dist", "mod
const terminalPath = piTuiRoot ? resolve(piTuiRoot, "dist", "terminal.js") : null;
const editorPath = piTuiRoot ? resolve(piTuiRoot, "dist", "components", "editor.js") : null;
const workspaceRoot = resolve(appRoot, ".feynman", "npm", "node_modules");
const piSubagentsRoot = resolve(workspaceRoot, "pi-subagents");
const webAccessPath = resolve(workspaceRoot, "pi-web-access", "index.ts");
const sessionSearchIndexerPath = resolve(
workspaceRoot,
@@ -243,6 +245,19 @@ function ensurePandoc() {
ensurePandoc();
if (existsSync(piSubagentsRoot)) {
for (const relativePath of PI_SUBAGENTS_PATCH_TARGETS) {
const entryPath = resolve(piSubagentsRoot, relativePath);
if (!existsSync(entryPath)) continue;
const source = readFileSync(entryPath, "utf8");
const patched = patchPiSubagentsSource(relativePath, source);
if (patched !== source) {
writeFileSync(entryPath, patched, "utf8");
}
}
}
if (packageJsonPath && existsSync(packageJsonPath)) {
const pkg = JSON.parse(readFileSync(packageJsonPath, "utf8"));
if (pkg.piConfig?.name !== "feynman" || pkg.piConfig?.configDir !== ".feynman") {