2 files changed, 15 insertions, 0 deletions
diff --git a/src/Utility/Utility.c b/src/Utility/Utility.c
index 3be4ef4..b4ad91d 100644
--- a/src/Utility/Utility.c
+++ b/src/Utility/Utility.c
@@ -1,4 +1,7 @@
 #include "Utility.h"
+#include <beaker.h>
+#include <stdlib.h>
+#include <string.h>
 
 int hex_to_int(char c) {
   if (c >= '0' && c <= '9')
@@ -9,3 +12,14 @@ int hex_to_int(char c) {
     return c - 'A' + 10;
   return -1;
 }
+
+char *get_theme(const char *default_theme) {
+  char *cookie = get_cookie("theme");
+  if (cookie &&
+      (strcmp(cookie, "light") == 0 ||
+       strcmp(cookie, "dark") == 0)) {
+    return cookie;
+  }
+  free(cookie);
+  return strdup(default_theme);
+}
diff --git a/src/Utility/Utility.h b/src/Utility/Utility.h
index 3b0181c..e67282f 100644
--- a/src/Utility/Utility.h
+++ b/src/Utility/Utility.h
@@ -2,5 +2,6 @@
 #define UTILITY_H
 
 int hex_to_int(char c);
+char *get_theme(const char *default_theme);
 
 #endif