From 18f8a4ae1ab79095bf1e2bcbdbf8804f1e0b2654 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Wed, 18 Jul 2018 09:15:58 -0500
Subject: [PATCH] Add latest_only and sort options for subscription feed

---
 src/invidious.cr                    | 35 ++++++++++++++++++++++++++---
 src/invidious/helpers.cr            |  4 ++++
 src/invidious/views/preferences.ecr | 14 ++++++++++++
 3 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/src/invidious.cr b/src/invidious.cr
index 0e135af7..39f8a2a8 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -740,6 +740,13 @@ post "/preferences" do |env|
     max_results = env.params.body["max_results"]?.try &.as(String).to_i
     max_results ||= 40
 
+    sort = env.params.body["sort"]?.try &.as(String)
+    sort ||= "published"
+
+    latest_only = env.params.body["latest_only"]?.try &.as(String)
+    latest_only ||= "off"
+    latest_only = latest_only == "on"
+
     preferences = {
       "video_loop"  => video_loop,
       "autoplay"    => autoplay,
@@ -748,6 +755,8 @@ post "/preferences" do |env|
       "volume"      => volume,
       "dark_mode"   => dark_mode,
       "max_results" => max_results,
+      "sort"        => sort,
+      "latest_only" => latest_only,
     }.to_json
 
     PG_DB.exec("UPDATE users SET preferences = $1 WHERE email = $2", preferences, user.email)
@@ -762,6 +771,7 @@ get "/feed/subscriptions" do |env|
 
   if user
     user = user.as(User)
+    preferences = user.preferences
 
     # Refresh account
     headers = HTTP::Headers.new
@@ -770,7 +780,7 @@ get "/feed/subscriptions" do |env|
     client = make_client(YT_URL)
     user = get_user(user.id, client, headers, PG_DB)
 
-    max_results = user.preferences.max_results
+    max_results = preferences.max_results
     max_results ||= env.params.query["maxResults"]?.try &.to_i
     max_results ||= 40
 
@@ -785,10 +795,29 @@ get "/feed/subscriptions" do |env|
       offset = (page - 1) * max_results
     end
 
-    args = arg_array(user.subscriptions, 3)
-    videos = PG_DB.query_all("SELECT * FROM channel_videos WHERE ucid IN (#{args}) \
+    if preferences.latest_only
+      args = arg_array(user.subscriptions)
+      videos = PG_DB.query_all("SELECT DISTINCT ON (ucid) * FROM channel_videos WHERE \
+      ucid IN (#{args}) ORDER BY ucid, published DESC", user.subscriptions, as: ChannelVideo)
+      videos.sort_by! { |video| video.published }.reverse!
+    else
+      args = arg_array(user.subscriptions, 3)
+      videos = PG_DB.query_all("SELECT * FROM channel_videos WHERE ucid IN (#{args}) \
     ORDER BY published DESC LIMIT $1 OFFSET $2", [limit, offset] + user.subscriptions, as: ChannelVideo)
+    end
 
+    case preferences.sort
+    when "alphabetically"
+      videos.sort_by! { |video| video.title }
+    when "alphabetically - reverse"
+      videos.sort_by! { |video| video.title }.reverse!
+    when "channel name"
+      videos.sort_by! { |video| video.author }
+    when "channel name - reverse"
+      videos.sort_by! { |video| video.author }.reverse!
+    end
+
+    # TODO: Add option to disable picking out notifications from regular feed
     notifications = PG_DB.query_one("SELECT notifications FROM users WHERE email = $1", user.email, as: Array(String))
 
     notifications = videos.select { |v| notifications.includes? v.id }
diff --git a/src/invidious/helpers.cr b/src/invidious/helpers.cr
index 923b6d1c..d9d64bf0 100644
--- a/src/invidious/helpers.cr
+++ b/src/invidious/helpers.cr
@@ -25,6 +25,8 @@ DEFAULT_USER_PREFERENCES = Preferences.from_json({
   "volume"      => 100,
   "dark_mode"   => false,
   "max_results" => 40,
+  "sort"        => "published",
+  "latest_only" => false,
 }.to_json)
 
 class Config
@@ -144,6 +146,8 @@ class Preferences
     volume:      Int32,
     dark_mode:   Bool,
     max_results: Int32,
+    sort:        String,
+    latest_only: Bool,
   })
 end
 
diff --git a/src/invidious/views/preferences.ecr b/src/invidious/views/preferences.ecr
index 10e58a4b..10c016af 100644
--- a/src/invidious/views/preferences.ecr
+++ b/src/invidious/views/preferences.ecr
@@ -58,6 +58,20 @@ function update_value(element) {
             <input name="max_results" id="max_results" type="number" value="<%= user.preferences.max_results %>">
         </div>
 
+        <div class="pure-control-group">
+            <label for="sort">Sort videos by: </label>
+            <select name="sort" id="sort">
+            <% ["published", "alphabetically", "alphabetically - reverse", "channel name", "channel name - reverse"].each do |option| %>
+                <option <% if user.preferences.sort == option %> selected <% end %>><%= option %></option>
+            <% end %>
+            </select>
+        </div>
+
+        <div class="pure-control-group">
+            <label for="latest_only">Only show latest video from channel: </label>
+            <input name="latest_only" id="latest_only" type="checkbox" <% if user.preferences.latest_only %>checked<% end %>>
+        </div>
+
         <div class="pure-controls">
             <button type="submit" class="pure-button pure-button-primary">Save preferences</button>
         </div>