import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
// The adapter is a prebuilt .js bundle, not a TypeScript source.
import piExtension from "../node_modules/context-mode/build/adapters/pi/extension.js";

import { filterContextModeMessages } from "./context-mode-filter.ts";
import { guardExtensionApi } from "./prompt-guard.ts";

/**
 * Load context-mode through the prompt guard: the bundle keeps sole ownership
 * of the system prompt, so the upstream routing anchor appended to the
 * `context` message list is filtered out (bundle routing lives in
 * `config/prompt/routing.jsonc`), while upstream active-memory/resume parts
 * pass through untouched. The upstream code this filter depends on is pinned
 * by `tests/system-prompt.test.ts` against the exact `context-mode` version.
 */
export default function bundledContextModeExtension(pi: ExtensionAPI): void {
	piExtension(guardExtensionApi(pi, { contextFilter: filterContextModeMessages }));
}