Add cube conundrum tests and implementation

vicainelli · vicainelli · commit bc60cb0336c4 · 2023-12-07T11:40:42.000+01:00
diff --git a/src/day-02/cube-conundrum.test.ts b/src/day-02/cube-conundrum.test.ts
@@ -0,0 +1,31 @@
+import { describe, expect, test } from "bun:test";
+import { cubeConundrumPartOne, cubeConundrumPartTwo } from "./cube-conundrum";
+
+const example = Bun.file("src/day-02/example.txt");
+const input = Bun.file("src/day-02/input.txt");
+
+const content_input = await input.text();
+const content_example = await example.text();
+
+type maxCountType = {
+  [key: string]: number;
+};
+
+const maxCount: maxCountType = {
+  red: 12,
+  green: 13,
+  blue: 14,
+};
+
+describe("Day 02, cube conundrum", () => {
+  describe("Part One", () => {
+    test("it should return 8, example", () => {
+      const result = cubeConundrumPartOne(content_example);
+      expect(result).toEqual(8);
+    });
+    test("it should return 2716, input", () => {
+      const result = cubeConundrumPartOne(content_input);
+      expect(result).toEqual(2716);
+    });
+  });
+});
diff --git a/src/day-02/cube-conundrum.ts b/src/day-02/cube-conundrum.ts
@@ -0,0 +1,41 @@
+type maxCountType = {
+  [key: string]: number;
+};
+
+const maxCount: maxCountType = {
+  red: 12,
+  green: 13,
+  blue: 14,
+};
+
+export function cubeConundrumPartOne(content: string): number {
+  // Split the content into lines
+  const lines = content.trim().split("\n");
+
+  // Process each line
+  return (
+    lines
+      .map((line) => {
+        // Split the line into sets
+        const sets = line.split(": ")[1].split("; ");
+
+        // Check if all pulls in a set are valid
+        const areAllPullsValid = sets.map((set) => {
+          const pulls = set.split(", ");
+
+          // Check if a pull is valid
+          return pulls.every((pull) => {
+            const [count, color] = pull.split(" ");
+            return maxCount[color] >= Number(count);
+          });
+        });
+
+        // Check if all sets are valid
+        return areAllPullsValid.every((p) => p);
+      })
+      // Sum up the results
+      .reduce((sum, result, index) => {
+        return result ? sum + (index + 1) : sum;
+      }, 0)
+  );
+}
