import { promises as fs } from "node:fs";
import * as tc from "@actions/tool-cache";
export async function updateChecksums(
  filePath: string,
  downloadUrls: string[],
): Promise<void> {
  await fs.rm(filePath);
  await fs.appendFile(
    filePath,
    "// AUTOGENERATED_DO_NOT_EDIT\nexport const KNOWN_CHECKSUMS: {[key: string]: string} = {\n",
  );
  let firstLine = true;
  for (const downloadUrl of downloadUrls) {
    const content = await downloadAssetContent(downloadUrl);
    const checksum = content.split(" ")[0].trim();
    const key = getKey(downloadUrl);
    if (!firstLine) {
      await fs.appendFile(filePath, ",\n");
    }
    await fs.appendFile(filePath, `  '${key}':\n    '${checksum}'`);
    firstLine = false;
  }
  await fs.appendFile(filePath, "}\n");
}

function getKey(downloadUrl: string): string {
  // https://github.com/astral-sh/uv/releases/download/0.3.2/uv-aarch64-apple-darwin.tar.gz.sha256
  const parts = downloadUrl.split("/");
  const fileName = parts[parts.length - 1];
  const name = fileName.split(".")[0].split("uv-")[1];
  const version = parts[parts.length - 2];
  return `${name}-${version}`;
}

async function downloadAssetContent(downloadUrl: string): Promise<string> {
  const downloadPath = await tc.downloadTool(downloadUrl);
  const content = await fs.readFile(downloadPath, "utf8");
  return content;
}