import std/[strutils, strscans]

import ../aoc

type
  Color {.pure.} = enum
    red, blue, green

  Cubes = array[Color, int]
  Game = object
    turns: seq[Cubes]

proc parseInput(input: string): seq[Game] =
  for line in input.splitLines:
    result.add Game()
    let turns = line.split(":")[1]
    for turn in turns.split(";"):
      var cubes: Cubes
      for cubeStr in turn.split(","):
        var cnt: int
        var color: string
        discard cubeStr.strip().scanf("$i $w", cnt, color)
        cubes[parseEnum[Color](color)] = cnt
      result[^1].turns.add cubes

proc isValid(game: Game): bool =
  const limits: Cubes = [red: 12, blue: 14, green: 13]
  for turn in game.turns:
    for col, cnt in turn:
      if cnt > limits[col]:
        return false
  return true

proc partOne*(input: string): int =
  var games = parseInput(input)
  for i, game in games:
    if game.isValid():
      result += (i+1)

proc partTwo*(input: string): int =
  var games = parseInput(input)
  for i, game in games:
    var mins: Cubes
    for turn in game.turns:
      for col, cnt in turn:
        mins[col] = max(mins[col], cnt)
    result += mins[red] * mins[green] * mins[blue]

solve:
  example:
    partOne: 8
    partTwo: 2286
  input:
    partOne: 2239
    partTwo: 83435