Fix warnings with latest version of Crystal
This commit is contained in:
parent
92f337c67e
commit
452d1e8307
@ -1203,7 +1203,7 @@ post "/playlist_ajax" do |env|
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
playlist_video = PlaylistVideo.new(
|
playlist_video = PlaylistVideo.new({
|
||||||
title: video.title,
|
title: video.title,
|
||||||
id: video.id,
|
id: video.id,
|
||||||
author: video.author,
|
author: video.author,
|
||||||
@ -1212,8 +1212,8 @@ post "/playlist_ajax" do |env|
|
|||||||
published: video.published,
|
published: video.published,
|
||||||
plid: playlist_id,
|
plid: playlist_id,
|
||||||
live_now: video.live_now,
|
live_now: video.live_now,
|
||||||
index: Random::Secure.rand(0_i64..Int64::MAX)
|
index: Random::Secure.rand(0_i64..Int64::MAX),
|
||||||
)
|
})
|
||||||
|
|
||||||
video_array = playlist_video.to_a
|
video_array = playlist_video.to_a
|
||||||
args = arg_array(video_array)
|
args = arg_array(video_array)
|
||||||
@ -1839,8 +1839,8 @@ post "/login" do |env|
|
|||||||
sid = Base64.urlsafe_encode(Random::Secure.random_bytes(32))
|
sid = Base64.urlsafe_encode(Random::Secure.random_bytes(32))
|
||||||
user, sid = create_user(sid, email, password)
|
user, sid = create_user(sid, email, password)
|
||||||
user_array = user.to_a
|
user_array = user.to_a
|
||||||
|
user_array[4] = user_array[4].to_json # User preferences
|
||||||
|
|
||||||
user_array[4] = user_array[4].to_json
|
|
||||||
args = arg_array(user_array)
|
args = arg_array(user_array)
|
||||||
|
|
||||||
PG_DB.exec("INSERT INTO users VALUES (#{args})", args: user_array)
|
PG_DB.exec("INSERT INTO users VALUES (#{args})", args: user_array)
|
||||||
@ -2519,7 +2519,7 @@ post "/data_control" do |env|
|
|||||||
if user
|
if user
|
||||||
user = user.as(User)
|
user = user.as(User)
|
||||||
|
|
||||||
# TODO: Find better way to prevent timeout
|
# TODO: Find a way to prevent browser timeout
|
||||||
|
|
||||||
HTTP::FormData.parse(env.request) do |part|
|
HTTP::FormData.parse(env.request) do |part|
|
||||||
body = part.body.gets_to_end
|
body = part.body.gets_to_end
|
||||||
@ -2546,7 +2546,7 @@ post "/data_control" do |env|
|
|||||||
end
|
end
|
||||||
|
|
||||||
if body["preferences"]?
|
if body["preferences"]?
|
||||||
user.preferences = Preferences.from_json(body["preferences"].to_json, user.preferences)
|
user.preferences = Preferences.from_json(body["preferences"].to_json)
|
||||||
PG_DB.exec("UPDATE users SET preferences = $1 WHERE email = $2", user.preferences.to_json, user.email)
|
PG_DB.exec("UPDATE users SET preferences = $1 WHERE email = $2", user.preferences.to_json, user.email)
|
||||||
end
|
end
|
||||||
|
|
||||||
@ -2573,7 +2573,7 @@ post "/data_control" do |env|
|
|||||||
next
|
next
|
||||||
end
|
end
|
||||||
|
|
||||||
playlist_video = PlaylistVideo.new(
|
playlist_video = PlaylistVideo.new({
|
||||||
title: video.title,
|
title: video.title,
|
||||||
id: video.id,
|
id: video.id,
|
||||||
author: video.author,
|
author: video.author,
|
||||||
@ -2582,8 +2582,8 @@ post "/data_control" do |env|
|
|||||||
published: video.published,
|
published: video.published,
|
||||||
plid: playlist.id,
|
plid: playlist.id,
|
||||||
live_now: video.live_now,
|
live_now: video.live_now,
|
||||||
index: Random::Secure.rand(0_i64..Int64::MAX)
|
index: Random::Secure.rand(0_i64..Int64::MAX),
|
||||||
)
|
})
|
||||||
|
|
||||||
video_array = playlist_video.to_a
|
video_array = playlist_video.to_a
|
||||||
args = arg_array(video_array)
|
args = arg_array(video_array)
|
||||||
@ -3154,7 +3154,7 @@ get "/feed/channel/:ucid" do |env|
|
|||||||
description_html = entry.xpath_node("group/description").not_nil!.to_s
|
description_html = entry.xpath_node("group/description").not_nil!.to_s
|
||||||
views = entry.xpath_node("group/community/statistics").not_nil!.["views"].to_i64
|
views = entry.xpath_node("group/community/statistics").not_nil!.["views"].to_i64
|
||||||
|
|
||||||
SearchVideo.new(
|
SearchVideo.new({
|
||||||
title: title,
|
title: title,
|
||||||
id: video_id,
|
id: video_id,
|
||||||
author: author,
|
author: author,
|
||||||
@ -3166,8 +3166,8 @@ get "/feed/channel/:ucid" do |env|
|
|||||||
live_now: false,
|
live_now: false,
|
||||||
paid: false,
|
paid: false,
|
||||||
premium: false,
|
premium: false,
|
||||||
premiere_timestamp: nil
|
premiere_timestamp: nil,
|
||||||
)
|
})
|
||||||
end
|
end
|
||||||
|
|
||||||
XML.build(indent: " ", encoding: "UTF-8") do |xml|
|
XML.build(indent: " ", encoding: "UTF-8") do |xml|
|
||||||
@ -3397,7 +3397,7 @@ post "/feed/webhook/:token" do |env|
|
|||||||
}.to_json
|
}.to_json
|
||||||
PG_DB.exec("NOTIFY notifications, E'#{payload}'")
|
PG_DB.exec("NOTIFY notifications, E'#{payload}'")
|
||||||
|
|
||||||
video = ChannelVideo.new(
|
video = ChannelVideo.new({
|
||||||
id: id,
|
id: id,
|
||||||
title: video.title,
|
title: video.title,
|
||||||
published: published,
|
published: published,
|
||||||
@ -3408,7 +3408,7 @@ post "/feed/webhook/:token" do |env|
|
|||||||
live_now: video.live_now,
|
live_now: video.live_now,
|
||||||
premiere_timestamp: video.premiere_timestamp,
|
premiere_timestamp: video.premiere_timestamp,
|
||||||
views: video.views,
|
views: video.views,
|
||||||
)
|
})
|
||||||
|
|
||||||
PG_DB.query_all("UPDATE users SET feed_needs_update = true, notifications = array_append(notifications, $1) \
|
PG_DB.query_all("UPDATE users SET feed_needs_update = true, notifications = array_append(notifications, $1) \
|
||||||
WHERE updated < $2 AND $3 = ANY(subscriptions) AND $1 <> ALL(notifications)",
|
WHERE updated < $2 AND $3 = ANY(subscriptions) AND $1 <> ALL(notifications)",
|
||||||
@ -4666,7 +4666,7 @@ post "/api/v1/auth/preferences" do |env|
|
|||||||
user = env.get("user").as(User)
|
user = env.get("user").as(User)
|
||||||
|
|
||||||
begin
|
begin
|
||||||
preferences = Preferences.from_json(env.request.body || "{}", user.preferences)
|
preferences = Preferences.from_json(env.request.body || "{}")
|
||||||
rescue
|
rescue
|
||||||
preferences = user.preferences
|
preferences = user.preferences
|
||||||
end
|
end
|
||||||
@ -4920,7 +4920,7 @@ post "/api/v1/auth/playlists/:plid/videos" do |env|
|
|||||||
next error_message
|
next error_message
|
||||||
end
|
end
|
||||||
|
|
||||||
playlist_video = PlaylistVideo.new(
|
playlist_video = PlaylistVideo.new({
|
||||||
title: video.title,
|
title: video.title,
|
||||||
id: video.id,
|
id: video.id,
|
||||||
author: video.author,
|
author: video.author,
|
||||||
@ -4929,8 +4929,8 @@ post "/api/v1/auth/playlists/:plid/videos" do |env|
|
|||||||
published: video.published,
|
published: video.published,
|
||||||
plid: plid,
|
plid: plid,
|
||||||
live_now: video.live_now,
|
live_now: video.live_now,
|
||||||
index: Random::Secure.rand(0_i64..Int64::MAX)
|
index: Random::Secure.rand(0_i64..Int64::MAX),
|
||||||
)
|
})
|
||||||
|
|
||||||
video_array = playlist_video.to_a
|
video_array = playlist_video.to_a
|
||||||
args = arg_array(video_array)
|
args = arg_array(video_array)
|
||||||
|
@ -1,14 +1,27 @@
|
|||||||
struct InvidiousChannel
|
struct InvidiousChannel
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
id: String,
|
|
||||||
author: String,
|
property id : String
|
||||||
updated: Time,
|
property author : String
|
||||||
deleted: Bool,
|
property updated : Time
|
||||||
subscribed: Time?,
|
property deleted : Bool
|
||||||
})
|
property subscribed : Time?
|
||||||
end
|
end
|
||||||
|
|
||||||
struct ChannelVideo
|
struct ChannelVideo
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property id : String
|
||||||
|
property title : String
|
||||||
|
property published : Time
|
||||||
|
property updated : Time
|
||||||
|
property ucid : String
|
||||||
|
property author : String
|
||||||
|
property length_seconds : Int32 = 0
|
||||||
|
property live_now : Bool = false
|
||||||
|
property premiere_timestamp : Time? = nil
|
||||||
|
property views : Int64? = nil
|
||||||
|
|
||||||
def to_json(locale, json : JSON::Builder)
|
def to_json(locale, json : JSON::Builder)
|
||||||
json.object do
|
json.object do
|
||||||
json.field "type", "shortVideo"
|
json.field "type", "shortVideo"
|
||||||
@ -84,49 +97,36 @@ struct ChannelVideo
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
id: String,
|
|
||||||
title: String,
|
|
||||||
published: Time,
|
|
||||||
updated: Time,
|
|
||||||
ucid: String,
|
|
||||||
author: String,
|
|
||||||
length_seconds: {type: Int32, default: 0},
|
|
||||||
live_now: {type: Bool, default: false},
|
|
||||||
premiere_timestamp: {type: Time?, default: nil},
|
|
||||||
views: {type: Int64?, default: nil},
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct AboutRelatedChannel
|
struct AboutRelatedChannel
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
ucid: String,
|
|
||||||
author: String,
|
property ucid : String
|
||||||
author_url: String,
|
property author : String
|
||||||
author_thumbnail: String,
|
property author_url : String
|
||||||
})
|
property author_thumbnail : String
|
||||||
end
|
end
|
||||||
|
|
||||||
# TODO: Refactor into either SearchChannel or InvidiousChannel
|
# TODO: Refactor into either SearchChannel or InvidiousChannel
|
||||||
struct AboutChannel
|
struct AboutChannel
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
ucid: String,
|
|
||||||
author: String,
|
property ucid : String
|
||||||
auto_generated: Bool,
|
property author : String
|
||||||
author_url: String,
|
property auto_generated : Bool
|
||||||
author_thumbnail: String,
|
property author_url : String
|
||||||
banner: String?,
|
property author_thumbnail : String
|
||||||
description_html: String,
|
property banner : String?
|
||||||
paid: Bool,
|
property description_html : String
|
||||||
total_views: Int64,
|
property paid : Bool
|
||||||
sub_count: Int32,
|
property total_views : Int64
|
||||||
joined: Time,
|
property sub_count : Int32
|
||||||
is_family_friendly: Bool,
|
property joined : Time
|
||||||
allowed_regions: Array(String),
|
property is_family_friendly : Bool
|
||||||
related_channels: Array(AboutRelatedChannel),
|
property allowed_regions : Array(String)
|
||||||
tabs: Array(String),
|
property related_channels : Array(AboutRelatedChannel)
|
||||||
})
|
property tabs : Array(String)
|
||||||
end
|
end
|
||||||
|
|
||||||
class ChannelRedirect < Exception
|
class ChannelRedirect < Exception
|
||||||
@ -248,7 +248,7 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
|
|||||||
|
|
||||||
premiere_timestamp = channel_video.try &.premiere_timestamp
|
premiere_timestamp = channel_video.try &.premiere_timestamp
|
||||||
|
|
||||||
video = ChannelVideo.new(
|
video = ChannelVideo.new({
|
||||||
id: video_id,
|
id: video_id,
|
||||||
title: title,
|
title: title,
|
||||||
published: published,
|
published: published,
|
||||||
@ -259,7 +259,7 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
|
|||||||
live_now: live_now,
|
live_now: live_now,
|
||||||
premiere_timestamp: premiere_timestamp,
|
premiere_timestamp: premiere_timestamp,
|
||||||
views: views,
|
views: views,
|
||||||
)
|
})
|
||||||
|
|
||||||
emails = db.query_all("UPDATE users SET notifications = array_append(notifications, $1) \
|
emails = db.query_all("UPDATE users SET notifications = array_append(notifications, $1) \
|
||||||
WHERE updated < $2 AND $3 = ANY(subscriptions) AND $1 <> ALL(notifications) RETURNING email",
|
WHERE updated < $2 AND $3 = ANY(subscriptions) AND $1 <> ALL(notifications) RETURNING email",
|
||||||
@ -298,7 +298,7 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
|
|||||||
videos = extract_videos(initial_data.as_h, author, ucid)
|
videos = extract_videos(initial_data.as_h, author, ucid)
|
||||||
|
|
||||||
count = videos.size
|
count = videos.size
|
||||||
videos = videos.map { |video| ChannelVideo.new(
|
videos = videos.map { |video| ChannelVideo.new({
|
||||||
id: video.id,
|
id: video.id,
|
||||||
title: video.title,
|
title: video.title,
|
||||||
published: video.published,
|
published: video.published,
|
||||||
@ -308,8 +308,8 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
|
|||||||
length_seconds: video.length_seconds,
|
length_seconds: video.length_seconds,
|
||||||
live_now: video.live_now,
|
live_now: video.live_now,
|
||||||
premiere_timestamp: video.premiere_timestamp,
|
premiere_timestamp: video.premiere_timestamp,
|
||||||
views: video.views
|
views: video.views,
|
||||||
) }
|
}) }
|
||||||
|
|
||||||
videos.each do |video|
|
videos.each do |video|
|
||||||
ids << video.id
|
ids << video.id
|
||||||
@ -352,7 +352,13 @@ def fetch_channel(ucid, db, pull_all_videos = true, locale = nil)
|
|||||||
db.exec("DELETE FROM channel_videos * WHERE NOT id = ANY ('{#{ids.map { |id| %("#{id}") }.join(",")}}') AND ucid = $1", ucid)
|
db.exec("DELETE FROM channel_videos * WHERE NOT id = ANY ('{#{ids.map { |id| %("#{id}") }.join(",")}}') AND ucid = $1", ucid)
|
||||||
end
|
end
|
||||||
|
|
||||||
channel = InvidiousChannel.new(ucid, author, Time.utc, false, nil)
|
channel = InvidiousChannel.new({
|
||||||
|
id: ucid,
|
||||||
|
author: author,
|
||||||
|
updated: Time.utc,
|
||||||
|
deleted: false,
|
||||||
|
subscribed: nil,
|
||||||
|
})
|
||||||
|
|
||||||
return channel
|
return channel
|
||||||
end
|
end
|
||||||
@ -396,11 +402,11 @@ def produce_channel_videos_url(ucid, page = 1, auto_generated = nil, sort_by = "
|
|||||||
"2:string" => ucid,
|
"2:string" => ucid,
|
||||||
"3:base64" => {
|
"3:base64" => {
|
||||||
"2:string" => "videos",
|
"2:string" => "videos",
|
||||||
"6:varint": 2_i64,
|
"6:varint" => 2_i64,
|
||||||
"7:varint": 1_i64,
|
"7:varint" => 1_i64,
|
||||||
"12:varint": 1_i64,
|
"12:varint" => 1_i64,
|
||||||
"13:string": "",
|
"13:string" => "",
|
||||||
"23:varint": 0_i64,
|
"23:varint" => 0_i64,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@ -445,11 +451,11 @@ def produce_channel_playlists_url(ucid, cursor, sort = "newest", auto_generated
|
|||||||
"2:string" => ucid,
|
"2:string" => ucid,
|
||||||
"3:base64" => {
|
"3:base64" => {
|
||||||
"2:string" => "playlists",
|
"2:string" => "playlists",
|
||||||
"6:varint": 2_i64,
|
"6:varint" => 2_i64,
|
||||||
"7:varint": 1_i64,
|
"7:varint" => 1_i64,
|
||||||
"12:varint": 1_i64,
|
"12:varint" => 1_i64,
|
||||||
"13:string": "",
|
"13:string" => "",
|
||||||
"23:varint": 0_i64,
|
"23:varint" => 0_i64,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@ -849,12 +855,12 @@ def get_about_info(ucid, locale)
|
|||||||
related_author_thumbnail = node.xpath_node(%q(.//img)).try &.["data-thumb"]
|
related_author_thumbnail = node.xpath_node(%q(.//img)).try &.["data-thumb"]
|
||||||
related_author_thumbnail ||= ""
|
related_author_thumbnail ||= ""
|
||||||
|
|
||||||
AboutRelatedChannel.new(
|
AboutRelatedChannel.new({
|
||||||
ucid: related_id,
|
ucid: related_id,
|
||||||
author: related_title,
|
author: related_title,
|
||||||
author_url: related_author_url,
|
author_url: related_author_url,
|
||||||
author_thumbnail: related_author_thumbnail,
|
author_thumbnail: related_author_thumbnail,
|
||||||
)
|
})
|
||||||
end
|
end
|
||||||
|
|
||||||
joined = about.xpath_node(%q(//span[contains(., "Joined")]))
|
joined = about.xpath_node(%q(//span[contains(., "Joined")]))
|
||||||
@ -876,7 +882,7 @@ def get_about_info(ucid, locale)
|
|||||||
|
|
||||||
tabs = about.xpath_nodes(%q(//ul[@id="channel-navigation-menu"]/li/a/span)).map { |node| node.content.downcase }
|
tabs = about.xpath_nodes(%q(//ul[@id="channel-navigation-menu"]/li/a/span)).map { |node| node.content.downcase }
|
||||||
|
|
||||||
AboutChannel.new(
|
AboutChannel.new({
|
||||||
ucid: ucid,
|
ucid: ucid,
|
||||||
author: author,
|
author: author,
|
||||||
auto_generated: auto_generated,
|
auto_generated: auto_generated,
|
||||||
@ -891,8 +897,8 @@ def get_about_info(ucid, locale)
|
|||||||
is_family_friendly: is_family_friendly,
|
is_family_friendly: is_family_friendly,
|
||||||
allowed_regions: allowed_regions,
|
allowed_regions: allowed_regions,
|
||||||
related_channels: related_channels,
|
related_channels: related_channels,
|
||||||
tabs: tabs
|
tabs: tabs,
|
||||||
)
|
})
|
||||||
end
|
end
|
||||||
|
|
||||||
def get_60_videos(ucid, author, page, auto_generated, sort_by = "newest")
|
def get_60_videos(ucid, author, page, auto_generated, sort_by = "newest")
|
||||||
|
@ -1,11 +1,23 @@
|
|||||||
class RedditThing
|
class RedditThing
|
||||||
JSON.mapping({
|
include JSON::Serializable
|
||||||
kind: String,
|
|
||||||
data: RedditComment | RedditLink | RedditMore | RedditListing,
|
property kind : String
|
||||||
})
|
property data : RedditComment | RedditLink | RedditMore | RedditListing
|
||||||
end
|
end
|
||||||
|
|
||||||
class RedditComment
|
class RedditComment
|
||||||
|
include JSON::Serializable
|
||||||
|
|
||||||
|
property author : String
|
||||||
|
property body_html : String
|
||||||
|
property replies : RedditThing | String
|
||||||
|
property score : Int32
|
||||||
|
property depth : Int32
|
||||||
|
property permalink : String
|
||||||
|
|
||||||
|
@[JSON::Field(converter: RedditComment::TimeConverter)]
|
||||||
|
property created_utc : Time
|
||||||
|
|
||||||
module TimeConverter
|
module TimeConverter
|
||||||
def self.from_json(value : JSON::PullParser) : Time
|
def self.from_json(value : JSON::PullParser) : Time
|
||||||
Time.unix(value.read_float.to_i)
|
Time.unix(value.read_float.to_i)
|
||||||
@ -15,46 +27,33 @@ class RedditComment
|
|||||||
json.number(value.to_unix)
|
json.number(value.to_unix)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
JSON.mapping({
|
|
||||||
author: String,
|
|
||||||
body_html: String,
|
|
||||||
replies: RedditThing | String,
|
|
||||||
score: Int32,
|
|
||||||
depth: Int32,
|
|
||||||
permalink: String,
|
|
||||||
created_utc: {
|
|
||||||
type: Time,
|
|
||||||
converter: RedditComment::TimeConverter,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct RedditLink
|
struct RedditLink
|
||||||
JSON.mapping({
|
include JSON::Serializable
|
||||||
author: String,
|
|
||||||
score: Int32,
|
property author : String
|
||||||
subreddit: String,
|
property score : Int32
|
||||||
num_comments: Int32,
|
property subreddit : String
|
||||||
id: String,
|
property num_comments : Int32
|
||||||
permalink: String,
|
property id : String
|
||||||
title: String,
|
property permalink : String
|
||||||
})
|
property title : String
|
||||||
end
|
end
|
||||||
|
|
||||||
struct RedditMore
|
struct RedditMore
|
||||||
JSON.mapping({
|
include JSON::Serializable
|
||||||
children: Array(String),
|
|
||||||
count: Int32,
|
property children : Array(String)
|
||||||
depth: Int32,
|
property count : Int32
|
||||||
})
|
property depth : Int32
|
||||||
end
|
end
|
||||||
|
|
||||||
class RedditListing
|
class RedditListing
|
||||||
JSON.mapping({
|
include JSON::Serializable
|
||||||
children: Array(RedditThing),
|
|
||||||
modhash: String,
|
property children : Array(RedditThing)
|
||||||
})
|
property modhash : String
|
||||||
end
|
end
|
||||||
|
|
||||||
def fetch_youtube_comments(id, db, cursor, format, locale, thin_mode, region, sort_by = "top")
|
def fetch_youtube_comments(id, db, cursor, format, locale, thin_mode, region, sort_by = "top")
|
||||||
|
@ -1,219 +1,100 @@
|
|||||||
require "./macros"
|
require "./macros"
|
||||||
|
|
||||||
struct Nonce
|
struct Nonce
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
nonce: String,
|
|
||||||
expire: Time,
|
property nonce : String
|
||||||
})
|
property expire : Time
|
||||||
end
|
end
|
||||||
|
|
||||||
struct SessionId
|
struct SessionId
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
id: String,
|
|
||||||
email: String,
|
property id : String
|
||||||
issued: String,
|
property email : String
|
||||||
})
|
property issued : String
|
||||||
end
|
end
|
||||||
|
|
||||||
struct Annotation
|
struct Annotation
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
id: String,
|
|
||||||
annotations: String,
|
property id : String
|
||||||
})
|
property annotations : String
|
||||||
end
|
end
|
||||||
|
|
||||||
struct ConfigPreferences
|
struct ConfigPreferences
|
||||||
module StringToArray
|
include YAML::Serializable
|
||||||
def self.to_json(value : Array(String), json : JSON::Builder)
|
|
||||||
json.array do
|
|
||||||
value.each do |element|
|
|
||||||
json.string element
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_json(value : JSON::PullParser) : Array(String)
|
property annotations : Bool = false
|
||||||
begin
|
property annotations_subscribed : Bool = false
|
||||||
result = [] of String
|
property autoplay : Bool = false
|
||||||
value.read_array do
|
property captions : Array(String) = ["", "", ""]
|
||||||
result << HTML.escape(value.read_string[0, 100])
|
property comments : Array(String) = ["youtube", ""]
|
||||||
end
|
property continue : Bool = false
|
||||||
rescue ex
|
property continue_autoplay : Bool = true
|
||||||
result = [HTML.escape(value.read_string[0, 100]), ""]
|
property dark_mode : String = ""
|
||||||
end
|
property latest_only : Bool = false
|
||||||
|
property listen : Bool = false
|
||||||
|
property local : Bool = false
|
||||||
|
property locale : String = "en-US"
|
||||||
|
property max_results : Int32 = 40
|
||||||
|
property notifications_only : Bool = false
|
||||||
|
property player_style : String = "invidious"
|
||||||
|
property quality : String = "hd720"
|
||||||
|
property default_home : String = "Popular"
|
||||||
|
property feed_menu : Array(String) = ["Popular", "Trending", "Subscriptions", "Playlists"]
|
||||||
|
property related_videos : Bool = true
|
||||||
|
property sort : String = "published"
|
||||||
|
property speed : Float32 = 1.0_f32
|
||||||
|
property thin_mode : Bool = false
|
||||||
|
property unseen_only : Bool = false
|
||||||
|
property video_loop : Bool = false
|
||||||
|
property volume : Int32 = 100
|
||||||
|
|
||||||
result
|
def to_tuple
|
||||||
|
{% begin %}
|
||||||
|
{
|
||||||
|
{{*@type.instance_vars.map { |var| "#{var.name}: #{var.name}".id }}}
|
||||||
|
}
|
||||||
|
{% end %}
|
||||||
end
|
end
|
||||||
|
|
||||||
def self.to_yaml(value : Array(String), yaml : YAML::Nodes::Builder)
|
|
||||||
yaml.sequence do
|
|
||||||
value.each do |element|
|
|
||||||
yaml.scalar element
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : Array(String)
|
|
||||||
begin
|
|
||||||
unless node.is_a?(YAML::Nodes::Sequence)
|
|
||||||
node.raise "Expected sequence, not #{node.class}"
|
|
||||||
end
|
|
||||||
|
|
||||||
result = [] of String
|
|
||||||
node.nodes.each do |item|
|
|
||||||
unless item.is_a?(YAML::Nodes::Scalar)
|
|
||||||
node.raise "Expected scalar, not #{item.class}"
|
|
||||||
end
|
|
||||||
|
|
||||||
result << HTML.escape(item.value[0, 100])
|
|
||||||
end
|
|
||||||
rescue ex
|
|
||||||
if node.is_a?(YAML::Nodes::Scalar)
|
|
||||||
result = [HTML.escape(node.value[0, 100]), ""]
|
|
||||||
else
|
|
||||||
result = ["", ""]
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
result
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
module BoolToString
|
|
||||||
def self.to_json(value : String, json : JSON::Builder)
|
|
||||||
json.string value
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_json(value : JSON::PullParser) : String
|
|
||||||
begin
|
|
||||||
result = value.read_string
|
|
||||||
|
|
||||||
if result.empty?
|
|
||||||
CONFIG.default_user_preferences.dark_mode
|
|
||||||
else
|
|
||||||
result
|
|
||||||
end
|
|
||||||
rescue ex
|
|
||||||
if value.read_bool
|
|
||||||
"dark"
|
|
||||||
else
|
|
||||||
"light"
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.to_yaml(value : String, yaml : YAML::Nodes::Builder)
|
|
||||||
yaml.scalar value
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : String
|
|
||||||
unless node.is_a?(YAML::Nodes::Scalar)
|
|
||||||
node.raise "Expected scalar, not #{node.class}"
|
|
||||||
end
|
|
||||||
|
|
||||||
case node.value
|
|
||||||
when "true"
|
|
||||||
"dark"
|
|
||||||
when "false"
|
|
||||||
"light"
|
|
||||||
when ""
|
|
||||||
CONFIG.default_user_preferences.dark_mode
|
|
||||||
else
|
|
||||||
node.value
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
yaml_mapping({
|
|
||||||
annotations: {type: Bool, default: false},
|
|
||||||
annotations_subscribed: {type: Bool, default: false},
|
|
||||||
autoplay: {type: Bool, default: false},
|
|
||||||
captions: {type: Array(String), default: ["", "", ""], converter: StringToArray},
|
|
||||||
comments: {type: Array(String), default: ["youtube", ""], converter: StringToArray},
|
|
||||||
continue: {type: Bool, default: false},
|
|
||||||
continue_autoplay: {type: Bool, default: true},
|
|
||||||
dark_mode: {type: String, default: "", converter: BoolToString},
|
|
||||||
latest_only: {type: Bool, default: false},
|
|
||||||
listen: {type: Bool, default: false},
|
|
||||||
local: {type: Bool, default: false},
|
|
||||||
locale: {type: String, default: "en-US"},
|
|
||||||
max_results: {type: Int32, default: 40},
|
|
||||||
notifications_only: {type: Bool, default: false},
|
|
||||||
player_style: {type: String, default: "invidious"},
|
|
||||||
quality: {type: String, default: "hd720"},
|
|
||||||
default_home: {type: String, default: "Popular"},
|
|
||||||
feed_menu: {type: Array(String), default: ["Popular", "Trending", "Subscriptions", "Playlists"]},
|
|
||||||
related_videos: {type: Bool, default: true},
|
|
||||||
sort: {type: String, default: "published"},
|
|
||||||
speed: {type: Float32, default: 1.0_f32},
|
|
||||||
thin_mode: {type: Bool, default: false},
|
|
||||||
unseen_only: {type: Bool, default: false},
|
|
||||||
video_loop: {type: Bool, default: false},
|
|
||||||
volume: {type: Int32, default: 100},
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct Config
|
struct Config
|
||||||
module ConfigPreferencesConverter
|
include YAML::Serializable
|
||||||
def self.to_yaml(value : Preferences, yaml : YAML::Nodes::Builder)
|
|
||||||
value.to_yaml(yaml)
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : Preferences
|
property channel_threads : Int32 # Number of threads to use for crawling videos from channels (for updating subscriptions)
|
||||||
Preferences.new(*ConfigPreferences.new(ctx, node).to_tuple)
|
property feed_threads : Int32 # Number of threads to use for updating feeds
|
||||||
end
|
property db : DBConfig # Database configuration
|
||||||
end
|
property full_refresh : Bool # Used for crawling channels: threads should check all videos uploaded by a channel
|
||||||
|
property https_only : Bool? # Used to tell Invidious it is behind a proxy, so links to resources should be https://
|
||||||
|
property hmac_key : String? # HMAC signing key for CSRF tokens and verifying pubsub subscriptions
|
||||||
|
property domain : String? # Domain to be used for links to resources on the site where an absolute URL is required
|
||||||
|
property use_pubsub_feeds : Bool | Int32 = false # Subscribe to channels using PubSubHubbub (requires domain, hmac_key)
|
||||||
|
property captcha_enabled : Bool = true
|
||||||
|
property login_enabled : Bool = true
|
||||||
|
property registration_enabled : Bool = true
|
||||||
|
property statistics_enabled : Bool = false
|
||||||
|
property admins : Array(String) = [] of String
|
||||||
|
property external_port : Int32? = nil
|
||||||
|
property default_user_preferences : ConfigPreferences
|
||||||
|
property dmca_content : Array(String) = [] of String # For compliance with DMCA, disables download widget using list of video IDs
|
||||||
|
property check_tables : Bool = false # Check table integrity, automatically try to add any missing columns, create tables, etc.
|
||||||
|
property cache_annotations : Bool = false # Cache annotations requested from IA, will not cache empty annotations or annotations that only contain cards
|
||||||
|
property banner : String? = nil # Optional banner to be displayed along top of page for announcements, etc.
|
||||||
|
property hsts : Bool? = true # Enables 'Strict-Transport-Security'. Ensure that `domain` and all subdomains are served securely
|
||||||
|
property disable_proxy : Bool? | Array(String)? = false # Disable proxying server-wide: options: 'dash', 'livestreams', 'downloads', 'local'
|
||||||
|
|
||||||
module FamilyConverter
|
@[YAML::Field(converter: Preferences::FamilyConverter)]
|
||||||
def self.to_yaml(value : Socket::Family, yaml : YAML::Nodes::Builder)
|
property force_resolve : Socket::Family = Socket::Family::UNSPEC # Connect to YouTube over 'ipv6', 'ipv4'. Will sometimes resolve fix issues with rate-limiting (see https://github.com/ytdl-org/youtube-dl/issues/21729)
|
||||||
case value
|
property port : Int32 = 3000 # Port to listen for connections (overrided by command line argument)
|
||||||
when Socket::Family::UNSPEC
|
property host_binding : String = "0.0.0.0" # Host to bind (overrided by command line argument)
|
||||||
yaml.scalar nil
|
property pool_size : Int32 = 100 # Pool size for HTTP requests to youtube.com and ytimg.com (each domain has a separate pool of `pool_size`)
|
||||||
when Socket::Family::INET
|
property admin_email : String = "omarroth@protonmail.com" # Email for bug reports
|
||||||
yaml.scalar "ipv4"
|
|
||||||
when Socket::Family::INET6
|
|
||||||
yaml.scalar "ipv6"
|
|
||||||
when Socket::Family::UNIX
|
|
||||||
raise "Invalid socket family #{value}"
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : Socket::Family
|
@[YAML::Field(converter: Preferences::StringToCookies)]
|
||||||
if node.is_a?(YAML::Nodes::Scalar)
|
property cookies : HTTP::Cookies = HTTP::Cookies.new # Saved cookies in "name1=value1; name2=value2..." format
|
||||||
case node.value.downcase
|
property captcha_key : String? = nil # Key for Anti-Captcha
|
||||||
when "ipv4"
|
|
||||||
Socket::Family::INET
|
|
||||||
when "ipv6"
|
|
||||||
Socket::Family::INET6
|
|
||||||
else
|
|
||||||
Socket::Family::UNSPEC
|
|
||||||
end
|
|
||||||
else
|
|
||||||
node.raise "Expected scalar, not #{node.class}"
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
module StringToCookies
|
|
||||||
def self.to_yaml(value : HTTP::Cookies, yaml : YAML::Nodes::Builder)
|
|
||||||
(value.map { |c| "#{c.name}=#{c.value}" }).join("; ").to_yaml(yaml)
|
|
||||||
end
|
|
||||||
|
|
||||||
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : HTTP::Cookies
|
|
||||||
unless node.is_a?(YAML::Nodes::Scalar)
|
|
||||||
node.raise "Expected scalar, not #{node.class}"
|
|
||||||
end
|
|
||||||
|
|
||||||
cookies = HTTP::Cookies.new
|
|
||||||
node.value.split(";").each do |cookie|
|
|
||||||
next if cookie.strip.empty?
|
|
||||||
name, value = cookie.split("=", 2)
|
|
||||||
cookies << HTTP::Cookie.new(name.strip, value.strip)
|
|
||||||
end
|
|
||||||
|
|
||||||
cookies
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def disabled?(option)
|
def disabled?(option)
|
||||||
case disabled = CONFIG.disable_proxy
|
case disabled = CONFIG.disable_proxy
|
||||||
@ -229,50 +110,16 @@ struct Config
|
|||||||
return false
|
return false
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
YAML.mapping({
|
|
||||||
channel_threads: Int32, # Number of threads to use for crawling videos from channels (for updating subscriptions)
|
|
||||||
feed_threads: Int32, # Number of threads to use for updating feeds
|
|
||||||
db: DBConfig, # Database configuration
|
|
||||||
full_refresh: Bool, # Used for crawling channels: threads should check all videos uploaded by a channel
|
|
||||||
https_only: Bool?, # Used to tell Invidious it is behind a proxy, so links to resources should be https://
|
|
||||||
hmac_key: String?, # HMAC signing key for CSRF tokens and verifying pubsub subscriptions
|
|
||||||
domain: String?, # Domain to be used for links to resources on the site where an absolute URL is required
|
|
||||||
use_pubsub_feeds: {type: Bool | Int32, default: false}, # Subscribe to channels using PubSubHubbub (requires domain, hmac_key)
|
|
||||||
captcha_enabled: {type: Bool, default: true},
|
|
||||||
login_enabled: {type: Bool, default: true},
|
|
||||||
registration_enabled: {type: Bool, default: true},
|
|
||||||
statistics_enabled: {type: Bool, default: false},
|
|
||||||
admins: {type: Array(String), default: [] of String},
|
|
||||||
external_port: {type: Int32?, default: nil},
|
|
||||||
default_user_preferences: {type: Preferences,
|
|
||||||
default: Preferences.new(*ConfigPreferences.from_yaml("").to_tuple),
|
|
||||||
converter: ConfigPreferencesConverter,
|
|
||||||
},
|
|
||||||
dmca_content: {type: Array(String), default: [] of String}, # For compliance with DMCA, disables download widget using list of video IDs
|
|
||||||
check_tables: {type: Bool, default: false}, # Check table integrity, automatically try to add any missing columns, create tables, etc.
|
|
||||||
cache_annotations: {type: Bool, default: false}, # Cache annotations requested from IA, will not cache empty annotations or annotations that only contain cards
|
|
||||||
banner: {type: String?, default: nil}, # Optional banner to be displayed along top of page for announcements, etc.
|
|
||||||
hsts: {type: Bool?, default: true}, # Enables 'Strict-Transport-Security'. Ensure that `domain` and all subdomains are served securely
|
|
||||||
disable_proxy: {type: Bool? | Array(String)?, default: false}, # Disable proxying server-wide: options: 'dash', 'livestreams', 'downloads', 'local'
|
|
||||||
force_resolve: {type: Socket::Family, default: Socket::Family::UNSPEC, converter: FamilyConverter}, # Connect to YouTube over 'ipv6', 'ipv4'. Will sometimes resolve fix issues with rate-limiting (see https://github.com/ytdl-org/youtube-dl/issues/21729)
|
|
||||||
port: {type: Int32, default: 3000}, # Port to listen for connections (overrided by command line argument)
|
|
||||||
host_binding: {type: String, default: "0.0.0.0"}, # Host to bind (overrided by command line argument)
|
|
||||||
pool_size: {type: Int32, default: 100}, # Pool size for HTTP requests to youtube.com and ytimg.com (each domain has a separate pool of `pool_size`)
|
|
||||||
admin_email: {type: String, default: "omarroth@protonmail.com"}, # Email for bug reports
|
|
||||||
cookies: {type: HTTP::Cookies, default: HTTP::Cookies.new, converter: StringToCookies}, # Saved cookies in "name1=value1; name2=value2..." format
|
|
||||||
captcha_key: {type: String?, default: nil}, # Key for Anti-Captcha
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct DBConfig
|
struct DBConfig
|
||||||
yaml_mapping({
|
include YAML::Serializable
|
||||||
user: String,
|
|
||||||
password: String,
|
property user : String
|
||||||
host: String,
|
property password : String
|
||||||
port: Int32,
|
property host : String
|
||||||
dbname: String,
|
property port : Int32
|
||||||
})
|
property dbname : String
|
||||||
end
|
end
|
||||||
|
|
||||||
def login_req(f_req)
|
def login_req(f_req)
|
||||||
@ -365,7 +212,7 @@ def extract_items(initial_data : Hash(String, JSON::Any), author_fallback : Stri
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
items << SearchVideo.new(
|
items << SearchVideo.new({
|
||||||
title: title,
|
title: title,
|
||||||
id: video_id,
|
id: video_id,
|
||||||
author: author,
|
author: author,
|
||||||
@ -377,8 +224,8 @@ def extract_items(initial_data : Hash(String, JSON::Any), author_fallback : Stri
|
|||||||
live_now: live_now,
|
live_now: live_now,
|
||||||
paid: paid,
|
paid: paid,
|
||||||
premium: premium,
|
premium: premium,
|
||||||
premiere_timestamp: premiere_timestamp
|
premiere_timestamp: premiere_timestamp,
|
||||||
)
|
})
|
||||||
elsif i = item["channelRenderer"]?
|
elsif i = item["channelRenderer"]?
|
||||||
author = i["title"]["simpleText"]?.try &.as_s || author_fallback || ""
|
author = i["title"]["simpleText"]?.try &.as_s || author_fallback || ""
|
||||||
author_id = i["channelId"]?.try &.as_s || author_id_fallback || ""
|
author_id = i["channelId"]?.try &.as_s || author_id_fallback || ""
|
||||||
@ -391,7 +238,7 @@ def extract_items(initial_data : Hash(String, JSON::Any), author_fallback : Stri
|
|||||||
video_count = i["videoCountText"]?.try &.["runs"].as_a[0]?.try &.["text"].as_s.gsub(/\D/, "").to_i || 0
|
video_count = i["videoCountText"]?.try &.["runs"].as_a[0]?.try &.["text"].as_s.gsub(/\D/, "").to_i || 0
|
||||||
description_html = i["descriptionSnippet"]?.try { |t| parse_content(t) } || ""
|
description_html = i["descriptionSnippet"]?.try { |t| parse_content(t) } || ""
|
||||||
|
|
||||||
items << SearchChannel.new(
|
items << SearchChannel.new({
|
||||||
author: author,
|
author: author,
|
||||||
ucid: author_id,
|
ucid: author_id,
|
||||||
author_thumbnail: author_thumbnail,
|
author_thumbnail: author_thumbnail,
|
||||||
@ -399,7 +246,7 @@ def extract_items(initial_data : Hash(String, JSON::Any), author_fallback : Stri
|
|||||||
video_count: video_count,
|
video_count: video_count,
|
||||||
description_html: description_html,
|
description_html: description_html,
|
||||||
auto_generated: auto_generated,
|
auto_generated: auto_generated,
|
||||||
)
|
})
|
||||||
elsif i = item["gridPlaylistRenderer"]?
|
elsif i = item["gridPlaylistRenderer"]?
|
||||||
title = i["title"]["runs"].as_a[0]?.try &.["text"].as_s || ""
|
title = i["title"]["runs"].as_a[0]?.try &.["text"].as_s || ""
|
||||||
plid = i["playlistId"]?.try &.as_s || ""
|
plid = i["playlistId"]?.try &.as_s || ""
|
||||||
@ -407,15 +254,15 @@ def extract_items(initial_data : Hash(String, JSON::Any), author_fallback : Stri
|
|||||||
video_count = i["videoCountText"]["runs"].as_a[0]?.try &.["text"].as_s.gsub(/\D/, "").to_i || 0
|
video_count = i["videoCountText"]["runs"].as_a[0]?.try &.["text"].as_s.gsub(/\D/, "").to_i || 0
|
||||||
playlist_thumbnail = i["thumbnail"]["thumbnails"][0]?.try &.["url"]?.try &.as_s || ""
|
playlist_thumbnail = i["thumbnail"]["thumbnails"][0]?.try &.["url"]?.try &.as_s || ""
|
||||||
|
|
||||||
items << SearchPlaylist.new(
|
items << SearchPlaylist.new({
|
||||||
title: title,
|
title: title,
|
||||||
id: plid,
|
id: plid,
|
||||||
author: author_fallback || "",
|
author: author_fallback || "",
|
||||||
ucid: author_id_fallback || "",
|
ucid: author_id_fallback || "",
|
||||||
video_count: video_count,
|
video_count: video_count,
|
||||||
videos: [] of SearchPlaylistVideo,
|
videos: [] of SearchPlaylistVideo,
|
||||||
thumbnail: playlist_thumbnail
|
thumbnail: playlist_thumbnail,
|
||||||
)
|
})
|
||||||
elsif i = item["playlistRenderer"]?
|
elsif i = item["playlistRenderer"]?
|
||||||
title = i["title"]["simpleText"]?.try &.as_s || ""
|
title = i["title"]["simpleText"]?.try &.as_s || ""
|
||||||
plid = i["playlistId"]?.try &.as_s || ""
|
plid = i["playlistId"]?.try &.as_s || ""
|
||||||
@ -432,24 +279,24 @@ def extract_items(initial_data : Hash(String, JSON::Any), author_fallback : Stri
|
|||||||
v_title = v["title"]["simpleText"]?.try &.as_s || ""
|
v_title = v["title"]["simpleText"]?.try &.as_s || ""
|
||||||
v_id = v["videoId"]?.try &.as_s || ""
|
v_id = v["videoId"]?.try &.as_s || ""
|
||||||
v_length_seconds = v["lengthText"]?.try &.["simpleText"]?.try { |t| decode_length_seconds(t.as_s) } || 0
|
v_length_seconds = v["lengthText"]?.try &.["simpleText"]?.try { |t| decode_length_seconds(t.as_s) } || 0
|
||||||
SearchPlaylistVideo.new(
|
SearchPlaylistVideo.new({
|
||||||
title: v_title,
|
title: v_title,
|
||||||
id: v_id,
|
id: v_id,
|
||||||
length_seconds: v_length_seconds
|
length_seconds: v_length_seconds,
|
||||||
)
|
})
|
||||||
end || [] of SearchPlaylistVideo
|
end || [] of SearchPlaylistVideo
|
||||||
|
|
||||||
# TODO: i["publishedTimeText"]?
|
# TODO: i["publishedTimeText"]?
|
||||||
|
|
||||||
items << SearchPlaylist.new(
|
items << SearchPlaylist.new({
|
||||||
title: title,
|
title: title,
|
||||||
id: plid,
|
id: plid,
|
||||||
author: author,
|
author: author,
|
||||||
ucid: author_id,
|
ucid: author_id,
|
||||||
video_count: video_count,
|
video_count: video_count,
|
||||||
videos: videos,
|
videos: videos,
|
||||||
thumbnail: playlist_thumbnail
|
thumbnail: playlist_thumbnail,
|
||||||
)
|
})
|
||||||
elsif i = item["radioRenderer"]? # Mix
|
elsif i = item["radioRenderer"]? # Mix
|
||||||
# TODO
|
# TODO
|
||||||
elsif i = item["showRenderer"]? # Show
|
elsif i = item["showRenderer"]? # Show
|
||||||
@ -465,6 +312,7 @@ end
|
|||||||
|
|
||||||
def check_enum(db, logger, enum_name, struct_type = nil)
|
def check_enum(db, logger, enum_name, struct_type = nil)
|
||||||
return # TODO
|
return # TODO
|
||||||
|
|
||||||
if !db.query_one?("SELECT true FROM pg_type WHERE typname = $1", enum_name, as: Bool)
|
if !db.query_one?("SELECT true FROM pg_type WHERE typname = $1", enum_name, as: Bool)
|
||||||
logger.puts("CREATE TYPE #{enum_name}")
|
logger.puts("CREATE TYPE #{enum_name}")
|
||||||
|
|
||||||
@ -488,7 +336,7 @@ def check_table(db, logger, table_name, struct_type = nil)
|
|||||||
|
|
||||||
return if !struct_type
|
return if !struct_type
|
||||||
|
|
||||||
struct_array = struct_type.to_type_tuple
|
struct_array = struct_type.type_array
|
||||||
column_array = get_column_array(db, table_name)
|
column_array = get_column_array(db, table_name)
|
||||||
column_types = File.read("config/sql/#{table_name}.sql").match(/CREATE TABLE public\.#{table_name}\n\((?<types>[\d\D]*?)\);/)
|
column_types = File.read("config/sql/#{table_name}.sql").match(/CREATE TABLE public\.#{table_name}\n\((?<types>[\d\D]*?)\);/)
|
||||||
.try &.["types"].split(",").map { |line| line.strip }.reject &.starts_with?("CONSTRAINT")
|
.try &.["types"].split(",").map { |line| line.strip }.reject &.starts_with?("CONSTRAINT")
|
||||||
|
@ -67,7 +67,7 @@ def refresh_feeds(db, logger, config)
|
|||||||
begin
|
begin
|
||||||
# Drop outdated views
|
# Drop outdated views
|
||||||
column_array = get_column_array(db, view_name)
|
column_array = get_column_array(db, view_name)
|
||||||
ChannelVideo.to_type_tuple.each_with_index do |name, i|
|
ChannelVideo.type_array.each_with_index do |name, i|
|
||||||
if name != column_array[i]?
|
if name != column_array[i]?
|
||||||
logger.puts("DROP MATERIALIZED VIEW #{view_name}")
|
logger.puts("DROP MATERIALIZED VIEW #{view_name}")
|
||||||
db.exec("DROP MATERIALIZED VIEW #{view_name}")
|
db.exec("DROP MATERIALIZED VIEW #{view_name}")
|
||||||
|
@ -1,43 +1,51 @@
|
|||||||
macro db_mapping(mapping)
|
module DB::Serializable
|
||||||
def initialize({{*mapping.keys.map { |id| "@#{id}".id }}})
|
macro included
|
||||||
|
{% verbatim do %}
|
||||||
|
macro finished
|
||||||
|
def self.type_array
|
||||||
|
\{{ @type.instance_vars
|
||||||
|
.reject { |var| var.annotation(::DB::Field) && var.annotation(::DB::Field)[:ignore] }
|
||||||
|
.map { |name| name.stringify }
|
||||||
|
}}
|
||||||
|
end
|
||||||
|
|
||||||
|
def initialize(tuple)
|
||||||
|
\{% for var in @type.instance_vars %}
|
||||||
|
\{% ann = var.annotation(::DB::Field) %}
|
||||||
|
\{% if ann && ann[:ignore] %}
|
||||||
|
\{% else %}
|
||||||
|
@\{{var.name}} = tuple[:\{{var.name.id}}]
|
||||||
|
\{% end %}
|
||||||
|
\{% end %}
|
||||||
end
|
end
|
||||||
|
|
||||||
def to_a
|
def to_a
|
||||||
return [ {{*mapping.keys.map { |id| "@#{id}".id }}} ]
|
\{{ @type.instance_vars
|
||||||
|
.reject { |var| var.annotation(::DB::Field) && var.annotation(::DB::Field)[:ignore] }
|
||||||
|
.map { |name| name }
|
||||||
|
}}
|
||||||
|
end
|
||||||
|
end
|
||||||
|
{% end %}
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def self.to_type_tuple
|
module JSON::Serializable
|
||||||
return { {{*mapping.keys.map { |id| "#{id}" }}} }
|
macro included
|
||||||
|
{% verbatim do %}
|
||||||
|
macro finished
|
||||||
|
def initialize(tuple)
|
||||||
|
\{% for var in @type.instance_vars %}
|
||||||
|
\{% ann = var.annotation(::JSON::Field) %}
|
||||||
|
\{% if ann && ann[:ignore] %}
|
||||||
|
\{% else %}
|
||||||
|
@\{{var.name}} = tuple[:\{{var.name.id}}]
|
||||||
|
\{% end %}
|
||||||
|
\{% end %}
|
||||||
end
|
end
|
||||||
|
|
||||||
DB.mapping( {{mapping}} )
|
|
||||||
end
|
end
|
||||||
|
{% end %}
|
||||||
macro json_mapping(mapping)
|
|
||||||
def initialize({{*mapping.keys.map { |id| "@#{id}".id }}})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
def to_a
|
|
||||||
return [ {{*mapping.keys.map { |id| "@#{id}".id }}} ]
|
|
||||||
end
|
|
||||||
|
|
||||||
patched_json_mapping( {{mapping}} )
|
|
||||||
YAML.mapping( {{mapping}} )
|
|
||||||
end
|
|
||||||
|
|
||||||
macro yaml_mapping(mapping)
|
|
||||||
def initialize({{*mapping.keys.map { |id| "@#{id}".id }}})
|
|
||||||
end
|
|
||||||
|
|
||||||
def to_a
|
|
||||||
return [ {{*mapping.keys.map { |id| "@#{id}".id }}} ]
|
|
||||||
end
|
|
||||||
|
|
||||||
def to_tuple
|
|
||||||
return { {{*mapping.keys.map { |id| "@#{id}".id }}} }
|
|
||||||
end
|
|
||||||
|
|
||||||
YAML.mapping({{mapping}})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
macro templated(filename, template = "template")
|
macro templated(filename, template = "template")
|
||||||
|
@ -1,166 +0,0 @@
|
|||||||
# Overloads https://github.com/crystal-lang/crystal/blob/0.28.0/src/json/from_json.cr#L24
|
|
||||||
def Object.from_json(string_or_io, default) : self
|
|
||||||
parser = JSON::PullParser.new(string_or_io)
|
|
||||||
new parser, default
|
|
||||||
end
|
|
||||||
|
|
||||||
# Adds configurable 'default'
|
|
||||||
macro patched_json_mapping(_properties_, strict = false)
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
{% _properties_[key] = {type: value} unless value.is_a?(HashLiteral) || value.is_a?(NamedTupleLiteral) %}
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
{% _properties_[key][:key_id] = key.id.gsub(/\?$/, "") %}
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
@{{value[:key_id]}} : {{value[:type]}}{{ (value[:nilable] ? "?" : "").id }}
|
|
||||||
|
|
||||||
{% if value[:setter] == nil ? true : value[:setter] %}
|
|
||||||
def {{value[:key_id]}}=(_{{value[:key_id]}} : {{value[:type]}}{{ (value[:nilable] ? "?" : "").id }})
|
|
||||||
@{{value[:key_id]}} = _{{value[:key_id]}}
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:getter] == nil ? true : value[:getter] %}
|
|
||||||
def {{key.id}} : {{value[:type]}}{{ (value[:nilable] ? "?" : "").id }}
|
|
||||||
@{{value[:key_id]}}
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:presence] %}
|
|
||||||
@{{value[:key_id]}}_present : Bool = false
|
|
||||||
|
|
||||||
def {{value[:key_id]}}_present?
|
|
||||||
@{{value[:key_id]}}_present
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
def initialize(%pull : ::JSON::PullParser, default = nil)
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
%var{key.id} = nil
|
|
||||||
%found{key.id} = false
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
%location = %pull.location
|
|
||||||
begin
|
|
||||||
%pull.read_begin_object
|
|
||||||
rescue exc : ::JSON::ParseException
|
|
||||||
raise ::JSON::MappingError.new(exc.message, self.class.to_s, nil, *%location, exc)
|
|
||||||
end
|
|
||||||
until %pull.kind.end_object?
|
|
||||||
%key_location = %pull.location
|
|
||||||
key = %pull.read_object_key
|
|
||||||
case key
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
when {{value[:key] || value[:key_id].stringify}}
|
|
||||||
%found{key.id} = true
|
|
||||||
begin
|
|
||||||
%var{key.id} =
|
|
||||||
{% if value[:nilable] || value[:default] != nil %} %pull.read_null_or { {% end %}
|
|
||||||
|
|
||||||
{% if value[:root] %}
|
|
||||||
%pull.on_key!({{value[:root]}}) do
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:converter] %}
|
|
||||||
{{value[:converter]}}.from_json(%pull)
|
|
||||||
{% elsif value[:type].is_a?(Path) || value[:type].is_a?(Generic) %}
|
|
||||||
{{value[:type]}}.new(%pull)
|
|
||||||
{% else %}
|
|
||||||
::Union({{value[:type]}}).new(%pull)
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:root] %}
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:nilable] || value[:default] != nil %} } {% end %}
|
|
||||||
rescue exc : ::JSON::ParseException
|
|
||||||
raise ::JSON::MappingError.new(exc.message, self.class.to_s, {{value[:key] || value[:key_id].stringify}}, *%key_location, exc)
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
else
|
|
||||||
{% if strict %}
|
|
||||||
raise ::JSON::MappingError.new("Unknown JSON attribute: #{key}", self.class.to_s, nil, *%key_location, nil)
|
|
||||||
{% else %}
|
|
||||||
%pull.skip
|
|
||||||
{% end %}
|
|
||||||
end
|
|
||||||
end
|
|
||||||
%pull.read_next
|
|
||||||
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
{% unless value[:nilable] || value[:default] != nil %}
|
|
||||||
if %var{key.id}.nil? && !%found{key.id} && !::Union({{value[:type]}}).nilable?
|
|
||||||
raise ::JSON::MappingError.new("Missing JSON attribute: {{(value[:key] || value[:key_id]).id}}", self.class.to_s, nil, *%location, nil)
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:nilable] %}
|
|
||||||
{% if value[:default] != nil %}
|
|
||||||
@{{value[:key_id]}} = %found{key.id} ? %var{key.id} : (default.responds_to?(:{{value[:key_id]}}) ? default.{{value[:key_id]}} : {{value[:default]}})
|
|
||||||
{% else %}
|
|
||||||
@{{value[:key_id]}} = %var{key.id}
|
|
||||||
{% end %}
|
|
||||||
{% elsif value[:default] != nil %}
|
|
||||||
@{{value[:key_id]}} = %var{key.id}.nil? ? (default.responds_to?(:{{value[:key_id]}}) ? default.{{value[:key_id]}} : {{value[:default]}}) : %var{key.id}
|
|
||||||
{% else %}
|
|
||||||
@{{value[:key_id]}} = (%var{key.id}).as({{value[:type]}})
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:presence] %}
|
|
||||||
@{{value[:key_id]}}_present = %found{key.id}
|
|
||||||
{% end %}
|
|
||||||
{% end %}
|
|
||||||
end
|
|
||||||
|
|
||||||
def to_json(json : ::JSON::Builder)
|
|
||||||
json.object do
|
|
||||||
{% for key, value in _properties_ %}
|
|
||||||
_{{value[:key_id]}} = @{{value[:key_id]}}
|
|
||||||
|
|
||||||
{% unless value[:emit_null] %}
|
|
||||||
unless _{{value[:key_id]}}.nil?
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
json.field({{value[:key] || value[:key_id].stringify}}) do
|
|
||||||
{% if value[:root] %}
|
|
||||||
{% if value[:emit_null] %}
|
|
||||||
if _{{value[:key_id]}}.nil?
|
|
||||||
nil.to_json(json)
|
|
||||||
else
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
json.object do
|
|
||||||
json.field({{value[:root]}}) do
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:converter] %}
|
|
||||||
if _{{value[:key_id]}}
|
|
||||||
{{ value[:converter] }}.to_json(_{{value[:key_id]}}, json)
|
|
||||||
else
|
|
||||||
nil.to_json(json)
|
|
||||||
end
|
|
||||||
{% else %}
|
|
||||||
_{{value[:key_id]}}.to_json(json)
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
{% if value[:root] %}
|
|
||||||
{% if value[:emit_null] %}
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
end
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
end
|
|
||||||
|
|
||||||
{% unless value[:emit_null] %}
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
{% end %}
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
@ -1,21 +1,21 @@
|
|||||||
struct MixVideo
|
struct MixVideo
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
title: String,
|
|
||||||
id: String,
|
property title : String
|
||||||
author: String,
|
property id : String
|
||||||
ucid: String,
|
property author : String
|
||||||
length_seconds: Int32,
|
property ucid : String
|
||||||
index: Int32,
|
property length_seconds : Int32
|
||||||
rdid: String,
|
property index : Int32
|
||||||
})
|
property rdid : String
|
||||||
end
|
end
|
||||||
|
|
||||||
struct Mix
|
struct Mix
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
title: String,
|
|
||||||
id: String,
|
property title : String
|
||||||
videos: Array(MixVideo),
|
property id : String
|
||||||
})
|
property videos : Array(MixVideo)
|
||||||
end
|
end
|
||||||
|
|
||||||
def fetch_mix(rdid, video_id, cookies = nil, locale = nil)
|
def fetch_mix(rdid, video_id, cookies = nil, locale = nil)
|
||||||
@ -48,23 +48,22 @@ def fetch_mix(rdid, video_id, cookies = nil, locale = nil)
|
|||||||
|
|
||||||
id = item["videoId"].as_s
|
id = item["videoId"].as_s
|
||||||
title = item["title"]?.try &.["simpleText"].as_s
|
title = item["title"]?.try &.["simpleText"].as_s
|
||||||
if !title
|
next if !title
|
||||||
next
|
|
||||||
end
|
|
||||||
author = item["longBylineText"]["runs"][0]["text"].as_s
|
author = item["longBylineText"]["runs"][0]["text"].as_s
|
||||||
ucid = item["longBylineText"]["runs"][0]["navigationEndpoint"]["browseEndpoint"]["browseId"].as_s
|
ucid = item["longBylineText"]["runs"][0]["navigationEndpoint"]["browseEndpoint"]["browseId"].as_s
|
||||||
length_seconds = decode_length_seconds(item["lengthText"]["simpleText"].as_s)
|
length_seconds = decode_length_seconds(item["lengthText"]["simpleText"].as_s)
|
||||||
index = item["navigationEndpoint"]["watchEndpoint"]["index"].as_i
|
index = item["navigationEndpoint"]["watchEndpoint"]["index"].as_i
|
||||||
|
|
||||||
videos << MixVideo.new(
|
videos << MixVideo.new({
|
||||||
title,
|
title: title,
|
||||||
id,
|
id: id,
|
||||||
author,
|
author: author,
|
||||||
ucid,
|
ucid: ucid,
|
||||||
length_seconds,
|
length_seconds: length_seconds,
|
||||||
index,
|
index: index,
|
||||||
rdid
|
rdid: rdid,
|
||||||
)
|
})
|
||||||
end
|
end
|
||||||
|
|
||||||
if !cookies
|
if !cookies
|
||||||
@ -74,7 +73,11 @@ def fetch_mix(rdid, video_id, cookies = nil, locale = nil)
|
|||||||
|
|
||||||
videos.uniq! { |video| video.id }
|
videos.uniq! { |video| video.id }
|
||||||
videos = videos.first(50)
|
videos = videos.first(50)
|
||||||
return Mix.new(mix_title, rdid, videos)
|
return Mix.new({
|
||||||
|
title: mix_title,
|
||||||
|
id: rdid,
|
||||||
|
videos: videos,
|
||||||
|
})
|
||||||
end
|
end
|
||||||
|
|
||||||
def template_mix(mix)
|
def template_mix(mix)
|
||||||
|
@ -1,4 +1,16 @@
|
|||||||
struct PlaylistVideo
|
struct PlaylistVideo
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property title : String
|
||||||
|
property id : String
|
||||||
|
property author : String
|
||||||
|
property ucid : String
|
||||||
|
property length_seconds : Int32
|
||||||
|
property published : Time
|
||||||
|
property plid : String
|
||||||
|
property index : Int64
|
||||||
|
property live_now : Bool
|
||||||
|
|
||||||
def to_xml(auto_generated, xml : XML::Builder)
|
def to_xml(auto_generated, xml : XML::Builder)
|
||||||
xml.element("entry") do
|
xml.element("entry") do
|
||||||
xml.element("id") { xml.text "yt:video:#{self.id}" }
|
xml.element("id") { xml.text "yt:video:#{self.id}" }
|
||||||
@ -78,21 +90,22 @@ struct PlaylistVideo
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
title: String,
|
|
||||||
id: String,
|
|
||||||
author: String,
|
|
||||||
ucid: String,
|
|
||||||
length_seconds: Int32,
|
|
||||||
published: Time,
|
|
||||||
plid: String,
|
|
||||||
index: Int64,
|
|
||||||
live_now: Bool,
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct Playlist
|
struct Playlist
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property title : String
|
||||||
|
property id : String
|
||||||
|
property author : String
|
||||||
|
property author_thumbnail : String
|
||||||
|
property ucid : String
|
||||||
|
property description : String
|
||||||
|
property video_count : Int32
|
||||||
|
property views : Int64
|
||||||
|
property updated : Time
|
||||||
|
property thumbnail : String?
|
||||||
|
|
||||||
def to_json(offset, locale, json : JSON::Builder, continuation : String? = nil)
|
def to_json(offset, locale, json : JSON::Builder, continuation : String? = nil)
|
||||||
json.object do
|
json.object do
|
||||||
json.field "type", "playlist"
|
json.field "type", "playlist"
|
||||||
@ -147,19 +160,6 @@ struct Playlist
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
title: String,
|
|
||||||
id: String,
|
|
||||||
author: String,
|
|
||||||
author_thumbnail: String,
|
|
||||||
ucid: String,
|
|
||||||
description: String,
|
|
||||||
video_count: Int32,
|
|
||||||
views: Int64,
|
|
||||||
updated: Time,
|
|
||||||
thumbnail: String?,
|
|
||||||
})
|
|
||||||
|
|
||||||
def privacy
|
def privacy
|
||||||
PlaylistPrivacy::Public
|
PlaylistPrivacy::Public
|
||||||
end
|
end
|
||||||
@ -176,6 +176,29 @@ enum PlaylistPrivacy
|
|||||||
end
|
end
|
||||||
|
|
||||||
struct InvidiousPlaylist
|
struct InvidiousPlaylist
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property title : String
|
||||||
|
property id : String
|
||||||
|
property author : String
|
||||||
|
property description : String = ""
|
||||||
|
property video_count : Int32
|
||||||
|
property created : Time
|
||||||
|
property updated : Time
|
||||||
|
|
||||||
|
@[DB::Field(converter: InvidiousPlaylist::PlaylistPrivacyConverter)]
|
||||||
|
property privacy : PlaylistPrivacy = PlaylistPrivacy::Private
|
||||||
|
property index : Array(Int64)
|
||||||
|
|
||||||
|
@[DB::Field(ignore: true)]
|
||||||
|
property thumbnail_id : String?
|
||||||
|
|
||||||
|
module PlaylistPrivacyConverter
|
||||||
|
def self.from_rs(rs)
|
||||||
|
return PlaylistPrivacy.parse(String.new(rs.read(Slice(UInt8))))
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
def to_json(offset, locale, json : JSON::Builder, continuation : String? = nil)
|
def to_json(offset, locale, json : JSON::Builder, continuation : String? = nil)
|
||||||
json.object do
|
json.object do
|
||||||
json.field "type", "invidiousPlaylist"
|
json.field "type", "invidiousPlaylist"
|
||||||
@ -216,26 +239,6 @@ struct InvidiousPlaylist
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
property thumbnail_id
|
|
||||||
|
|
||||||
module PlaylistPrivacyConverter
|
|
||||||
def self.from_rs(rs)
|
|
||||||
return PlaylistPrivacy.parse(String.new(rs.read(Slice(UInt8))))
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
title: String,
|
|
||||||
id: String,
|
|
||||||
author: String,
|
|
||||||
description: {type: String, default: ""},
|
|
||||||
video_count: Int32,
|
|
||||||
created: Time,
|
|
||||||
updated: Time,
|
|
||||||
privacy: {type: PlaylistPrivacy, default: PlaylistPrivacy::Private, converter: PlaylistPrivacyConverter},
|
|
||||||
index: Array(Int64),
|
|
||||||
})
|
|
||||||
|
|
||||||
def thumbnail
|
def thumbnail
|
||||||
@thumbnail_id ||= PG_DB.query_one?("SELECT id FROM playlist_videos WHERE plid = $1 ORDER BY array_position($2, index) LIMIT 1", self.id, self.index, as: String) || "-----------"
|
@thumbnail_id ||= PG_DB.query_one?("SELECT id FROM playlist_videos WHERE plid = $1 ORDER BY array_position($2, index) LIMIT 1", self.id, self.index, as: String) || "-----------"
|
||||||
"/vi/#{@thumbnail_id}/mqdefault.jpg"
|
"/vi/#{@thumbnail_id}/mqdefault.jpg"
|
||||||
@ -261,7 +264,7 @@ end
|
|||||||
def create_playlist(db, title, privacy, user)
|
def create_playlist(db, title, privacy, user)
|
||||||
plid = "IVPL#{Random::Secure.urlsafe_base64(24)[0, 31]}"
|
plid = "IVPL#{Random::Secure.urlsafe_base64(24)[0, 31]}"
|
||||||
|
|
||||||
playlist = InvidiousPlaylist.new(
|
playlist = InvidiousPlaylist.new({
|
||||||
title: title.byte_slice(0, 150),
|
title: title.byte_slice(0, 150),
|
||||||
id: plid,
|
id: plid,
|
||||||
author: user.email,
|
author: user.email,
|
||||||
@ -271,7 +274,7 @@ def create_playlist(db, title, privacy, user)
|
|||||||
updated: Time.utc,
|
updated: Time.utc,
|
||||||
privacy: privacy,
|
privacy: privacy,
|
||||||
index: [] of Int64,
|
index: [] of Int64,
|
||||||
)
|
})
|
||||||
|
|
||||||
playlist_array = playlist.to_a
|
playlist_array = playlist.to_a
|
||||||
args = arg_array(playlist_array)
|
args = arg_array(playlist_array)
|
||||||
@ -282,7 +285,7 @@ def create_playlist(db, title, privacy, user)
|
|||||||
end
|
end
|
||||||
|
|
||||||
def subscribe_playlist(db, user, playlist)
|
def subscribe_playlist(db, user, playlist)
|
||||||
playlist = InvidiousPlaylist.new(
|
playlist = InvidiousPlaylist.new({
|
||||||
title: playlist.title.byte_slice(0, 150),
|
title: playlist.title.byte_slice(0, 150),
|
||||||
id: playlist.id,
|
id: playlist.id,
|
||||||
author: user.email,
|
author: user.email,
|
||||||
@ -292,7 +295,7 @@ def subscribe_playlist(db, user, playlist)
|
|||||||
updated: playlist.updated,
|
updated: playlist.updated,
|
||||||
privacy: PlaylistPrivacy::Private,
|
privacy: PlaylistPrivacy::Private,
|
||||||
index: [] of Int64,
|
index: [] of Int64,
|
||||||
)
|
})
|
||||||
|
|
||||||
playlist_array = playlist.to_a
|
playlist_array = playlist.to_a
|
||||||
args = arg_array(playlist_array)
|
args = arg_array(playlist_array)
|
||||||
@ -393,7 +396,7 @@ def fetch_playlist(plid, locale)
|
|||||||
author = author_info["title"]["runs"][0]["text"]?.try &.as_s || ""
|
author = author_info["title"]["runs"][0]["text"]?.try &.as_s || ""
|
||||||
ucid = author_info["title"]["runs"][0]["navigationEndpoint"]["browseEndpoint"]["browseId"]?.try &.as_s || ""
|
ucid = author_info["title"]["runs"][0]["navigationEndpoint"]["browseEndpoint"]["browseId"]?.try &.as_s || ""
|
||||||
|
|
||||||
return Playlist.new(
|
return Playlist.new({
|
||||||
title: title,
|
title: title,
|
||||||
id: plid,
|
id: plid,
|
||||||
author: author,
|
author: author,
|
||||||
@ -403,8 +406,8 @@ def fetch_playlist(plid, locale)
|
|||||||
video_count: video_count,
|
video_count: video_count,
|
||||||
views: views,
|
views: views,
|
||||||
updated: updated,
|
updated: updated,
|
||||||
thumbnail: thumbnail
|
thumbnail: thumbnail,
|
||||||
)
|
})
|
||||||
end
|
end
|
||||||
|
|
||||||
def get_playlist_videos(db, playlist, offset, locale = nil, continuation = nil)
|
def get_playlist_videos(db, playlist, offset, locale = nil, continuation = nil)
|
||||||
@ -471,7 +474,7 @@ def extract_playlist_videos(initial_data : Hash(String, JSON::Any))
|
|||||||
length_seconds = 0
|
length_seconds = 0
|
||||||
end
|
end
|
||||||
|
|
||||||
videos << PlaylistVideo.new(
|
videos << PlaylistVideo.new({
|
||||||
title: title,
|
title: title,
|
||||||
id: video_id,
|
id: video_id,
|
||||||
author: author,
|
author: author,
|
||||||
@ -480,8 +483,8 @@ def extract_playlist_videos(initial_data : Hash(String, JSON::Any))
|
|||||||
published: Time.utc,
|
published: Time.utc,
|
||||||
plid: plid,
|
plid: plid,
|
||||||
live_now: live,
|
live_now: live,
|
||||||
index: index - 1
|
index: index - 1,
|
||||||
)
|
})
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
@ -1,4 +1,19 @@
|
|||||||
struct SearchVideo
|
struct SearchVideo
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property title : String
|
||||||
|
property id : String
|
||||||
|
property author : String
|
||||||
|
property ucid : String
|
||||||
|
property published : Time
|
||||||
|
property views : Int64
|
||||||
|
property description_html : String
|
||||||
|
property length_seconds : Int32
|
||||||
|
property live_now : Bool
|
||||||
|
property paid : Bool
|
||||||
|
property premium : Bool
|
||||||
|
property premiere_timestamp : Time?
|
||||||
|
|
||||||
def to_xml(auto_generated, query_params, xml : XML::Builder)
|
def to_xml(auto_generated, query_params, xml : XML::Builder)
|
||||||
query_params["v"] = self.id
|
query_params["v"] = self.id
|
||||||
|
|
||||||
@ -99,32 +114,27 @@ struct SearchVideo
|
|||||||
def is_upcoming
|
def is_upcoming
|
||||||
premiere_timestamp ? true : false
|
premiere_timestamp ? true : false
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
title: String,
|
|
||||||
id: String,
|
|
||||||
author: String,
|
|
||||||
ucid: String,
|
|
||||||
published: Time,
|
|
||||||
views: Int64,
|
|
||||||
description_html: String,
|
|
||||||
length_seconds: Int32,
|
|
||||||
live_now: Bool,
|
|
||||||
paid: Bool,
|
|
||||||
premium: Bool,
|
|
||||||
premiere_timestamp: Time?,
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct SearchPlaylistVideo
|
struct SearchPlaylistVideo
|
||||||
db_mapping({
|
include DB::Serializable
|
||||||
title: String,
|
|
||||||
id: String,
|
property title : String
|
||||||
length_seconds: Int32,
|
property id : String
|
||||||
})
|
property length_seconds : Int32
|
||||||
end
|
end
|
||||||
|
|
||||||
struct SearchPlaylist
|
struct SearchPlaylist
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property title : String
|
||||||
|
property id : String
|
||||||
|
property author : String
|
||||||
|
property ucid : String
|
||||||
|
property video_count : Int32
|
||||||
|
property videos : Array(SearchPlaylistVideo)
|
||||||
|
property thumbnail : String?
|
||||||
|
|
||||||
def to_json(locale, json : JSON::Builder)
|
def to_json(locale, json : JSON::Builder)
|
||||||
json.object do
|
json.object do
|
||||||
json.field "type", "playlist"
|
json.field "type", "playlist"
|
||||||
@ -164,19 +174,19 @@ struct SearchPlaylist
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
title: String,
|
|
||||||
id: String,
|
|
||||||
author: String,
|
|
||||||
ucid: String,
|
|
||||||
video_count: Int32,
|
|
||||||
videos: Array(SearchPlaylistVideo),
|
|
||||||
thumbnail: String?,
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct SearchChannel
|
struct SearchChannel
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property author : String
|
||||||
|
property ucid : String
|
||||||
|
property author_thumbnail : String
|
||||||
|
property subscriber_count : Int32
|
||||||
|
property video_count : Int32
|
||||||
|
property description_html : String
|
||||||
|
property auto_generated : Bool
|
||||||
|
|
||||||
def to_json(locale, json : JSON::Builder)
|
def to_json(locale, json : JSON::Builder)
|
||||||
json.object do
|
json.object do
|
||||||
json.field "type", "channel"
|
json.field "type", "channel"
|
||||||
@ -216,16 +226,6 @@ struct SearchChannel
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
author: String,
|
|
||||||
ucid: String,
|
|
||||||
author_thumbnail: String,
|
|
||||||
subscriber_count: Int32,
|
|
||||||
video_count: Int32,
|
|
||||||
description_html: String,
|
|
||||||
auto_generated: Bool,
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
alias SearchItem = SearchVideo | SearchChannel | SearchPlaylist
|
alias SearchItem = SearchVideo | SearchChannel | SearchPlaylist
|
||||||
|
@ -4,6 +4,20 @@ require "crypto/bcrypt/password"
|
|||||||
MATERIALIZED_VIEW_SQL = ->(email : String) { "SELECT cv.* FROM channel_videos cv WHERE EXISTS (SELECT subscriptions FROM users u WHERE cv.ucid = ANY (u.subscriptions) AND u.email = E'#{email.gsub({'\'' => "\\'", '\\' => "\\\\"})}') ORDER BY published DESC" }
|
MATERIALIZED_VIEW_SQL = ->(email : String) { "SELECT cv.* FROM channel_videos cv WHERE EXISTS (SELECT subscriptions FROM users u WHERE cv.ucid = ANY (u.subscriptions) AND u.email = E'#{email.gsub({'\'' => "\\'", '\\' => "\\\\"})}') ORDER BY published DESC" }
|
||||||
|
|
||||||
struct User
|
struct User
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property updated : Time
|
||||||
|
property notifications : Array(String)
|
||||||
|
property subscriptions : Array(String)
|
||||||
|
property email : String
|
||||||
|
|
||||||
|
@[DB::Field(converter: User::PreferencesConverter)]
|
||||||
|
property preferences : Preferences
|
||||||
|
property password : String?
|
||||||
|
property token : String
|
||||||
|
property watched : Array(String)
|
||||||
|
property feed_needs_update : Bool?
|
||||||
|
|
||||||
module PreferencesConverter
|
module PreferencesConverter
|
||||||
def self.from_rs(rs)
|
def self.from_rs(rs)
|
||||||
begin
|
begin
|
||||||
@ -13,31 +27,78 @@ struct User
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
updated: Time,
|
|
||||||
notifications: Array(String),
|
|
||||||
subscriptions: Array(String),
|
|
||||||
email: String,
|
|
||||||
preferences: {
|
|
||||||
type: Preferences,
|
|
||||||
converter: PreferencesConverter,
|
|
||||||
},
|
|
||||||
password: String?,
|
|
||||||
token: String,
|
|
||||||
watched: Array(String),
|
|
||||||
feed_needs_update: Bool?,
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct Preferences
|
struct Preferences
|
||||||
module ProcessString
|
include JSON::Serializable
|
||||||
|
include YAML::Serializable
|
||||||
|
|
||||||
|
property annotations : Bool = CONFIG.default_user_preferences.annotations
|
||||||
|
property annotations_subscribed : Bool = CONFIG.default_user_preferences.annotations_subscribed
|
||||||
|
property autoplay : Bool = CONFIG.default_user_preferences.autoplay
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::StringToArray)]
|
||||||
|
@[YAML::Field(converter: Preferences::StringToArray)]
|
||||||
|
property captions : Array(String) = CONFIG.default_user_preferences.captions
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::StringToArray)]
|
||||||
|
@[YAML::Field(converter: Preferences::StringToArray)]
|
||||||
|
property comments : Array(String) = CONFIG.default_user_preferences.comments
|
||||||
|
property continue : Bool = CONFIG.default_user_preferences.continue
|
||||||
|
property continue_autoplay : Bool = CONFIG.default_user_preferences.continue_autoplay
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::BoolToString)]
|
||||||
|
@[YAML::Field(converter: Preferences::BoolToString)]
|
||||||
|
property dark_mode : String = CONFIG.default_user_preferences.dark_mode
|
||||||
|
property latest_only : Bool = CONFIG.default_user_preferences.latest_only
|
||||||
|
property listen : Bool = CONFIG.default_user_preferences.listen
|
||||||
|
property local : Bool = CONFIG.default_user_preferences.local
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::ProcessString)]
|
||||||
|
property locale : String = CONFIG.default_user_preferences.locale
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::ClampInt)]
|
||||||
|
property max_results : Int32 = CONFIG.default_user_preferences.max_results
|
||||||
|
property notifications_only : Bool = CONFIG.default_user_preferences.notifications_only
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::ProcessString)]
|
||||||
|
property player_style : String = CONFIG.default_user_preferences.player_style
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::ProcessString)]
|
||||||
|
property quality : String = CONFIG.default_user_preferences.quality
|
||||||
|
property default_home : String = CONFIG.default_user_preferences.default_home
|
||||||
|
property feed_menu : Array(String) = CONFIG.default_user_preferences.feed_menu
|
||||||
|
property related_videos : Bool = CONFIG.default_user_preferences.related_videos
|
||||||
|
|
||||||
|
@[JSON::Field(converter: Preferences::ProcessString)]
|
||||||
|
property sort : String = CONFIG.default_user_preferences.sort
|
||||||
|
property speed : Float32 = CONFIG.default_user_preferences.speed
|
||||||
|
property thin_mode : Bool = CONFIG.default_user_preferences.thin_mode
|
||||||
|
property unseen_only : Bool = CONFIG.default_user_preferences.unseen_only
|
||||||
|
property video_loop : Bool = CONFIG.default_user_preferences.video_loop
|
||||||
|
property volume : Int32 = CONFIG.default_user_preferences.volume
|
||||||
|
|
||||||
|
module BoolToString
|
||||||
def self.to_json(value : String, json : JSON::Builder)
|
def self.to_json(value : String, json : JSON::Builder)
|
||||||
json.string value
|
json.string value
|
||||||
end
|
end
|
||||||
|
|
||||||
def self.from_json(value : JSON::PullParser) : String
|
def self.from_json(value : JSON::PullParser) : String
|
||||||
HTML.escape(value.read_string[0, 100])
|
begin
|
||||||
|
result = value.read_string
|
||||||
|
|
||||||
|
if result.empty?
|
||||||
|
CONFIG.default_user_preferences.dark_mode
|
||||||
|
else
|
||||||
|
result
|
||||||
|
end
|
||||||
|
rescue ex
|
||||||
|
if value.read_bool
|
||||||
|
"dark"
|
||||||
|
else
|
||||||
|
"light"
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def self.to_yaml(value : String, yaml : YAML::Nodes::Builder)
|
def self.to_yaml(value : String, yaml : YAML::Nodes::Builder)
|
||||||
@ -45,7 +106,20 @@ struct Preferences
|
|||||||
end
|
end
|
||||||
|
|
||||||
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : String
|
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : String
|
||||||
HTML.escape(node.value[0, 100])
|
unless node.is_a?(YAML::Nodes::Scalar)
|
||||||
|
node.raise "Expected scalar, not #{node.class}"
|
||||||
|
end
|
||||||
|
|
||||||
|
case node.value
|
||||||
|
when "true"
|
||||||
|
"dark"
|
||||||
|
when "false"
|
||||||
|
"light"
|
||||||
|
when ""
|
||||||
|
CONFIG.default_user_preferences.dark_mode
|
||||||
|
else
|
||||||
|
node.value
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
@ -67,33 +141,130 @@ struct Preferences
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
json_mapping({
|
module FamilyConverter
|
||||||
annotations: {type: Bool, default: CONFIG.default_user_preferences.annotations},
|
def self.to_yaml(value : Socket::Family, yaml : YAML::Nodes::Builder)
|
||||||
annotations_subscribed: {type: Bool, default: CONFIG.default_user_preferences.annotations_subscribed},
|
case value
|
||||||
autoplay: {type: Bool, default: CONFIG.default_user_preferences.autoplay},
|
when Socket::Family::UNSPEC
|
||||||
captions: {type: Array(String), default: CONFIG.default_user_preferences.captions, converter: ConfigPreferences::StringToArray},
|
yaml.scalar nil
|
||||||
comments: {type: Array(String), default: CONFIG.default_user_preferences.comments, converter: ConfigPreferences::StringToArray},
|
when Socket::Family::INET
|
||||||
continue: {type: Bool, default: CONFIG.default_user_preferences.continue},
|
yaml.scalar "ipv4"
|
||||||
continue_autoplay: {type: Bool, default: CONFIG.default_user_preferences.continue_autoplay},
|
when Socket::Family::INET6
|
||||||
dark_mode: {type: String, default: CONFIG.default_user_preferences.dark_mode, converter: ConfigPreferences::BoolToString},
|
yaml.scalar "ipv6"
|
||||||
latest_only: {type: Bool, default: CONFIG.default_user_preferences.latest_only},
|
when Socket::Family::UNIX
|
||||||
listen: {type: Bool, default: CONFIG.default_user_preferences.listen},
|
raise "Invalid socket family #{value}"
|
||||||
local: {type: Bool, default: CONFIG.default_user_preferences.local},
|
end
|
||||||
locale: {type: String, default: CONFIG.default_user_preferences.locale, converter: ProcessString},
|
end
|
||||||
max_results: {type: Int32, default: CONFIG.default_user_preferences.max_results, converter: ClampInt},
|
|
||||||
notifications_only: {type: Bool, default: CONFIG.default_user_preferences.notifications_only},
|
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : Socket::Family
|
||||||
player_style: {type: String, default: CONFIG.default_user_preferences.player_style, converter: ProcessString},
|
if node.is_a?(YAML::Nodes::Scalar)
|
||||||
quality: {type: String, default: CONFIG.default_user_preferences.quality, converter: ProcessString},
|
case node.value.downcase
|
||||||
default_home: {type: String, default: CONFIG.default_user_preferences.default_home},
|
when "ipv4"
|
||||||
feed_menu: {type: Array(String), default: CONFIG.default_user_preferences.feed_menu},
|
Socket::Family::INET
|
||||||
related_videos: {type: Bool, default: CONFIG.default_user_preferences.related_videos},
|
when "ipv6"
|
||||||
sort: {type: String, default: CONFIG.default_user_preferences.sort, converter: ProcessString},
|
Socket::Family::INET6
|
||||||
speed: {type: Float32, default: CONFIG.default_user_preferences.speed},
|
else
|
||||||
thin_mode: {type: Bool, default: CONFIG.default_user_preferences.thin_mode},
|
Socket::Family::UNSPEC
|
||||||
unseen_only: {type: Bool, default: CONFIG.default_user_preferences.unseen_only},
|
end
|
||||||
video_loop: {type: Bool, default: CONFIG.default_user_preferences.video_loop},
|
else
|
||||||
volume: {type: Int32, default: CONFIG.default_user_preferences.volume},
|
node.raise "Expected scalar, not #{node.class}"
|
||||||
})
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
module ProcessString
|
||||||
|
def self.to_json(value : String, json : JSON::Builder)
|
||||||
|
json.string value
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.from_json(value : JSON::PullParser) : String
|
||||||
|
HTML.escape(value.read_string[0, 100])
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.to_yaml(value : String, yaml : YAML::Nodes::Builder)
|
||||||
|
yaml.scalar value
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : String
|
||||||
|
HTML.escape(node.value[0, 100])
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
module StringToArray
|
||||||
|
def self.to_json(value : Array(String), json : JSON::Builder)
|
||||||
|
json.array do
|
||||||
|
value.each do |element|
|
||||||
|
json.string element
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.from_json(value : JSON::PullParser) : Array(String)
|
||||||
|
begin
|
||||||
|
result = [] of String
|
||||||
|
value.read_array do
|
||||||
|
result << HTML.escape(value.read_string[0, 100])
|
||||||
|
end
|
||||||
|
rescue ex
|
||||||
|
result = [HTML.escape(value.read_string[0, 100]), ""]
|
||||||
|
end
|
||||||
|
|
||||||
|
result
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.to_yaml(value : Array(String), yaml : YAML::Nodes::Builder)
|
||||||
|
yaml.sequence do
|
||||||
|
value.each do |element|
|
||||||
|
yaml.scalar element
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : Array(String)
|
||||||
|
begin
|
||||||
|
unless node.is_a?(YAML::Nodes::Sequence)
|
||||||
|
node.raise "Expected sequence, not #{node.class}"
|
||||||
|
end
|
||||||
|
|
||||||
|
result = [] of String
|
||||||
|
node.nodes.each do |item|
|
||||||
|
unless item.is_a?(YAML::Nodes::Scalar)
|
||||||
|
node.raise "Expected scalar, not #{item.class}"
|
||||||
|
end
|
||||||
|
|
||||||
|
result << HTML.escape(item.value[0, 100])
|
||||||
|
end
|
||||||
|
rescue ex
|
||||||
|
if node.is_a?(YAML::Nodes::Scalar)
|
||||||
|
result = [HTML.escape(node.value[0, 100]), ""]
|
||||||
|
else
|
||||||
|
result = ["", ""]
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
result
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
module StringToCookies
|
||||||
|
def self.to_yaml(value : HTTP::Cookies, yaml : YAML::Nodes::Builder)
|
||||||
|
(value.map { |c| "#{c.name}=#{c.value}" }).join("; ").to_yaml(yaml)
|
||||||
|
end
|
||||||
|
|
||||||
|
def self.from_yaml(ctx : YAML::ParseContext, node : YAML::Nodes::Node) : HTTP::Cookies
|
||||||
|
unless node.is_a?(YAML::Nodes::Scalar)
|
||||||
|
node.raise "Expected scalar, not #{node.class}"
|
||||||
|
end
|
||||||
|
|
||||||
|
cookies = HTTP::Cookies.new
|
||||||
|
node.value.split(";").each do |cookie|
|
||||||
|
next if cookie.strip.empty?
|
||||||
|
name, value = cookie.split("=", 2)
|
||||||
|
cookies << HTTP::Cookie.new(name.strip, value.strip)
|
||||||
|
end
|
||||||
|
|
||||||
|
cookies
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
def get_user(sid, headers, db, refresh = true)
|
def get_user(sid, headers, db, refresh = true)
|
||||||
@ -103,8 +274,7 @@ def get_user(sid, headers, db, refresh = true)
|
|||||||
if refresh && Time.utc - user.updated > 1.minute
|
if refresh && Time.utc - user.updated > 1.minute
|
||||||
user, sid = fetch_user(sid, headers, db)
|
user, sid = fetch_user(sid, headers, db)
|
||||||
user_array = user.to_a
|
user_array = user.to_a
|
||||||
|
user_array[4] = user_array[4].to_json # User preferences
|
||||||
user_array[4] = user_array[4].to_json
|
|
||||||
args = arg_array(user_array)
|
args = arg_array(user_array)
|
||||||
|
|
||||||
db.exec("INSERT INTO users VALUES (#{args}) \
|
db.exec("INSERT INTO users VALUES (#{args}) \
|
||||||
@ -122,8 +292,7 @@ def get_user(sid, headers, db, refresh = true)
|
|||||||
else
|
else
|
||||||
user, sid = fetch_user(sid, headers, db)
|
user, sid = fetch_user(sid, headers, db)
|
||||||
user_array = user.to_a
|
user_array = user.to_a
|
||||||
|
user_array[4] = user_array[4].to_json # User preferences
|
||||||
user_array[4] = user_array[4].to_json
|
|
||||||
args = arg_array(user.to_a)
|
args = arg_array(user.to_a)
|
||||||
|
|
||||||
db.exec("INSERT INTO users VALUES (#{args}) \
|
db.exec("INSERT INTO users VALUES (#{args}) \
|
||||||
@ -166,7 +335,17 @@ def fetch_user(sid, headers, db)
|
|||||||
|
|
||||||
token = Base64.urlsafe_encode(Random::Secure.random_bytes(32))
|
token = Base64.urlsafe_encode(Random::Secure.random_bytes(32))
|
||||||
|
|
||||||
user = User.new(Time.utc, [] of String, channels, email, CONFIG.default_user_preferences, nil, token, [] of String, true)
|
user = User.new({
|
||||||
|
updated: Time.utc,
|
||||||
|
notifications: [] of String,
|
||||||
|
subscriptions: channels,
|
||||||
|
email: email,
|
||||||
|
preferences: Preferences.new(CONFIG.default_user_preferences.to_tuple),
|
||||||
|
password: nil,
|
||||||
|
token: token,
|
||||||
|
watched: [] of String,
|
||||||
|
feed_needs_update: true,
|
||||||
|
})
|
||||||
return user, sid
|
return user, sid
|
||||||
end
|
end
|
||||||
|
|
||||||
@ -174,7 +353,17 @@ def create_user(sid, email, password)
|
|||||||
password = Crypto::Bcrypt::Password.create(password, cost: 10)
|
password = Crypto::Bcrypt::Password.create(password, cost: 10)
|
||||||
token = Base64.urlsafe_encode(Random::Secure.random_bytes(32))
|
token = Base64.urlsafe_encode(Random::Secure.random_bytes(32))
|
||||||
|
|
||||||
user = User.new(Time.utc, [] of String, [] of String, email, CONFIG.default_user_preferences, password.to_s, token, [] of String, true)
|
user = User.new({
|
||||||
|
updated: Time.utc,
|
||||||
|
notifications: [] of String,
|
||||||
|
subscriptions: [] of String,
|
||||||
|
email: email,
|
||||||
|
preferences: Preferences.new(CONFIG.default_user_preferences.to_tuple),
|
||||||
|
password: password.to_s,
|
||||||
|
token: token,
|
||||||
|
watched: [] of String,
|
||||||
|
feed_needs_update: true,
|
||||||
|
})
|
||||||
|
|
||||||
return user, sid
|
return user, sid
|
||||||
end
|
end
|
||||||
@ -281,48 +470,6 @@ def subscribe_ajax(channel_id, action, env_headers)
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
# TODO: Playlist stub, sync with YouTube for Google accounts
|
|
||||||
# def playlist_ajax(video_ids, source_playlist_id, name, privacy, action, env_headers)
|
|
||||||
# headers = HTTP::Headers.new
|
|
||||||
# headers["Cookie"] = env_headers["Cookie"]
|
|
||||||
#
|
|
||||||
# html = YT_POOL.client &.get("/view_all_playlists", headers)
|
|
||||||
#
|
|
||||||
# cookies = HTTP::Cookies.from_headers(headers)
|
|
||||||
# html.cookies.each do |cookie|
|
|
||||||
# if {"VISITOR_INFO1_LIVE", "YSC", "SIDCC"}.includes? cookie.name
|
|
||||||
# if cookies[cookie.name]?
|
|
||||||
# cookies[cookie.name] = cookie
|
|
||||||
# else
|
|
||||||
# cookies << cookie
|
|
||||||
# end
|
|
||||||
# end
|
|
||||||
# end
|
|
||||||
# headers = cookies.add_request_headers(headers)
|
|
||||||
#
|
|
||||||
# if match = html.body.match(/'XSRF_TOKEN': "(?<session_token>[^"]+)"/)
|
|
||||||
# session_token = match["session_token"]
|
|
||||||
#
|
|
||||||
# headers["content-type"] = "application/x-www-form-urlencoded"
|
|
||||||
#
|
|
||||||
# post_req = {
|
|
||||||
# video_ids: [] of String,
|
|
||||||
# source_playlist_id: "",
|
|
||||||
# n: name,
|
|
||||||
# p: privacy,
|
|
||||||
# session_token: session_token,
|
|
||||||
# }
|
|
||||||
# post_url = "/playlist_ajax?#{action}=1"
|
|
||||||
#
|
|
||||||
# response = client.post(post_url, headers, form: post_req)
|
|
||||||
# if response.status_code == 200
|
|
||||||
# return JSON.parse(response.body)["result"]["playlistId"].as_s
|
|
||||||
# else
|
|
||||||
# return nil
|
|
||||||
# end
|
|
||||||
# end
|
|
||||||
# end
|
|
||||||
|
|
||||||
def get_subscription_feed(db, user, max_results = 40, page = 1)
|
def get_subscription_feed(db, user, max_results = 40, page = 1)
|
||||||
limit = max_results.clamp(0, MAX_ITEMS_PER_PAGE)
|
limit = max_results.clamp(0, MAX_ITEMS_PER_PAGE)
|
||||||
offset = (page - 1) * limit
|
offset = (page - 1) * limit
|
||||||
|
@ -222,30 +222,50 @@ VIDEO_FORMATS = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
struct VideoPreferences
|
struct VideoPreferences
|
||||||
json_mapping({
|
include JSON::Serializable
|
||||||
annotations: Bool,
|
|
||||||
autoplay: Bool,
|
property annotations : Bool
|
||||||
comments: Array(String),
|
property autoplay : Bool
|
||||||
continue: Bool,
|
property comments : Array(String)
|
||||||
continue_autoplay: Bool,
|
property continue : Bool
|
||||||
controls: Bool,
|
property continue_autoplay : Bool
|
||||||
listen: Bool,
|
property controls : Bool
|
||||||
local: Bool,
|
property listen : Bool
|
||||||
preferred_captions: Array(String),
|
property local : Bool
|
||||||
player_style: String,
|
property preferred_captions : Array(String)
|
||||||
quality: String,
|
property player_style : String
|
||||||
raw: Bool,
|
property quality : String
|
||||||
region: String?,
|
property raw : Bool
|
||||||
related_videos: Bool,
|
property region : String?
|
||||||
speed: (Float32 | Float64),
|
property related_videos : Bool
|
||||||
video_end: (Float64 | Int32),
|
property speed : Float32 | Float64
|
||||||
video_loop: Bool,
|
property video_end : Float64 | Int32
|
||||||
video_start: (Float64 | Int32),
|
property video_loop : Bool
|
||||||
volume: Int32,
|
property video_start : Float64 | Int32
|
||||||
})
|
property volume : Int32
|
||||||
end
|
end
|
||||||
|
|
||||||
struct Video
|
struct Video
|
||||||
|
include DB::Serializable
|
||||||
|
|
||||||
|
property id : String
|
||||||
|
|
||||||
|
@[DB::Field(converter: Video::JSONConverter)]
|
||||||
|
property info : Hash(String, JSON::Any)
|
||||||
|
property updated : Time
|
||||||
|
|
||||||
|
@[DB::Field(ignore: true)]
|
||||||
|
property captions : Array(Caption)?
|
||||||
|
|
||||||
|
@[DB::Field(ignore: true)]
|
||||||
|
property adaptive_fmts : Array(Hash(String, JSON::Any))?
|
||||||
|
|
||||||
|
@[DB::Field(ignore: true)]
|
||||||
|
property fmt_stream : Array(Hash(String, JSON::Any))?
|
||||||
|
|
||||||
|
@[DB::Field(ignore: true)]
|
||||||
|
property description : String?
|
||||||
|
|
||||||
module JSONConverter
|
module JSONConverter
|
||||||
def self.from_rs(rs)
|
def self.from_rs(rs)
|
||||||
JSON.parse(rs.read(String)).as_h
|
JSON.parse(rs.read(String)).as_h
|
||||||
@ -552,6 +572,7 @@ struct Video
|
|||||||
|
|
||||||
def fmt_stream
|
def fmt_stream
|
||||||
return @fmt_stream.as(Array(Hash(String, JSON::Any))) if @fmt_stream
|
return @fmt_stream.as(Array(Hash(String, JSON::Any))) if @fmt_stream
|
||||||
|
|
||||||
fmt_stream = info["streamingData"]?.try &.["formats"]?.try &.as_a.map &.as_h || [] of Hash(String, JSON::Any)
|
fmt_stream = info["streamingData"]?.try &.["formats"]?.try &.as_a.map &.as_h || [] of Hash(String, JSON::Any)
|
||||||
fmt_stream.each do |fmt|
|
fmt_stream.each do |fmt|
|
||||||
if s = (fmt["cipher"]? || fmt["signatureCipher"]?).try { |h| HTTP::Params.parse(h.as_s) }
|
if s = (fmt["cipher"]? || fmt["signatureCipher"]?).try { |h| HTTP::Params.parse(h.as_s) }
|
||||||
@ -751,30 +772,20 @@ struct Video
|
|||||||
def session_token : String?
|
def session_token : String?
|
||||||
info["sessionToken"]?.try &.as_s?
|
info["sessionToken"]?.try &.as_s?
|
||||||
end
|
end
|
||||||
|
|
||||||
db_mapping({
|
|
||||||
id: String,
|
|
||||||
info: {type: Hash(String, JSON::Any), converter: Video::JSONConverter},
|
|
||||||
updated: Time,
|
|
||||||
})
|
|
||||||
|
|
||||||
@captions : Array(Caption)?
|
|
||||||
@adaptive_fmts : Array(Hash(String, JSON::Any))?
|
|
||||||
@fmt_stream : Array(Hash(String, JSON::Any))?
|
|
||||||
end
|
|
||||||
|
|
||||||
struct Caption
|
|
||||||
json_mapping({
|
|
||||||
name: CaptionName,
|
|
||||||
baseUrl: String,
|
|
||||||
languageCode: String,
|
|
||||||
})
|
|
||||||
end
|
end
|
||||||
|
|
||||||
struct CaptionName
|
struct CaptionName
|
||||||
json_mapping({
|
include JSON::Serializable
|
||||||
simpleText: String,
|
|
||||||
})
|
property simpleText : String
|
||||||
|
end
|
||||||
|
|
||||||
|
struct Caption
|
||||||
|
include JSON::Serializable
|
||||||
|
|
||||||
|
property name : CaptionName
|
||||||
|
property baseUrl : String
|
||||||
|
property languageCode : String
|
||||||
end
|
end
|
||||||
|
|
||||||
class VideoRedirect < Exception
|
class VideoRedirect < Exception
|
||||||
@ -990,7 +1001,12 @@ def fetch_video(id, region)
|
|||||||
|
|
||||||
raise info["reason"]?.try &.as_s || "" if !info["videoDetails"]?
|
raise info["reason"]?.try &.as_s || "" if !info["videoDetails"]?
|
||||||
|
|
||||||
video = Video.new(id, info, Time.utc)
|
video = Video.new({
|
||||||
|
id: id,
|
||||||
|
info: info,
|
||||||
|
updated: Time.utc,
|
||||||
|
})
|
||||||
|
|
||||||
return video
|
return video
|
||||||
end
|
end
|
||||||
|
|
||||||
@ -1097,7 +1113,7 @@ def process_video_params(query, preferences)
|
|||||||
controls ||= 1
|
controls ||= 1
|
||||||
controls = controls >= 1
|
controls = controls >= 1
|
||||||
|
|
||||||
params = VideoPreferences.new(
|
params = VideoPreferences.new({
|
||||||
annotations: annotations,
|
annotations: annotations,
|
||||||
autoplay: autoplay,
|
autoplay: autoplay,
|
||||||
comments: comments,
|
comments: comments,
|
||||||
@ -1117,7 +1133,7 @@ def process_video_params(query, preferences)
|
|||||||
video_loop: video_loop,
|
video_loop: video_loop,
|
||||||
video_start: video_start,
|
video_start: video_start,
|
||||||
volume: volume,
|
volume: volume,
|
||||||
)
|
})
|
||||||
|
|
||||||
return params
|
return params
|
||||||
end
|
end
|
||||||
|
Loading…
Reference in New Issue
Block a user