From e358607a332411cf00119c98d1ccb1258e173e35 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ga=C3=ABl=20Depreeuw?= <gael@depreeuw.dev>
Date: Sun, 6 Dec 2020 18:13:51 +0100
Subject: [PATCH] Solve day 2, part 2

---
 src/Day2.hs | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/src/Day2.hs b/src/Day2.hs
index 4d3433f..5826ded 100644
--- a/src/Day2.hs
+++ b/src/Day2.hs
@@ -4,10 +4,12 @@ module Day2 (day2) where
 
 import qualified Data.Text as T
 import qualified Data.Attoparsec.Text as P
+import Data.List (elemIndices)
 
 type Max = Int
 type Min = Int
 type Letter = Char
+type Validator = Policy -> String -> Bool
 
 data Policy = Policy { lBound :: Min
                      , uBound :: Max
@@ -24,20 +26,34 @@ parserPolicy = do
     _ <- P.string ": "
     return $ Policy lB uB le
 
-validatePolicy :: Policy -> String -> Bool
+validatePolicy :: Validator
 validatePolicy p s =
     let l = length $ filter (== letter p) s in
     l >= lBound p && l <= uBound p
 
-validateLine :: String -> Bool
-validateLine s =
+validateLine :: Validator -> String -> Bool
+validateLine v s =
     case P.parse parserPolicy (T.pack s) of
         P.Fail {} -> error "day2 - parser error"
         P.Partial _ -> error "day2 - input error"
-        P.Done i r -> validatePolicy r (T.unpack i)
+        P.Done i r -> v r (T.unpack i)
+
+xor :: Bool -> Bool -> Bool
+xor True = not
+xor False = id
+
+validateCorrectPolicy :: Validator
+validateCorrectPolicy p s =
+    let index1 = lBound p - 1
+        index2 = uBound p - 1
+        indices = elemIndices (letter p) s in
+        xor (index1 `elem` indices) (index2 `elem` indices)
+
 
 day2 :: IO ()
 day2 = do
-  r <- readFile "./input/day2"
+  rs <- lines <$> readFile "./input/day2"
   putStr "[Day 2-1] # valid passwords: "
-  print . length . filter (==True) . fmap validateLine . lines $ r
\ No newline at end of file
+  print . length . filter (==True) . fmap (validateLine validatePolicy) $ rs
+  putStr "[Day 2-2] # valid passwords: "
+  print . length . filter (==True) . fmap (validateLine validateCorrectPolicy) $ rs