2018-09-04 19:52:10 +05:30
# "Invidious" (which is an alternative front-end to YouTube)
2019-03-15 22:14:53 +05:30
# Copyright (C) 2019 Omar Roth
2018-01-28 23:02:40 +05:30
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
2018-11-23 00:56:08 +05:30
require " digest/md5 "
2019-01-24 01:45:19 +05:30
require " file_utils "
2017-11-23 13:18:55 +05:30
require " kemal "
2018-07-19 00:56:02 +05:30
require " openssl/hmac "
2018-02-04 03:43:14 +05:30
require " option_parser "
2017-11-24 09:36:43 +05:30
require " pg "
2018-11-22 04:42:13 +05:30
require " sqlite3 "
2018-01-17 01:32:35 +05:30
require " xml "
2018-03-10 00:12:23 +05:30
require " yaml "
2020-06-16 04:27:20 +05:30
require " compress/zip "
2019-10-27 23:20:42 +05:30
require " protodec/utils "
2018-08-05 02:00:44 +05:30
require " ./invidious/helpers/* "
2018-07-06 18:29:56 +05:30
require " ./invidious/* "
2021-07-14 21:16:12 +05:30
require " ./invidious/channels/* "
2020-10-06 10:11:18 +05:30
require " ./invidious/routes/** "
require " ./invidious/jobs/** "
2017-11-30 03:03:46 +05:30
2021-01-23 23:28:13 +05:30
CONFIG = Config . load
HMAC_KEY = CONFIG . hmac_key || Random :: Secure . hex ( 32 )
2018-03-10 00:12:23 +05:30
2021-01-30 20:22:48 +05:30
PG_DB = DB . open CONFIG . database_url
2019-06-23 19:09:14 +05:30
ARCHIVE_URL = URI . parse ( " https://archive.org " )
LOGIN_URL = URI . parse ( " https://accounts.google.com " )
PUBSUB_URL = URI . parse ( " https://pubsubhubbub.appspot.com " )
REDDIT_URL = URI . parse ( " https://www.reddit.com " )
2020-03-10 20:42:11 +05:30
TEXTCAPTCHA_URL = URI . parse ( " https://textcaptcha.com " )
2019-06-23 19:09:14 +05:30
YT_URL = URI . parse ( " https://www.youtube.com " )
2021-01-24 00:09:04 +05:30
HOST_URL = make_host_url ( Kemal . config )
2019-06-23 19:09:14 +05:30
2019-06-07 23:09:12 +05:30
CHARS_SAFE = " ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_ "
TEST_IDS = { " AgbeGFYluEA " , " BaW_jenozKc " , " a9LDPn-MO4I " , " ddFvjfvPnqk " , " iqKdEhx-dD4 " }
2019-06-09 02:34:55 +05:30
MAX_ITEMS_PER_PAGE = 1500
2018-03-05 09:55:03 +05:30
2019-11-25 00:11:47 +05:30
REQUEST_HEADERS_WHITELIST = { " accept " , " accept-encoding " , " cache-control " , " content-length " , " if-none-match " , " range " }
RESPONSE_HEADERS_BLACKLIST = { " access-control-allow-origin " , " alt-svc " , " server " }
2019-07-05 02:00:00 +05:30
HTTP_CHUNK_SIZE = 10485760 # ~10MB
2019-06-23 19:09:14 +05:30
2020-02-16 00:22:28 +05:30
CURRENT_BRANCH = {{ " #{ ` git branch | sed -n '/* /s///p' ` . strip } " }}
2019-06-23 19:09:14 +05:30
CURRENT_COMMIT = {{ " #{ ` git rev-list HEAD --max-count=1 --abbrev-commit ` . strip } " }}
2020-12-06 00:36:24 +05:30
CURRENT_VERSION = {{ " #{ ` git log -1 --format=%ci | awk '{print $1}' | sed s/-/./g ` . strip } " }}
2019-06-23 19:09:14 +05:30
2019-05-09 22:22:37 +05:30
# This is used to determine the `?v=` on the end of file URLs (for cache busting). We
# only need to expire modified assets, so we can use this to find the last commit that changes
# any assets
ASSET_COMMIT = {{ " #{ ` git rev-list HEAD --max-count=1 --abbrev-commit -- assets ` . strip } " }}
2019-04-06 18:58:53 +05:30
SOFTWARE = {
" name " = > " invidious " ,
" version " = > " #{ CURRENT_VERSION } - #{ CURRENT_COMMIT } " ,
" branch " = > " #{ CURRENT_BRANCH } " ,
}
2021-04-04 01:41:35 +05:30
YT_POOL = YoutubeConnectionPool . new ( YT_URL , capacity : CONFIG . pool_size , timeout : 2.0 , use_quic : CONFIG . use_quic )
2019-10-25 22:28:16 +05:30
2021-01-04 21:21:06 +05:30
# CLI
2019-04-06 18:58:53 +05:30
Kemal . config . extra_options do | parser |
parser . banner = " Usage: invidious [arguments] "
2021-01-04 21:21:06 +05:30
parser . on ( " -c THREADS " , " --channel-threads=THREADS " , " Number of threads for refreshing channels (default: #{ CONFIG . channel_threads } ) " ) do | number |
2019-04-06 18:58:53 +05:30
begin
2021-01-04 21:21:06 +05:30
CONFIG . channel_threads = number . to_i
2019-04-06 18:58:53 +05:30
rescue ex
puts " THREADS must be integer "
exit
end
end
2021-01-04 21:21:06 +05:30
parser . on ( " -f THREADS " , " --feed-threads=THREADS " , " Number of threads for refreshing feeds (default: #{ CONFIG . feed_threads } ) " ) do | number |
2019-04-06 18:58:53 +05:30
begin
2021-01-04 21:21:06 +05:30
CONFIG . feed_threads = number . to_i
2019-04-06 18:58:53 +05:30
rescue ex
puts " THREADS must be integer "
exit
end
end
2021-01-04 21:21:06 +05:30
parser . on ( " -o OUTPUT " , " --output=OUTPUT " , " Redirect output (default: #{ CONFIG . output } ) " ) do | output |
CONFIG . output = output
2019-04-06 18:58:53 +05:30
end
2021-01-04 21:21:06 +05:30
parser . on ( " -l LEVEL " , " --log-level=LEVEL " , " Log level, one of #{ LogLevel . values } (default: #{ CONFIG . log_level } ) " ) do | log_level |
CONFIG . log_level = LogLevel . parse ( log_level )
2020-12-21 20:35:35 +05:30
end
parser . on ( " -v " , " --version " , " Print version " ) do
2019-04-06 18:58:53 +05:30
puts SOFTWARE . to_pretty_json
exit
end
end
Kemal :: CLI . new ARGV
2021-01-04 21:21:06 +05:30
if CONFIG . output . upcase != " STDOUT "
FileUtils . mkdir_p ( File . dirname ( CONFIG . output ) )
2021-01-04 20:35:15 +05:30
end
2021-01-04 21:21:06 +05:30
OUTPUT = CONFIG . output . upcase == " STDOUT " ? STDOUT : File . open ( CONFIG . output , mode : " a " )
LOGGER = Invidious :: LogHandler . new ( OUTPUT , CONFIG . log_level )
2021-01-04 20:35:15 +05:30
2019-04-15 21:43:09 +05:30
# Check table integrity
2019-04-11 22:43:25 +05:30
if CONFIG . check_tables
2021-01-04 21:21:06 +05:30
check_enum ( PG_DB , " privacy " , PlaylistPrivacy )
2019-08-06 05:19:13 +05:30
2021-01-04 21:21:06 +05:30
check_table ( PG_DB , " channels " , InvidiousChannel )
check_table ( PG_DB , " channel_videos " , ChannelVideo )
check_table ( PG_DB , " playlists " , InvidiousPlaylist )
check_table ( PG_DB , " playlist_videos " , PlaylistVideo )
check_table ( PG_DB , " nonces " , Nonce )
check_table ( PG_DB , " session_ids " , SessionId )
check_table ( PG_DB , " users " , User )
check_table ( PG_DB , " videos " , Video )
2019-04-15 21:43:09 +05:30
if CONFIG . cache_annotations
2021-01-04 21:21:06 +05:30
check_table ( PG_DB , " annotations " , Annotation )
2019-04-15 21:43:09 +05:30
end
2019-04-11 22:43:25 +05:30
end
2018-03-26 08:48:29 +05:30
2019-04-11 02:53:37 +05:30
# Start jobs
2019-05-15 22:56:29 +05:30
2021-01-24 00:11:50 +05:30
if CONFIG . channel_threads > 0
Invidious :: Jobs . register Invidious :: Jobs :: RefreshChannelsJob . new ( PG_DB )
end
if CONFIG . feed_threads > 0
Invidious :: Jobs . register Invidious :: Jobs :: RefreshFeedsJob . new ( PG_DB )
end
2020-09-27 22:49:44 +05:30
DECRYPT_FUNCTION = DecryptFunction . new ( CONFIG . decrypt_polling )
2021-01-24 00:09:04 +05:30
if CONFIG . decrypt_polling
2021-01-04 21:21:06 +05:30
Invidious :: Jobs . register Invidious :: Jobs :: UpdateDecryptFunctionJob . new
2020-09-27 22:49:44 +05:30
end
2019-03-04 06:48:23 +05:30
2021-01-24 00:09:04 +05:30
if CONFIG . statistics_enabled
Invidious :: Jobs . register Invidious :: Jobs :: StatisticsRefreshJob . new ( PG_DB , SOFTWARE )
2020-10-17 17:55:57 +05:30
end
2019-08-27 18:38:26 +05:30
2021-01-24 00:09:04 +05:30
if ( CONFIG . use_pubsub_feeds . is_a? ( Bool ) && CONFIG . use_pubsub_feeds . as ( Bool ) ) || ( CONFIG . use_pubsub_feeds . is_a? ( Int32 ) && CONFIG . use_pubsub_feeds . as ( Int32 ) > 0 )
Invidious :: Jobs . register Invidious :: Jobs :: SubscribeToFeedsJob . new ( PG_DB , HMAC_KEY )
2021-01-08 00:45:26 +05:30
end
2021-01-24 00:09:04 +05:30
if CONFIG . popular_enabled
2020-12-27 10:42:43 +05:30
Invidious :: Jobs . register Invidious :: Jobs :: PullPopularVideosJob . new ( PG_DB )
end
2021-01-24 00:09:04 +05:30
if CONFIG . captcha_key
Invidious :: Jobs . register Invidious :: Jobs :: BypassCaptchaJob . new
2019-03-02 06:55:16 +05:30
end
2019-03-02 04:17:06 +05:30
2020-10-17 17:55:57 +05:30
connection_channel = Channel ( { Bool , Channel ( PQ :: Notification ) } ) . new ( 32 )
2021-01-30 20:22:48 +05:30
Invidious :: Jobs . register Invidious :: Jobs :: NotificationJob . new ( connection_channel , CONFIG . database_url )
2020-10-17 17:55:57 +05:30
2020-10-06 10:11:18 +05:30
Invidious :: Jobs . start_all
def popular_videos
Invidious :: Jobs :: PullPopularVideosJob :: POPULAR_VIDEOS . get
2018-11-09 07:38:03 +05:30
end
2018-03-25 09:26:41 +05:30
before_all do | env |
2020-10-16 15:53:18 +05:30
preferences = begin
2021-05-24 19:22:55 +05:30
Preferences . from_json ( URI . decode_www_form ( env . request . cookies [ " PREFS " ]? . try & . value || " {} " ) )
2020-03-16 03:16:08 +05:30
rescue
2020-10-16 15:53:18 +05:30
Preferences . from_json ( " {} " )
2020-03-16 03:16:08 +05:30
end
2020-10-16 15:53:18 +05:30
env . set " preferences " , preferences
2019-05-11 03:18:38 +05:30
env . response . headers [ " X-XSS-Protection " ] = " 1; mode=block "
2018-09-06 08:21:40 +05:30
env . response . headers [ " X-Content-Type-Options " ] = " nosniff "
2021-06-17 23:15:20 +05:30
2021-06-19 13:10:33 +05:30
# Allow media resources to be loaded from google servers
2021-06-17 23:15:20 +05:30
# TODO: check if *.youtube.com can be removed
2020-03-16 03:16:08 +05:30
if CONFIG . disabled? ( " local " ) || ! preferences . local
2021-06-17 23:15:20 +05:30
extra_media_csp = " https://*.googlevideo.com:443 https://*.youtube.com:443 "
else
extra_media_csp = " "
end
# Only allow the pages at /embed/* to be embedded
if env . request . resource . starts_with? ( " /embed " )
frame_ancestors = " 'self' http: https: "
else
2021-06-20 22:13:00 +05:30
frame_ancestors = " 'none' "
2020-03-16 03:16:08 +05:30
end
2021-06-17 23:15:20 +05:30
# TODO: Remove style-src's 'unsafe-inline', requires to remove all
# inline styles (<style> [..] </style>, style=" [..] ")
env . response . headers [ " Content-Security-Policy " ] = {
" default-src 'none' " ,
" script-src 'self' " ,
" style-src 'self' 'unsafe-inline' " ,
" img-src 'self' data: " ,
" font-src 'self' data: " ,
" connect-src 'self' " ,
" manifest-src 'self' " ,
" media-src 'self' blob: " + extra_media_csp ,
" child-src 'self' blob: " ,
" frame-src 'self' " ,
" frame-ancestors " + frame_ancestors ,
} . join ( " ; " )
2019-04-08 00:31:08 +05:30
env . response . headers [ " Referrer-Policy " ] = " same-origin "
2021-06-17 23:15:20 +05:30
# Ask the chrom*-based browsers to disable FLoC
# See: https://blog.runcloud.io/google-floc/
env . response . headers [ " Permissions-Policy " ] = " interest-cohort=() "
2021-01-24 00:09:04 +05:30
if ( Kemal . config . ssl || CONFIG . https_only ) && CONFIG . hsts
2019-05-01 07:23:56 +05:30
env . response . headers [ " Strict-Transport-Security " ] = " max-age=31536000; includeSubDomains; preload "
2019-04-08 00:31:08 +05:30
end
2019-03-29 00:13:40 +05:30
2019-11-20 22:33:52 +05:30
next if {
" /sb/ " ,
" /vi/ " ,
" /s_p/ " ,
" /yts/ " ,
" /ggpht/ " ,
" /api/manifest/ " ,
" /videoplayback " ,
" /latest_version " ,
} . any? { | r | env . request . resource . starts_with? r }
2018-07-16 21:54:24 +05:30
if env . request . cookies . has_key? " SID "
2018-04-01 05:39:27 +05:30
sid = env . request . cookies [ " SID " ] . value
2018-07-06 05:13:26 +05:30
2019-04-19 02:53:50 +05:30
if sid . starts_with? " v1: "
raise " Cannot use token as SID "
end
2018-07-19 00:56:02 +05:30
# Invidious users only have SID
if ! env . request . cookies . has_key? " SSID "
2019-04-16 09:53:40 +05:30
if email = PG_DB . query_one? ( " SELECT email FROM session_ids WHERE id = $1 " , sid , as : String )
2019-02-12 08:22:47 +05:30
user = PG_DB . query_one ( " SELECT * FROM users WHERE email = $1 " , email , as : User )
2019-08-06 05:19:13 +05:30
csrf_token = generate_response ( sid , {
" :authorize_token " ,
" :playlist_ajax " ,
" :signout " ,
" :subscription_ajax " ,
" :token_ajax " ,
" :watch_ajax " ,
} , HMAC_KEY , PG_DB , 1 . week )
2018-11-09 05:12:25 +05:30
2019-03-11 23:14:25 +05:30
preferences = user . preferences
2020-10-16 15:53:18 +05:30
env . set " preferences " , preferences
2019-03-11 23:14:25 +05:30
2018-08-15 23:10:42 +05:30
env . set " sid " , sid
2019-04-19 02:53:50 +05:30
env . set " csrf_token " , csrf_token
2019-04-16 09:53:40 +05:30
env . set " user " , user
2018-07-19 00:56:02 +05:30
end
else
2019-04-16 09:53:40 +05:30
headers = HTTP :: Headers . new
headers [ " Cookie " ] = env . request . headers [ " Cookie " ]
2018-07-19 00:56:02 +05:30
begin
2021-01-04 21:21:06 +05:30
user , sid = get_user ( sid , headers , PG_DB , false )
2019-08-06 05:19:13 +05:30
csrf_token = generate_response ( sid , {
" :authorize_token " ,
" :playlist_ajax " ,
" :signout " ,
" :subscription_ajax " ,
" :token_ajax " ,
" :watch_ajax " ,
} , HMAC_KEY , PG_DB , 1 . week )
2018-11-16 07:53:17 +05:30
2019-03-11 23:14:25 +05:30
preferences = user . preferences
2020-10-16 15:53:18 +05:30
env . set " preferences " , preferences
2019-03-11 23:14:25 +05:30
2018-08-15 23:10:42 +05:30
env . set " sid " , sid
2019-04-19 02:53:50 +05:30
env . set " csrf_token " , csrf_token
2019-04-16 09:53:40 +05:30
env . set " user " , user
2018-07-19 00:56:02 +05:30
rescue ex
end
2018-07-16 23:20:41 +05:30
end
2018-04-14 08:02:14 +05:30
end
2018-08-17 20:49:20 +05:30
2019-08-15 21:59:55 +05:30
dark_mode = convert_theme ( env . params . query [ " dark_mode " ]? ) || preferences . dark_mode . to_s
2019-03-11 23:14:25 +05:30
thin_mode = env . params . query [ " thin_mode " ]? || preferences . thin_mode . to_s
thin_mode = thin_mode == " true "
locale = env . params . query [ " hl " ]? || preferences . locale
preferences . dark_mode = dark_mode
preferences . thin_mode = thin_mode
preferences . locale = locale
2021-03-18 00:37:38 +05:30
env . set " preferences " , preferences
2018-12-21 03:02:09 +05:30
2018-08-17 20:49:20 +05:30
current_page = env . request . path
if env . request . query
query = HTTP :: Params . parse ( env . request . query . not_nil! )
if query [ " referer " ]?
query [ " referer " ] = get_referer ( env , " / " )
end
current_page += " ? #{ query } "
end
2019-09-24 23:01:33 +05:30
env . set " current_page " , URI . encode_www_form ( current_page )
2018-03-22 23:14:36 +05:30
end
2021-02-02 10:48:16 +05:30
Invidious :: Routing . get " / " , Invidious :: Routes :: Misc , :home
Invidious :: Routing . get " /privacy " , Invidious :: Routes :: Misc , :privacy
Invidious :: Routing . get " /licenses " , Invidious :: Routes :: Misc , :licenses
2021-04-07 08:37:19 +05:30
2021-08-04 03:16:15 +05:30
Invidious :: Routing . get " /channel/:ucid " , Invidious :: Routes :: Channels , :home
Invidious :: Routing . get " /channel/:ucid/home " , Invidious :: Routes :: Channels , :home
Invidious :: Routing . get " /channel/:ucid/videos " , Invidious :: Routes :: Channels , :videos
Invidious :: Routing . get " /channel/:ucid/playlists " , Invidious :: Routes :: Channels , :playlists
Invidious :: Routing . get " /channel/:ucid/community " , Invidious :: Routes :: Channels , :community
Invidious :: Routing . get " /channel/:ucid/about " , Invidious :: Routes :: Channels , :about
[ " " , " /videos " , " /playlists " , " /community " , " /about " ] . each do | path |
# /c/LinusTechTips
Invidious :: Routing . get " /c/:user #{ path } " , Invidious :: Routes :: Channels , :brand_redirect
# /user/linustechtips | Not always the same as /c/
Invidious :: Routing . get " /user/:user #{ path } " , Invidious :: Routes :: Channels , :brand_redirect
# /attribution_link?a=anything&u=/channel/UCZYTClx2T1of7BRZ86-8fow
Invidious :: Routing . get " /attribution_link #{ path } " , Invidious :: Routes :: Channels , :brand_redirect
# /profile?user=linustechtips
Invidious :: Routing . get " /profile/ #{ path } " , Invidious :: Routes :: Channels , :profile
end
2021-04-07 08:37:19 +05:30
Invidious :: Routing . get " /watch " , Invidious :: Routes :: Watch , :handle
Invidious :: Routing . get " /watch/:id " , Invidious :: Routes :: Watch , :redirect
Invidious :: Routing . get " /shorts/:id " , Invidious :: Routes :: Watch , :redirect
Invidious :: Routing . get " /w/:id " , Invidious :: Routes :: Watch , :redirect
Invidious :: Routing . get " /v/:id " , Invidious :: Routes :: Watch , :redirect
Invidious :: Routing . get " /e/:id " , Invidious :: Routes :: Watch , :redirect
2021-03-27 07:06:40 +05:30
Invidious :: Routing . get " /redirect " , Invidious :: Routes :: Misc , :cross_instance_redirect
2021-04-07 08:37:19 +05:30
2021-02-02 10:48:16 +05:30
Invidious :: Routing . get " /embed/ " , Invidious :: Routes :: Embed , :redirect
Invidious :: Routing . get " /embed/:id " , Invidious :: Routes :: Embed , :show
2021-04-07 08:37:19 +05:30
2020-11-20 06:51:48 +05:30
Invidious :: Routing . get " /view_all_playlists " , Invidious :: Routes :: Playlists , :index
Invidious :: Routing . get " /create_playlist " , Invidious :: Routes :: Playlists , :new
Invidious :: Routing . post " /create_playlist " , Invidious :: Routes :: Playlists , :create
Invidious :: Routing . get " /subscribe_playlist " , Invidious :: Routes :: Playlists , :subscribe
Invidious :: Routing . get " /delete_playlist " , Invidious :: Routes :: Playlists , :delete_page
Invidious :: Routing . post " /delete_playlist " , Invidious :: Routes :: Playlists , :delete
Invidious :: Routing . get " /edit_playlist " , Invidious :: Routes :: Playlists , :edit
Invidious :: Routing . post " /edit_playlist " , Invidious :: Routes :: Playlists , :update
Invidious :: Routing . get " /add_playlist_items " , Invidious :: Routes :: Playlists , :add_playlist_items_page
Invidious :: Routing . post " /playlist_ajax " , Invidious :: Routes :: Playlists , :playlist_ajax
Invidious :: Routing . get " /playlist " , Invidious :: Routes :: Playlists , :show
Invidious :: Routing . get " /mix " , Invidious :: Routes :: Playlists , :mix
2021-04-07 08:37:19 +05:30
2020-12-02 05:32:39 +05:30
Invidious :: Routing . get " /opensearch.xml " , Invidious :: Routes :: Search , :opensearch
Invidious :: Routing . get " /results " , Invidious :: Routes :: Search , :results
Invidious :: Routing . get " /search " , Invidious :: Routes :: Search , :search
2021-04-07 08:37:19 +05:30
2020-12-09 08:20:39 +05:30
Invidious :: Routing . get " /login " , Invidious :: Routes :: Login , :login_page
Invidious :: Routing . post " /login " , Invidious :: Routes :: Login , :login
Invidious :: Routing . post " /signout " , Invidious :: Routes :: Login , :signout
2021-04-07 08:37:19 +05:30
2021-02-02 10:48:16 +05:30
Invidious :: Routing . get " /preferences " , Invidious :: Routes :: PreferencesRoute , :show
Invidious :: Routing . post " /preferences " , Invidious :: Routes :: PreferencesRoute , :update
Invidious :: Routing . get " /toggle_theme " , Invidious :: Routes :: PreferencesRoute , :toggle_theme
2018-07-22 07:26:11 +05:30
2021-07-22 10:04:16 +05:30
define_v1_api_routes ( )
2021-08-13 12:01:12 +05:30
define_api_manifest_routes ( )
define_video_playback_routes ( )
2021-07-22 10:04:16 +05:30
2018-08-05 02:00:44 +05:30
# Users
2018-07-22 07:26:11 +05:30
2019-04-16 09:53:40 +05:30
post " /watch_ajax " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-11-20 09:36:59 +05:30
user = env . get? " user "
2019-04-16 09:53:40 +05:30
sid = env . get? " sid "
2018-11-20 09:36:59 +05:30
referer = get_referer ( env , " /feed/subscriptions " )
redirect = env . params . query [ " redirect " ]?
2019-04-16 09:53:40 +05:30
redirect || = " true "
2018-11-20 09:36:59 +05:30
redirect = redirect == " true "
2019-04-16 09:53:40 +05:30
if ! user
2019-04-19 02:53:50 +05:30
if redirect
next env . redirect referer
else
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " No such user " )
2019-04-19 02:53:50 +05:30
end
2018-11-20 09:36:59 +05:30
end
2018-12-21 03:02:09 +05:30
2019-04-16 09:53:40 +05:30
user = user . as ( User )
sid = sid . as ( String )
2019-04-19 02:53:50 +05:30
token = env . params . body [ " csrf_token " ]?
2018-11-20 09:36:59 +05:30
id = env . params . query [ " id " ]?
if ! id
2019-03-23 20:54:30 +05:30
env . response . status_code = 400
next
2018-11-20 09:36:59 +05:30
end
2019-04-16 09:53:40 +05:30
begin
2019-04-19 02:53:50 +05:30
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
2019-04-16 09:53:40 +05:30
rescue ex
if redirect
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2019-04-16 09:53:40 +05:30
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , ex )
2019-04-16 09:53:40 +05:30
end
end
if env . params . query [ " action_mark_watched " ]?
action = " action_mark_watched "
elsif env . params . query [ " action_mark_unwatched " ]?
action = " action_mark_unwatched "
else
next env . redirect referer
end
case action
when " action_mark_watched "
if ! user . watched . includes? id
2020-02-28 22:16:24 +05:30
PG_DB . exec ( " UPDATE users SET watched = array_append(watched, $1) WHERE email = $2 " , id , user . email )
2019-04-16 09:53:40 +05:30
end
when " action_mark_unwatched "
2018-11-22 04:42:13 +05:30
PG_DB . exec ( " UPDATE users SET watched = array_remove(watched, $1) WHERE email = $2 " , id , user . email )
2020-04-09 22:48:09 +05:30
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Unsupported action #{ action } " )
2018-11-20 09:36:59 +05:30
end
if redirect
env . redirect referer
else
env . response . content_type = " application/json "
" {} "
end
end
2018-08-05 09:37:38 +05:30
# /modify_notifications
# will "ding" all subscriptions.
2018-08-05 02:00:44 +05:30
# /modify_notifications?receive_all_updates=false&receive_no_updates=false
# will "unding" all subscriptions.
get " /modify_notifications " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-08-05 02:00:44 +05:30
user = env . get? " user "
2019-04-07 23:29:12 +05:30
sid = env . get? " sid "
referer = get_referer ( env , " / " )
2018-07-14 19:06:31 +05:30
2019-04-07 23:29:12 +05:30
redirect = env . params . query [ " redirect " ]?
redirect || = " false "
redirect = redirect == " true "
2018-07-30 07:35:40 +05:30
2019-04-19 02:53:50 +05:30
if ! user
if redirect
next env . redirect referer
else
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " No such user " )
2019-04-19 02:53:50 +05:30
end
2019-04-07 23:29:12 +05:30
end
user = user . as ( User )
if ! user . password
2018-08-05 02:00:44 +05:30
channel_req = { } of String = > String
2018-02-27 06:29:02 +05:30
2018-08-05 02:00:44 +05:30
channel_req [ " receive_all_updates " ] = env . params . query [ " receive_all_updates " ]? || " true "
channel_req [ " receive_no_updates " ] = env . params . query [ " receive_no_updates " ]? || " "
channel_req [ " receive_post_updates " ] = env . params . query [ " receive_post_updates " ]? || " true "
2018-01-07 23:12:24 +05:30
2018-08-05 02:00:44 +05:30
channel_req . reject! { | k , v | v != " true " && v != " false " }
2018-01-07 08:09:24 +05:30
2018-08-05 02:00:44 +05:30
headers = HTTP :: Headers . new
headers [ " Cookie " ] = env . request . headers [ " Cookie " ]
2017-12-31 02:51:43 +05:30
2019-10-25 22:28:16 +05:30
html = YT_POOL . client & . get ( " /subscription_manager?disable_polymer=1 " , headers )
2019-04-07 23:29:12 +05:30
2021-05-24 19:15:50 +05:30
cookies = HTTP :: Cookies . from_client_headers ( headers )
2019-04-07 23:29:12 +05:30
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 )
2020-06-16 04:03:23 +05:30
if match = html . body . match ( / 'XSRF_TOKEN': "(?<session_token>[^"]+)" / )
2018-08-05 02:00:44 +05:30
session_token = match [ " session_token " ]
else
next env . redirect referer
end
2018-07-19 00:56:02 +05:30
2019-04-07 23:29:12 +05:30
headers [ " content-type " ] = " application/x-www-form-urlencoded "
2018-08-05 02:00:44 +05:30
channel_req [ " session_token " ] = session_token
2018-04-08 08:06:09 +05:30
2019-04-07 23:29:12 +05:30
subs = XML . parse_html ( html . body )
2018-08-05 02:00:44 +05:30
subs . xpath_nodes ( % q ( / / a [ @class = " subscription-title yt-uix-sessionlink " ] / @href ) ) . each do | channel |
channel_id = channel . content . lstrip ( " /channel/ " ) . not_nil!
channel_req [ " channel_id " ] = channel_id
2019-10-25 22:28:16 +05:30
YT_POOL . client & . post ( " /subscription_ajax?action_update_subscription_preferences=1 " , headers , form : channel_req )
2018-08-05 02:00:44 +05:30
end
2018-07-19 00:56:02 +05:30
end
2019-04-07 23:29:12 +05:30
if redirect
env . redirect referer
else
env . response . content_type = " application/json "
" {} "
end
end
2019-04-16 09:53:40 +05:30
post " /subscription_ajax " do | env |
2019-04-07 23:29:12 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env , " / " )
redirect = env . params . query [ " redirect " ]?
2019-04-16 09:53:40 +05:30
redirect || = " true "
2019-04-07 23:29:12 +05:30
redirect = redirect == " true "
2019-04-16 09:53:40 +05:30
if ! user
2019-04-19 02:53:50 +05:30
if redirect
next env . redirect referer
else
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " No such user " )
2019-04-19 02:53:50 +05:30
end
2019-04-07 23:29:12 +05:30
end
user = user . as ( User )
2019-04-16 09:53:40 +05:30
sid = sid . as ( String )
2019-04-19 02:53:50 +05:30
token = env . params . body [ " csrf_token " ]?
2019-04-16 09:53:40 +05:30
begin
2019-04-19 02:53:50 +05:30
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
2019-04-16 09:53:40 +05:30
rescue ex
if redirect
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2019-04-16 09:53:40 +05:30
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , ex )
2019-04-16 09:53:40 +05:30
end
end
2019-04-07 23:29:12 +05:30
2019-06-08 06:26:41 +05:30
if env . params . query [ " action_create_subscription_to_channel " ]? . try & . to_i? . try & . == 1
2019-04-07 23:29:12 +05:30
action = " action_create_subscription_to_channel "
2019-06-08 06:26:41 +05:30
elsif env . params . query [ " action_remove_subscriptions " ]? . try & . to_i? . try & . == 1
2019-04-07 23:29:12 +05:30
action = " action_remove_subscriptions "
else
next env . redirect referer
end
channel_id = env . params . query [ " c " ]?
channel_id || = " "
if ! user . password
2019-04-15 04:38:00 +05:30
# Sync subscriptions with YouTube
2019-05-15 22:56:29 +05:30
subscribe_ajax ( channel_id , action , env . request . headers )
2019-04-07 23:29:12 +05:30
end
2019-05-15 22:56:29 +05:30
email = user . email
2019-04-07 23:29:12 +05:30
case action
2019-06-08 06:26:41 +05:30
when " action_create_subscription_to_channel "
2019-04-07 23:29:12 +05:30
if ! user . subscriptions . includes? channel_id
2021-01-04 21:21:06 +05:30
get_channel ( channel_id , PG_DB , false , false )
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = array_append(subscriptions, $1) WHERE email = $2 " , channel_id , email )
2019-04-07 23:29:12 +05:30
end
2019-06-08 06:26:41 +05:30
when " action_remove_subscriptions "
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = array_remove(subscriptions, $1) WHERE email = $2 " , channel_id , email )
2020-04-09 22:48:09 +05:30
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Unsupported action #{ action } " )
2019-04-07 23:29:12 +05:30
end
if redirect
env . redirect referer
else
env . response . content_type = " application/json "
" {} "
end
2018-08-05 02:00:44 +05:30
end
2018-04-29 20:10:33 +05:30
2018-08-05 02:00:44 +05:30
get " /subscription_manager " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-08-05 02:00:44 +05:30
user = env . get? " user "
2019-02-11 00:03:29 +05:30
sid = env . get? " sid "
2019-04-19 02:53:50 +05:30
referer = get_referer ( env )
2018-08-09 06:56:02 +05:30
2019-04-19 02:53:50 +05:30
if ! user
2018-08-09 06:56:02 +05:30
next env . redirect referer
2018-04-28 19:57:05 +05:30
end
2018-08-05 02:00:44 +05:30
user = user . as ( User )
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
if ! user . password
# Refresh account
headers = HTTP :: Headers . new
headers [ " Cookie " ] = env . request . headers [ " Cookie " ]
2018-04-08 08:06:09 +05:30
2021-01-04 21:21:06 +05:30
user , sid = get_user ( sid , headers , PG_DB )
2018-08-05 02:00:44 +05:30
end
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
action_takeout = env . params . query [ " action_takeout " ]? . try & . to_i?
action_takeout || = 0
action_takeout = action_takeout == 1
2018-07-19 00:56:02 +05:30
2018-08-05 02:00:44 +05:30
format = env . params . query [ " format " ]?
format || = " rss "
2018-07-19 00:56:02 +05:30
2019-04-22 21:10:29 +05:30
if user . subscriptions . empty?
values = " '{}' "
else
values = " VALUES #{ user . subscriptions . map { | id | %( ( ' #{ id } ' ) ) } . join ( " , " ) } "
end
subscriptions = PG_DB . query_all ( " SELECT * FROM channels WHERE id = ANY( #{ values } ) " , as : InvidiousChannel )
2018-08-05 02:00:44 +05:30
subscriptions . sort_by! { | channel | channel . author . downcase }
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
if action_takeout
if format == " json "
env . response . content_type = " application/json "
env . response . headers [ " content-disposition " ] = " attachment "
2020-07-25 23:00:28 +05:30
playlists = PG_DB . query_all ( " SELECT * FROM playlists WHERE author = $1 AND id LIKE 'IV%' ORDER BY created " , user . email , as : InvidiousPlaylist )
next JSON . build do | json |
json . object do
json . field " subscriptions " , user . subscriptions
json . field " watch_history " , user . watched
json . field " preferences " , user . preferences
json . field " playlists " do
json . array do
playlists . each do | playlist |
json . object do
json . field " title " , playlist . title
json . field " description " , html_to_content ( playlist . description_html )
json . field " privacy " , playlist . privacy . to_s
json . field " videos " do
json . array do
2020-07-29 02:51:39 +05:30
PG_DB . query_all ( " SELECT id FROM playlist_videos WHERE plid = $1 ORDER BY array_position($2, index) LIMIT 500 " , playlist . id , playlist . index , as : String ) . each do | video_id |
json . string video_id
2020-07-25 23:00:28 +05:30
end
end
end
end
end
end
end
end
end
2018-08-05 02:00:44 +05:30
else
env . response . content_type = " application/xml "
env . response . headers [ " content-disposition " ] = " attachment "
export = XML . build do | xml |
xml . element ( " opml " , version : " 1.1 " ) do
xml . element ( " body " ) do
if format == " newpipe "
title = " YouTube Subscriptions "
else
title = " Invidious Subscriptions "
end
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
xml . element ( " outline " , text : title , title : title ) do
subscriptions . each do | channel |
if format == " newpipe "
xmlUrl = " https://www.youtube.com/feeds/videos.xml?channel_id= #{ channel . id } "
else
2020-06-16 03:40:30 +05:30
xmlUrl = " #{ HOST_URL } /feed/channel/ #{ channel . id } "
2018-08-05 02:00:44 +05:30
end
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
xml . element ( " outline " , text : channel . author , title : channel . author ,
" type " : " rss " , xmlUrl : xmlUrl )
end
end
end
2018-07-19 00:56:02 +05:30
end
2018-03-16 22:10:29 +05:30
end
2018-08-05 02:00:44 +05:30
next export . gsub ( %( <?xml version="1.0"?> \n ) , " " )
end
end
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
templated " subscription_manager "
end
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
get " /data_control " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-08-05 02:00:44 +05:30
user = env . get? " user "
2018-08-09 06:56:02 +05:30
referer = get_referer ( env )
2018-03-16 22:10:29 +05:30
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
2018-08-05 02:00:44 +05:30
end
2019-07-13 07:30:50 +05:30
user = user . as ( User )
templated " data_control "
2018-08-05 02:00:44 +05:30
end
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
post " /data_control " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-08-05 02:00:44 +05:30
user = env . get? " user "
2018-08-09 06:56:02 +05:30
referer = get_referer ( env )
2018-03-16 22:10:29 +05:30
2018-08-05 02:00:44 +05:30
if user
user = user . as ( User )
2018-04-29 20:10:33 +05:30
2020-07-26 20:28:50 +05:30
# TODO: Find a way to prevent browser timeout
2019-04-25 06:48:35 +05:30
2018-08-05 02:00:44 +05:30
HTTP :: FormData . parse ( env . request ) do | part |
body = part . body . gets_to_end
2020-07-25 23:00:28 +05:30
next if body . empty?
2018-04-18 04:24:33 +05:30
2020-04-09 22:48:09 +05:30
# TODO: Unify into single import based on content-type
2018-08-05 02:00:44 +05:30
case part . name
when " import_invidious "
body = JSON . parse ( body )
2018-07-26 20:50:15 +05:30
2018-11-10 04:55:24 +05:30
if body [ " subscriptions " ]?
user . subscriptions += body [ " subscriptions " ] . as_a . map { | a | a . as_s }
user . subscriptions . uniq!
2021-01-04 21:21:06 +05:30
user . subscriptions = get_batch_channels ( user . subscriptions , PG_DB , false , false )
2018-11-10 04:55:24 +05:30
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = $1 WHERE email = $2 " , user . subscriptions , user . email )
2018-08-05 02:00:44 +05:30
end
2018-07-26 20:50:15 +05:30
2018-11-09 04:13:28 +05:30
if body [ " watch_history " ]?
2018-11-10 04:55:24 +05:30
user . watched += body [ " watch_history " ] . as_a . map { | a | a . as_s }
user . watched . uniq!
PG_DB . exec ( " UPDATE users SET watched = $1 WHERE email = $2 " , user . watched , user . email )
2018-07-26 20:50:15 +05:30
end
2018-04-29 20:10:33 +05:30
2018-11-09 04:05:26 +05:30
if body [ " preferences " ]?
2020-07-26 20:28:50 +05:30
user . preferences = Preferences . from_json ( body [ " preferences " ] . to_json )
2018-11-10 04:55:24 +05:30
PG_DB . exec ( " UPDATE users SET preferences = $1 WHERE email = $2 " , user . preferences . to_json , user . email )
2018-11-09 04:05:26 +05:30
end
2020-07-25 23:00:28 +05:30
if playlists = body [ " playlists " ]? . try & . as_a?
playlists . each do | item |
title = item [ " title " ]? . try & . as_s? . try & . delete ( " <> " )
description = item [ " description " ]? . try & . as_s? . try & . delete ( " \ r " )
privacy = item [ " privacy " ]? . try & . as_s? . try { | privacy | PlaylistPrivacy . parse? privacy }
next if ! title
next if ! description
next if ! privacy
playlist = create_playlist ( PG_DB , title , privacy , user )
PG_DB . exec ( " UPDATE playlists SET description = $1 WHERE id = $2 " , description , playlist . id )
2020-07-29 02:51:39 +05:30
videos = item [ " videos " ]? . try & . as_a? . try & . each_with_index do | video_id , idx |
2020-11-30 15:29:21 +05:30
raise InfoException . new ( " Playlist cannot have more than 500 videos " ) if idx > 500
2020-07-29 02:51:39 +05:30
2020-07-25 23:00:28 +05:30
video_id = video_id . try & . as_s?
next if ! video_id
begin
video = get_video ( video_id , PG_DB )
rescue ex
next
end
2020-07-26 20:28:50 +05:30
playlist_video = PlaylistVideo . new ( {
title : video . title ,
id : video . id ,
author : video . author ,
ucid : video . ucid ,
2020-07-25 23:00:28 +05:30
length_seconds : video . length_seconds ,
2020-07-26 20:28:50 +05:30
published : video . published ,
plid : playlist . id ,
live_now : video . live_now ,
index : Random :: Secure . rand ( 0 _i64 .. Int64 :: MAX ) ,
} )
2020-07-25 23:00:28 +05:30
video_array = playlist_video . to_a
args = arg_array ( video_array )
PG_DB . exec ( " INSERT INTO playlist_videos VALUES ( #{ args } ) " , args : video_array )
2020-11-01 09:44:46 +05:30
PG_DB . exec ( " UPDATE playlists SET index = array_append(index, $1), video_count = cardinality(index) + 1, updated = $2 WHERE id = $3 " , playlist_video . index , Time . utc , playlist . id )
2020-07-25 23:00:28 +05:30
end
end
end
2018-08-05 02:00:44 +05:30
when " import_youtube "
2021-02-28 02:29:09 +05:30
if body [ 0 .. 4 ] == " <opml "
2021-02-27 23:28:55 +05:30
subscriptions = XML . parse ( body )
user . subscriptions += subscriptions . xpath_nodes ( % q ( / / outline [ @type = " rss " ] ) ) . map do | channel |
channel [ " xmlUrl " ] . match ( / UC[a-zA-Z0-9_-]{22} / ) . not_nil! [ 0 ]
end
else
subscriptions = JSON . parse ( body )
user . subscriptions += subscriptions . as_a . compact_map do | entry |
entry [ " snippet " ] [ " resourceId " ] [ " channelId " ] . as_s
2021-02-27 23:42:01 +05:30
end
2018-11-10 04:55:24 +05:30
end
user . subscriptions . uniq!
2021-01-04 21:21:06 +05:30
user . subscriptions = get_batch_channels ( user . subscriptions , PG_DB , false , false )
2018-10-07 04:49:47 +05:30
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = $1 WHERE email = $2 " , user . subscriptions , user . email )
2018-11-10 04:55:24 +05:30
when " import_freetube "
user . subscriptions += body . scan ( / "channelId":"(?<channel_id>[a-zA-Z0-9_-]{24})" / ) . map do | md |
md [ " channel_id " ]
end
user . subscriptions . uniq!
2021-01-04 21:21:06 +05:30
user . subscriptions = get_batch_channels ( user . subscriptions , PG_DB , false , false )
2018-11-10 04:55:24 +05:30
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = $1 WHERE email = $2 " , user . subscriptions , user . email )
2018-08-05 02:00:44 +05:30
when " import_newpipe_subscriptions "
body = JSON . parse ( body )
2019-04-23 02:09:57 +05:30
user . subscriptions += body [ " subscriptions " ] . as_a . compact_map do | channel |
if match = channel [ " url " ] . as_s . match ( / \/ channel \/ (?<channel>UC[a-zA-Z0-9_-]{22}) / )
next match [ " channel " ]
elsif match = channel [ " url " ] . as_s . match ( / \/ user \/ (?<user>.+) / )
2019-10-25 22:28:16 +05:30
response = YT_POOL . client & . get ( " /user/ #{ match [ " user " ] } ?disable_polymer=1&hl=en&gl=US " )
2020-01-14 18:51:17 +05:30
html = XML . parse_html ( response . body )
ucid = html . xpath_node ( % q ( / / link [ @rel = " canonical " ] ) ) . try & . [ " href " ] . split ( " / " ) [ - 1 ]
next ucid if ucid
2019-04-23 02:09:57 +05:30
end
nil
2018-11-10 04:55:24 +05:30
end
user . subscriptions . uniq!
2021-01-04 21:21:06 +05:30
user . subscriptions = get_batch_channels ( user . subscriptions , PG_DB , false , false )
2018-11-10 04:55:24 +05:30
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = $1 WHERE email = $2 " , user . subscriptions , user . email )
2018-08-05 02:00:44 +05:30
when " import_newpipe "
2020-06-16 04:27:20 +05:30
Compress :: Zip :: Reader . open ( IO :: Memory . new ( body ) ) do | file |
2018-08-05 02:00:44 +05:30
file . each_entry do | entry |
if entry . filename == " newpipe.db "
2018-11-22 04:42:13 +05:30
tempfile = File . tempfile ( " .db " )
File . write ( tempfile . path , entry . io . gets_to_end )
db = DB . open ( " sqlite3:// " + tempfile . path )
2018-04-29 20:10:33 +05:30
2018-11-22 04:42:13 +05:30
user . watched += db . query_all ( " SELECT url FROM streams " , as : String ) . map { | url | url . lchop ( " https://www.youtube.com/watch?v= " ) }
2018-11-10 04:55:24 +05:30
user . watched . uniq!
2018-07-19 00:56:02 +05:30
2018-11-10 04:55:24 +05:30
PG_DB . exec ( " UPDATE users SET watched = $1 WHERE email = $2 " , user . watched , user . email )
2018-10-07 04:49:47 +05:30
2018-11-22 04:42:13 +05:30
user . subscriptions += db . query_all ( " SELECT url FROM subscriptions " , as : String ) . map { | url | url . lchop ( " https://www.youtube.com/channel/ " ) }
2018-11-10 04:55:24 +05:30
user . subscriptions . uniq!
2021-01-04 21:21:06 +05:30
user . subscriptions = get_batch_channels ( user . subscriptions , PG_DB , false , false )
2018-11-10 04:55:24 +05:30
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = $1 WHERE email = $2 " , user . subscriptions , user . email )
2018-11-22 04:42:13 +05:30
db . close
tempfile . delete
2018-08-05 02:00:44 +05:30
end
2018-07-19 00:56:02 +05:30
end
2018-07-08 19:27:06 +05:30
end
2020-04-09 22:48:09 +05:30
else nil # Ignore
2018-07-19 00:56:02 +05:30
end
2018-08-05 02:00:44 +05:30
end
end
2018-07-19 00:56:02 +05:30
2018-08-05 02:00:44 +05:30
env . redirect referer
end
2018-07-19 00:56:02 +05:30
2019-04-22 20:48:17 +05:30
get " /change_password " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env )
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
2019-04-22 20:48:17 +05:30
end
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
csrf_token = generate_response ( sid , { " :change_password " } , HMAC_KEY , PG_DB )
templated " change_password "
2019-04-22 20:48:17 +05:30
end
post " /change_password " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env )
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
token = env . params . body [ " csrf_token " ]?
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
# We don't store passwords for Google accounts
if ! user . password
2020-11-30 15:29:21 +05:30
next error_template ( 400 , " Cannot change password for Google accounts " )
2019-07-13 07:30:50 +05:30
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
begin
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
rescue ex
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2019-07-13 07:30:50 +05:30
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
password = env . params . body [ " password " ]?
if ! password
2020-11-30 15:29:21 +05:30
next error_template ( 401 , " Password is a required field " )
2019-07-13 07:30:50 +05:30
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
new_passwords = env . params . body . select { | k , v | k . match ( / ^new_password \ [ \ d+ \ ]$ / ) } . map { | k , v | v }
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
if new_passwords . size <= 1 || new_passwords . uniq . size != 1
2020-11-30 15:29:21 +05:30
next error_template ( 400 , " New passwords must match " )
2019-07-13 07:30:50 +05:30
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
new_password = new_passwords . uniq [ 0 ]
if new_password . empty?
2020-11-30 15:29:21 +05:30
next error_template ( 401 , " Password cannot be empty " )
2019-07-13 07:30:50 +05:30
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
if new_password . bytesize > 55
2020-11-30 15:29:21 +05:30
next error_template ( 400 , " Password cannot be longer than 55 characters " )
2019-07-13 07:30:50 +05:30
end
2019-04-22 20:48:17 +05:30
2019-07-13 07:30:50 +05:30
if ! Crypto :: Bcrypt :: Password . new ( user . password . not_nil! ) . verify ( password . byte_slice ( 0 , 55 ) )
2020-11-30 15:29:21 +05:30
next error_template ( 401 , " Incorrect password " )
2019-04-22 20:48:17 +05:30
end
2019-07-13 07:30:50 +05:30
new_password = Crypto :: Bcrypt :: Password . create ( new_password , cost : 10 )
PG_DB . exec ( " UPDATE users SET password = $1 WHERE email = $2 " , new_password . to_s , user . email )
2019-04-22 20:48:17 +05:30
env . redirect referer
end
2018-11-08 11:42:14 +05:30
get " /delete_account " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-11-08 11:42:14 +05:30
user = env . get? " user "
2019-04-16 09:53:40 +05:30
sid = env . get? " sid "
2018-11-08 11:42:14 +05:30
referer = get_referer ( env )
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
2018-11-08 11:42:14 +05:30
end
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
csrf_token = generate_response ( sid , { " :delete_account " } , HMAC_KEY , PG_DB )
templated " delete_account "
2018-11-08 11:42:14 +05:30
end
post " /delete_account " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-11-08 11:42:14 +05:30
user = env . get? " user "
2019-04-16 09:53:40 +05:30
sid = env . get? " sid "
2018-11-08 11:42:14 +05:30
referer = get_referer ( env )
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
end
2018-11-08 11:42:14 +05:30
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
token = env . params . body [ " csrf_token " ]?
2018-11-08 11:42:14 +05:30
2019-07-13 07:30:50 +05:30
begin
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
rescue ex
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2019-07-13 07:30:50 +05:30
end
2018-11-08 11:42:14 +05:30
2019-07-13 07:30:50 +05:30
view_name = " subscriptions_ #{ sha256 ( user . email ) } "
PG_DB . exec ( " DELETE FROM users * WHERE email = $1 " , user . email )
PG_DB . exec ( " DELETE FROM session_ids * WHERE email = $1 " , user . email )
PG_DB . exec ( " DROP MATERIALIZED VIEW #{ view_name } " )
env . request . cookies . each do | cookie |
cookie . expires = Time . utc ( 1990 , 1 , 1 )
env . response . cookies << cookie
2018-11-08 11:42:14 +05:30
end
env . redirect referer
end
2018-08-05 02:00:44 +05:30
get " /clear_watch_history " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-08-05 02:00:44 +05:30
user = env . get? " user "
2019-04-16 09:53:40 +05:30
sid = env . get? " sid "
2018-11-08 11:42:14 +05:30
referer = get_referer ( env )
2018-08-09 06:56:02 +05:30
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
2018-11-08 11:42:14 +05:30
end
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
csrf_token = generate_response ( sid , { " :clear_watch_history " } , HMAC_KEY , PG_DB )
templated " clear_watch_history "
2018-11-08 11:42:14 +05:30
end
post " /clear_watch_history " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-11-08 11:42:14 +05:30
user = env . get? " user "
2019-04-16 09:53:40 +05:30
sid = env . get? " sid "
2018-08-09 06:56:02 +05:30
referer = get_referer ( env )
2018-03-16 22:10:29 +05:30
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
end
2018-11-08 11:42:14 +05:30
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
token = env . params . body [ " csrf_token " ]?
2018-11-08 11:42:14 +05:30
2019-07-13 07:30:50 +05:30
begin
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
rescue ex
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2018-08-05 02:00:44 +05:30
end
2019-07-13 07:30:50 +05:30
PG_DB . exec ( " UPDATE users SET watched = '{}' WHERE email = $1 " , user . email )
2018-08-05 02:00:44 +05:30
env . redirect referer
end
2019-05-15 22:56:29 +05:30
get " /authorize_token " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env )
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
end
2019-05-15 22:56:29 +05:30
2019-07-13 07:30:50 +05:30
user = user . as ( User )
sid = sid . as ( String )
csrf_token = generate_response ( sid , { " :authorize_token " } , HMAC_KEY , PG_DB )
2019-05-15 22:56:29 +05:30
2019-07-13 07:30:50 +05:30
scopes = env . params . query [ " scopes " ]? . try & . split ( " , " )
scopes || = [ ] of String
2019-05-15 22:56:29 +05:30
2019-07-13 07:30:50 +05:30
callback_url = env . params . query [ " callback_url " ]?
if callback_url
callback_url = URI . parse ( callback_url )
2019-05-15 22:56:29 +05:30
end
2019-07-13 07:30:50 +05:30
expire = env . params . query [ " expire " ]? . try & . to_i?
templated " authorize_token "
2019-05-15 22:56:29 +05:30
end
2019-04-19 02:53:50 +05:30
post " /authorize_token " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env )
2019-07-13 07:30:50 +05:30
if ! user
next env . redirect referer
end
2019-04-19 02:53:50 +05:30
2019-07-13 07:30:50 +05:30
user = env . get ( " user " ) . as ( User )
sid = sid . as ( String )
token = env . params . body [ " csrf_token " ]?
2019-04-19 02:53:50 +05:30
2019-07-13 07:30:50 +05:30
begin
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
rescue ex
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2019-07-13 07:30:50 +05:30
end
2019-04-19 02:53:50 +05:30
2019-07-13 07:30:50 +05:30
scopes = env . params . body . select { | k , v | k . match ( / ^scopes \ [ \ d+ \ ]$ / ) } . map { | k , v | v }
callback_url = env . params . body [ " callbackUrl " ]?
expire = env . params . body [ " expire " ]? . try & . to_i?
2019-04-19 02:53:50 +05:30
2019-07-13 07:30:50 +05:30
access_token = generate_token ( user . email , scopes , expire , HMAC_KEY , PG_DB )
2019-04-19 02:53:50 +05:30
2019-07-13 07:30:50 +05:30
if callback_url
2019-09-24 23:01:33 +05:30
access_token = URI . encode_www_form ( access_token )
2019-07-13 07:30:50 +05:30
url = URI . parse ( callback_url )
2019-04-19 02:53:50 +05:30
2019-07-13 07:30:50 +05:30
if url . query
query = HTTP :: Params . parse ( url . query . not_nil! )
2019-04-19 02:53:50 +05:30
else
2019-07-13 07:30:50 +05:30
query = HTTP :: Params . new
2019-04-19 02:53:50 +05:30
end
2019-07-13 07:30:50 +05:30
query [ " token " ] = access_token
url . query = query . to_s
env . redirect url . to_s
else
csrf_token = " "
env . set " access_token " , access_token
templated " authorize_token "
2019-04-19 02:53:50 +05:30
end
end
get " /token_manager " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env , " /subscription_manager " )
if ! user
next env . redirect referer
end
user = user . as ( User )
tokens = PG_DB . query_all ( " SELECT id, issued FROM session_ids WHERE email = $1 ORDER BY issued DESC " , user . email , as : { session : String , issued : Time } )
templated " token_manager "
end
post " /token_ajax " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
user = env . get? " user "
sid = env . get? " sid "
referer = get_referer ( env )
redirect = env . params . query [ " redirect " ]?
redirect || = " true "
redirect = redirect == " true "
if ! user
if redirect
next env . redirect referer
else
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " No such user " )
2019-04-19 02:53:50 +05:30
end
end
user = user . as ( User )
sid = sid . as ( String )
token = env . params . body [ " csrf_token " ]?
begin
validate_request ( token , sid , env . request , HMAC_KEY , PG_DB , locale )
rescue ex
if redirect
2020-11-30 15:29:21 +05:30
next error_template ( 400 , ex )
2019-04-19 02:53:50 +05:30
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , ex )
2019-04-19 02:53:50 +05:30
end
end
if env . params . query [ " action_revoke_token " ]?
action = " action_revoke_token "
else
next env . redirect referer
end
session = env . params . query [ " session " ]?
session || = " "
case action
when . starts_with? " action_revoke_token "
PG_DB . exec ( " DELETE FROM session_ids * WHERE id = $1 AND email = $2 " , session , user . email )
2020-04-09 22:48:09 +05:30
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Unsupported action #{ action } " )
2019-04-19 02:53:50 +05:30
end
if redirect
env . redirect referer
else
env . response . content_type = " application/json "
" {} "
end
end
2018-08-05 02:00:44 +05:30
# Feeds
2020-11-20 06:51:48 +05:30
get " /feed/playlists " do | env |
env . redirect " /view_all_playlists "
end
2018-11-26 22:20:34 +05:30
get " /feed/top " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2020-12-27 10:42:43 +05:30
message = translate ( locale , " The Top feed has been removed from Invidious. " )
templated " message "
2018-11-26 22:20:34 +05:30
end
get " /feed/popular " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2021-01-24 00:09:04 +05:30
if CONFIG . popular_enabled
2020-12-27 10:42:43 +05:30
templated " popular "
else
message = translate ( locale , " The Popular feed has been disabled by the administrator. " )
templated " message "
end
2018-11-26 22:20:34 +05:30
end
2018-11-20 22:48:12 +05:30
get " /feed/trending " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-11-20 22:48:12 +05:30
trending_type = env . params . query [ " type " ]?
2018-12-21 04:18:45 +05:30
trending_type || = " Default "
2018-11-20 22:48:12 +05:30
region = env . params . query [ " region " ]?
2018-12-21 04:18:45 +05:30
region || = " US "
2018-11-20 22:48:12 +05:30
begin
2019-06-29 07:47:56 +05:30
trending , plid = fetch_trending ( trending_type , region , locale )
2018-11-20 22:48:12 +05:30
rescue ex
2020-11-30 15:29:21 +05:30
next error_template ( 500 , ex )
2018-11-20 22:48:12 +05:30
end
templated " trending "
end
2018-08-05 02:00:44 +05:30
get " /feed/subscriptions " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-08-05 02:00:44 +05:30
user = env . get? " user "
2019-02-11 00:03:29 +05:30
sid = env . get? " sid "
2018-08-09 06:56:02 +05:30
referer = get_referer ( env )
2018-08-05 02:00:44 +05:30
2019-06-07 23:09:12 +05:30
if ! user
next env . redirect referer
end
2019-06-07 23:12:07 +05:30
user = user . as ( User )
sid = sid . as ( String )
token = user . token
2018-11-22 00:36:29 +05:30
2019-06-07 23:09:12 +05:30
if user . preferences . unseen_only
2019-06-07 23:12:07 +05:30
env . set " show_watched " , true
end
2018-08-05 02:00:44 +05:30
2019-06-07 23:12:07 +05:30
# Refresh account
headers = HTTP :: Headers . new
headers [ " Cookie " ] = env . request . headers [ " Cookie " ]
2018-08-05 02:00:44 +05:30
2019-06-07 23:12:07 +05:30
if ! user . password
2021-01-04 21:21:06 +05:30
user , sid = get_user ( sid , headers , PG_DB )
2019-06-07 23:12:07 +05:30
end
2018-03-26 08:51:24 +05:30
2019-06-09 02:34:55 +05:30
max_results = env . params . query [ " max_results " ]? . try & . to_i? . try & . clamp ( 0 , MAX_ITEMS_PER_PAGE )
max_results || = user . preferences . max_results
max_results || = CONFIG . default_user_preferences . max_results
2018-08-05 02:00:44 +05:30
2019-06-07 23:12:07 +05:30
page = env . params . query [ " page " ]? . try & . to_i?
page || = 1
2018-08-05 02:00:44 +05:30
2019-06-07 23:09:12 +05:30
videos , notifications = get_subscription_feed ( PG_DB , user , max_results , page )
2019-02-18 22:59:57 +05:30
2019-06-07 23:12:07 +05:30
# "updated" here is used for delivering new notifications, so if
# we know a user has looked at their feed e.g. in the past 10 minutes,
# they've already seen a video posted 20 minutes ago, and don't need
# to be notified.
2019-06-08 06:26:41 +05:30
PG_DB . exec ( " UPDATE users SET notifications = $1, updated = $2 WHERE email = $3 " , [ ] of String , Time . utc ,
2019-06-07 23:12:07 +05:30
user . email )
user . notifications = [ ] of String
env . set " user " , user
2018-08-05 02:00:44 +05:30
2019-06-07 23:12:07 +05:30
templated " subscriptions "
end
2018-03-16 22:10:29 +05:30
2018-11-20 09:36:59 +05:30
get " /feed/history " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-11-20 09:36:59 +05:30
user = env . get? " user "
referer = get_referer ( env )
page = env . params . query [ " page " ]? . try & . to_i?
page || = 1
2019-06-07 23:09:12 +05:30
if ! user
next env . redirect referer
end
2019-06-07 23:12:07 +05:30
user = user . as ( User )
2018-11-10 08:07:46 +05:30
2019-06-09 02:34:55 +05:30
max_results = env . params . query [ " max_results " ]? . try & . to_i? . try & . clamp ( 0 , MAX_ITEMS_PER_PAGE )
max_results || = user . preferences . max_results
max_results || = CONFIG . default_user_preferences . max_results
if user . watched [ ( page - 1 ) * max_results ]?
watched = user . watched . reverse [ ( page - 1 ) * max_results , max_results ]
2018-11-20 09:36:59 +05:30
end
2019-06-09 02:34:55 +05:30
watched || = [ ] of String
2018-11-10 04:55:24 +05:30
2019-06-07 23:12:07 +05:30
templated " history "
end
2018-08-05 02:00:44 +05:30
get " /feed/channel/:ucid " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2019-06-08 08:09:32 +05:30
env . response . content_type = " application/atom+xml "
2019-03-06 00:26:59 +05:30
2018-08-05 02:00:44 +05:30
ucid = env . params . url [ " ucid " ]
2018-04-08 08:06:09 +05:30
2019-09-07 21:15:37 +05:30
params = HTTP :: Params . parse ( env . params . query [ " params " ]? || " " )
2018-09-21 20:10:04 +05:30
begin
2019-06-29 07:18:24 +05:30
channel = get_about_info ( ucid , locale )
2019-09-08 21:38:59 +05:30
rescue ex : ChannelRedirect
next env . redirect env . request . resource . gsub ( ucid , ex . channel_id )
2018-09-21 20:10:04 +05:30
rescue ex
2020-11-30 15:29:21 +05:30
next error_atom ( 500 , ex )
2018-09-05 07:34:40 +05:30
end
2020-04-10 22:19:51 +05:30
response = YT_POOL . client & . get ( " /feeds/videos.xml?channel_id= #{ channel . ucid } " )
rss = XML . parse_html ( response . body )
2019-02-19 03:36:00 +05:30
2020-04-08 00:04:40 +05:30
videos = rss . xpath_nodes ( " //feed/entry " ) . map do | entry |
2019-02-19 03:36:00 +05:30
video_id = entry . xpath_node ( " videoid " ) . not_nil! . content
title = entry . xpath_node ( " title " ) . not_nil! . content
2019-03-08 08:43:54 +05:30
published = Time . parse_rfc3339 ( entry . xpath_node ( " published " ) . not_nil! . content )
updated = Time . parse_rfc3339 ( entry . xpath_node ( " updated " ) . not_nil! . content )
2019-02-19 03:36:00 +05:30
author = entry . xpath_node ( " author/name " ) . not_nil! . content
ucid = entry . xpath_node ( " channelid " ) . not_nil! . content
2019-06-09 01:38:27 +05:30
description_html = entry . xpath_node ( " group/description " ) . not_nil! . to_s
2019-02-19 03:36:00 +05:30
views = entry . xpath_node ( " group/community/statistics " ) . not_nil! . [ " views " ] . to_i64
2020-07-26 20:28:50 +05:30
SearchVideo . new ( {
title : title ,
id : video_id ,
author : author ,
ucid : ucid ,
published : published ,
views : views ,
description_html : description_html ,
length_seconds : 0 ,
live_now : false ,
premium : false ,
premiere_timestamp : nil ,
} )
2019-02-19 03:36:00 +05:30
end
2018-07-16 21:54:24 +05:30
2019-06-07 23:09:12 +05:30
XML . build ( indent : " " , encoding : " UTF-8 " ) do | xml |
2018-08-05 02:00:44 +05:30
xml . element ( " feed " , " xmlns:yt " : " http://www.youtube.com/xml/schemas/2015 " ,
2018-12-23 23:37:04 +05:30
" xmlns:media " : " http://search.yahoo.com/mrss/ " , xmlns : " http://www.w3.org/2005/Atom " ,
" xml:lang " : " en-US " ) do
2020-06-16 03:40:30 +05:30
xml . element ( " link " , rel : " self " , href : " #{ HOST_URL } #{ env . request . resource } " )
2019-06-29 07:18:24 +05:30
xml . element ( " id " ) { xml . text " yt:channel: #{ channel . ucid } " }
xml . element ( " yt:channelId " ) { xml . text channel . ucid }
2020-12-14 23:55:39 +05:30
xml . element ( " icon " ) { xml . text channel . author_thumbnail }
2019-06-29 07:18:24 +05:30
xml . element ( " title " ) { xml . text channel . author }
2020-06-16 03:40:30 +05:30
xml . element ( " link " , rel : " alternate " , href : " #{ HOST_URL } /channel/ #{ channel . ucid } " )
2018-07-28 20:19:58 +05:30
2018-08-05 02:00:44 +05:30
xml . element ( " author " ) do
2019-06-29 07:18:24 +05:30
xml . element ( " name " ) { xml . text channel . author }
2020-06-16 03:40:30 +05:30
xml . element ( " uri " ) { xml . text " #{ HOST_URL } /channel/ #{ channel . ucid } " }
2018-08-05 02:00:44 +05:30
end
2018-07-29 09:01:02 +05:30
2018-09-05 07:34:40 +05:30
videos . each do | video |
2020-06-16 03:40:30 +05:30
video . to_xml ( channel . auto_generated , params , xml )
2019-06-07 23:12:07 +05:30
end
end
end
end
2018-07-16 21:54:24 +05:30
2018-08-05 02:00:44 +05:30
get " /feed/private " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2019-06-08 08:09:32 +05:30
env . response . content_type = " application/atom+xml "
2019-03-06 00:26:59 +05:30
2018-08-05 02:00:44 +05:30
token = env . params . query [ " token " ]?
2018-07-16 21:54:24 +05:30
2018-08-05 02:00:44 +05:30
if ! token
2019-03-23 20:54:30 +05:30
env . response . status_code = 403
next
2018-08-05 02:00:44 +05:30
end
2018-03-25 09:08:35 +05:30
2018-08-05 02:00:44 +05:30
user = PG_DB . query_one? ( " SELECT * FROM users WHERE token = $1 " , token . strip , as : User )
if ! user
2019-03-23 20:54:30 +05:30
env . response . status_code = 403
next
2018-08-05 02:00:44 +05:30
end
2018-07-17 18:49:45 +05:30
2019-06-09 02:34:55 +05:30
max_results = env . params . query [ " max_results " ]? . try & . to_i? . try & . clamp ( 0 , MAX_ITEMS_PER_PAGE )
max_results || = user . preferences . max_results
max_results || = CONFIG . default_user_preferences . max_results
2018-07-17 18:49:45 +05:30
2018-08-05 02:00:44 +05:30
page = env . params . query [ " page " ]? . try & . to_i?
page || = 1
2018-03-25 09:08:35 +05:30
2019-09-07 21:15:37 +05:30
params = HTTP :: Params . parse ( env . params . query [ " params " ]? || " " )
2019-06-07 23:09:12 +05:30
videos , notifications = get_subscription_feed ( PG_DB , user , max_results , page )
2018-07-31 21:14:07 +05:30
2019-06-07 23:09:12 +05:30
XML . build ( indent : " " , encoding : " UTF-8 " ) do | xml |
2018-12-23 23:37:04 +05:30
xml . element ( " feed " , " xmlns:yt " : " http://www.youtube.com/xml/schemas/2015 " ,
" xmlns:media " : " http://search.yahoo.com/mrss/ " , xmlns : " http://www.w3.org/2005/Atom " ,
2018-08-05 02:00:44 +05:30
" xml:lang " : " en-US " ) do
2020-06-16 03:40:30 +05:30
xml . element ( " link " , " type " : " text/html " , rel : " alternate " , href : " #{ HOST_URL } /feed/subscriptions " )
2019-06-07 23:09:12 +05:30
xml . element ( " link " , " type " : " application/atom+xml " , rel : " self " ,
2020-06-16 03:40:30 +05:30
href : " #{ HOST_URL } #{ env . request . resource } " )
2018-12-21 03:02:09 +05:30
xml . element ( " title " ) { xml . text translate ( locale , " Invidious Private Feed for `x` " , user . email ) }
2018-07-18 19:45:58 +05:30
2019-06-08 07:57:37 +05:30
( notifications + videos ) . each do | video |
2020-06-16 03:40:30 +05:30
video . to_xml ( locale , params , xml )
2018-08-05 02:00:44 +05:30
end
2018-04-18 02:57:55 +05:30
end
2018-08-05 02:00:44 +05:30
end
end
2018-03-25 09:08:35 +05:30
2018-09-18 04:43:24 +05:30
get " /feed/playlist/:plid " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2019-06-08 08:09:32 +05:30
env . response . content_type = " application/atom+xml "
2019-03-06 00:26:59 +05:30
2018-09-18 04:43:24 +05:30
plid = env . params . url [ " plid " ]
2019-09-07 21:15:37 +05:30
params = HTTP :: Params . parse ( env . params . query [ " params " ]? || " " )
2018-09-18 04:43:24 +05:30
path = env . request . path
2019-08-06 05:19:13 +05:30
if plid . starts_with? " IV "
if playlist = PG_DB . query_one? ( " SELECT * FROM playlists WHERE id = $1 " , plid , as : InvidiousPlaylist )
videos = get_playlist_videos ( PG_DB , playlist , offset : 0 , locale : locale )
next XML . build ( indent : " " , encoding : " UTF-8 " ) do | xml |
xml . element ( " feed " , " xmlns:yt " : " http://www.youtube.com/xml/schemas/2015 " ,
" xmlns:media " : " http://search.yahoo.com/mrss/ " , xmlns : " http://www.w3.org/2005/Atom " ,
" xml:lang " : " en-US " ) do
2020-06-16 03:40:30 +05:30
xml . element ( " link " , rel : " self " , href : " #{ HOST_URL } #{ env . request . resource } " )
2019-08-06 05:19:13 +05:30
xml . element ( " id " ) { xml . text " iv:playlist: #{ plid } " }
xml . element ( " iv:playlistId " ) { xml . text plid }
xml . element ( " title " ) { xml . text playlist . title }
2020-06-16 03:40:30 +05:30
xml . element ( " link " , rel : " alternate " , href : " #{ HOST_URL } /playlist?list= #{ plid } " )
2019-08-06 05:19:13 +05:30
xml . element ( " author " ) do
xml . element ( " name " ) { xml . text playlist . author }
end
videos . each do | video |
2020-06-16 03:40:30 +05:30
video . to_xml ( false , xml )
2019-08-06 05:19:13 +05:30
end
end
end
else
env . response . status_code = 404
next
end
end
2019-10-25 22:28:16 +05:30
response = YT_POOL . client & . get ( " /feeds/videos.xml?playlist_id= #{ plid } " )
2018-09-18 04:43:24 +05:30
document = XML . parse ( response . body )
document . xpath_nodes ( % q ( / / * [ @href ] | / / * [ @url ] ) ) . each do | node |
node . attributes . each do | attribute |
case attribute . name
2019-09-07 21:15:37 +05:30
when " url " , " href "
2021-02-01 00:22:32 +05:30
request_target = URI . parse ( node [ attribute . name ] ) . request_target
query_string_opt = request_target . starts_with? ( " /watch?v= " ) ? " & #{ params } " : " "
node [ attribute . name ] = " #{ HOST_URL } #{ request_target } #{ query_string_opt } "
2020-04-09 22:48:09 +05:30
else nil # Skip
2018-09-18 04:43:24 +05:30
end
end
end
document = document . to_xml ( options : XML :: SaveOptions :: NO_DECL )
document . scan ( / <uri>(?<url>[^<]+)< \/ uri> / ) . each do | match |
2021-02-01 00:22:32 +05:30
content = " #{ HOST_URL } #{ URI . parse ( match [ " url " ] ) . request_target } "
2018-09-18 04:43:24 +05:30
document = document . gsub ( match [ 0 ] , " <uri> #{ content } </uri> " )
end
document
end
2019-03-30 02:20:18 +05:30
get " /feeds/videos.xml " do | env |
if ucid = env . params . query [ " channel_id " ]?
env . redirect " /feed/channel/ #{ ucid } "
elsif user = env . params . query [ " user " ]?
env . redirect " /feed/channel/ #{ user } "
elsif plid = env . params . query [ " playlist_id " ]?
env . redirect " /feed/playlist/ #{ plid } "
end
end
2019-03-04 22:16:58 +05:30
# Support push notifications via PubSubHubbub
2019-03-04 06:48:23 +05:30
2019-03-04 08:10:24 +05:30
get " /feed/webhook/:token " do | env |
verify_token = env . params . url [ " token " ]
2019-07-10 21:52:10 +05:30
mode = env . params . query [ " hub.mode " ]?
topic = env . params . query [ " hub.topic " ]?
challenge = env . params . query [ " hub.challenge " ]?
if ! mode || ! topic || ! challenge
env . response . status_code = 400
next
else
mode = mode . not_nil!
topic = topic . not_nil!
challenge = challenge . not_nil!
end
2019-03-04 06:48:23 +05:30
2019-06-08 06:26:41 +05:30
case verify_token
when . starts_with? " v1 "
2019-03-04 19:23:31 +05:30
_ , time , nonce , signature = verify_token . split ( " : " )
data = " #{ time } : #{ nonce } "
2019-06-08 06:26:41 +05:30
when . starts_with? " v2 "
2019-03-04 19:23:31 +05:30
time , signature = verify_token . split ( " : " )
data = " #{ time } "
2019-06-08 06:26:41 +05:30
else
env . response . status_code = 400
next
2019-03-04 19:23:31 +05:30
end
2019-03-04 06:48:23 +05:30
2019-04-04 18:19:53 +05:30
# The hub will sometimes check if we're still subscribed after delivery errors,
# so we reply with a 200 as long as the request hasn't expired
2019-06-08 06:26:41 +05:30
if Time . utc . to_unix - time . to_i > 432000
2019-03-23 20:54:30 +05:30
env . response . status_code = 400
next
2019-03-04 06:48:23 +05:30
end
2019-03-04 19:23:31 +05:30
if OpenSSL :: HMAC . hexdigest ( :sha1 , HMAC_KEY , data ) != signature
2019-03-23 20:54:30 +05:30
env . response . status_code = 400
next
2019-03-04 06:48:23 +05:30
end
2019-06-08 06:26:41 +05:30
if ucid = HTTP :: Params . parse ( URI . parse ( topic ) . query . not_nil! ) [ " channel_id " ]?
PG_DB . exec ( " UPDATE channels SET subscribed = $1 WHERE id = $2 " , Time . utc , ucid )
elsif plid = HTTP :: Params . parse ( URI . parse ( topic ) . query . not_nil! ) [ " playlist_id " ]?
PG_DB . exec ( " UPDATE playlists SET subscribed = $1 WHERE id = $2 " , Time . utc , ucid )
else
env . response . status_code = 400
next
end
2019-03-04 06:48:23 +05:30
2019-03-23 20:54:30 +05:30
env . response . status_code = 200
2019-06-08 06:26:41 +05:30
challenge
2019-03-04 06:48:23 +05:30
end
2019-03-04 08:10:24 +05:30
post " /feed/webhook/:token " do | env |
2019-04-11 04:28:42 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2019-03-04 22:37:27 +05:30
token = env . params . url [ " token " ]
2019-03-04 06:48:23 +05:30
body = env . request . body . not_nil! . gets_to_end
signature = env . request . headers [ " X-Hub-Signature " ] . lchop ( " sha1= " )
if signature != OpenSSL :: HMAC . hexdigest ( :sha1 , HMAC_KEY , body )
2021-01-04 21:21:06 +05:30
LOGGER . error ( " /feed/webhook/ #{ token } : Invalid signature " )
2019-03-23 20:54:30 +05:30
env . response . status_code = 200
next
2019-03-04 06:48:23 +05:30
end
2019-03-04 22:37:27 +05:30
spawn do
rss = XML . parse_html ( body )
rss . xpath_nodes ( " //feed/entry " ) . each do | entry |
id = entry . xpath_node ( " videoid " ) . not_nil! . content
2019-04-04 18:19:53 +05:30
author = entry . xpath_node ( " author/name " ) . not_nil! . content
2019-03-08 09:19:52 +05:30
published = Time . parse_rfc3339 ( entry . xpath_node ( " published " ) . not_nil! . content )
2019-03-08 08:43:54 +05:30
updated = Time . parse_rfc3339 ( entry . xpath_node ( " updated " ) . not_nil! . content )
2019-03-04 06:48:23 +05:30
2019-06-29 07:47:56 +05:30
video = get_video ( id , PG_DB , force_refresh : true )
2019-04-11 04:28:42 +05:30
# Deliver notifications to `/api/v1/auth/notifications`
payload = {
2019-04-20 23:11:51 +05:30
" topic " = > video . ucid ,
" videoId " = > video . id ,
" published " = > published . to_unix ,
2019-04-11 04:28:42 +05:30
} . to_json
PG_DB . exec ( " NOTIFY notifications, E' #{ payload } ' " )
2020-07-26 20:28:50 +05:30
video = ChannelVideo . new ( {
id : id ,
title : video . title ,
published : published ,
updated : updated ,
ucid : video . ucid ,
author : author ,
length_seconds : video . length_seconds ,
live_now : video . live_now ,
2019-04-11 04:28:42 +05:30
premiere_timestamp : video . premiere_timestamp ,
2020-07-26 20:28:50 +05:30
views : video . views ,
} )
2019-03-04 06:48:23 +05:30
2021-02-28 02:29:09 +05:30
was_insert = PG_DB . query_one ( " INSERT INTO channel_videos VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
ON CONFLICT ( id ) DO UPDATE SET title = $2 , published = $3 ,
updated = $4 , ucid = $5 , author = $6 , length_seconds = $7 ,
2020-09-10 04:33:27 +05:30
live_now = $8 , premiere_timestamp = $9 , views = $10 returning ( xmax = 0 ) as was_insert " , *video.to_tuple, as: Bool)
2021-02-28 02:29:09 +05:30
PG_DB . exec ( " UPDATE users SET notifications = array_append(notifications, $1),
2020-09-10 04:33:27 +05:30
feed_needs_update = true WHERE $2 = ANY ( subscriptions ) " , video.id, video.ucid) if was_insert
2019-03-04 22:37:27 +05:30
end
2019-03-04 06:48:23 +05:30
end
2019-03-04 07:20:23 +05:30
2019-03-23 20:54:30 +05:30
env . response . status_code = 200
next
2019-03-04 06:48:23 +05:30
end
2018-08-05 02:00:44 +05:30
# Channels
2019-04-28 22:17:16 +05:30
{ " /channel/:ucid/live " , " /user/:user/live " , " /c/:user/live " } . each do | route |
get route do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
# Appears to be a bug in routing, having several routes configured
# as `/a/:a`, `/b/:a`, `/c/:a` results in 404
value = env . request . resource . split ( " / " ) [ 2 ]
body = " "
{ " channel " , " user " , " c " } . each do | type |
2019-10-25 22:28:16 +05:30
response = YT_POOL . client & . get ( " / #{ type } / #{ value } /live?disable_polymer=1 " )
2019-04-28 22:17:16 +05:30
if response . status_code == 200
body = response . body
end
end
video_id = body . match ( / 'VIDEO_ID': "(?<id>[a-zA-Z0-9_-]{11})" / ) . try & . [ " id " ]?
if video_id
params = [ ] of String
env . params . query . each do | k , v |
params << " #{ k } = #{ v } "
end
params = params . join ( " & " )
url = " /watch?v= #{ video_id } "
if ! params . empty?
url += " & #{ params } "
end
env . redirect url
else
env . redirect " /channel/ #{ value } "
end
end
end
2018-08-05 02:00:44 +05:30
# API Endpoints
2018-07-28 18:54:53 +05:30
2019-08-06 05:19:13 +05:30
{ " /api/v1/playlists/:plid " , " /api/v1/auth/playlists/:plid " } . each do | route |
get route do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-10-08 07:41:33 +05:30
2019-08-06 05:19:13 +05:30
env . response . content_type = " application/json "
plid = env . params . url [ " plid " ]
2018-10-07 08:48:50 +05:30
2019-08-06 05:19:13 +05:30
offset = env . params . query [ " index " ]? . try & . to_i?
offset || = env . params . query [ " page " ]? . try & . to_i? . try { | page | ( page - 1 ) * 100 }
offset || = 0
2018-08-15 20:52:36 +05:30
2019-08-06 05:19:13 +05:30
continuation = env . params . query [ " continuation " ]?
2018-09-28 20:24:45 +05:30
2019-08-06 05:19:13 +05:30
format = env . params . query [ " format " ]?
format || = " json "
2018-08-15 20:52:36 +05:30
2019-08-06 05:19:13 +05:30
if plid . starts_with? " RD "
next env . redirect " /api/v1/mixes/ #{ plid } "
end
2018-08-15 20:52:36 +05:30
2019-08-06 05:19:13 +05:30
begin
playlist = get_playlist ( PG_DB , plid , locale )
2021-01-04 10:05:59 +05:30
rescue ex : InfoException
next error_json ( 404 , ex )
2019-08-06 05:19:13 +05:30
rescue ex
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not exist. " )
2019-08-06 05:19:13 +05:30
end
2018-09-25 20:58:40 +05:30
2019-08-06 05:19:13 +05:30
user = env . get? ( " user " ) . try & . as ( User )
2019-10-16 17:51:26 +05:30
if ! playlist || playlist . privacy . private? && playlist . author != user . try & . email
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not exist. " )
2019-08-06 05:19:13 +05:30
end
2018-09-25 20:58:40 +05:30
2020-06-16 03:40:30 +05:30
response = playlist . to_json ( offset , locale , continuation : continuation )
2018-08-15 20:52:36 +05:30
2019-08-06 05:19:13 +05:30
if format == " html "
response = JSON . parse ( response )
playlist_html = template_playlist ( response )
2019-10-22 07:10:03 +05:30
index , next_video = response [ " videos " ] . as_a . skip ( 1 ) . select { | video | ! video [ " author " ] . as_s . empty? } [ 0 ]? . try { | v | { v [ " index " ] , v [ " videoId " ] } } || { nil , nil }
2018-08-15 20:52:36 +05:30
2019-08-06 05:19:13 +05:30
response = {
" playlistHtml " = > playlist_html ,
" index " = > index ,
" nextVideo " = > next_video ,
} . to_json
2018-08-15 20:52:36 +05:30
end
2018-10-08 07:41:33 +05:30
2019-08-06 05:19:13 +05:30
response
2018-10-08 07:41:33 +05:30
end
2019-03-24 00:35:13 +05:30
end
2018-08-15 20:52:36 +05:30
2018-09-29 09:42:35 +05:30
get " /api/v1/mixes/:rdid " do | env |
2019-03-11 23:14:25 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2018-12-21 03:02:09 +05:30
2018-09-29 09:42:35 +05:30
env . response . content_type = " application/json "
rdid = env . params . url [ " rdid " ]
continuation = env . params . query [ " continuation " ]?
2019-02-16 04:58:54 +05:30
continuation || = rdid . lchop ( " RD " ) [ 0 , 11 ]
2018-09-29 09:42:35 +05:30
2018-10-08 07:41:33 +05:30
format = env . params . query [ " format " ]?
format || = " json "
2018-09-29 09:42:35 +05:30
begin
2018-12-21 03:02:09 +05:30
mix = fetch_mix ( rdid , continuation , locale : locale )
2018-10-31 19:45:17 +05:30
if ! rdid . ends_with? continuation
mix = fetch_mix ( rdid , mix . videos [ 1 ] . id )
index = mix . videos . index ( mix . videos . select { | video | video . id == continuation } [ 0 ]? )
end
mix . videos = mix . videos [ index .. - 1 ]
2018-09-29 09:42:35 +05:30
rescue ex
2020-11-30 15:29:21 +05:30
next error_json ( 500 , ex )
2018-09-29 09:42:35 +05:30
end
response = JSON . build do | json |
json . object do
json . field " title " , mix . title
json . field " mixId " , mix . id
json . field " videos " do
json . array do
mix . videos . each do | video |
json . object do
json . field " title " , video . title
json . field " videoId " , video . id
json . field " author " , video . author
json . field " authorId " , video . ucid
json . field " authorUrl " , " /channel/ #{ video . ucid } "
json . field " videoThumbnails " do
json . array do
2020-06-16 03:40:30 +05:30
generate_thumbnails ( json , video . id )
2018-09-29 09:42:35 +05:30
end
end
json . field " index " , video . index
json . field " lengthSeconds " , video . length_seconds
end
end
end
end
end
end
2018-10-08 07:41:33 +05:30
if format == " html "
response = JSON . parse ( response )
playlist_html = template_mix ( response )
2019-10-22 04:30:56 +05:30
next_video = response [ " videos " ] . as_a . select { | video | ! video [ " author " ] . as_s . empty? } [ 0 ]? . try & . [ " videoId " ]
2018-10-08 07:41:33 +05:30
response = {
" playlistHtml " = > playlist_html ,
" nextVideo " = > next_video ,
} . to_json
end
2019-03-23 20:54:30 +05:30
response
2018-09-29 09:42:35 +05:30
end
2019-06-07 23:09:12 +05:30
# Authenticated endpoints
2019-04-11 04:28:42 +05:30
get " /api/v1/auth/notifications " do | env |
2019-06-02 18:11:53 +05:30
env . response . content_type = " text/event-stream "
2019-04-11 04:28:42 +05:30
topics = env . params . query [ " topics " ]? . try & . split ( " , " ) . uniq . first ( 1000 )
topics || = [ ] of String
2020-06-16 03:40:30 +05:30
create_notification_stream ( env , topics , connection_channel )
2019-05-21 19:31:17 +05:30
end
2019-04-11 04:28:42 +05:30
2019-05-21 19:31:17 +05:30
post " /api/v1/auth/notifications " do | env |
2019-06-02 18:11:53 +05:30
env . response . content_type = " text/event-stream "
2019-05-21 19:31:17 +05:30
topics = env . params . body [ " topics " ]? . try & . split ( " , " ) . uniq . first ( 1000 )
topics || = [ ] of String
2019-04-11 04:28:42 +05:30
2020-06-16 03:40:30 +05:30
create_notification_stream ( env , topics , connection_channel )
2019-04-11 04:28:42 +05:30
end
2019-05-01 07:31:57 +05:30
get " /api/v1/auth/preferences " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
user . preferences . to_json
end
2019-04-19 02:53:50 +05:30
2019-05-01 07:31:57 +05:30
post " /api/v1/auth/preferences " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
begin
2020-07-26 20:28:50 +05:30
preferences = Preferences . from_json ( env . request . body || " {} " )
2019-05-01 07:31:57 +05:30
rescue
preferences = user . preferences
end
PG_DB . exec ( " UPDATE users SET preferences = $1 WHERE email = $2 " , preferences . to_json , user . email )
env . response . status_code = 204
end
2019-04-19 02:53:50 +05:30
2019-06-07 23:09:12 +05:30
get " /api/v1/auth/feed " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2019-06-09 02:34:55 +05:30
max_results = env . params . query [ " max_results " ]? . try & . to_i?
max_results || = user . preferences . max_results
max_results || = CONFIG . default_user_preferences . max_results
2019-06-07 23:09:12 +05:30
page = env . params . query [ " page " ]? . try & . to_i?
page || = 1
videos , notifications = get_subscription_feed ( PG_DB , user , max_results , page )
JSON . build do | json |
json . object do
json . field " notifications " do
json . array do
notifications . each do | video |
2020-06-16 03:40:30 +05:30
video . to_json ( locale , json )
2019-06-07 23:09:12 +05:30
end
end
end
json . field " videos " do
json . array do
videos . each do | video |
2020-06-16 03:40:30 +05:30
video . to_json ( locale , json )
2019-06-07 23:09:12 +05:30
end
end
end
end
end
end
2019-04-22 21:10:29 +05:30
get " /api/v1/auth/subscriptions " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
2019-04-19 02:53:50 +05:30
2019-04-22 21:10:29 +05:30
if user . subscriptions . empty?
values = " '{}' "
else
values = " VALUES #{ user . subscriptions . map { | id | %( ( ' #{ id } ' ) ) } . join ( " , " ) } "
end
2019-04-19 02:53:50 +05:30
2019-04-22 21:10:29 +05:30
subscriptions = PG_DB . query_all ( " SELECT * FROM channels WHERE id = ANY( #{ values } ) " , as : InvidiousChannel )
JSON . build do | json |
json . array do
subscriptions . each do | subscription |
json . object do
json . field " author " , subscription . author
json . field " authorId " , subscription . id
end
end
end
end
end
post " /api/v1/auth/subscriptions/:ucid " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
ucid = env . params . url [ " ucid " ]
if ! user . subscriptions . includes? ucid
2021-01-04 21:21:06 +05:30
get_channel ( ucid , PG_DB , false , false )
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = array_append(subscriptions,$1) WHERE email = $2 " , ucid , user . email )
2019-04-22 21:10:29 +05:30
end
2019-05-15 22:56:29 +05:30
# For Google accounts, access tokens don't have enough information to
# make a request on the user's behalf, which is why we don't sync with
# YouTube.
2019-04-22 21:10:29 +05:30
env . response . status_code = 204
end
delete " /api/v1/auth/subscriptions/:ucid " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
ucid = env . params . url [ " ucid " ]
2019-06-01 20:49:18 +05:30
PG_DB . exec ( " UPDATE users SET feed_needs_update = true, subscriptions = array_remove(subscriptions, $1) WHERE email = $2 " , ucid , user . email )
2019-04-22 21:10:29 +05:30
env . response . status_code = 204
end
2019-04-19 02:53:50 +05:30
2019-08-06 05:19:13 +05:30
get " /api/v1/auth/playlists " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
playlists = PG_DB . query_all ( " SELECT * FROM playlists WHERE author = $1 " , user . email , as : InvidiousPlaylist )
JSON . build do | json |
json . array do
playlists . each do | playlist |
2020-06-16 03:40:30 +05:30
playlist . to_json ( 0 , locale , json )
2019-08-06 05:19:13 +05:30
end
end
end
end
post " /api/v1/auth/playlists " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
title = env . params . json [ " title " ]? . try & . as ( String ) . delete ( " <> " ) . byte_slice ( 0 , 150 )
if ! title
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Invalid title. " )
2019-08-06 05:19:13 +05:30
end
privacy = env . params . json [ " privacy " ]? . try { | privacy | PlaylistPrivacy . parse ( privacy . as ( String ) . downcase ) }
if ! privacy
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Invalid privacy setting. " )
2019-08-06 05:19:13 +05:30
end
if PG_DB . query_one ( " SELECT count(*) FROM playlists WHERE author = $1 " , user . email , as : Int64 ) >= 100
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " User cannot have more than 100 playlists. " )
2019-08-06 05:19:13 +05:30
end
playlist = create_playlist ( PG_DB , title , privacy , user )
2020-06-16 03:40:30 +05:30
env . response . headers [ " Location " ] = " #{ HOST_URL } /api/v1/auth/playlists/ #{ playlist . id } "
2019-08-06 05:19:13 +05:30
env . response . status_code = 201
{
" title " = > title ,
" playlistId " = > playlist . id ,
} . to_json
end
patch " /api/v1/auth/playlists/:plid " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
plid = env . params . url [ " plid " ]
playlist = PG_DB . query_one? ( " SELECT * FROM playlists WHERE id = $1 " , plid , as : InvidiousPlaylist )
2019-10-16 17:51:26 +05:30
if ! playlist || playlist . author != user . email && playlist . privacy . private?
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not exist. " )
2019-08-06 05:19:13 +05:30
end
if playlist . author != user . email
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " Invalid user " )
2019-08-06 05:19:13 +05:30
end
title = env . params . json [ " title " ] . try & . as ( String ) . delete ( " <> " ) . byte_slice ( 0 , 150 ) || playlist . title
privacy = env . params . json [ " privacy " ]? . try { | privacy | PlaylistPrivacy . parse ( privacy . as ( String ) . downcase ) } || playlist . privacy
description = env . params . json [ " description " ]? . try & . as ( String ) . delete ( " \ r " ) || playlist . description
if title != playlist . title ||
privacy != playlist . privacy ||
description != playlist . description
updated = Time . utc
else
updated = playlist . updated
end
PG_DB . exec ( " UPDATE playlists SET title = $1, privacy = $2, description = $3, updated = $4 WHERE id = $5 " , title , privacy , description , updated , plid )
env . response . status_code = 204
end
delete " /api/v1/auth/playlists/:plid " do | env |
2020-11-30 15:29:21 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2019-08-06 05:19:13 +05:30
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
plid = env . params . url [ " plid " ]
playlist = PG_DB . query_one? ( " SELECT * FROM playlists WHERE id = $1 " , plid , as : InvidiousPlaylist )
2019-10-16 17:51:26 +05:30
if ! playlist || playlist . author != user . email && playlist . privacy . private?
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not exist. " )
2019-08-06 05:19:13 +05:30
end
if playlist . author != user . email
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " Invalid user " )
2019-08-06 05:19:13 +05:30
end
PG_DB . exec ( " DELETE FROM playlist_videos * WHERE plid = $1 " , plid )
PG_DB . exec ( " DELETE FROM playlists * WHERE id = $1 " , plid )
env . response . status_code = 204
end
post " /api/v1/auth/playlists/:plid/videos " do | env |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
plid = env . params . url [ " plid " ]
playlist = PG_DB . query_one? ( " SELECT * FROM playlists WHERE id = $1 " , plid , as : InvidiousPlaylist )
2019-10-16 17:51:26 +05:30
if ! playlist || playlist . author != user . email && playlist . privacy . private?
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not exist. " )
2019-08-06 05:19:13 +05:30
end
if playlist . author != user . email
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " Invalid user " )
2019-08-06 05:19:13 +05:30
end
if playlist . index . size >= 500
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Playlist cannot have more than 500 videos " )
2019-08-06 05:19:13 +05:30
end
video_id = env . params . json [ " videoId " ] . try & . as ( String )
if ! video_id
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " Invalid videoId " )
2019-08-06 05:19:13 +05:30
end
begin
video = get_video ( video_id , PG_DB )
rescue ex
2020-11-30 15:29:21 +05:30
next error_json ( 500 , ex )
2019-08-06 05:19:13 +05:30
end
2020-07-26 20:28:50 +05:30
playlist_video = PlaylistVideo . new ( {
title : video . title ,
id : video . id ,
author : video . author ,
ucid : video . ucid ,
2019-08-06 05:19:13 +05:30
length_seconds : video . length_seconds ,
2020-07-26 20:28:50 +05:30
published : video . published ,
plid : plid ,
live_now : video . live_now ,
index : Random :: Secure . rand ( 0 _i64 .. Int64 :: MAX ) ,
} )
2019-08-06 05:19:13 +05:30
video_array = playlist_video . to_a
args = arg_array ( video_array )
PG_DB . exec ( " INSERT INTO playlist_videos VALUES ( #{ args } ) " , args : video_array )
2020-11-01 09:44:46 +05:30
PG_DB . exec ( " UPDATE playlists SET index = array_append(index, $1), video_count = cardinality(index) + 1, updated = $2 WHERE id = $3 " , playlist_video . index , Time . utc , plid )
2019-08-06 05:19:13 +05:30
2020-06-16 03:40:30 +05:30
env . response . headers [ " Location " ] = " #{ HOST_URL } /api/v1/auth/playlists/ #{ plid } /videos/ #{ playlist_video . index . to_u64 . to_s ( 16 ) . upcase } "
2019-08-06 05:19:13 +05:30
env . response . status_code = 201
2020-06-16 03:40:30 +05:30
playlist_video . to_json ( locale , index : playlist . index . size )
2019-08-06 05:19:13 +05:30
end
delete " /api/v1/auth/playlists/:plid/videos/:index " do | env |
2020-11-30 15:29:21 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2019-08-06 05:19:13 +05:30
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
plid = env . params . url [ " plid " ]
index = env . params . url [ " index " ] . to_i64 ( 16 )
playlist = PG_DB . query_one? ( " SELECT * FROM playlists WHERE id = $1 " , plid , as : InvidiousPlaylist )
2019-10-16 17:51:26 +05:30
if ! playlist || playlist . author != user . email && playlist . privacy . private?
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not exist. " )
2019-08-06 05:19:13 +05:30
end
if playlist . author != user . email
2020-11-30 15:29:21 +05:30
next error_json ( 403 , " Invalid user " )
2019-08-06 05:19:13 +05:30
end
if ! playlist . index . includes? index
2020-11-30 15:29:21 +05:30
next error_json ( 404 , " Playlist does not contain index " )
2019-08-06 05:19:13 +05:30
end
PG_DB . exec ( " DELETE FROM playlist_videos * WHERE index = $1 " , index )
2020-11-01 09:44:46 +05:30
PG_DB . exec ( " UPDATE playlists SET index = array_remove(index, $1), video_count = cardinality(index) - 1, updated = $2 WHERE id = $3 " , index , Time . utc , plid )
2019-08-06 05:19:13 +05:30
env . response . status_code = 204
end
# patch "/api/v1/auth/playlists/:plid/videos/:index" do |env|
# TODO: Playlist stub
# end
2019-04-19 02:53:50 +05:30
get " /api/v1/auth/tokens " do | env |
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
scopes = env . get ( " scopes " ) . as ( Array ( String ) )
tokens = PG_DB . query_all ( " SELECT id, issued FROM session_ids WHERE email = $1 " , user . email , as : { session : String , issued : Time } )
JSON . build do | json |
json . array do
tokens . each do | token |
json . object do
json . field " session " , token [ :session ]
json . field " issued " , token [ :issued ] . to_unix
end
end
end
end
end
post " /api/v1/auth/tokens/register " do | env |
user = env . get ( " user " ) . as ( User )
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
case env . request . headers [ " Content-Type " ]?
when " application/x-www-form-urlencoded "
scopes = env . params . body . select { | k , v | k . match ( / ^scopes \ [ \ d+ \ ]$ / ) } . map { | k , v | v }
callback_url = env . params . body [ " callbackUrl " ]?
expire = env . params . body [ " expire " ]? . try & . to_i?
when " application/json "
scopes = env . params . json [ " scopes " ] . as ( Array ) . map { | v | v . as_s }
callback_url = env . params . json [ " callbackUrl " ]? . try & . as ( String )
expire = env . params . json [ " expire " ]? . try & . as ( Int64 )
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Invalid or missing header 'Content-Type' " )
2019-04-19 02:53:50 +05:30
end
if callback_url && callback_url . empty?
callback_url = nil
end
if callback_url
callback_url = URI . parse ( callback_url )
end
if sid = env . get? ( " sid " ) . try & . as ( String )
env . response . content_type = " text/html "
csrf_token = generate_response ( sid , { " :authorize_token " } , HMAC_KEY , PG_DB , use_nonce : true )
next templated " authorize_token "
else
env . response . content_type = " application/json "
superset_scopes = env . get ( " scopes " ) . as ( Array ( String ) )
authorized_scopes = [ ] of String
scopes . each do | scope |
if scopes_include_scope ( superset_scopes , scope )
authorized_scopes << scope
end
end
access_token = generate_token ( user . email , authorized_scopes , expire , HMAC_KEY , PG_DB )
if callback_url
2019-09-24 23:01:33 +05:30
access_token = URI . encode_www_form ( access_token )
2019-04-19 02:53:50 +05:30
if query = callback_url . query
query = HTTP :: Params . parse ( query . not_nil! )
else
query = HTTP :: Params . new
end
query [ " token " ] = access_token
callback_url . query = query . to_s
env . redirect callback_url . to_s
else
access_token
end
end
end
post " /api/v1/auth/tokens/unregister " do | env |
2020-11-30 15:29:21 +05:30
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
2019-04-19 02:53:50 +05:30
env . response . content_type = " application/json "
user = env . get ( " user " ) . as ( User )
scopes = env . get ( " scopes " ) . as ( Array ( String ) )
session = env . params . json [ " session " ]? . try & . as ( String )
session || = env . get ( " session " ) . as ( String )
# Allow tokens to revoke other tokens with correct scope
if session == env . get ( " session " ) . as ( String )
PG_DB . exec ( " DELETE FROM session_ids * WHERE id = $1 " , session )
elsif scopes_include_scope ( scopes , " GET:tokens " )
PG_DB . exec ( " DELETE FROM session_ids * WHERE id = $1 " , session )
else
2020-11-30 15:29:21 +05:30
next error_json ( 400 , " Cannot revoke session #{ session } " )
2019-04-19 02:53:50 +05:30
end
env . response . status_code = 204
end
2018-09-18 05:09:28 +05:30
get " /ggpht/* " do | env |
url = env . request . path . lchop ( " /ggpht " )
2020-05-08 19:30:53 +05:30
headers = HTTP :: Headers { " :authority " = > " yt3.ggpht.com " }
2019-06-23 19:09:14 +05:30
REQUEST_HEADERS_WHITELIST . each do | header |
2019-04-12 03:30:00 +05:30
if env . request . headers [ header ]?
headers [ header ] = env . request . headers [ header ]
end
end
2019-05-26 20:11:12 +05:30
begin
2020-03-07 00:23:35 +05:30
YT_POOL . client & . get ( url , headers ) do | response |
2019-07-03 23:43:40 +05:30
env . response . status_code = response . status_code
2019-05-26 21:23:56 +05:30
response . headers . each do | key , value |
2019-11-25 00:11:47 +05:30
if ! RESPONSE_HEADERS_BLACKLIST . includes? ( key . downcase )
2019-05-26 21:23:56 +05:30
env . response . headers [ key ] = value
end
2019-05-19 17:42:45 +05:30
end
2019-04-12 03:30:00 +05:30
2019-07-03 23:43:40 +05:30
env . response . headers [ " Access-Control-Allow-Origin " ] = " * "
if response . status_code >= 300
2019-07-04 00:24:15 +05:30
env . response . headers . delete ( " Transfer-Encoding " )
2019-05-26 21:23:56 +05:30
break
end
2019-04-12 03:30:00 +05:30
2019-05-26 21:23:56 +05:30
proxy_file ( response , env )
end
2019-05-26 20:11:12 +05:30
rescue ex
end
2019-04-12 03:30:00 +05:30
end
2020-10-25 14:05:16 +05:30
options " /sb/:authority/:id/:storyboard/:index " do | env |
2019-05-03 00:50:19 +05:30
env . response . headers [ " Access-Control-Allow-Origin " ] = " * "
env . response . headers [ " Access-Control-Allow-Methods " ] = " GET, OPTIONS "
env . response . headers [ " Access-Control-Allow-Headers " ] = " Content-Type, Range "
end
2020-10-25 14:05:16 +05:30
get " /sb/:authority/:id/:storyboard/:index " do | env |
authority = env . params . url [ " authority " ]
2019-04-12 03:30:00 +05:30
id = env . params . url [ " id " ]
storyboard = env . params . url [ " storyboard " ]
index = env . params . url [ " index " ]
2020-03-07 00:23:35 +05:30
url = " /sb/ #{ id } / #{ storyboard } / #{ index } ? #{ env . params . query } "
headers = HTTP :: Headers . new
2020-10-25 14:05:16 +05:30
headers [ " :authority " ] = " #{ authority } .ytimg.com "
2019-04-12 03:30:00 +05:30
2019-06-23 19:09:14 +05:30
REQUEST_HEADERS_WHITELIST . each do | header |
2019-03-11 22:13:48 +05:30
if env . request . headers [ header ]?
headers [ header ] = env . request . headers [ header ]
end
end
2018-09-18 05:09:28 +05:30
2019-05-26 20:11:12 +05:30
begin
2020-03-07 00:23:35 +05:30
YT_POOL . client & . get ( url , headers ) do | response |
2019-05-26 21:23:56 +05:30
env . response . status_code = response . status_code
response . headers . each do | key , value |
2019-11-25 00:11:47 +05:30
if ! RESPONSE_HEADERS_BLACKLIST . includes? ( key . downcase )
2019-05-26 21:23:56 +05:30
env . response . headers [ key ] = value
end
2019-05-19 17:42:45 +05:30
end
2018-09-18 05:09:28 +05:30
2020-03-05 02:06:39 +05:30
env . response . headers [ " Connection " ] = " close "
2019-07-03 23:43:40 +05:30
env . response . headers [ " Access-Control-Allow-Origin " ] = " * "
if response . status_code >= 300
2019-07-04 00:24:15 +05:30
env . response . headers . delete ( " Transfer-Encoding " )
2019-05-26 21:23:56 +05:30
break
end
2018-09-18 05:09:28 +05:30
2019-05-26 21:23:56 +05:30
proxy_file ( response , env )
end
2019-05-26 20:11:12 +05:30
rescue ex
end
2018-09-18 05:09:28 +05:30
end
2019-08-17 02:16:37 +05:30
get " /s_p/:id/:name " do | env |
id = env . params . url [ " id " ]
name = env . params . url [ " name " ]
url = env . request . resource
2020-05-08 19:30:53 +05:30
headers = HTTP :: Headers { " :authority " = > " i9.ytimg.com " }
2019-08-17 02:16:37 +05:30
REQUEST_HEADERS_WHITELIST . each do | header |
if env . request . headers [ header ]?
headers [ header ] = env . request . headers [ header ]
end
end
begin
2020-03-07 00:23:35 +05:30
YT_POOL . client & . get ( url , headers ) do | response |
2019-08-17 02:16:37 +05:30
env . response . status_code = response . status_code
response . headers . each do | key , value |
2019-11-25 00:11:47 +05:30
if ! RESPONSE_HEADERS_BLACKLIST . includes? ( key . downcase )
2019-08-17 02:16:37 +05:30
env . response . headers [ key ] = value
end
2019-11-01 21:32:38 +05:30
end
env . response . headers [ " Access-Control-Allow-Origin " ] = " * "
if response . status_code >= 300 && response . status_code != 404
env . response . headers . delete ( " Transfer-Encoding " )
break
end
proxy_file ( response , env )
end
rescue ex
end
end
get " /yts/img/:name " do | env |
headers = HTTP :: Headers . new
REQUEST_HEADERS_WHITELIST . each do | header |
if env . request . headers [ header ]?
headers [ header ] = env . request . headers [ header ]
end
end
begin
YT_POOL . client & . get ( env . request . resource , headers ) do | response |
env . response . status_code = response . status_code
response . headers . each do | key , value |
2019-11-25 00:11:47 +05:30
if ! RESPONSE_HEADERS_BLACKLIST . includes? ( key . downcase )
2019-11-01 21:32:38 +05:30
env . response . headers [ key ] = value
end
2019-08-17 02:16:37 +05:30
end
env . response . headers [ " Access-Control-Allow-Origin " ] = " * "
if response . status_code >= 300 && response . status_code != 404
env . response . headers . delete ( " Transfer-Encoding " )
break
end
proxy_file ( response , env )
end
rescue ex
end
end
2018-09-15 07:54:28 +05:30
get " /vi/:id/:name " do | env |
id = env . params . url [ " id " ]
name = env . params . url [ " name " ]
2020-05-08 19:30:53 +05:30
headers = HTTP :: Headers { " :authority " = > " i.ytimg.com " }
2020-03-07 00:23:35 +05:30
2018-09-15 07:54:28 +05:30
if name == " maxres.jpg "
2020-06-16 03:40:30 +05:30
build_thumbnails ( id ) . each do | thumb |
2020-03-07 00:23:35 +05:30
if YT_POOL . client & . head ( " /vi/ #{ id } / #{ thumb [ :url ] } .jpg " , headers ) . status_code == 200
2018-09-15 07:54:28 +05:30
name = thumb [ :url ] + " .jpg "
break
end
end
end
url = " /vi/ #{ id } / #{ name } "
2019-06-23 19:09:14 +05:30
REQUEST_HEADERS_WHITELIST . each do | header |
2019-03-11 22:13:48 +05:30
if env . request . headers [ header ]?
headers [ header ] = env . request . headers [ header ]
end
end
2018-09-15 07:54:28 +05:30
2019-05-26 20:11:12 +05:30
begin
2020-03-07 00:23:35 +05:30
YT_POOL . client & . get ( url , headers ) do | response |
2019-05-26 21:23:56 +05:30
env . response . status_code = response . status_code
response . headers . each do | key , value |
2019-11-25 00:11:47 +05:30
if ! RESPONSE_HEADERS_BLACKLIST . includes? ( key . downcase )
2019-05-26 21:23:56 +05:30
env . response . headers [ key ] = value
end
2019-05-19 17:42:45 +05:30
end
2018-09-15 07:54:28 +05:30
2019-07-03 23:43:40 +05:30
env . response . headers [ " Access-Control-Allow-Origin " ] = " * "
2019-07-04 00:24:15 +05:30
if response . status_code >= 300 && response . status_code != 404
env . response . headers . delete ( " Transfer-Encoding " )
2019-05-26 21:23:56 +05:30
break
end
2018-09-15 07:54:28 +05:30
2019-05-26 21:23:56 +05:30
proxy_file ( response , env )
end
2019-05-26 20:11:12 +05:30
rescue ex
end
2018-04-16 07:17:37 +05:30
end
2019-10-27 09:49:05 +05:30
get " /Captcha " do | env |
2020-05-08 19:30:53 +05:30
headers = HTTP :: Headers { " :authority " = > " accounts.google.com " }
response = YT_POOL . client & . get ( env . request . resource , headers )
2019-10-27 09:49:05 +05:30
env . response . headers [ " Content-Type " ] = response . headers [ " Content-Type " ]
response . body
end
2019-08-22 04:53:20 +05:30
# Undocumented, creates anonymous playlist with specified 'video_ids', max 50 videos
2019-05-03 19:41:27 +05:30
get " /watch_videos " do | env |
2019-10-25 22:28:16 +05:30
response = YT_POOL . client & . get ( env . request . resource )
2019-05-03 19:41:27 +05:30
if url = response . headers [ " Location " ]?
2021-02-01 00:22:32 +05:30
url = URI . parse ( url ) . request_target
2019-05-03 19:41:27 +05:30
next env . redirect url
end
env . response . status_code = response . status_code
end
2018-02-10 20:45:23 +05:30
error 404 do | env |
2019-03-27 15:58:53 +05:30
if md = env . request . path . match ( / ^ \/ (?<id>([a-zA-Z0-9_-]{11})|( \ w+))$ / )
2019-04-18 01:16:00 +05:30
item = md [ " id " ]
2018-10-07 08:49:36 +05:30
2019-04-18 01:16:00 +05:30
# Check if item is branding URL e.g. https://youtube.com/gaming
2019-10-25 22:28:16 +05:30
response = YT_POOL . client & . get ( " / #{ item } " )
2019-03-27 15:58:53 +05:30
if response . status_code == 301
2021-02-01 00:22:32 +05:30
response = YT_POOL . client & . get ( URI . parse ( response . headers [ " Location " ] ) . request_target )
2019-03-27 15:58:53 +05:30
end
2019-06-07 23:12:07 +05:30
if response . body . empty?
env . response . headers [ " Location " ] = " / "
halt env , status_code : 302
end
2019-03-27 15:58:53 +05:30
html = XML . parse_html ( response . body )
2020-01-14 18:51:17 +05:30
ucid = html . xpath_node ( % q ( / / link [ @rel = " canonical " ] ) ) . try & . [ " href " ] . split ( " / " ) [ - 1 ]
2019-03-27 15:58:53 +05:30
if ucid
2020-01-14 18:51:17 +05:30
env . response . headers [ " Location " ] = " /channel/ #{ ucid } "
2019-03-27 15:58:53 +05:30
halt env , status_code : 302
end
2018-10-07 08:49:36 +05:30
params = [ ] of String
env . params . query . each do | k , v |
params << " #{ k } = #{ v } "
end
params = params . join ( " & " )
2019-04-18 01:16:00 +05:30
url = " /watch?v= #{ item } "
2018-10-07 08:49:36 +05:30
if ! params . empty?
url += " & #{ params } "
end
2019-04-18 01:16:00 +05:30
# Check if item is video ID
2019-10-25 22:28:16 +05:30
if item . match ( / ^[a-zA-Z0-9_-]{11}$ / ) && YT_POOL . client & . head ( " /watch?v= #{ item } " ) . status_code != 404
2019-02-22 02:37:22 +05:30
env . response . headers [ " Location " ] = url
halt env , status_code : 302
end
end
2019-01-13 00:48:08 +05:30
env . response . headers [ " Location " ] = " / "
halt env , status_code : 302
2017-12-31 02:51:43 +05:30
end
2020-11-30 15:29:21 +05:30
error 500 do | env , ex |
locale = LOCALES [ env . get ( " preferences " ) . as ( Preferences ) . locale ]?
error_template ( 500 , ex )
2017-12-31 02:51:43 +05:30
end
2018-03-09 22:58:57 +05:30
static_headers do | response , filepath , filestat |
2019-05-08 19:28:10 +05:30
response . headers . add ( " Cache-Control " , " max-age=2629800 " )
2018-03-09 22:58:57 +05:30
end
2017-11-23 13:18:55 +05:30
public_folder " assets "
2018-04-16 09:26:58 +05:30
2018-07-31 05:12:45 +05:30
Kemal . config . powered_by_header = false
2018-04-16 09:26:58 +05:30
add_handler FilteredCompressHandler . new
2019-02-03 10:18:47 +05:30
add_handler APIHandler . new
2019-04-19 02:53:50 +05:30
add_handler AuthHandler . new
2019-03-23 20:54:30 +05:30
add_handler DenyFrame . new
2019-04-19 02:53:50 +05:30
add_context_storage_type ( Array ( String ) )
2019-02-24 21:19:48 +05:30
add_context_storage_type ( Preferences )
2019-04-19 02:53:50 +05:30
add_context_storage_type ( User )
2017-11-23 13:18:55 +05:30
2021-01-04 21:21:06 +05:30
Kemal . config . logger = LOGGER
2019-09-23 22:35:29 +05:30
Kemal . config . host_binding = Kemal . config . host_binding != " 0.0.0.0 " ? Kemal . config . host_binding : CONFIG . host_binding
Kemal . config . port = Kemal . config . port != 3000 ? Kemal . config . port : CONFIG . port
2017-11-23 13:18:55 +05:30
Kemal . run