From b5c92c1a2fe1ce0ac680db5b9e5bb3eb57d7ad01 Mon Sep 17 00:00:00 2001 From: Omar Roth Date: Sat, 28 Jul 2018 20:40:59 -0500 Subject: [PATCH] Add watch history --- config/sql/users.sql | 1 + src/invidious.cr | 80 ++++++++++++++++++++++++----- src/invidious/helpers.cr | 24 +++++++-- src/invidious/views/preferences.ecr | 14 ++++- 4 files changed, 101 insertions(+), 18 deletions(-) diff --git a/config/sql/users.sql b/config/sql/users.sql index 54f74219..2abcb991 100644 --- a/config/sql/users.sql +++ b/config/sql/users.sql @@ -12,6 +12,7 @@ CREATE TABLE public.users preferences text COLLATE pg_catalog."default", password text COLLATE pg_catalog."default", token text COLLATE pg_catalog."default", + watched text[] COLLATE pg_catalog."default", CONSTRAINT users_email_key UNIQUE (email), CONSTRAINT users_id_key UNIQUE (id) ) diff --git a/src/invidious.cr b/src/invidious.cr index 137ce72a..35169ce1 100644 --- a/src/invidious.cr +++ b/src/invidious.cr @@ -270,20 +270,24 @@ get "/" do |env| end get "/watch" do |env| + if env.params.query["v"]? + id = env.params.query["v"] + else + next env.redirect "/" + end + user = env.get? "user" if user user = user.as(User) + if user.watched != ["N/A"] && !user.watched.includes? id + PG_DB.exec("UPDATE users SET watched = watched || $1 WHERE id = $2", [id], user.id) + end + preferences = user.preferences subscriptions = user.subscriptions.as(Array(String)) end subscriptions ||= [] of String - if env.params.query["v"]? - id = env.params.query["v"] - else - next env.redirect "/" - end - autoplay = env.params.query["autoplay"]?.try &.to_i video_loop = env.params.query["video_loop"]?.try &.to_i @@ -1618,8 +1622,12 @@ post "/login" do |env| sid = Base64.encode(Random::Secure.random_bytes(50)) user = create_user(sid, email, password) + if user.watched = ["N/A"] + user_array = user.to_a[0..-2] + else + user_array = user.to_a + end - user_array = user.to_a user_array[5] = user_array[5].to_json args = arg_array(user_array) @@ -1719,6 +1727,10 @@ post "/preferences" do |env| latest_only ||= "off" latest_only = latest_only == "on" + unseen_only = env.params.body["unseen_only"]?.try &.as(String) + unseen_only ||= "off" + unseen_only = unseen_only == "on" + preferences = { "video_loop" => video_loop, "autoplay" => autoplay, @@ -1732,6 +1744,7 @@ post "/preferences" do |env| "max_results" => max_results, "sort" => sort, "latest_only" => latest_only, + "unseen_only" => unseen_only, }.to_json PG_DB.exec("UPDATE users SET preferences = $1 WHERE email = $2", preferences, user.email) @@ -1773,14 +1786,41 @@ get "/feed/subscriptions" do |env| end 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) + if preferences.unseen_only + ucids = arg_array(user.subscriptions) + if user.watched.empty? + watched = "'{}'" + else + watched = arg_array(user.watched, user.subscriptions.size + 1) + end + + videos = PG_DB.query_all("SELECT DISTINCT ON (ucid) * FROM channel_videos WHERE \ + ucid IN (#{ucids}) AND id NOT IN (#{watched}) ORDER BY ucid, published DESC", + user.subscriptions + user.watched, as: ChannelVideo) + else + 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) + end + 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) + if preferences.unseen_only + ucids = arg_array(user.subscriptions, 3) + if user.watched.empty? + watched = "'{}'" + else + watched = arg_array(user.watched, user.subscriptions.size + 3) + end + + videos = PG_DB.query_all("SELECT * FROM channel_videos WHERE ucid IN (#{ucids}) \ + AND id NOT IN (#{watched}) ORDER BY published DESC LIMIT $1 OFFSET $2", + [limit, offset] + user.subscriptions + user.watched, as: ChannelVideo) + 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 end case preferences.sort @@ -2167,6 +2207,20 @@ get "/subscription_ajax" do |env| env.redirect referer end +get "/clear_watch_history" do |env| + user = env.get? "user" + referer = env.request.headers["referer"]? + referer ||= "/" + + if user + user = user.as(User) + + PG_DB.exec("UPDATE users SET watched = '{}' WHERE id = $1", user.id) + end + + env.redirect referer +end + get "/user/:user" do |env| user = env.params.url["user"] env.redirect "/channel/#{user}" diff --git a/src/invidious/helpers.cr b/src/invidious/helpers.cr index 370e1efd..d978ec29 100644 --- a/src/invidious/helpers.cr +++ b/src/invidious/helpers.cr @@ -29,6 +29,7 @@ DEFAULT_USER_PREFERENCES = Preferences.from_json({ "max_results" => 40, "sort" => "published", "latest_only" => false, + "unseen_only" => false, }.to_json) class Config @@ -147,6 +148,10 @@ class User }, password: String?, token: String, + watched: { + type: Array(String), + default: ["N/A"], + }, }) end @@ -177,6 +182,7 @@ class Preferences max_results: Int32, sort: String, latest_only: Bool, + unseen_only: Bool, }) end @@ -817,7 +823,12 @@ def get_user(sid, client, headers, db, refresh = true) if refresh && Time.now - user.updated > 1.minute user = fetch_user(sid, client, headers, db) - user_array = user.to_a + if user.watched = ["N/A"] + user_array = user.to_a[0..-2] + else + user_array = user.to_a + end + user_array[5] = user_array[5].to_json args = arg_array(user_array) @@ -826,7 +837,12 @@ def get_user(sid, client, headers, db, refresh = true) end else user = fetch_user(sid, client, headers, db) - user_array = user.to_a + if user.watched = ["N/A"] + user_array = user.to_a[0..-2] + else + user_array = user.to_a + end + user_array[5] = user_array[5].to_json args = arg_array(user.to_a) @@ -864,7 +880,7 @@ def fetch_user(sid, client, headers, db) token = Base64.encode(Random::Secure.random_bytes(32)) - user = User.new(sid, Time.now, [] of String, channels, email, DEFAULT_USER_PREFERENCES, nil, token) + user = User.new(sid, Time.now, [] of String, channels, email, DEFAULT_USER_PREFERENCES, nil, token, [] of String) return user end @@ -872,7 +888,7 @@ def create_user(sid, email, password) password = Crypto::Bcrypt::Password.create(password, cost: 10) token = Base64.encode(Random::Secure.random_bytes(32)) - user = User.new(sid, Time.now, [] of String, [] of String, email, DEFAULT_USER_PREFERENCES, password.to_s, token) + user = User.new(sid, Time.now, [] of String, [] of String, email, DEFAULT_USER_PREFERENCES, password.to_s, token, [] of String) return user end diff --git a/src/invidious/views/preferences.ecr b/src/invidious/views/preferences.ecr index 7f675bcc..70f4d73a 100644 --- a/src/invidious/views/preferences.ecr +++ b/src/invidious/views/preferences.ecr @@ -88,10 +88,22 @@ function update_value(element) {
- + checked<% end %>>
+
+ + checked<% end %>> +
+ + Data preferences +
+
+