import * as fs from "node:fs";
import * as path from "node:path";
import type { RunArtifact } from "./types";

export function artifactFileName(model: string, suite: "baseline" | "coding-lite" | "test-all" = "baseline", thinking: "default" | "max" = "default", date = new Date(), tag?: string): string {
  const safeModel = model.replace(/[^a-zA-Z0-9._-]+/g, "-").replace(/^-+|-+$/g, "") || "model";
  const stamp = date.toISOString().replace(/:/g, "-").replace(/\.\d{3}/, "");
  const suiteName = suite === "baseline" ? "3ptest" : suite;
  return `simplebench${tag ? `-${tag}-` : "--"}${suiteName}-${safeModel}-${thinking}-${stamp}.json`;
}

export function writeArtifactBundle(artifact: RunArtifact, files: Record<string, string>): string {
  const name = artifactFileName(artifact.benchmark.model, artifact.benchmark.suite || "baseline", artifact.benchmark.thinking.requested, new Date(), artifact.benchmark.tag).replace(/\.json$/, "");
  for (let suffix = 0; ; suffix += 1) {
    const output = path.resolve(process.cwd(), suffix ? `${name}-${suffix}` : name);
    try {
      fs.mkdirSync(output);
      fs.writeFileSync(path.join(output, "result.json"), JSON.stringify(artifact, null, 2) + "\n", "utf8");
      for (const [file, content] of Object.entries(files)) fs.writeFileSync(path.join(output, path.basename(file)), content, "utf8");
      return output;
    } catch (error: any) {
      if (error?.code !== "EEXIST") throw error;
    }
  }
}

export function writeArtifact(artifact: RunArtifact): string {
  const name = artifactFileName(artifact.benchmark.model, artifact.benchmark.suite || "baseline", artifact.benchmark.thinking.requested, new Date(), artifact.benchmark.tag);
  const extension = path.extname(name);
  const base = name.slice(0, -extension.length);
  const contents = JSON.stringify(artifact, null, 2) + "\n";
  for (let suffix = 0; ; suffix += 1) {
    const output = path.resolve(process.cwd(), suffix ? `${base}-${suffix}${extension}` : name);
    try {
      fs.writeFileSync(output, contents, { encoding: "utf8", flag: "wx" });
      return output;
    } catch (error: any) {
      if (error?.code !== "EEXIST") throw error;
    }
  }
}