import { describe, it, expect, beforeEach, afterEach } from "vitest";
import * as fs from "node:fs";
import * as os from "node:os";
import * as path from "node:path";
import { BaselineStore, type Baseline } from "../src/baselineStore";

let dir: string;

beforeEach(() => {
  dir = fs.mkdtempSync(path.join(os.tmpdir(), "baseline-store-"));
});
afterEach(() => {
  fs.rmSync(dir, { recursive: true, force: true });
});

const KEY = "a1b2c3"; // a stand-in for the controller's sha256(uri) key

function sample(uri = "file:///ws/notes/chapter-1.md"): Baseline {
  return { uri, text: "hello\nworld\n", capturedAt: "2026-06-11T00:00:00.000Z", reason: "opened" };
}

describe("BaselineStore", () => {
  it("returns null for a key with no baseline", () => {
    expect(new BaselineStore(dir).load(KEY)).toBeNull();
  });

  it("computes the baseline path as baselines/<key>.json", () => {
    const store = new BaselineStore(dir);
    expect(store.baselinePath(KEY)).toBe(path.join(dir, "baselines", "a1b2c3.json"));
  });

  it("save then load round-trips the baseline (including the uri identity)", () => {
    const store = new BaselineStore(dir);
    const b = sample();
    store.save(KEY, b);
    expect(store.load(KEY)).toEqual(b);
  });

  it("overwrites in place: the newest epoch wins, no history kept", () => {
    const store = new BaselineStore(dir);
    store.save(KEY, { uri: "untitled:Untitled-1", text: "v1", capturedAt: "2026-06-11T00:00:00.000Z", reason: "opened" });
    store.save(KEY, { uri: "untitled:Untitled-1", text: "v2", capturedAt: "2026-06-11T00:01:00.000Z", reason: "pinned" });
    expect(store.load(KEY)).toEqual({ uri: "untitled:Untitled-1", text: "v2", capturedAt: "2026-06-11T00:01:00.000Z", reason: "pinned" });
  });

  it("writes pretty JSON with a trailing newline", () => {
    const store = new BaselineStore(dir);
    const b = sample();
    store.save(KEY, b);
    const raw = fs.readFileSync(store.baselinePath(KEY), "utf8");
    expect(raw).toBe(JSON.stringify(b, null, 2) + "\n");
  });

  it("distinct keys are independent files (no collision)", () => {
    const store = new BaselineStore(dir);
    store.save("k1", sample("file:///a.md"));
    store.save("k2", sample("file:///b.md"));
    expect(store.load("k1")!.uri).toBe("file:///a.md");
    expect(store.load("k2")!.uri).toBe("file:///b.md");
  });
});