import type { Context, PluginDefinition } from "@yaakapp/api";
import { createMcpServer } from "./server.js";

const serverPort = parseInt(process.env.YAAK_PLUGIN_MCP_SERVER_PORT ?? "64343", 10);

let mcpServer: Awaited<ReturnType<typeof createMcpServer>> | null = null;

export const plugin: PluginDefinition = {
  async init(ctx: Context) {
    // Start the server after waiting, so there's an active window open to do things
    // like show the startup toast.
    console.log("Initializing MCP Server plugin");
    setTimeout(async () => {
      try {
        mcpServer = createMcpServer({ yaak: ctx }, serverPort);
      } catch (err) {
        console.error("Failed to start MCP server:", err);
        void ctx.toast.show({
          message: `Failed to start MCP Server: ${err instanceof Error ? err.message : String(err)}`,
          icon: "alert_triangle",
          color: "danger",
          timeout: 10000,
        });
      }
    }, 5000);
  },

  async dispose() {
    console.log("Disposing MCP Server plugin");

    if (mcpServer) {
      await mcpServer.close();
      mcpServer = null;
    }
  },
};