iv-org
/
invidious
mirror of https://github.com/iv-org/invidious.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use youtubei API for trending

This commit is contained in:
bopol 2021-06-24 00:08:40 +02:00
parent 5a8825d016
commit 50267a6dd6
2 changed files with 15 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/invidious/helpers/youtube_api.cr
View File

@ -25,12 +25,14 @@ end
####################################################################
# request_youtube_api_browse(continuation)
# request_youtube_api_browse(browse_id, params)
# request_youtube_api_browse(browse_id, params, region)
#
# Requests the youtubei/v1/browse endpoint with the required headers
# and POST data in order to get a JSON reply in english US that can
# and POST data in order to get a JSON reply in english that can
# be easily parsed.
#
# The region can be provided, default is US.
#
# The requested data can either be:
#
# - A continuation token (ctoken). Depending on this token's
@ -49,11 +51,11 @@ def request_youtube_api_browse(continuation : String)
return _youtube_api_post_json("/youtubei/v1/browse", data)
end
def request_youtube_api_browse(browse_id : String, params : String)
def request_youtube_api_browse(browse_id : String, params : String, region : String = "US")
# JSON Request data, required by the API
data = {
"browseId" => browse_id,
"context" => make_youtube_api_context("US"),
"context" => make_youtube_api_context(region),
}
# Append the additionnal parameters if those were provided

30
src/invidious/trending.cr
View File

@ -2,31 +2,19 @@ def fetch_trending(trending_type, region, locale)
region ||= "US"
region = region.upcase
trending = ""
plid = nil
if trending_type && trending_type != "Default"
if trending_type == "Music"
trending_type = 1
elsif trending_type == "Gaming"
trending_type = 2
elsif trending_type == "Movies"
trending_type = 3
end
response = YT_POOL.client &.get("/feed/trending?gl=#{region}&hl=en").body
initial_data = extract_initial_data(response)
url = initial_data["contents"]["twoColumnBrowseResultsRenderer"]["tabs"][trending_type]["tabRenderer"]["endpoint"]["commandMetadata"]["webCommandMetadata"]["url"]
url = "#{url}&gl=#{region}&hl=en"
trending = YT_POOL.client &.get(url).body
plid = extract_plid(url)
else
trending = YT_POOL.client &.get("/feed/trending?gl=#{region}&hl=en").body
if trending_type == "Music"
params = "4gINGgt5dG1hX2NoYXJ0cw%3D%3D"
elsif trending_type == "Gaming"
params = "4gIcGhpnYW1pbmdfY29ycHVzX21vc3RfcG9wdWxhcg%3D%3D"
elsif trending_type == "Movies"
params = "4gIKGgh0cmFpbGVycw%3D%3D"
else # Default
params = ""
end
initial_data = extract_initial_data(trending)
initial_data = request_youtube_api_browse("FEtrending", params: params, region: region)
trending = extract_videos(initial_data)
return {trending, plid}