import { Interval } from "./ast";
import { detectRepeats } from "./detectRepeats";
import { Duration } from "./Duration";
import { ConstantIntensity, RangeIntensity } from "./Intensity";

describe("detectRepeats()", () => {
  it("does nothing with empty array", () => {
    expect(detectRepeats([])).toEqual([]);
  });

  it("does nothing when no interval repeats", () => {
    const intervals: Interval[] = [
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Interval", duration: new Duration(30), intensity: new ConstantIntensity(1.2), comments: [] },
      { type: "Cooldown", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
    ];
    expect(detectRepeats(intervals)).toEqual(intervals);
  });

  it("detects whole workout consisting of repetitions", () => {
    const intervals: Interval[] = [
      { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
    ];
    expect(detectRepeats(intervals)).toEqual([
      {
        type: "repeat",
        times: 4,
        intervals: [
          { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
          { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
        ],
        comments: [],
      },
    ]);
  });

  it("detects repetitions in the middle of workout", () => {
    const intervals: Interval[] = [
      { type: "Warmup", duration: new Duration(60), intensity: new RangeIntensity(0.5, 1), comments: [] },
      { type: "Rest", duration: new Duration(120), intensity: new ConstantIntensity(0.2), comments: [] },
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Rest", duration: new Duration(120), intensity: new ConstantIntensity(0.2), comments: [] },
      { type: "Cooldown", duration: new Duration(60), intensity: new RangeIntensity(1, 0.5), comments: [] },
    ];
    expect(detectRepeats(intervals)).toEqual([
      { type: "Warmup", duration: new Duration(60), intensity: new RangeIntensity(0.5, 1), comments: [] },
      { type: "Rest", duration: new Duration(120), intensity: new ConstantIntensity(0.2), comments: [] },
      {
        type: "repeat",
        times: 4,
        intervals: [
          { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
          { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
        ],
        comments: [],
      },
      { type: "Rest", duration: new Duration(120), intensity: new ConstantIntensity(0.2), comments: [] },
      { type: "Cooldown", duration: new Duration(60), intensity: new RangeIntensity(1, 0.5), comments: [] },
    ]);
  });

  it("detects multiple repetitions", () => {
    const intervals: Interval[] = [
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(100), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(100), intensity: new ConstantIntensity(0.5), comments: [] },
      { type: "Interval", duration: new Duration(100), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(100), intensity: new ConstantIntensity(0.5), comments: [] },
    ];
    expect(detectRepeats(intervals)).toEqual([
      {
        type: "repeat",
        times: 2,
        intervals: [
          { type: "Interval", duration: new Duration(60), intensity: new ConstantIntensity(1), comments: [] },
          { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
        ],
        comments: [],
      },
      {
        type: "repeat",
        times: 2,
        intervals: [
          { type: "Interval", duration: new Duration(100), intensity: new ConstantIntensity(1), comments: [] },
          { type: "Rest", duration: new Duration(100), intensity: new ConstantIntensity(0.5), comments: [] },
        ],
        comments: [],
      },
    ]);
  });

  it("takes cadence differences into account", () => {
    const intervals: Interval[] = [
      { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      {
        type: "Interval",
        duration: new Duration(120),
        intensity: new ConstantIntensity(1),
        cadence: 100,
        comments: [],
      },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), cadence: 80, comments: [] },
      {
        type: "Interval",
        duration: new Duration(120),
        intensity: new ConstantIntensity(1),
        cadence: 100,
        comments: [],
      },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), cadence: 80, comments: [] },
    ];
    expect(detectRepeats(intervals)).toEqual([
      { type: "Interval", duration: new Duration(120), intensity: new ConstantIntensity(1), comments: [] },
      { type: "Rest", duration: new Duration(60), intensity: new ConstantIntensity(0.5), comments: [] },
      {
        type: "repeat",
        times: 2,
        intervals: [
          {
            type: "Interval",
            duration: new Duration(120),
            intensity: new ConstantIntensity(1),
            cadence: 100,
            comments: [],
          },
          {
            type: "Rest",
            duration: new Duration(60),
            intensity: new ConstantIntensity(0.5),
            cadence: 80,
            comments: [],
          },
        ],
        comments: [],
      },
    ]);
  });

  it("does not consider warmup/cooldown-range intervals to be repeatable", () => {
    const intervals: Interval[] = [
      { type: "Warmup", duration: new Duration(60), intensity: new RangeIntensity(0.1, 1), comments: [] },
      { type: "Cooldown", duration: new Duration(120), intensity: new RangeIntensity(1, 0.5), comments: [] },
      { type: "Warmup", duration: new Duration(60), intensity: new RangeIntensity(0.1, 1), comments: [] },
      { type: "Cooldown", duration: new Duration(120), intensity: new RangeIntensity(1, 0.5), comments: [] },
      { type: "Warmup", duration: new Duration(60), intensity: new RangeIntensity(0.1, 1), comments: [] },
      { type: "Cooldown", duration: new Duration(120), intensity: new RangeIntensity(1, 0.5), comments: [] },
      { type: "Warmup", duration: new Duration(60), intensity: new RangeIntensity(0.1, 1), comments: [] },
      { type: "Cooldown", duration: new Duration(120), intensity: new RangeIntensity(1, 0.5), comments: [] },
    ];
    expect(detectRepeats(intervals)).toEqual(intervals);
  });

  it("gathers comments together", () => {
    const intervals: Interval[] = [
      {
        type: "Interval",
        duration: new Duration(100),
        intensity: new ConstantIntensity(1),
        comments: [
          { offset: new Duration(0), text: "Let's start", loc: { row: 1, col: 1 } },
          { offset: new Duration(20), text: "Stay strong!", loc: { row: 2, col: 1 } },
          { offset: new Duration(90), text: "Finish it!", loc: { row: 3, col: 1 } },
        ],
      },
      {
        type: "Rest",
        duration: new Duration(100),
        intensity: new ConstantIntensity(0.5),
        comments: [
          { offset: new Duration(0), text: "Huh... have a rest", loc: { row: 4, col: 1 } },
          { offset: new Duration(80), text: "Ready for next?", loc: { row: 5, col: 1 } },
        ],
      },
      {
        type: "Interval",
        duration: new Duration(100),
        intensity: new ConstantIntensity(1),
        comments: [
          { offset: new Duration(0), text: "Bring it on again!", loc: { row: 6, col: 1 } },
          { offset: new Duration(50), text: "Half way", loc: { row: 7, col: 1 } },
          { offset: new Duration(90), text: "Almost there!", loc: { row: 8, col: 1 } },
        ],
      },
      {
        type: "Rest",
        duration: new Duration(100),
        intensity: new ConstantIntensity(0.5),
        comments: [
          { offset: new Duration(30), text: "Wow... you did it!", loc: { row: 9, col: 1 } },
          { offset: new Duration(40), text: "Nice job.", loc: { row: 10, col: 1 } },
          { offset: new Duration(50), text: "Until next time...", loc: { row: 11, col: 1 } },
        ],
      },
    ];
    expect(detectRepeats(intervals)).toEqual([
      {
        type: "repeat",
        times: 2,
        intervals: [
          { type: "Interval", duration: new Duration(100), intensity: new ConstantIntensity(1), comments: [] },
          { type: "Rest", duration: new Duration(100), intensity: new ConstantIntensity(0.5), comments: [] },
        ],
        comments: [
          { offset: new Duration(0), text: "Let's start", loc: { row: 1, col: 1 } },
          { offset: new Duration(20), text: "Stay strong!", loc: { row: 2, col: 1 } },
          { offset: new Duration(90), text: "Finish it!", loc: { row: 3, col: 1 } },

          { offset: new Duration(100), text: "Huh... have a rest", loc: { row: 4, col: 1 } },
          { offset: new Duration(180), text: "Ready for next?", loc: { row: 5, col: 1 } },

          { offset: new Duration(200), text: "Bring it on again!", loc: { row: 6, col: 1 } },
          { offset: new Duration(250), text: "Half way", loc: { row: 7, col: 1 } },
          { offset: new Duration(290), text: "Almost there!", loc: { row: 8, col: 1 } },

          { offset: new Duration(330), text: "Wow... you did it!", loc: { row: 9, col: 1 } },
          { offset: new Duration(340), text: "Nice job.", loc: { row: 10, col: 1 } },
          { offset: new Duration(350), text: "Until next time...", loc: { row: 11, col: 1 } },
        ],
      },
    ]);
  });
});