Merge pull request #9 from omarroth/add-preferences

Add user preferences
This commit is contained in:
Omar Roth 2018-07-16 18:46:16 -05:00 committed by GitHub
commit cae8e1b883
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 242 additions and 95 deletions

View file

@ -21,6 +21,10 @@ body {
color: #f0f0f0; color: #f0f0f0;
} }
.pure-form > fieldset > input { .pure-form > fieldset > input, .pure-control-group > input {
color: #101010;
}
.pure-form > fieldset > select, .pure-control-group > select {
color: #101010; color: #101010;
} }

View file

@ -9,6 +9,7 @@ CREATE TABLE public.users
notifications text[] COLLATE pg_catalog."default", notifications text[] COLLATE pg_catalog."default",
subscriptions text[] COLLATE pg_catalog."default", subscriptions text[] COLLATE pg_catalog."default",
email text COLLATE pg_catalog."default" NOT NULL, email text COLLATE pg_catalog."default" NOT NULL,
preferences text COLLAGE pg_catalog."default",
CONSTRAINT users_email_key UNIQUE (email), CONSTRAINT users_email_key UNIQUE (email),
CONSTRAINT users_id_key UNIQUE (id) CONSTRAINT users_id_key UNIQUE (id)
) )

View file

@ -214,24 +214,19 @@ spawn do
end end
before_all do |env| before_all do |env|
if env.request.cookies.has_key?("SID") if env.request.cookies.has_key? "SID"
env.set "authorized", true headers = HTTP::Headers.new
headers["Cookie"] = env.request.headers["Cookie"]
sid = env.request.cookies["SID"].value sid = env.request.cookies["SID"].value
env.set "sid", sid
subscriptions = PG_DB.query_one?("SELECT subscriptions FROM users WHERE id = $1", sid, as: Array(String)) begin
subscriptions ||= [] of String client = make_client(YT_URL)
env.set "subscriptions", subscriptions user = get_user(sid, client, headers, PG_DB, false)
notifications = PG_DB.query_one?("SELECT cardinality(notifications) FROM users WHERE id = $1", sid, as: Int32) env.set "user", user
notifications ||= 0 rescue ex
env.set "notifications", notifications
end end
if env.request.cookies.has_key?("darktheme") && env.request.cookies["darktheme"].value == "true"
env.set "darktheme", true
end end
end end
@ -240,9 +235,11 @@ get "/" do |env|
end end
get "/watch" do |env| get "/watch" do |env|
authorized = env.get? "authorized" user = env.get? "user"
if authorized if user
subscriptions = env.get("subscriptions").as(Array(String)) user = user.as(User)
preferences = user.preferences
subscriptions = user.subscriptions.as(Array(String))
end end
subscriptions ||= [] of String subscriptions ||= [] of String
@ -679,11 +676,77 @@ get "/signout" do |env|
env.redirect referer env.redirect referer
end end
get "/preferences" do |env|
user = env.get? "user"
referer = env.request.headers["referer"]?
referer ||= "/preferences"
if referer.size > 64
puts "nope"
referer = "/preferences"
end
if user
user = user.as(User)
templated "preferences"
else
env.redirect referer
end
end
post "/preferences" do |env|
user = env.get? "user"
referer = env.params.query["referer"]?
referer ||= "/preferences"
if user
user = user.as(User)
video_loop = env.params.body["video_loop"]?.try &.as(String)
video_loop ||= "off"
video_loop = video_loop == "on"
autoplay = env.params.body["autoplay"]?.try &.as(String)
autoplay ||= "off"
autoplay = autoplay == "on"
speed = env.params.body["speed"]?.try &.as(String).to_f
speed ||= 1.0
quality = env.params.body["quality"]?.try &.as(String)
quality ||= "hd720"
volume = env.params.body["volume"]?.try &.as(String).to_i
volume ||= 100
dark_mode = env.params.body["dark_mode"]?.try &.as(String)
dark_mode ||= "off"
dark_mode = dark_mode == "on"
preferences = {
"video_loop" => video_loop,
"autoplay" => autoplay,
"speed" => speed,
"quality" => quality,
"volume" => volume,
"dark_mode" => dark_mode,
}.to_json
PG_DB.exec("UPDATE users SET preferences = $1 WHERE email = $2", preferences, user.email)
end
env.redirect referer
end
# Get subscriptions for authorized user # Get subscriptions for authorized user
get "/feed/subscriptions" do |env| get "/feed/subscriptions" do |env|
authorized = env.get? "authorized" user = env.get? "user"
if user
user = user.as(User)
if authorized
max_results = env.params.query["maxResults"]?.try &.to_i || 40 max_results = env.params.query["maxResults"]?.try &.to_i || 40
page = env.params.query["page"]?.try &.to_i page = env.params.query["page"]?.try &.to_i
@ -697,14 +760,6 @@ get "/feed/subscriptions" do |env|
offset = (page - 1) * max_results offset = (page - 1) * max_results
end end
headers = HTTP::Headers.new
headers["Cookie"] = env.request.headers["Cookie"]
sid = env.get("sid").as(String)
client = make_client(YT_URL)
user = get_user(sid, client, headers, PG_DB)
args = arg_array(user.subscriptions, 3) args = arg_array(user.subscriptions, 3)
videos = PG_DB.query_all("SELECT * FROM channel_videos WHERE ucid IN (#{args}) \ 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) ORDER BY published DESC LIMIT $1 OFFSET $2", [limit, offset] + user.subscriptions, as: ChannelVideo)
@ -718,7 +773,7 @@ get "/feed/subscriptions" do |env|
videos = videos[0..max_results] videos = videos[0..max_results]
end end
PG_DB.exec("UPDATE users SET notifications = $1 WHERE id = $2", [] of String, sid) PG_DB.exec("UPDATE users SET notifications = $1 WHERE id = $2", [] of String, user.id)
env.set "notifications", 0 env.set "notifications", 0
templated "subscriptions" templated "subscriptions"
@ -735,12 +790,14 @@ end
# /modify_notifications?receive_all_updates=false&receive_no_updates=false # /modify_notifications?receive_all_updates=false&receive_no_updates=false
# will "unding" all subscriptions. # will "unding" all subscriptions.
get "/modify_notifications" do |env| get "/modify_notifications" do |env|
authorized = env.get? "authorized" user = env.get? "user"
referer = env.request.headers["referer"]? referer = env.request.headers["referer"]?
referer ||= "/" referer ||= "/"
if authorized if user
user = user.as(User)
channel_req = {} of String => String channel_req = {} of String => String
channel_req["receive_all_updates"] = env.params.query["receive_all_updates"]? || "true" channel_req["receive_all_updates"] = env.params.query["receive_all_updates"]? || "true"
@ -779,12 +836,14 @@ get "/modify_notifications" do |env|
end end
get "/subscription_manager" do |env| get "/subscription_manager" do |env|
authorized = env.get? "authorized" user = env.get? "user"
if !authorized
if !user
next env.redirect "/" next env.redirect "/"
end end
subscriptions = env.get?("subscriptions").as(Array(String)) user = user.as(User)
subscriptions = user.subscriptions
subscriptions ||= [] of String subscriptions ||= [] of String
client = make_client(YT_URL) client = make_client(YT_URL)
@ -797,11 +856,13 @@ get "/subscription_manager" do |env|
end end
get "/subscription_ajax" do |env| get "/subscription_ajax" do |env|
authorized = env.get? "authorized" user = env.get? "user"
referer = env.request.headers["referer"]? referer = env.request.headers["referer"]?
referer ||= "/" referer ||= "/"
if authorized if user
user = user.as(User)
if env.params.query["action_create_subscription_to_channel"]? if env.params.query["action_create_subscription_to_channel"]?
action = "action_create_subscription_to_channel" action = "action_create_subscription_to_channel"
elsif env.params.query["action_remove_subscriptions"]? elsif env.params.query["action_remove_subscriptions"]?
@ -836,7 +897,7 @@ get "/subscription_ajax" do |env|
# Update user # Update user
if client.post(post_url, headers, post_req).status_code == 200 if client.post(post_url, headers, post_req).status_code == 200
sid = env.get("sid").as(String) sid = user.id
case action case action
when .starts_with? "action_create" when .starts_with? "action_create"
@ -856,11 +917,10 @@ get "/user/:user" do |env|
end end
get "/channel/:ucid" do |env| get "/channel/:ucid" do |env|
authorized = env.get? "authorized" user = env.get? "user"
if authorized if user
sid = env.get("sid").as(String) user = user.as(User)
subscriptions = user.subscriptions
subscriptions = PG_DB.query_one?("SELECT subscriptions FROM users WHERE id = $1", sid, as: Array(String))
end end
subscriptions ||= [] of String subscriptions ||= [] of String
@ -898,20 +958,6 @@ get "/channel/:ucid" do |env|
templated "channel" templated "channel"
end end
get "/modify_theme" do |env|
referer = env.request.headers["referer"]?
referer ||= "/"
if env.params.query["dark"]?
env.response.cookies["darktheme"] = "true"
elsif env.params.query["light"]?
env.request.cookies["darktheme"].expires = Time.new(1990, 1, 1)
env.request.cookies.add_response_headers(env.response.headers)
end
env.redirect referer
end
get "/redirect" do |env| get "/redirect" do |env|
if env.params.query["q"]? if env.params.query["q"]?
env.redirect env.params.query["q"] env.redirect env.params.query["q"]
@ -1167,6 +1213,6 @@ end
public_folder "assets" public_folder "assets"
add_handler FilteredCompressHandler.new add_handler FilteredCompressHandler.new
add_context_storage_type(Array(String)) add_context_storage_type(User)
Kemal.run Kemal.run

View file

@ -17,6 +17,15 @@ macro rendered(filename)
render "src/invidious/views/#{{{filename}}}.ecr" render "src/invidious/views/#{{{filename}}}.ecr"
end end
DEFAULT_USER_PREFERENCES = Preferences.from_json({
"video_loop" => false,
"autoplay" => false,
"speed" => 1.0,
"quality" => "hd720",
"volume" => 100,
"dark_mode" => false,
}.to_json)
class Config class Config
YAML.mapping({ YAML.mapping({
crawl_threads: Int32, crawl_threads: Int32,
@ -86,12 +95,6 @@ class Video
end end
class InvidiousChannel class InvidiousChannel
module XMLConverter
def self.from_rs(rs)
XML.parse_html(rs.read(String))
end
end
add_mapping({ add_mapping({
id: String, id: String,
author: String, author: String,
@ -111,12 +114,34 @@ class ChannelVideo
end end
class User class User
module PreferencesConverter
def self.from_rs(rs)
Preferences.from_json(rs.read(String))
end
end
add_mapping({ add_mapping({
id: String, id: String,
updated: Time, updated: Time,
notifications: Array(String), notifications: Array(String),
subscriptions: Array(String), subscriptions: Array(String),
email: String, email: String,
preferences: {
type: Preferences,
default: DEFAULT_USER_PREFERENCES,
converter: PreferencesConverter,
},
})
end
class Preferences
JSON.mapping({
video_loop: Bool,
autoplay: Bool,
speed: Float32,
quality: String,
volume: Int32,
dark_mode: Bool,
}) })
end end
@ -674,11 +699,11 @@ def fetch_channel(ucid, client, db, pull_all_videos = true)
return channel return channel
end end
def get_user(sid, client, headers, db) def get_user(sid, client, headers, db, refresh = true)
if db.query_one?("SELECT EXISTS (SELECT true FROM users WHERE id = $1)", sid, as: Bool) if db.query_one?("SELECT EXISTS (SELECT true FROM users WHERE id = $1)", sid, as: Bool)
user = db.query_one("SELECT * FROM users WHERE id = $1", sid, as: User) user = db.query_one("SELECT * FROM users WHERE id = $1", sid, as: User)
if Time.now - user.updated > 1.minute if refresh && Time.now - user.updated > 1.minute
user = fetch_user(sid, client, headers, db) user = fetch_user(sid, client, headers, db)
user_array = user.to_a user_array = user.to_a
args = arg_array(user_array) args = arg_array(user_array)
@ -723,7 +748,7 @@ def fetch_user(sid, client, headers, db)
email = "" email = ""
end end
user = User.new(sid, Time.now, [] of String, channels, email) user = User.new(sid, Time.now, [] of String, channels, email, DEFAULT_USER_PREFERENCES)
return user return user
end end

View file

@ -3,7 +3,7 @@
<% end %> <% end %>
<h1><%= author %></h1> <h1><%= author %></h1>
<% if authorized %> <% if user %>
<% if subscriptions.includes? ucid %> <% if subscriptions.includes? ucid %>
<p> <p>
<a href="/subscription_ajax?action_remove_subscriptions=1&c=<%= ucid %>"> <a href="/subscription_ajax?action_remove_subscriptions=1&c=<%= ucid %>">

View file

@ -8,7 +8,7 @@
<link rel="stylesheet" href="https://unpkg.com/purecss@1.0.0/build/grids-responsive-min.css"> <link rel="stylesheet" href="https://unpkg.com/purecss@1.0.0/build/grids-responsive-min.css">
<link rel="stylesheet" href="https://use.fontawesome.com/releases/v5.0.6/css/all.css"> <link rel="stylesheet" href="https://use.fontawesome.com/releases/v5.0.6/css/all.css">
<link rel="stylesheet" href="/css/default.css"> <link rel="stylesheet" href="/css/default.css">
<% if env.get? "darktheme" %> <% if env.get?("user") && env.get("user").as(User).preferences.dark_mode %>
<link rel="stylesheet" href="/css/darktheme.css"> <link rel="stylesheet" href="/css/darktheme.css">
<% else %> <% else %>
<link rel="stylesheet" href="/css/lighttheme.css"> <link rel="stylesheet" href="/css/lighttheme.css">
@ -21,22 +21,22 @@
<div class="pure-u-1 pure-u-md-1-5"></div> <div class="pure-u-1 pure-u-md-1-5"></div>
<div class="pure-u-1 pure-u-md-3-5"> <div class="pure-u-1 pure-u-md-3-5">
<div class="pure-g" style="padding:1em;"> <div class="pure-g" style="padding:1em;">
<div class="pure-u-1 pure-u-md-1-5"> <div class="pure-u-1 pure-u-md-4-24">
<center><a href="/" class="pure-menu-heading">Invidious</a></center> <center><a href="/" class="pure-menu-heading">Invidious</a></center>
</div> </div>
<div class="pure-u-1 pure-u-md-3-5"> <div class="pure-u-1 pure-u-md-12-24">
<form class="pure-form" action="/search" method="get"> <form class="pure-form" action="/search" method="get">
<fieldset> <fieldset>
<input type="search" style="width:100%;" name="q" placeholder="search" value="<%= env.params.query["q"]? %>"> <input type="search" style="width:100%;" name="q" placeholder="search" value="<%= env.params.query["q"]? %>">
</fieldset> </fieldset>
</form> </form>
</div> </div>
<div class="pure-u-1 pure-u-md-1-5"> <div class="pure-u-1 pure-u-md-8-24">
<% if env.get? "authorized" %> <% if env.get? "user" %>
<div class="pure-g"> <div class="pure-g">
<div class="pure-u-1-3"> <div class="pure-u-1-3">
<a href="/feed/subscriptions" class="pure-menu-heading"> <a href="/feed/subscriptions" class="pure-menu-heading">
<% notifications = env.get("notifications").as(Int32) %> <% notifications = env.get("user").as(User).notifications.size %>
<% if notifications > 0 %> <% if notifications > 0 %>
<center><%= notifications %> <i class="fas fa-bell"></i></center> <center><%= notifications %> <i class="fas fa-bell"></i></center>
<% else %> <% else %>
@ -44,8 +44,15 @@
<% end %> <% end %>
</a> </a>
</div> </div>
<div class="pure-u-2-3"> <div class="pure-u-1-3">
<center><a href="/signout" class="pure-menu-heading">Sign out</a></center> <a href="/preferences" class="pure-menu-heading">
<center><i class="fas fa-cog"></i></center>
</a>
</div>
<div class="pure-u-1-3">
<a href="/signout" class="pure-menu-heading">
<center>Sign out</center>
</a>
</div> </div>
</div> </div>
<% else %> <% else %>
@ -54,13 +61,6 @@
</div> </div>
</div> </div>
<%= content %> <%= content %>
<center class="h-box">
<% if env.get? "darktheme" %>
<a href="/modify_theme?light">Light theme</a>
<% else %>
<a href="/modify_theme?dark">Dark theme</a>
<% end %>
</center>
<center class="h-box"> <center class="h-box">
Released under AGPLv3 by <a href="https://github.com/omarroth">Omar Roth</a> - Released under AGPLv3 by <a href="https://github.com/omarroth">Omar Roth</a> -
source available <a href="https://github.com/omarroth/invidious">here</a> source available <a href="https://github.com/omarroth/invidious">here</a>

View file

@ -0,0 +1,59 @@
<% content_for "header" do %>
<title>Preferences - Invidious</title>
<% end %>
<script>
function update_value(element) {
document.getElementById('volume-value').innerText = element.value;
}
</script>
<form class="pure-form pure-form-aligned" action="/preferences?referer=<%= referer %>" method="post">
<fieldset>
<legend>Player preferences</legend>
<div class="pure-control-group">
<label for="video_loop">Always loop: </label>
<input name="video_loop" id="video_loop" type="checkbox" <% if user.preferences.video_loop %>checked<% end %>>
</div>
<div class="pure-control-group">
<label for="autoplay">Autoplay: </label>
<input name="autoplay" id="autoplay" type="checkbox" <% if user.preferences.autoplay %>checked<% end %>>
</div>
<div class="pure-control-group">
<label for="speed">Default speed: </label>
<select name="speed" id="speed">
<% [2.0, 1.5, 1.0, 0.5].each do |option| %>
<option <% if user.preferences.speed == option %> selected <% end %>><%= option %></option>
<% end %>
</select>
</div>
<div class="pure-control-group">
<label for="quality">Preferred video quality: </label>
<select name="quality" id="quality">
<% ["hd720", "medium", "small"].each do |option| %>
<option <% if user.preferences.quality == option %> selected <% end %>><%= option %></option>
<% end %>
</select>
</div>
<div class="pure-control-group">
<label for="volume">Player volume: </label>
<input name="volume" id="volume" oninput="update_value(this);" type="range" min="0" max="100" step="5" value="<%= user.preferences.volume %>">
<span class="pure-form-message-inline" id="volume-value"><%= user.preferences.volume %></span>
</div>
<legend>Visual preferences</legend>
<div class="pure-control-group">
<label for="dark_mode">Dark mode: </label>
<input name="dark_mode" id="dark_mode" type="checkbox" <% if user.preferences.dark_mode %>checked<% end %>>
</div>
<div class="pure-controls">
<button type="submit" class="pure-button pure-button-primary">Save preferences</button>
</div>
</fieldset>
</form>

View file

@ -17,15 +17,23 @@
<% end %> <% end %>
<% else %> <% else %>
<% fmt_stream.each_with_index do |fmt, i| %> <% fmt_stream.each_with_index do |fmt, i| %>
<% if preferences %>
<source src="<%= fmt["url"] %>" type='<%= fmt["type"] %>' label="<%= fmt["label"] %>" selected="<%= preferences.quality == fmt["label"].split(" - ")[0] %>">
<% else %>
<source src="<%= fmt["url"] %>" type='<%= fmt["type"] %>' label="<%= fmt["label"] %>" selected="<%= i == 0 ? true : false %>"> <source src="<%= fmt["url"] %>" type='<%= fmt["type"] %>' label="<%= fmt["label"] %>" selected="<%= i == 0 ? true : false %>">
<% end %> <% end %>
<% end %> <% end %>
<% end %>
</video> </video>
</div> </div>
<script> <script>
var options = { var options = {
preload: "auto", preload: "auto",
<% if preferences %>
<% if preferences.autoplay %>autoplay: true, <% end %>
<% if preferences.video_loop %>loop: true, <% end %>
<% end %>
playbackRates: [0.5, 1, 1.5, 2], playbackRates: [0.5, 1, 1.5, 2],
controlBar: { controlBar: {
children: [ children: [
@ -81,6 +89,10 @@ var player = videojs('player', options, function() {
}); });
}); });
<% if preferences %>
player.volume(<%= preferences.volume.to_f / 100 %>);
player.playbackRate(<%= preferences.speed %>);
<% end %>
player.offset({ player.offset({
start: <%= video_start %>, start: <%= video_start %>,
end: <%= video_end %> end: <%= video_end %>
@ -165,7 +177,7 @@ player.src(currentSources);
<h3><%= video.author %></h3> <h3><%= video.author %></h3>
</a> </a>
</p> </p>
<% if authorized %> <% if user %>
<% if subscriptions.includes? video.ucid %> <% if subscriptions.includes? video.ucid %>
<p> <p>
<a href="/subscription_ajax?action_remove_subscriptions=1&c=<%= video.ucid %>"> <a href="/subscription_ajax?action_remove_subscriptions=1&c=<%= video.ucid %>">