rdelaage
/
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(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 # 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. # be easily parsed.
# #
# The region can be provided, default is US.
#
# The requested data can either be: # The requested data can either be:
# #
# - A continuation token (ctoken). Depending on this token's # - 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) return _youtube_api_post_json("/youtubei/v1/browse", data)
end 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 # JSON Request data, required by the API
data = { data = {
"browseId" => browse_id, "browseId" => browse_id,
"context" => make_youtube_api_context("US"), "context" => make_youtube_api_context(region),
} }
# Append the additionnal parameters if those were provided # 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 ||= "US"
region = region.upcase region = region.upcase
trending = ""
plid = nil plid = nil
if trending_type && trending_type != "Default" if trending_type == "Music"
if trending_type == "Music" params = "4gINGgt5dG1hX2NoYXJ0cw%3D%3D"
trending_type = 1 elsif trending_type == "Gaming"
elsif trending_type == "Gaming" params = "4gIcGhpnYW1pbmdfY29ycHVzX21vc3RfcG9wdWxhcg%3D%3D"
trending_type = 2 elsif trending_type == "Movies"
elsif trending_type == "Movies" params = "4gIKGgh0cmFpbGVycw%3D%3D"
trending_type = 3 else # Default
end params = ""
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
end end
initial_data = extract_initial_data(trending) initial_data = request_youtube_api_browse("FEtrending", params: params, region: region)
trending = extract_videos(initial_data) trending = extract_videos(initial_data)
return {trending, plid} return {trending, plid}