package net.olegg.aoc.year2015.day16

import net.olegg.aoc.someday.SomeDay
import net.olegg.aoc.year2015.DayOf2015

/**
 * See [Year 2015, Day 16](https://adventofcode.com/2015/day/16)
 */
object Day16 : DayOf2015(16) {
  private val PATTERN = "^Sue (\\d+): (.*)$".toRegex()
  private val ANIMAL = "([a-z]+): (\\d+)".toRegex()

  private val SUES = lines
    .mapNotNull { line ->
      PATTERN.matchEntire(line)?.let { match ->
        val index = match.groupValues[1].toInt()
        val animals = match.groupValues[2]
        val own = ANIMAL.findAll(animals).associate {
          val (animal, count) = it.destructured
          animal to count.toInt()
        }
        return@let index to own
      }
    }

  private val FOOTPRINT = mapOf(
    "children" to 3,
    "cats" to 7,
    "samoyeds" to 2,
    "pomeranians" to 3,
    "akitas" to 0,
    "vizslas" to 0,
    "goldfish" to 5,
    "trees" to 3,
    "cars" to 2,
    "perfumes" to 1,
  )

  override fun first(): Any? {
    return SUES
      .first { sue -> sue.second.all { it.value == FOOTPRINT[it.key] } }
      .first
  }

  override fun second(): Any? {
    return SUES
      .first { (_, own) ->
        own.all { (key, value) ->
          val footprintValue = FOOTPRINT[key] ?: 0
          when (key) {
            "cats", "trees" -> value > footprintValue
            "pomeranians", "goldfish" -> value < footprintValue
            else -> value == footprintValue
          }
        }
      }
      .first
  }
}

fun main() = SomeDay.mainify(Day16)
