feat: add pi extension (#1)
This commit is contained in:
Hannes
@@ -0,0 +1,141 @@
|
||||
import { createRequire } from "node:module";
|
||||
|
||||
const require = createRequire(import.meta.url);
|
||||
const {
|
||||
DEFAULT_MODE,
|
||||
getDefaultMode,
|
||||
normalizeMode,
|
||||
normalizeConfigMode,
|
||||
normalizePersistedMode,
|
||||
writeDefaultMode,
|
||||
} = require("../hooks/ponytail-config.js");
|
||||
const { getPonytailInstructions, filterSkillBodyForMode } = require("../hooks/ponytail-instructions.js");
|
||||
|
||||
export { filterSkillBodyForMode };
|
||||
export const readDefaultMode = getDefaultMode;
|
||||
|
||||
export function resolveSessionMode(entries, fallbackMode = DEFAULT_MODE) {
|
||||
const fallback = normalizePersistedMode(fallbackMode) || DEFAULT_MODE;
|
||||
if (!Array.isArray(entries)) return fallback;
|
||||
|
||||
for (let i = entries.length - 1; i >= 0; i -= 1) {
|
||||
const entry = entries[i];
|
||||
if (entry?.type !== "custom" || entry?.customType !== "ponytail-mode") continue;
|
||||
|
||||
const mode = normalizePersistedMode(entry?.data?.mode);
|
||||
if (mode) return mode;
|
||||
}
|
||||
|
||||
return fallback;
|
||||
}
|
||||
|
||||
export function parsePonytailCommand(text, defaultMode = DEFAULT_MODE) {
|
||||
const fallback = normalizePersistedMode(defaultMode) || DEFAULT_MODE;
|
||||
const normalizedText = String(text || "").trim().toLowerCase();
|
||||
|
||||
if (!normalizedText) {
|
||||
return { type: "set-mode", mode: fallback === "off" ? "full" : fallback };
|
||||
}
|
||||
|
||||
const [primary, secondary] = normalizedText.split(/\s+/);
|
||||
|
||||
if (primary === "status") return { type: "status" };
|
||||
|
||||
if (primary === "default") {
|
||||
const mode = normalizeConfigMode(secondary);
|
||||
return mode ? { type: "set-default", mode } : { type: "invalid", reason: "invalid-default-mode" };
|
||||
}
|
||||
|
||||
const mode = normalizeMode(primary);
|
||||
return mode ? { type: "set-mode", mode } : { type: "invalid", reason: "invalid-mode", mode: primary };
|
||||
}
|
||||
|
||||
export { writeDefaultMode };
|
||||
|
||||
export default function ponytailExtension(pi) {
|
||||
let currentMode = DEFAULT_MODE;
|
||||
let configuredDefaultMode = getDefaultMode();
|
||||
|
||||
const setMode = (mode, ctx) => {
|
||||
const normalized = normalizePersistedMode(mode);
|
||||
if (!normalized) return;
|
||||
|
||||
currentMode = normalized;
|
||||
pi.appendEntry("ponytail-mode", { mode: normalized });
|
||||
ctx?.ui?.notify?.(`Ponytail mode set to ${normalized}.`, "info");
|
||||
};
|
||||
|
||||
const sendAlias = (skillName, args, ctx) => {
|
||||
const normalized = String(args || "").trim();
|
||||
const message = normalized ? `${skillName} ${normalized}` : skillName;
|
||||
|
||||
if (ctx?.isIdle?.() === false) {
|
||||
pi.sendUserMessage(message, { deliverAs: "followUp" });
|
||||
ctx?.ui?.notify?.(`${skillName} queued as follow-up.`, "info");
|
||||
return;
|
||||
}
|
||||
|
||||
pi.sendUserMessage(message);
|
||||
};
|
||||
|
||||
pi.registerCommand("ponytail", {
|
||||
description: "Set or report Ponytail mode",
|
||||
handler: async (args, ctx) => {
|
||||
const parsed = parsePonytailCommand(args, configuredDefaultMode);
|
||||
|
||||
if (parsed.type === "status") {
|
||||
ctx?.ui?.notify?.(`Ponytail: current ${currentMode} • default ${configuredDefaultMode}`, "info");
|
||||
return;
|
||||
}
|
||||
|
||||
if (parsed.type === "set-default") {
|
||||
const written = writeDefaultMode(parsed.mode);
|
||||
if (written) {
|
||||
configuredDefaultMode = getDefaultMode();
|
||||
const message = configuredDefaultMode === written
|
||||
? `Default Ponytail mode set to ${written}.`
|
||||
: `Saved default ${written}, but env override keeps default at ${configuredDefaultMode}.`;
|
||||
ctx?.ui?.notify?.(message, "info");
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
if (parsed.type === "set-mode") {
|
||||
setMode(parsed.mode, ctx);
|
||||
return;
|
||||
}
|
||||
|
||||
ctx?.ui?.notify?.("Unknown or unsupported /ponytail mode.", "warning");
|
||||
},
|
||||
});
|
||||
|
||||
pi.registerCommand("ponytail-review", {
|
||||
description: "Run /skill:ponytail-review",
|
||||
handler: (_args, ctx) => sendAlias("/skill:ponytail-review", "", ctx),
|
||||
});
|
||||
|
||||
pi.registerCommand("ponytail-help", {
|
||||
description: "Run /skill:ponytail-help",
|
||||
handler: (_args, ctx) => sendAlias("/skill:ponytail-help", "", ctx),
|
||||
});
|
||||
|
||||
pi.on("input", async (event) => {
|
||||
if (event?.source === "extension") return;
|
||||
|
||||
const text = String(event?.text || "");
|
||||
if (currentMode !== "off" && /\b(stop ponytail|normal mode)\b/i.test(text)) {
|
||||
setMode("off");
|
||||
}
|
||||
});
|
||||
|
||||
pi.on("session_start", async (_event, ctx) => {
|
||||
const entries = ctx?.sessionManager?.getBranch?.() || ctx?.sessionManager?.getEntries?.() || [];
|
||||
configuredDefaultMode = getDefaultMode();
|
||||
currentMode = resolveSessionMode(entries, configuredDefaultMode);
|
||||
});
|
||||
|
||||
pi.on("before_agent_start", async (event) => {
|
||||
if (!currentMode || currentMode === "off") return;
|
||||
return { systemPrompt: `${event.systemPrompt}\n\n${getPonytailInstructions(currentMode)}` };
|
||||
});
|
||||
}
|
||||
@@ -0,0 +1,8 @@
|
||||
{
|
||||
"name": "ponytail-pi-extension-dev",
|
||||
"private": true,
|
||||
"type": "module",
|
||||
"scripts": {
|
||||
"test": "node --test ./test/*.test.js"
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,119 @@
|
||||
import assert from "node:assert/strict";
|
||||
import { mkdtempSync, rmSync } from "node:fs";
|
||||
import { tmpdir } from "node:os";
|
||||
import { join } from "node:path";
|
||||
import test from "node:test";
|
||||
|
||||
import ponytailExtension from "../index.js";
|
||||
|
||||
function createPiHarness() {
|
||||
const events = new Map();
|
||||
const commands = new Map();
|
||||
const appendedEntries = [];
|
||||
const sentUserMessages = [];
|
||||
|
||||
const pi = {
|
||||
on(eventName, handler) {
|
||||
events.set(eventName, handler);
|
||||
},
|
||||
registerCommand(name, options) {
|
||||
commands.set(name, options);
|
||||
},
|
||||
appendEntry(customType, data) {
|
||||
appendedEntries.push({ customType, data });
|
||||
},
|
||||
sendUserMessage(text, options) {
|
||||
sentUserMessages.push({ text, options });
|
||||
},
|
||||
};
|
||||
|
||||
ponytailExtension(pi);
|
||||
return { events, commands, appendedEntries, sentUserMessages };
|
||||
}
|
||||
|
||||
function createCommandContext(overrides = {}) {
|
||||
return {
|
||||
isIdle: () => true,
|
||||
sessionManager: { getEntries: () => [] },
|
||||
ui: { notify() {} },
|
||||
...overrides,
|
||||
};
|
||||
}
|
||||
|
||||
function withTempConfig(fn) {
|
||||
const tempConfigHome = mkdtempSync(join(tmpdir(), "ponytail-test-"));
|
||||
const previousXdg = process.env.XDG_CONFIG_HOME;
|
||||
process.env.XDG_CONFIG_HOME = tempConfigHome;
|
||||
|
||||
return Promise.resolve()
|
||||
.then(fn)
|
||||
.finally(() => {
|
||||
if (previousXdg === undefined) delete process.env.XDG_CONFIG_HOME;
|
||||
else process.env.XDG_CONFIG_HOME = previousXdg;
|
||||
rmSync(tempConfigHome, { recursive: true, force: true });
|
||||
});
|
||||
}
|
||||
|
||||
test("extension registers Ponytail commands", () => {
|
||||
const { commands } = createPiHarness();
|
||||
|
||||
assert.deepEqual([...commands.keys()].sort(), ["ponytail", "ponytail-help", "ponytail-review"]);
|
||||
});
|
||||
|
||||
test("/ponytail updates session mode and injects instructions", async () => withTempConfig(async () => {
|
||||
const { commands, events, appendedEntries } = createPiHarness();
|
||||
const ctx = createCommandContext();
|
||||
|
||||
await events.get("session_start")({ reason: "startup" }, ctx);
|
||||
await commands.get("ponytail").handler("ultra", ctx);
|
||||
|
||||
assert.deepEqual(appendedEntries.at(-1), {
|
||||
customType: "ponytail-mode",
|
||||
data: { mode: "ultra" },
|
||||
});
|
||||
|
||||
const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
|
||||
assert.ok(result.systemPrompt.includes("PONYTAIL MODE ACTIVE"));
|
||||
assert.ok(result.systemPrompt.includes("ultra"));
|
||||
}));
|
||||
|
||||
test("session_start restores latest persisted mode", async () => withTempConfig(async () => {
|
||||
const { events } = createPiHarness();
|
||||
const ctx = createCommandContext({
|
||||
sessionManager: {
|
||||
getEntries: () => [
|
||||
{ type: "custom", customType: "ponytail-mode", data: { mode: "lite" } },
|
||||
],
|
||||
},
|
||||
});
|
||||
|
||||
await events.get("session_start")({ reason: "resume" }, ctx);
|
||||
const result = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
|
||||
|
||||
assert.ok(result.systemPrompt.includes("lite"));
|
||||
}));
|
||||
|
||||
test("skill alias commands delegate to Pi skill commands", async () => {
|
||||
const { commands, sentUserMessages } = createPiHarness();
|
||||
const ctx = createCommandContext();
|
||||
|
||||
await commands.get("ponytail-review").handler("", ctx);
|
||||
await commands.get("ponytail-help").handler("", ctx);
|
||||
|
||||
assert.deepEqual(sentUserMessages.map((entry) => entry.text), [
|
||||
"/skill:ponytail-review",
|
||||
"/skill:ponytail-help",
|
||||
]);
|
||||
});
|
||||
|
||||
test("normal mode disables persistent instructions", async () => withTempConfig(async () => {
|
||||
const { commands, events } = createPiHarness();
|
||||
const ctx = createCommandContext();
|
||||
|
||||
await events.get("session_start")({ reason: "startup" }, ctx);
|
||||
await commands.get("ponytail").handler("ultra", ctx);
|
||||
await events.get("input")({ text: "normal mode", source: "interactive" }, ctx);
|
||||
|
||||
const disabled = await events.get("before_agent_start")({ systemPrompt: "BASE" }, ctx);
|
||||
assert.equal(disabled, undefined);
|
||||
}));
|
||||
@@ -0,0 +1,68 @@
|
||||
import assert from "node:assert/strict";
|
||||
import { existsSync, mkdtempSync, readFileSync, rmSync } from "node:fs";
|
||||
import { tmpdir } from "node:os";
|
||||
import { join } from "node:path";
|
||||
import test from "node:test";
|
||||
|
||||
import {
|
||||
filterSkillBodyForMode,
|
||||
parsePonytailCommand,
|
||||
readDefaultMode,
|
||||
resolveSessionMode,
|
||||
writeDefaultMode,
|
||||
} from "../index.js";
|
||||
|
||||
test("parsePonytailCommand falls back to full when invoked bare and default is off", () => {
|
||||
assert.deepEqual(parsePonytailCommand("", "off"), { type: "set-mode", mode: "full" });
|
||||
});
|
||||
|
||||
test("parsePonytailCommand parses modes, status, and default subcommand", () => {
|
||||
assert.deepEqual(parsePonytailCommand("ultra", "full"), { type: "set-mode", mode: "ultra" });
|
||||
assert.deepEqual(parsePonytailCommand("status", "full"), { type: "status" });
|
||||
assert.deepEqual(parsePonytailCommand("default lite", "full"), { type: "set-default", mode: "lite" });
|
||||
});
|
||||
|
||||
test("resolveSessionMode prefers latest persisted session mode", () => {
|
||||
const entries = [
|
||||
{ type: "custom", customType: "ponytail-mode", data: { mode: "lite" } },
|
||||
{ type: "custom", customType: "ponytail-mode", data: { mode: "ultra" } },
|
||||
];
|
||||
|
||||
assert.equal(resolveSessionMode(entries, "full"), "ultra");
|
||||
});
|
||||
|
||||
test("readDefaultMode and writeDefaultMode use XDG config path", () => {
|
||||
const tempDir = mkdtempSync(join(tmpdir(), "ponytail-config-"));
|
||||
const previousXdg = process.env.XDG_CONFIG_HOME;
|
||||
const previousDefault = process.env.PONYTAIL_DEFAULT_MODE;
|
||||
const configPath = join(tempDir, "ponytail", "config.json");
|
||||
process.env.XDG_CONFIG_HOME = tempDir;
|
||||
delete process.env.PONYTAIL_DEFAULT_MODE;
|
||||
|
||||
try {
|
||||
assert.equal(readDefaultMode(), "full");
|
||||
assert.equal(writeDefaultMode("ultra"), "ultra");
|
||||
assert.equal(readDefaultMode(), "ultra");
|
||||
assert.ok(existsSync(configPath));
|
||||
assert.deepEqual(JSON.parse(readFileSync(configPath, "utf8")), { defaultMode: "ultra" });
|
||||
} finally {
|
||||
if (previousXdg === undefined) delete process.env.XDG_CONFIG_HOME;
|
||||
else process.env.XDG_CONFIG_HOME = previousXdg;
|
||||
if (previousDefault === undefined) delete process.env.PONYTAIL_DEFAULT_MODE;
|
||||
else process.env.PONYTAIL_DEFAULT_MODE = previousDefault;
|
||||
rmSync(tempDir, { recursive: true, force: true });
|
||||
}
|
||||
});
|
||||
|
||||
test("filterSkillBodyForMode keeps only requested intensity examples and rows", () => {
|
||||
const body = `---\nname: ponytail\n---\n| **lite** | keep lite |\n| **full** | keep full |\n| **ultra** | keep ultra |\n- lite: Lite example\n- full: Full example\n- ultra: Ultra example\nOther line`;
|
||||
|
||||
const filtered = filterSkillBodyForMode(body, "ultra");
|
||||
|
||||
assert.ok(!filtered.includes("keep lite"));
|
||||
assert.ok(!filtered.includes("keep full"));
|
||||
assert.ok(filtered.includes("keep ultra"));
|
||||
assert.ok(!filtered.includes("Lite example"));
|
||||
assert.ok(filtered.includes("Ultra example"));
|
||||
assert.ok(filtered.includes("Other line"));
|
||||
});
|
||||
Reference in New Issue
Block a user