From 1c4b158ac3cf066672dcade32fc0a739a9aed45b Mon Sep 17 00:00:00 2001
From: Bart Massey <bart.massey@gmail.com>
Date: Wed, 29 May 2024 09:47:23 -0700
Subject: [PATCH] improved cookie retrieval handling

---
 src/main.rs | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index d1459fc..9b8463c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -14,16 +14,24 @@ impl CookieProps {
     fn setup_cookie() -> Self {
         let cookie_options = wasm_cookies::CookieOptions::default()
             .expires_after(core::time::Duration::from_secs(52 * 7 * 24 * 60 * 60));
-        let cookie = wasm_cookies::get("test");
-        let cookie = if let Some(cookie) = cookie {
-            cookie.unwrap()
-        } else {
-            log!("setting cookie");
-            wasm_cookies::set("test", "123", &cookie_options);
-            "123".to_string()
-        };
-        log!(&cookie);
-        Self { cookie: Arc::new(cookie) }
+        match wasm_cookies::get("test") {
+            Some(Ok(cookie)) => {
+                log!("got cookie");
+                return Self { cookie: Arc::new(cookie) };
+            }
+            Some(Err(e)) => {
+                log!(format!("cookie error: {}", e));
+            }
+            None => {
+                log!("did not find cookie");
+            }
+        }
+        log!("setting cookie");
+        wasm_cookies::set("test", "123", &cookie_options);
+        let cookie = Arc::new("123".to_string());
+        // XXX Don't do this!! No secrets in logs!
+        // log!(&cookie);
+        Self { cookie }
     }
 }