From 82d454da8ec6543ef41ad5c088a37e236b6e239f Mon Sep 17 00:00:00 2001
From: Rick Hanlon <rickhanlonii@fb.com>
Date: Wed, 1 May 2024 10:47:44 -0400
Subject: [PATCH] add uww toggle

---
 src/components/Layout/Footer.tsx | 16 +++++++
 src/pages/_document.tsx          | 72 +++++++++++++++++++++-----------
 2 files changed, 63 insertions(+), 25 deletions(-)

diff --git a/src/components/Layout/Footer.tsx b/src/components/Layout/Footer.tsx
index 274f73228cd..5bcf9df984c 100644
--- a/src/components/Layout/Footer.tsx
+++ b/src/components/Layout/Footer.tsx
@@ -285,6 +285,22 @@ export function Footer() {
             dir="ltr">
             &copy;{new Date().getFullYear()}
           </div>
+          <div
+            className="uwu-visible text-xs cursor-pointer hover:text-link hover:dark:text-link-dark hover:underline"
+            onClick={() => {
+              // @ts-ignore
+              window.__setUwu(false);
+            }}>
+            no uwu plz
+          </div>
+          <div
+            className="uwu-hidden text-xs cursor-pointer hover:text-link hover:dark:text-link-dark hover:underline"
+            onClick={() => {
+              // @ts-ignore
+              window.__setUwu(true);
+            }}>
+            uwu?
+          </div>
           <div className="uwu-visible text-xs">
             Logo by
             <ExternalLink
diff --git a/src/pages/_document.tsx b/src/pages/_document.tsx
index c6512be8a8c..21e6e34aa48 100644
--- a/src/pages/_document.tsx
+++ b/src/pages/_document.tsx
@@ -22,37 +22,59 @@ const MyDocument = () => {
         <script
           dangerouslySetInnerHTML={{
             __html: `
-                (function () {
-                  try {
-                    var preferredUwu;
+              (function () {
+                try {
+                  let logShown = false;
+                  function setUwu(isUwu) {
                     try {
-                      preferredUwu = localStorage.getItem('uwu');
-                    } catch (err) { }
-  
-                    const isUwuValue = window.location 
-                      && window.location.search
-                      && window.location.search.match(/uwu=(true|false)/);
-                      
-                    if (isUwuValue) {
-                      const isUwu = isUwuValue[1] === 'true';
                       if (isUwu) {
-                        try {
-                          localStorage.setItem('uwu', true);
-                        } catch (err) { }
+                        localStorage.setItem('uwu', true);
                         document.documentElement.classList.add('uwu');
-                        console.log('uwu mode enabled. turn off with ?uwu=false')
-                        console.log('logo credit to @sawaratsuki1004 via https://github.com/SAWARATSUKI/ServiceLogos');
+                        if (!logShown) {
+                          console.log('uwu mode! turn off with ?uwu=0');
+                          console.log('logo credit to @sawaratsuki1004 via https://github.com/SAWARATSUKI/ServiceLogos');
+                          logShown = true;
+                        }
                       } else {
-                        try {
-                          localStorage.removeItem('uwu', false);
-                        } catch (err) { }
+                        localStorage.removeItem('uwu');
+                        document.documentElement.classList.remove('uwu');
+                        console.log('uwu mode off. turn on with ?uwu');
                       }
-                    } else if (preferredUwu) {
-                      document.documentElement.classList.add('uwu');
+                    } catch (err) { }
+                  }
+                  window.__setUwu = setUwu;
+                  function checkQueryParam() {
+                    const params = new URLSearchParams(window.location.search);
+                    const value = params.get('uwu');
+                    switch(value) {
+                      case '':
+                      case 'true':
+                      case '1':
+                        return true;
+                      case 'false':
+                      case '0':
+                        return false;
+                      default:
+                        return null;
                     }
-                  } catch (err) { }
-                })();
-              `,
+                  }
+                  function checkLocalStorage() {
+                    try {
+                      return localStorage.getItem('uwu') === 'true';
+                    } catch (err) {
+                      return false;
+                    }
+                  }
+                  const uwuQueryParam = checkQueryParam();
+                  console.log('uwuQueryParam', uwuQueryParam);
+                  if (uwuQueryParam != null) {
+                    setUwu(uwuQueryParam);
+                  } else if (checkLocalStorage()) {
+                    document.documentElement.classList.add('uwu');
+                  }
+                } catch (err) { }
+              })();
+            `,
           }}
         />
         <script