fix: Proper display of stars for 2022 Day 25

scarvalhojr · Nov 19, 2023 · 3becde5 · 3becde5
1 parent 8192ad5
commit 3becde5
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/aoc-client/src/lib.rs b/aoc-client/src/lib.rs
@@ -384,6 +384,12 @@ impl AocClient {
         ))
         .unwrap();
 
+        let all_stars = if main.contains("calendar calendar-perfect") {
+            true
+        } else {
+            false
+        };
+
         // Remove stars that have not been collected
         let calendar = cleaned_up
             .lines()
@@ -394,13 +400,14 @@ impl AocClient {
                     .map(|c| c.as_str())
                     .unwrap_or("");
 
-                let stars = if class.contains("calendar-verycomplete") {
-                    "**"
-                } else if class.contains("calendar-complete") {
-                    "*"
-                } else {
-                    ""
-                };
+                let stars =
+                    if class.contains("calendar-verycomplete") || all_stars {
+                        "**"
+                    } else if class.contains("calendar-complete") {
+                        "*"
+                    } else {
+                        ""
+                    };
 
                 star_regex.replace(line, stars)
             })