From 909d540d6471decb0cf6bb88134af10cd63e6eb8 Mon Sep 17 00:00:00 2001
From: Sven Heidemann <sven.heidemann@sh-edraft.de>
Date: Fri, 2 Dec 2022 07:21:20 +0100
Subject: [PATCH] day 2

---
 src/day2.py | 96 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 src/day2.py

diff --git a/src/day2.py b/src/day2.py
new file mode 100644
index 0000000..14049da
--- /dev/null
+++ b/src/day2.py
@@ -0,0 +1,96 @@
+from typing import Optional
+
+from cpl_core.console import Console
+from cpl_core.utils import String
+from cpl_query.enumerable import Enumerable
+from cpl_query.extension import List
+from cpl_core.pipes import *
+
+from aoc.aoc import get_input
+from day1 import aoc_input
+
+# global vars
+day = 2
+aoc_input = get_input(2022, day)
+rps = {'A': 'R', 'B': 'P', 'C': 'S', 'X': 'R', 'Y': 'P', 'Z': 'S'}
+res = {"X": 1, "Y": 2, "Z": 3}
+
+
+def get_score(opponent, me) -> int:
+    if opponent == 'R':
+        if me == 'R':
+            return 3
+        elif me == 'P':
+            return 6
+        elif me == 'S':
+            return 0
+    elif opponent == 'P':
+        if me == 'R':
+            return 0
+        elif me == 'P':
+            return 3
+        elif me == 'S':
+            return 6
+    elif opponent == 'S':
+        if me == 'R':
+            return 6
+        elif me == 'P':
+            return 0
+        elif me == 'S':
+            return 3
+
+
+def get_expected_score(opponent, self):
+    if self == 'Y':
+        to_me = {"A": "X", "B": "Y", "C": "Z"}
+        return to_me[opponent]
+    else:
+        if opponent == 'A':
+            if self == 'X':
+                return 'Z'
+            elif self == 'Z':
+                return 'Y'
+        elif opponent == 'B':
+            if self == 'X':
+                return 'X'
+            elif self == 'Z':
+                return 'Z'
+        elif opponent == 'C':
+            if self == 'X':
+                return 'Y'
+            elif self == 'Z':
+                return 'X'
+
+
+def part1(data) -> int:
+    score = 0
+    for x in data:
+        opponent, me = x.split()
+
+        score += res[me]
+        score += get_score(rps[opponent], rps[me])
+
+    return score
+
+
+def part2(data):
+    score = 0
+    for x in data:
+        opponent, me = x.split()
+
+        e = get_expected_score(opponent, me)
+        score += res[e]
+        score += get_score(rps[opponent], rps[e])
+
+    return score
+
+
+if __name__ == '__main__':
+    """
+        Das ist so fckng dreckige Lösung...
+        Warum bin ich um 6:30 aufgestanden...
+        Ich hab kein bock mehr.
+    """
+    Console.write_line(f'Advent of code day {day}')
+    Console.write_line(part1(aoc_input.splitlines()), part2(aoc_input.splitlines()))
+    Console.write_line()