2020-12-04 01:14:56 -06:00
|
|
|
import streams, strutils, sets, sequtils, sugar, re, tables
|
|
|
|
|
2020-12-04 10:21:14 -06:00
|
|
|
let validators = toTable({
|
|
|
|
"hcl": (v: string) => v.match re"^#[0-9a-f]{6}$",
|
|
|
|
"ecl": (v: string) => v.match re"^(amb|blu|brn|gry|grn|hzl|oth)$",
|
|
|
|
"pid": (v: string) => v.match re"^\d{9}$",
|
|
|
|
"byr": (v: string) => parseInt(v) in 1920..2002,
|
|
|
|
"iyr": (v: string) => parseInt(v) in 2010..2020,
|
|
|
|
"eyr": (v: string) => parseInt(v) in 2020..2030,
|
|
|
|
"hgt": proc (v: string): bool =
|
|
|
|
let (mn, mx) = if v.endsWith("cm"): (150, 193) else: (59, 76)
|
|
|
|
v.match(re"^\d+(cm|in)$") and v.replace(re"[^0-9]").parseInt in mn..mx,
|
2020-12-04 01:14:56 -06:00
|
|
|
"cid": (v: string) => true,
|
2020-12-04 10:21:14 -06:00
|
|
|
})
|
2020-12-04 01:14:56 -06:00
|
|
|
|
2020-12-04 10:21:14 -06:00
|
|
|
proc parsePassports(s: Stream): seq[Table[string, string]] =
|
|
|
|
s.readAll().split("\n\n").mapIt(it.replace("\n", " ").split(' ')
|
|
|
|
.filterIt(it != "").mapIt(it.split(':', 2)).mapIt((it[0], it[1])).toTable)
|
2020-12-04 01:14:56 -06:00
|
|
|
|
|
|
|
proc hasRequiredFields(p: Table[string, string]): bool =
|
2020-12-04 10:21:14 -06:00
|
|
|
toSeq(validators.keys).toHashSet.difference(["cid"].toHashSet) <= toSeq(p.keys).toHashSet
|
2020-12-03 22:49:54 -06:00
|
|
|
|
|
|
|
proc part1*(s: Stream): int =
|
2020-12-04 10:21:14 -06:00
|
|
|
s.parsePassports.filterIt(it.hasRequiredFields).len
|
2020-12-03 22:49:54 -06:00
|
|
|
|
|
|
|
proc part2*(s: Stream): int =
|
2020-12-04 10:21:14 -06:00
|
|
|
s.parsePassports.filterIt(it.hasRequiredFields and
|
|
|
|
toSeq(it.pairs).mapIt(validators[it[0]](it[1])).foldl(a and b)).len
|