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

Add CustomMessageEntry rendering infrastructure

Browse source 
- Add renderCustomMessage to HookAPI for registering custom renderers
- Add CustomMessageRenderer type and CustomMessageRenderOptions
- Store customMessageRenderers in LoadedHook
- Add getCustomMessageRenderer(customType) to HookRunner
- SessionContext.entries now aligned with messages (same length, corresponding indices)

TUI can now correlate messages with their source entries to identify
custom_message entries and use hook-provided renderers.
This commit is contained in:
Mario Zechner 2025-12-26 23:02:53 +01:00
parent 3ecaaa5937
commit 11a7845ceb
7 changed files with 83 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

18
packages/coding-agent/src/core/hooks/loader.ts
View file

@ -10,7 +10,7 @@ import { fileURLToPath } from "node:url";
import type { Attachment } from "@mariozechner/pi-agent-core";
import { createJiti } from "jiti";
import { getAgentDir } from "../../config.js";
import type { HookAPI, HookFactory } from "./types.js";
import type { CustomMessageRenderer, HookAPI, HookFactory } from "./types.js";
// Create require function to resolve module paths at runtime
const require = createRequire(import.meta.url);
@ -61,6 +61,8 @@ export interface LoadedHook {
resolvedPath: string;
/** Map of event type to handler functions */
handlers: Map<string, HandlerFn[]>;
/** Map of customType to custom message renderer */
customMessageRenderers: Map<string, CustomMessageRenderer>;
/** Set the send handler for this hook's pi.send() */
setSendHandler: (handler: SendHandler) => void;
}
@ -110,16 +112,18 @@ function resolveHookPath(hookPath: string, cwd: string): string {
}
/**
* Create a HookAPI instance that collects handlers.
* Returns the API and a function to set the send handler later.
* Create a HookAPI instance that collects handlers and renderers.
* Returns the API, renderers map, and a function to set the send handler later.
*/
function createHookAPI(handlers: Map<string, HandlerFn[]>): {
api: HookAPI;
customMessageRenderers: Map<string, CustomMessageRenderer>;
setSendHandler: (handler: SendHandler) => void;
} {
let sendHandler: SendHandler = () => {
// Default no-op until mode sets the handler
};
const customMessageRenderers = new Map<string, CustomMessageRenderer>();
const api: HookAPI = {
on(event: string, handler: HandlerFn): void {
@ -130,10 +134,14 @@ function createHookAPI(handlers: Map<string, HandlerFn[]>): {
send(text: string, attachments?: Attachment[]): void {
sendHandler(text, attachments);
},
renderCustomMessage(customType: string, renderer: CustomMessageRenderer): void {
customMessageRenderers.set(customType, renderer);
},
} as HookAPI;
return {
api,
customMessageRenderers,
setSendHandler: (handler: SendHandler) => {
sendHandler = handler;
},
@ -164,13 +172,13 @@ async function loadHook(hookPath: string, cwd: string): Promise<{ hook: LoadedHo
// Create handlers map and API
const handlers = new Map<string, HandlerFn[]>();
const { api, setSendHandler } = createHookAPI(handlers);
const { api, customMessageRenderers, setSendHandler } = createHookAPI(handlers);
// Call factory to register handlers
factory(api);
return {
hook: { path: hookPath, resolvedPath, handlers, setSendHandler },
hook: { path: hookPath, resolvedPath, handlers, customMessageRenderers, setSendHandler },
error: null,
};
} catch (err) {