shockbot/utils/repoSettings.ts

import type { AgentName, BashPermission, ToolPermission } from "../external.ts";
import type { RepoContext } from "./github.ts";

export interface Mode {
  id: string;
  name: string;
  description: string;
  prompt: string;
}

export interface RepoSettings {
  defaultAgent: AgentName | null;
  modes: Mode[];
  web: ToolPermission;
  search: ToolPermission;
  write: ToolPermission;
  bash: BashPermission;
}

/**
 * Fetch repository settings from the Pullfrog API
 * Returns defaults if repo doesn't exist or fetch fails
 */
export async function fetchRepoSettings(params: {
  token: string;
  repoContext: RepoContext;
}): Promise<RepoSettings> {
  const apiUrl = process.env.API_URL || "https://pullfrog.com";
  const timeoutMs = 30000;
  const controller = new AbortController();
  const timeoutId = setTimeout(() => controller.abort(), timeoutMs);

  try {
    const response = await fetch(
      `${apiUrl}/api/repo/${params.repoContext.owner}/${params.repoContext.name}/settings`,
      {
        method: "GET",
        headers: {
          Authorization: `Bearer ${params.token}`,
          "Content-Type": "application/json",
        },
        signal: controller.signal,
      }
    );

    clearTimeout(timeoutId);

    if (!response.ok) {
      return {
        defaultAgent: null,
        modes: [],
        web: "enabled",
        search: "enabled",
        write: "enabled",
        bash: "restricted",
      };
    }

    const settings = (await response.json()) as RepoSettings | null;
    if (settings === null) {
      return {
        defaultAgent: null,
        modes: [],
        web: "enabled",
        search: "enabled",
        write: "enabled",
        bash: "restricted",
      };
    }

    return settings;
  } catch {
    clearTimeout(timeoutId);
    return {
      defaultAgent: null,
      modes: [],
      web: "enabled",
      search: "enabled",
      write: "enabled",
      bash: "restricted",
    };
  }
}