server.ts

import {
  defineRpcContract,
  type BbPluginApi,
  type PluginProviderDeclaration,
} from "@get-bb/plugin-sdk";
import { z } from "zod";
import { loadUsageSnapshot } from "./lib/load-usage.ts";
import {
  enabledSidebarProviderIds,
  SIDEBAR_PROVIDER_IDS,
} from "./lib/preferences.ts";
import { PROVIDER_IDS } from "./lib/usage.ts";

const costSchema = z
  .object({
    usedUsdCents: z.number().finite(),
    limitUsdCents: z.number().finite(),
  })
  .strict();

const usageWindowSchema = z
  .object({
    label: z.string(),
    usedPercent: z.number().finite(),
    barPercent: z.number().finite().min(0).max(100),
    resetsAt: z.string().nullable(),
    cost: costSchema.nullable(),
  })
  .strict();

const providerSchema = z
  .object({
    id: z.enum(PROVIDER_IDS),
    name: z.string(),
    status: z.enum([
      "ok",
      "not_installed",
      "unauthenticated",
      "expired",
      "error",
    ]),
    accountEmail: z.string().nullable(),
    planLabel: z.string().nullable(),
    message: z.string().nullable(),
    windows: z.array(usageWindowSchema),
  })
  .strict();

export const usageRpcContract = defineRpcContract({
  getPreferences: {
    input: z.null(),
    output: z
      .object({
        enabledProviderIds: z.array(z.enum(SIDEBAR_PROVIDER_IDS)),
      })
      .strict(),
  },
  getUsage: {
    input: z
      .object({ threadId: z.string().trim().min(1).nullable() })
      .strict(),
    output: z
      .object({
        fetchedAt: z.string(),
        host: z
          .object({
            id: z.string().nullable(),
            name: z.string().nullable(),
          })
          .strict(),
        providers: z.array(providerSchema),
      })
      .strict(),
  },
});

const OPENCODE2_PROVIDER: PluginProviderDeclaration = {
  id: "acp-opencode2",
  displayName: "OpenCode 2",
  experimental_visibility: "installed",
  experimental_strings: {
    signInHint: "Run `opencode2 auth login` on the machine to sign in.",
    expiredHint:
      "Your OpenCode 2 session expired. Run `opencode2 auth login`, then reload.",
    installUrl: "https://opencode.ai/v2/docs",
    iconTint: { light: "#2563EB", dark: "#2563EB" },
  },
  experimental_serviceTiers: [
    { id: "default", label: "Default" },
    { id: "fast", label: "Fast" },
  ],
  experimental_bridgeOptions: {
    acpLaunchSpec: {
      displayName: "OpenCode 2",
      command: "opencode2",
      args: ["acp"],
      env: {},
    },
  },
  capabilities: {
    experimental_providerHealth: true,
    experimental_providerUsage: false,
    experimental_providerInstallation: false,
    supportsServiceTier: true,
    supportsNativeUserQuestion: false,
    fork: "tip",
    supportsManualCompaction: true,
    supportsThreadArchive: false,
    supportsThreadRename: false,
    permissionModes: ["accept-edits", "full"],
    reasoningLevels: ["low", "medium", "high", "xhigh", "max"],
  },
  // BB's ACP provider tier contributes the native `/` skills action.
  composerActions: [],
};

export default function plugin(bb: BbPluginApi) {
  const settings = bb.settings.define({
    enableClaudeCode: {
      type: "boolean",
      label: "Enable Claude Code",
      description: "Show Claude Code usage in the sidebar footer.",
      default: true,
    },
    enableCodex: {
      type: "boolean",
      label: "Enable Codex",
      description: "Show Codex usage in the sidebar footer.",
      default: true,
    },
    enableCursor: {
      type: "boolean",
      label: "Enable Cursor",
      description: "Show Cursor usage in the sidebar footer.",
      default: true,
    },
    enableOpenCodeGo: {
      type: "boolean",
      label: "Enable OpenCode Go",
      description: "Show the shared OpenCode and OpenCode 2 Go allowance in the sidebar footer.",
      default: true,
    },
    openCodeGoApiKey: {
      type: "string",
      label: "OpenCode Go API key",
      description:
        "Used only by the BB host to read the shared OpenCode and OpenCode 2 Go allowance.",
      secret: true,
    },
  });

  bb.providers.register(OPENCODE2_PROVIDER);

  bb.rpc.register(usageRpcContract, {
    async getPreferences() {
      return {
        enabledProviderIds: enabledSidebarProviderIds(await settings.get()),
      };
    },
    async getUsage({ threadId }) {
      const { openCodeGoApiKey } = await settings.get();
      return loadUsageSnapshot(
        bb.sdk,
        threadId,
        new Date(),
        openCodeGoApiKey ?? "",
      );
    },
  });
}