import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
import { createStore } from "./state";
import { createInbox } from "./inbox";
import { registerLifecycle } from "./lifecycle/lifecycle";
import { registerTools } from "./tools/index";
import { registerCommands } from "./commands/commands";
import { registerStorageFlags, resolveAdapter } from "./adapter/registry";
import type { ThreadingContext, ThreadingState } from "./context";

export default function (pi: ExtensionAPI) {
  pi.registerFlag("thread-id", {
    type: "string",
    description: "The ID of the current agent thread. Targetable via thread_send thread:<id>.",
  });
  pi.registerFlag("thread-parent", {
    type: "string",
    description: "The ID of the parent thread. This is used for Blocker request escalation.",
  });
  pi.registerFlag("thread-role", {
    type: "string",
    description: "Role label for this thread. Targetable via thread_send role:<role>",
  });
  pi.registerFlag("thread-journal", {
    type: "string",
    description: 'Journal entry cadence: "turn" (default), "done", or "off"',
  });
  pi.registerFlag("thread-journal-model", {
    type: "string",
    description: "Model for journal entry writing. Defaults to the thread's own model.",
  });
  pi.registerFlag("thread-storage", {
    type: "string",
    description: 'Storage backend: "local" (default)',
    default: "local",
  });

  // Register all flogs
  registerStorageFlags(pi);

  const state: ThreadingState = {
    active: undefined,
    toolUsedThisTurn: false,
    inFlightSince: null,
    compactingSince: null,
  };

  const store = createStore(pi, () => resolveAdapter(pi), state);
  const inbox = createInbox(pi, store, state);

  const threading: ThreadingContext = {
    pi,
    store,
    inbox,
    state,
  };

  registerLifecycle(threading);
  registerTools(threading);
  registerCommands(threading);
}