From cccf28aaf7cd199a5e9d8501b747c91c7f970e8b Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Tue, 7 Aug 2018 11:49:14 -0500
Subject: [PATCH] Add option to toggle light and dark to navbar

---
 src/invidious.cr               | 68 ++++++++++++++++++++++------------
 src/invidious/views/layout.ecr | 16 ++++++--
 2 files changed, 58 insertions(+), 26 deletions(-)

diff --git a/src/invidious.cr b/src/invidious.cr
index 1088e494..c1db9a6f 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -806,6 +806,28 @@ post "/preferences" do |env|
   env.redirect referer
 end
 
+get "/toggle_theme" do |env|
+  user = env.get? "user"
+
+  referer = env.params.query["referer"]?
+  referer ||= "/feed/subscriptions"
+
+  if user
+    user = user.as(User)
+    preferences = user.preferences
+
+    if preferences.dark_mode
+      preferences.dark_mode = false
+    else
+      preferences.dark_mode = true
+    end
+
+    PG_DB.exec("UPDATE users SET preferences = $1 WHERE email = $2", preferences.to_json, user.email)
+  end
+
+  env.redirect referer
+end
+
 # /modify_notifications
 # will "ding" all subscriptions.
 # /modify_notifications?receive_all_updates=false&receive_no_updates=false
@@ -1712,10 +1734,10 @@ get "/api/v1/comments/:id" do |env|
       env.response.content_type = "application/json"
 
       if format == "json"
-      next {"comments" => [] of String}.to_json
+        next {"comments" => [] of String}.to_json
       else
         next {"content_html" => ""}.to_json
-    end
+      end
     end
     ctoken = ctoken["ctoken"]
     itct = body.match(/itct=(?<itct>[^"]+)"/).not_nil!["itct"]
@@ -2664,35 +2686,35 @@ end
 get "/videoplayback/*" do |env|
   path = env.request.path
 
-    path = path.lchop("/videoplayback/")
-    path = path.rchop("/")
+  path = path.lchop("/videoplayback/")
+  path = path.rchop("/")
 
-    path = path.gsub(/mime\/\w+\/\w+/) do |mimetype|
-      mimetype = mimetype.split("/")
-      mimetype[0] + "/" + mimetype[1] + "%2F" + mimetype[2]
+  path = path.gsub(/mime\/\w+\/\w+/) do |mimetype|
+    mimetype = mimetype.split("/")
+    mimetype[0] + "/" + mimetype[1] + "%2F" + mimetype[2]
+  end
+
+  path = path.split("/")
+
+  raw_params = {} of String => Array(String)
+  path.each_slice(2) do |pair|
+    key, value = pair
+    value = URI.unescape(value)
+
+    if raw_params[key]?
+      raw_params[key] << value
+    else
+      raw_params[key] = [value]
     end
+  end
 
-    path = path.split("/")
-
-    raw_params = {} of String => Array(String)
-    path.each_slice(2) do |pair|
-      key, value = pair
-      value = URI.unescape(value)
-
-      if raw_params[key]?
-        raw_params[key] << value
-      else
-        raw_params[key] = [value]
-      end
-    end
-
-    query_params = HTTP::Params.new(raw_params)
+  query_params = HTTP::Params.new(raw_params)
 
   env.redirect "/videoplayback?#{query_params}"
 end
 
 get "/videoplayback" do |env|
-    query_params = env.params.query
+  query_params = env.params.query
 
   fvip = query_params["fvip"]
   mn = query_params["mn"].split(",")[0]
diff --git a/src/invidious/views/layout.ecr b/src/invidious/views/layout.ecr
index d87dc813..3646c6e1 100644
--- a/src/invidious/views/layout.ecr
+++ b/src/invidious/views/layout.ecr
@@ -33,7 +33,17 @@
         </div>
         <div class="pure-u-1 pure-u-md-8-24 user-field">
         <% if env.get? "user" %>
-        <div class="pure-u-1-3">
+        <div class="pure-u-1-4">
+          <a href="/toggle_theme" class="pure-menu-heading">
+            <% preferences = env.get("user").as(User).preferences %>
+            <% if preferences.dark_mode %>
+            <i class="icon ion-ios-sunny"></i>
+            <% else %>
+            <i class="icon ion-ios-moon"></i>
+            <% end %>
+          </a>
+        </div>
+        <div class="pure-u-1-4">
           <a href="/feed/subscriptions" class="pure-menu-heading">
             <% notification_count = env.get("user").as(User).notifications.size %>
             <% if notification_count > 0 %>
@@ -43,12 +53,12 @@
             <% end %>
           </a>
         </div>
-        <div class="pure-u-1-3">
+        <div class="pure-u-1-4">
           <a href="/preferences" class="pure-menu-heading">
             <i class="icon ion-ios-cog"></i>
           </a>
         </div>
-        <div class="pure-u-1-3">
+        <div class="pure-u-1-4">
           <a href="/signout" class="pure-menu-heading">Sign out</a>
         </div>
         <% else %>