shockbot/utils/runContext.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[];
  repoInstructions: string;
  web: ToolPermission;
  search: ToolPermission;
  write: ToolPermission;
  bash: BashPermission;
}

export interface RunContext {
  settings: RepoSettings;
  apiToken: string;
}

const defaultSettings: RepoSettings = {
  defaultAgent: null,
  modes: [],
  repoInstructions: "",
  web: "enabled",
  search: "enabled",
  write: "enabled",
  bash: "restricted",
};

const defaultRunContext: RunContext = {
  settings: defaultSettings,
  apiToken: "",
};

/**
 * fetch run context from Pullfrog API
 * returns settings + API token for subsequent calls
 * returns defaults if fetch fails
 */
export async function fetchRunContext(params: {
  token: string;
  repoContext: RepoContext;
}): Promise<RunContext> {
  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}/run-context`,
      {
        method: "GET",
        headers: {
          Authorization: `Bearer ${params.token}`,
          "Content-Type": "application/json",
        },
        signal: controller.signal,
      }
    );

    clearTimeout(timeoutId);

    if (!response.ok) {
      return defaultRunContext;
    }

    const data = (await response.json()) as {
      settings: RepoSettings | null;
      apiToken: string;
    } | null;

    if (data === null) {
      return defaultRunContext;
    }

    return {
      settings: data.settings ?? defaultSettings,
      apiToken: data.apiToken,
    };
  } catch {
    clearTimeout(timeoutId);
    return defaultRunContext;
  }
}