getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-17 13:05:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: update prompt-url widget

Browse source
This commit is contained in:
Mario Zechner 2026-01-09 22:09:05 +01:00
parent e2d14cdc23
commit 4c68ab6eab
1 changed files with 46 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

60
.pi/extensions/prompt-url-widget.ts
View file

@ -1,4 +1,4 @@
import { DynamicBorder, type ExtensionAPI } from "@mariozechner/pi-coding-agent"; import { DynamicBorder, type ExtensionAPI, type ExtensionContext } from "@mariozechner/pi-coding-agent";
import { Container, Text } from "@mariozechner/pi-tui"; import { Container, Text } from "@mariozechner/pi-tui";
const PR_PROMPT_PATTERN = /^You are given one or more GitHub PR URLs:\s*(\S+)/im; const PR_PROMPT_PATTERN = /^You are given one or more GitHub PR URLs:\s*(\S+)/im;
@ -35,13 +35,12 @@ async function fetchGhMetadata(
pi: ExtensionAPI, pi: ExtensionAPI,
kind: PromptMatch["kind"], kind: PromptMatch["kind"],
url: string, url: string,
signal?: AbortSignal,
): Promise<GhMetadata | undefined> { ): Promise<GhMetadata | undefined> {
const args = const args =
kind === "pr" ? ["pr", "view", url, "--json", "title,author"] : ["issue", "view", url, "--json", "title,author"]; kind === "pr" ? ["pr", "view", url, "--json", "title,author"] : ["issue", "view", url, "--json", "title,author"];
try { try {
const result = await pi.exec("gh", args, { signal }); const result = await pi.exec("gh", args);
if (result.code !== 0 || !result.stdout) return undefined; if (result.code !== 0 || !result.stdout) return undefined;
return JSON.parse(result.stdout) as GhMetadata; return JSON.parse(result.stdout) as GhMetadata;
} catch { } catch {
@ -60,18 +59,7 @@ function formatAuthor(author?: GhMetadata["author"]): string | undefined {
} }
export default function promptUrlWidgetExtension(pi: ExtensionAPI) { export default function promptUrlWidgetExtension(pi: ExtensionAPI) {
pi.on("before_agent_start", async (event, ctx) => { const setWidget = (ctx: ExtensionContext, match: PromptMatch, title?: string, authorText?: string) => {
if (!ctx.hasUI) return;
const match = extractPromptMatch(event.prompt);
if (!match) {
ctx.ui.setWidget("prompt-url", undefined);
return;
}
const meta = await fetchGhMetadata(pi, match.kind, match.url, event.signal);
const title = meta?.title?.trim();
const authorText = formatAuthor(meta?.author);
ctx.ui.setWidget("prompt-url", (_tui, thm) => { ctx.ui.setWidget("prompt-url", (_tui, thm) => {
const titleText = title ? thm.fg("accent", title) : thm.fg("accent", match.url); const titleText = title ? thm.fg("accent", title) : thm.fg("accent", match.url);
const authorLine = authorText ? thm.fg("muted", authorText) : undefined; const authorLine = authorText ? thm.fg("muted", authorText) : undefined;
@ -86,10 +74,52 @@ export default function promptUrlWidgetExtension(pi: ExtensionAPI) {
container.addChild(new Text(lines.join("\n"), 1, 0)); container.addChild(new Text(lines.join("\n"), 1, 0));
return container; return container;
}); });
};
pi.on("before_agent_start", async (event, ctx) => {
if (!ctx.hasUI) return;
const match = extractPromptMatch(event.prompt);
if (!match) {
ctx.ui.setWidget("prompt-url", undefined);
return;
}
setWidget(ctx, match);
void fetchGhMetadata(pi, match.kind, match.url).then((meta) => {
const title = meta?.title?.trim();
const authorText = formatAuthor(meta?.author);
setWidget(ctx, match, title, authorText);
});
}); });
pi.on("session_start", async (_event, ctx) => { pi.on("session_start", async (_event, ctx) => {
if (!ctx.hasUI) return; if (!ctx.hasUI) return;
const entries = ctx.sessionManager.getEntries();
const lastUser = [...entries].reverse().find((entry) => {
return entry.type === "message" && entry.message.role === "user";
});
const content = lastUser?.type === "message" && lastUser?.message.role === "user" ? lastUser.message.content : undefined;
const text =
typeof content === "string"
? content
: content
?.filter((block): block is { type: "text"; text: string } => block.type === "text")
.map((block) => block.text)
.join("\n") ?? "";
const match = text ? extractPromptMatch(text) : undefined;
if (!match) {
ctx.ui.setWidget("prompt-url", undefined); ctx.ui.setWidget("prompt-url", undefined);
return;
}
setWidget(ctx, match);
void fetchGhMetadata(pi, match.kind, match.url).then((meta) => {
const title = meta?.title?.trim();
const authorText = formatAuthor(meta?.author);
setWidget(ctx, match, title, authorText);
});
}); });
} }