import { expect, test } from "bun:test";
import type { Grid, TimedNote } from "../../src/core/types.ts";
import { repeatVote } from "../../src/pipeline/repeat.ts";

const grid: Grid = { bpm: 120, firstDownbeat: 0, beatsPerBar: 4 }; // 2 s bars
const riff: [number, number][] = [[0, 52], [0.5, 55], [1, 57], [1.5, 59], [1.75, 60]]; // [pos s, pitch]

function bars(n: number, f: (bar: number, pos: number, pitch: number) => boolean = () => true): TimedNote[] {
  const out: TimedNote[] = [];
  for (let b = 0; b < n; b++) for (const [pos, pitch] of riff) {
    if (f(b, pos, pitch)) out.push({ onset: b * 2 + pos, offset: b * 2 + pos + 0.2, pitch });
  }
  return out;
}
const key = (n: TimedNote) => `${n.onset.toFixed(3)}:${n.pitch}`;

test("adds a note missing from one repeat", () => {
  const notes = bars(6, (b, pos) => !(b === 3 && pos === 1));
  const out = repeatVote(notes, grid);
  expect(out.length).toBe(30);
  const added = out.find((n) => Math.abs(n.onset - 7) < 1e-6 && n.pitch === 57);
  expect(added).toBeDefined();
  expect(added!.offset - added!.onset).toBeCloseTo(0.2, 6);
});

test("drops a note that never recurs in a bar's repeats", () => {
  const notes = [...bars(6), { onset: 4.25, offset: 4.4, pitch: 70 }];
  const out = repeatVote(notes, grid);
  expect(out.map(key)).toEqual(bars(6).map(key));
});

test("leaves non-repeating material unchanged", () => {
  const notes: TimedNote[] = [];
  for (let b = 0; b < 6; b++) notes.push({ onset: b * 2 + 0.3 * (b % 4), offset: b * 2 + 1, pitch: 45 + 3 * b });
  expect(repeatVote(notes, grid).map(key)).toEqual(notes.map(key));
});

test("add/drop can be disabled and empty input is handled", () => {
  const notes = [...bars(6, (b, pos) => !(b === 3 && pos === 1)), { onset: 4.25, offset: 4.4, pitch: 70 }];
  expect(repeatVote(notes, grid, { add: false, drop: false }).length).toBe(notes.length);
  expect(repeatVote([], grid)).toEqual([]);
});

test("bars before a late first downbeat are handled", () => {
  const g: Grid = { ...grid, firstDownbeat: 0.5 };
  const notes = bars(6, (b, pos) => !(b === 2 && pos === 0.5)).map((n) => ({ ...n, onset: n.onset + 0.5, offset: n.offset + 0.5 }));
  const out = repeatVote(notes, g);
  expect(out.some((n) => Math.abs(n.onset - 5) < 1e-6 && n.pitch === 55)).toBe(true);
});