// Otty integration plugin for OpenCode (ESM) — the real logic.
//
// Reports agent state changes to the Otty app over IPC so a terminal pane can
// show the processing / idle / awaiting-input badge. This file ships as a
// readable, code-signed module inside Otty.app so you can audit exactly what
// runs. Otty installs a tiny shim at
//   ~/.config/opencode/plugins/otty-integration.js
// that sets the per-install paths on globalThis and re-exports this module.
//
// The per-install values (otty-cli path, IPC socket, marker) are read LAZILY
// inside OttyPlugin — NOT at module top level — because the shim sets them
// just before OpenCode invokes the plugin, which is after this module has
// already been evaluated by the `export *` in the shim.
import { spawn } from "child_process";

function sessionIdFrom(event) {
  return event?.properties?.sessionID
    || event?.properties?.info?.id
    || event?.sessionID
    || event?.sessionId
    || event?.session?.id
    || event?.properties?.session?.id
    || "";
}

function directoryFrom(event, fallbackDirectory, fallbackWorktree) {
  return event?.properties?.info?.directory
    || event?.properties?.directory
    || event?.directory
    || fallbackDirectory
    || fallbackWorktree
    || "";
}

function stateFromStatus(status) {
  const type = status?.type || status || "";
  if (type === "busy" || type === "retry") return "processing";
  if (type === "idle") return "idle";
  return null;
}

export const OttyPlugin = async ({ client, directory, worktree }) => {
  // Per-install values injected by the shim (see file header). Read here, at
  // invocation time, so the shim's assignments have already run.
  const CLI = globalThis.__OTTY_CLI || "otty-cli";
  const SOCKET = globalThis.__OTTY_SOCKET || "";

  function notify(sessionId, state, cwd, contextB64, onError) {
    if (!sessionId) return;
    const args = [
      "state:opencode",
      `session-id=${sessionId}`,
      `state=${state}`,
      // process.pid is the opencode process; Otty matches it against each
      // pane's process tree (pid-accurate), and correctly ignores opencode
      // instances running outside Otty (e.g. an editor).
      `agent-pid=${process.pid}`,
    ];
    if (cwd) {
      args.push(`cwd=${cwd}`);
    }
    if (contextB64) {
      args.push(`context-b64=${contextB64}`);
    }

    const env = { ...process.env };
    if (SOCKET) env.OTTY_SOCKET = SOCKET;
    const proc = spawn(CLI, args, {
      stdio: "ignore",
      detached: true,
      env,
    });
    proc.on("error", (error) => {
      if (onError) {
        void onError(error);
      }
    });
    proc.unref();
  }

  async function log(level, message, extra = {}) {
    try {
      await client.app.log({
        body: {
          service: "otty-plugin",
          level,
          message,
          extra,
        },
      });
    } catch {}
  }

  async function dispatch(sessionId, state, cwd, message, extra = {}, contextB64 = null) {
    if (!sessionId) return;
    await log("info", message, {
      sessionId,
      state,
      cwd,
      ...extra,
    });
    notify(sessionId, state, cwd, contextB64, async (error) => {
      await log("error", "Failed to spawn otty-cli", {
        sessionId,
        state,
        cwd,
        cli: CLI,
        socket: SOCKET,
        error: String(error),
      });
    });
  }

  await log("info", "OpenCode plugin initialized", {
    directory: directory || "",
    worktree: worktree || "",
    cli: CLI,
    socket: SOCKET,
  });

  // OpenCode has no dedicated "completed" event — a session is a state machine
  // and "busy → idle" IS task completion. The authoritative idle signals are
  // `session.idle` and `session.status` with {type:"idle"}; busy/retry mean
  // processing. `session.updated` is deliberately NOT here: its payload is the
  // whole Session object, so it fires on every metadata change (title, cost,
  // tokens, time) DURING active work and again as trailing flushes AFTER idle.
  // Treating it as idle made Otty see repeated processing→idle edges within a
  // single prompt — one spurious "Task complete" notification per tool/metadata
  // cycle (5+ per prompt) plus a flickering tab badge. Registration of the
  // session id is covered by chat.message / session.created / the app's HTTP-log
  // fallback, so dropping it loses no functionality.
  const interestingTypes = new Set([
    "session.created",
    "session.status",
    "session.idle",
    "tui.session.select",
  ]);

  return {
    event: async ({ event }) => {
      const type = event?.type || "";
      const sid = sessionIdFrom(event);
      const cwd = directoryFrom(event, directory, worktree);
      if (!interestingTypes.has(type)) return;

      if (!sid) {
        await log("warn", "OpenCode event missing session ID", {
          type,
          cwd,
        });
        return;
      }

      if (type === "session.created") {
        // A freshly created session is genuinely idle and has no preceding
        // work, so this never fires a "Task complete" notification; it just
        // seeds the session id early.
        await dispatch(sid, "idle", cwd, "Dispatching session lifecycle event", {
          type,
        });
        return;
      }

      if (type === "tui.session.select") {
        await dispatch(sid, "idle", cwd, "Dispatching TUI session selection", {
          type,
        });
        return;
      }

      if (type === "session.status") {
        const state = stateFromStatus(event?.properties?.status || event?.status);
        if (state) {
          await dispatch(sid, state, cwd, "Dispatching session status", {
            type,
            status: event?.properties?.status?.type || event?.properties?.status || event?.status?.type || event?.status || "",
          });
        } else {
          await log("warn", "OpenCode status event did not map to Otty state", {
            type,
            sessionId: sid,
            cwd,
            status: event?.properties?.status?.type || event?.properties?.status || event?.status?.type || event?.status || "",
          });
        }
        return;
      }

      if (type === "session.idle") {
        await dispatch(sid, "idle", cwd, "Dispatching session idle", {
          type,
        });
      }
    },
    "chat.message": async (input) => {
      const sid = input?.sessionID || "";
      const cwd = directory || worktree || "";
      if (!sid) {
        await log("warn", "chat.message missing session ID", {
          cwd,
        });
        return;
      }
      await dispatch(sid, "processing", cwd, "Dispatching chat message", {
        type: "chat.message",
        agent: input?.agent || "",
        model: input?.model?.providerID && input?.model?.modelID
          ? `${input.model.providerID}/${input.model.modelID}`
          : "",
        variant: input?.variant || "",
      });
    },
    "command.execute.before": async (input) => {
      const sid = input?.sessionID || "";
      const cwd = directory || worktree || "";
      if (!sid) {
        await log("warn", "command.execute.before missing session ID", {
          cwd,
          command: input?.command || "",
        });
        return;
      }
      await dispatch(sid, "processing", cwd, "Dispatching command execution", {
        type: "command.execute.before",
        command: input?.command || "",
      });
    },
    "tool.execute.before": async (input) => {
      const sid = input?.sessionID || "";
      const cwd = directory || worktree || "";
      if (!sid) {
        await log("warn", "tool.execute.before missing session ID", {
          cwd,
          tool: input?.tool || "",
        });
        return;
      }
      await dispatch(sid, "processing", cwd, "Dispatching tool execution", {
        type: "tool.execute.before",
        tool: input?.tool || "",
        callId: input?.callID || "",
      });
    },
    "permission.ask": async (input) => {
      const sid = input?.sessionID || "";
      if (!sid) {
        await log("warn", "permission.ask missing session ID", {
          cwd: directory || worktree || "",
        });
        return;
      }
      // Base64-encode the permission context for auto-approve
      let contextB64 = null;
      try {
        const ctx = JSON.stringify(input || {});
        contextB64 = Buffer.from(ctx).toString("base64");
      } catch {}
      await dispatch(sid, "awaiting", directory || worktree || "", "Dispatching permission ask", {
        type: "permission.ask",
      }, contextB64);
    },
  };
};
