import * as R from "remeda";
import { test, expect } from "vitest";
import fs from "fs-extra";
import path from 'path';
import { expectSolution } from './main.js';

const split = (splitter) => (text) => text.split(splitter);
const sum = R.sumBy(R.identity);

const input = R.pipe(
  fs.readFileSync( path.join( path.dirname(import.meta.url), "input").replace('file:',''), "utf8"),
  split("\n\n"),
  R.map((x) =>
    split("\n")(x)
      .map((x) => parseInt(x))
      .filter(R.isNumber)
  ),
  R.map(sum)
);


test("part 1", () => {
  const maxCalories = R.pipe(input, (calories) => Math.max(...calories));

  expectSolution(maxCalories).toEqual(71780);
});

test("part 2", () => {
  input.sort((a, b) => (a > b ? -1 : 1));

  expectSolution(R.pipe(input, R.take(3), R.sumBy(R.identity))).toEqual(212489);
});