setup-uv/src/utils/config-file.ts

import fs from "node:fs";
import * as toml from "smol-toml";

export function getConfigValueFromTomlFile(
  filePath: string,
  key: string,
): string | undefined {
  if (!fs.existsSync(filePath) || !filePath.endsWith(".toml")) {
    return undefined;
  }
  const fileContent = fs.readFileSync(filePath, "utf-8");

  if (filePath.endsWith("pyproject.toml")) {
    const tomlContent = toml.parse(fileContent) as {
      tool?: { uv?: Record<string, string | undefined> };
    };
    return tomlContent?.tool?.uv?.[key];
  }
  const tomlContent = toml.parse(fileContent) as Record<
    string,
    string | undefined
  >;
  return tomlContent[key];
}