feat: @mention routing via Gitea notifications API
- Replace per-repo event polling with notification-based polling - Collab repos default to mention-only mode (respond only when @mentioned) - Own repos default to all-events mode (respond to everything) - Parse @mentions and extract directive text for focused LLM context - Notification poll interval: 30s (configurable via PI_NOTIF_POLL_INTERVAL) - Mark notifications as read after processing (no ID tracking needed) - Add gitea_repo_config tool to switch repos between 'all' and 'mention' modes - Add gitea_tracked_repos tool to show all repos and their response modes - Persist per-repo configs to disk across reloads - Multi-bot coordination: issues can @mention different bots with directives
This commit is contained in:
@@ -2,22 +2,83 @@
|
||||
* pi-gitea Extension — entry point
|
||||
*
|
||||
* Registers Gitea tools (read + write) and optional webhook server.
|
||||
* Supports @mention routing for multi-bot coordination.
|
||||
*/
|
||||
|
||||
import registerReadTools from "./tools/read-tools.js";
|
||||
import registerWriteTools from "./tools/write-tools.js";
|
||||
import { startWebhookServer, stopWebhookServer, startPolling, stopPolling, setSendMessage } from "./webhook/server.js";
|
||||
import {
|
||||
startWebhookServer, stopWebhookServer,
|
||||
startPolling, stopPolling,
|
||||
setSendMessage, getTrackedRepos, setRepoConfig,
|
||||
} from "./webhook/server.js";
|
||||
import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
|
||||
import { Type } from "@sinclair/typebox";
|
||||
|
||||
export default function (pi: ExtensionAPI) {
|
||||
registerReadTools(pi);
|
||||
registerWriteTools(pi);
|
||||
|
||||
// ── Repo config tool ─────────────────────────────────────────────────────
|
||||
|
||||
pi.registerTool({
|
||||
name: "gitea_repo_config",
|
||||
label: "Gitea: Configure Repo Response Mode",
|
||||
description:
|
||||
'Set how the bot responds to events on a repo. ' +
|
||||
'"all" = respond to every event. ' +
|
||||
'"mention" = respond only when @mentioned or assigned. ' +
|
||||
'Collab repos default to "mention", own repos default to "all".',
|
||||
parameters: Type.Object({
|
||||
repo: Type.String({ description: "Repository (owner/name)" }),
|
||||
respondTo: Type.String({ description: '"all" or "mention"' }),
|
||||
}),
|
||||
async execute(_id, params) {
|
||||
const mode = params.respondTo as "all" | "mention";
|
||||
if (mode !== "all" && mode !== "mention") {
|
||||
return {
|
||||
content: [{ type: "text", text: `Invalid mode "${params.respondTo}". Use "all" or "mention".` }],
|
||||
};
|
||||
}
|
||||
const config = setRepoConfig(params.repo, { respondTo: mode });
|
||||
return {
|
||||
content: [{ type: "text", text: `✅ ${params.repo}: respondTo = ${config.respondTo}` }],
|
||||
details: { repo: params.repo, config },
|
||||
};
|
||||
},
|
||||
});
|
||||
|
||||
pi.registerTool({
|
||||
name: "gitea_tracked_repos",
|
||||
label: "Gitea: List Tracked Repos",
|
||||
description: "Show all tracked repos, their type (webhook/collab), and response mode.",
|
||||
parameters: Type.Object({}),
|
||||
async execute() {
|
||||
const { webhook, collab, configs } = getTrackedRepos();
|
||||
const lines: string[] = [];
|
||||
|
||||
for (const [name] of webhook) {
|
||||
const mode = configs.get(name)?.respondTo ?? "all";
|
||||
lines.push(`✅ ${name} — webhook (respondTo: ${mode})`);
|
||||
}
|
||||
for (const name of collab) {
|
||||
const mode = configs.get(name)?.respondTo ?? "mention";
|
||||
lines.push(`📋 ${name} — collab/notification (respondTo: ${mode})`);
|
||||
}
|
||||
|
||||
return {
|
||||
content: [{ type: "text", text: lines.length > 0 ? lines.join("\n") : "No repos tracked." }],
|
||||
details: { webhook: [...webhook.keys()], collab: [...collab], configs: Object.fromEntries(configs) },
|
||||
};
|
||||
},
|
||||
});
|
||||
|
||||
// ── Lifecycle ────────────────────────────────────────────────────────────
|
||||
|
||||
pi.on("session_start", async (_event, ctx) => {
|
||||
console.log("[pi-gitea] Session started");
|
||||
|
||||
const sendMessageFn = (msg: string) => {
|
||||
// Use followUp so events queue when the LLM is already processing
|
||||
ctx.sendUserMessage(msg, { deliverAs: "followUp" });
|
||||
return Promise.resolve();
|
||||
};
|
||||
@@ -25,9 +86,9 @@ export default function (pi: ExtensionAPI) {
|
||||
|
||||
try {
|
||||
await startWebhookServer(pi);
|
||||
startPolling(pi);
|
||||
await startPolling(pi);
|
||||
} catch (err) {
|
||||
console.error("[pi-gitea] Failed to start webhook server:", err);
|
||||
console.error("[pi-gitea] Failed to start:", err);
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
Reference in New Issue
Block a user