Function main

packages/core/script/generate-digitalocean.ts:590–733

()

// ---------------------------------------------------------------------------

async function main() {
  const args = process.argv.slice(2);
  const dryRun = args.includes("--dry-run");
  const newOnly = args.includes("--new-only");

  // Resolve API key
  const apiKeyArg = args.find((a) => a.startsWith("--api-key"));
  const apiKey =
    (apiKeyArg?.includes("=") ? apiKeyArg.split("=")[1] : args[args.indexOf(apiKeyArg!) + 1]) ??
    process.env.DIGITALOCEAN_API_TOKEN;

  if (!apiKey) {
    console.error("Error: DIGITALOCEAN_API_TOKEN is required (or pass --api-key=<key>)");
    console.error("Get one from: https://cloud.digitalocean.com/account/api/tokens");
    process.exit(1);
  }

  const modelsDir = path.join(import.meta.dirname, "..", "..", "..", "providers", "digitalocean", "models");

  const prefix = dryRun ? "[DRY RUN] " : "";
  console.log(`${prefix}Fetching DigitalOcean models from API...`);

  // Fetch both APIs in parallel
  const [modelsRes, pricingRes] = await Promise.all([
    fetch(MODELS_API, { headers: { Authorization: `Bearer ${apiKey}`, "Content-Type": "application/json" } }),
    fetch(PRICING_API, { headers: { "User-Agent": "models.dev/digitalocean-sync" } }),
  ]);

  if (!modelsRes.ok) {
    console.error(`Failed to fetch models API: ${modelsRes.status} ${modelsRes.statusText}`);
    if (modelsRes.status === 401 || modelsRes.status === 403)
      console.error("Check your DIGITALOCEAN_API_TOKEN has read access.");
    process.exit(1);
  }

  if (!pricingRes.ok) {
    console.error(`Failed to fetch pricing API: ${pricingRes.status} ${pricingRes.statusText}`);
    process.exit(1);
  }

  const modelsParsed = DoModelsResponse.safeParse(await modelsRes.json());
  if (!modelsParsed.success) {
    console.error("Unexpected models API response:", modelsParsed.error.errors);
    process.exit(1);
  }

  const pricingParsed = StaticContentResponse.safeParse(await pricingRes.json());
  if (!pricingParsed.success) {
    console.error("Unexpected pricing API response:", pricingParsed.error.errors);
    process.exit(1);
  }

  const apiModels = modelsParsed.data.models;
  const pricingMap = buildPricingMap(pricingParsed.data.gradient.models);

  // Collect existing TOML filenames for orphan detection
  const existingFiles = new Set<string>();
  for await (const file of new Bun.Glob("**/*.toml").scan({ cwd: modelsDir, absolute: false })) {