import {
  Agent,
  BasePlugin,
  type JsonObject,
  type PluginHooks,
} from "@downcity/agent";

class MessageAugmentPlugin extends BasePlugin {
  readonly name = "message_augment";
  readonly title = "Message Augment";
  readonly description = "Adds extra fields to inbound message payloads.";

  readonly hooks: PluginHooks = {
    pipeline: {
      "chat.enrich_message": [
        async ({ value }) => {
          const body = value as JsonObject;
          return {
            ...body,
            source: body.source || "web",
            pageTitle: body.pageTitle || "Untitled Page",
          };
        },
      ],
    },
  };
}

const agent = new Agent({
  id: "repo-helper",
  path: "/path/to/project",
  tools: {},
  plugins: [new MessageAugmentPlugin()],
});

const message = await agent.plugins.pipeline<JsonObject>("chat.enrich_message", {
  text: "Please summarize this page",
});