Add ability to automatically extract video signature funtion

This commit is contained in:
Omar Roth 2018-07-17 10:53:17 -05:00
parent e1ec1b6837
commit 0bb07e55f5
2 changed files with 73 additions and 10 deletions

View file

@ -173,7 +173,6 @@ video_threads.times do |i|
end
top_videos = [] of Video
spawn do
if CONFIG.dl_api_key
DetectLanguage.configure do |config|
@ -213,6 +212,20 @@ spawn do
end
end
# Refresh decrypt function
decrypt_function = [] of {name: String, value: Int32}
spawn do
loop do
begin
client = make_client(YT_URL)
decrypt_function = update_decrypt_function(client)
rescue ex
end
Fiber.yield
end
end
before_all do |env|
if env.request.cookies.has_key? "SID"
headers = HTTP::Headers.new
@ -296,11 +309,11 @@ get "/watch" do |env|
if adaptive_fmts[0]? && adaptive_fmts[0]["s"]?
adaptive_fmts.each do |fmt|
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"])
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"], decrypt_function)
end
fmt_stream.each do |fmt|
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"])
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"], decrypt_function)
end
end
@ -433,11 +446,11 @@ get "/embed/:id" do |env|
if adaptive_fmts[0]? && adaptive_fmts[0]["s"]?
adaptive_fmts.each do |fmt|
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"])
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"], decrypt_function)
end
fmt_stream.each do |fmt|
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"])
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"], decrypt_function)
end
end
@ -1044,7 +1057,7 @@ get "/api/manifest/dash/id/:id" do |env|
if adaptive_fmts[0]? && adaptive_fmts[0]["s"]?
adaptive_fmts.each do |fmt|
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"])
fmt["url"] += "&signature=" + decrypt_signature(fmt["s"], decrypt_function)
end
end