import * as R from "remeda";

import { readFile } from "../05/part1.js";

const readInput = (...args) =>
  R.pipe(
    readFile(...args),
    (lines) => lines.split("\n"),
    R.compact, // remove last line
    R.map((line) => line.split(" ").map((v, i) => (i ? parseInt(v) : v)))
  );

export const puzzleInput = readInput(import.meta.url, "input");
export const sample = readInput(import.meta.url, "sample");
export const sample2 = readInput(import.meta.url, "sample.2");

export const comsys = () => {
    let cycle = 1;
    let register = 1;
    const scheduled = [];

    return (instr) => {
        if( instr && (instr[0] === 'addx') ) {
                scheduled.push(0,instr[1]);
        } else {
            scheduled.push(0);
        }
        cycle++;
        register += scheduled.shift() ?? 0;
        return { cycle, register };
    }

}

export const gatherStrength = () => {
    let strength = 0;
    return ({cycle, signal, register}) => {
        if( (cycle - 20) % 40 === 0 ) {
            strength += signal;
        }
        return { strength, cycle, signal, register };
    }
}

export const addSignal = (input) => ({
    ...input,
    signal: input.cycle*input.register
});


export default (instructions) => {
    const sys = comsys();

    return R.pipe(
        R.range(0,220),
        R.map( () => sys(instructions.shift()) ),
        R.map( addSignal ),
        R.map( gatherStrength() ),
        R.last,
        R.prop('strength')
    )

};