Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

mp4decode download is working now #10

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

mp4decode download is working now #10

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
38 changes: 27 additions & 11 deletions train_portal/ice/videos.rb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,10 @@
require 'open-uri'
require 'zip'

if OpenSSL::SSL.const_defined?(:OP_IGNORE_UNEXPECTED_EOF)
OpenSSL::SSL::SSLContext::DEFAULT_PARAMS[:options] |= OpenSSL::SSL::OP_IGNORE_UNEXPECTED_EOF
end

module TrainPortal::Ice
module Videos
module_function
Expand Down Expand Up @@ -68,14 +72,16 @@ def download_video(manifests, file_name)
# mpd_manifest = read_mpd_manifest(manifests.detect { |url| url.end_with?('manifest.mpd') })
manifest_mpd_url = manifests.detect { |url| url.end_with?('manifest.mpd') }

file_name = full_path(file_name)
# yt_dlp(manifest_mpd_url, file_name)
file_name = Dir.glob("#{file_name}*.mp4").first
yt_dlp(manifest_mpd_url, full_path(file_name))
file_name = full_path(file_name, 'mp4')
mp4decode(file_name, manifest_mpd_url)
reencode(file_name)
end

def yt_dlp(download_url, file_name)
raise ArgumentError, 'file_name is required' if file_name.nil?
return if full_path(file_name, 'mp4') # file exists

unless system('which yt-dlp > /dev/null 2>&1')
raise "yt-dlp is not installed or not found in PATH"
end
Expand Down Expand Up @@ -126,15 +132,18 @@ def mp4decode(input_file, manifest_mpd_url)
mp4decrypt_path = File.join(tmp_dir, 'bin', 'mp4decrypt')
download_mp4decrypt(mp4decrypt_path) unless File.exist?(mp4decrypt_path)
end
binding.irb

# Extract the decryption key from the manifest.mpd
manifest = Nokogiri::XML(URI.open(manifest_mpd_url).read)
key_id = manifest.at_xpath('//cenc:default_KID', 'cenc' => 'urn:mpeg:cenc:2013').content
pssh = manifest.at_xpath('//cenc:pssh', 'cenc' => 'urn:mpeg:cenc:2013').content
manifest.remove_namespaces!
video_definitions = manifest.css('AdaptationSet[contentType="video"]')
max_quality_video = video_definitions.max_by { |v| v['maxWidth'] }
crypto_parameters = max_quality_video.css('ContentProtection[default_KID]').first
key_id = crypto_parameters['default_KID']
crypto_mode = crypto_parameters['value']
scheme_id_uri = crypto_parameters['schemeIdUri']
pssh = max_quality_video.css('ContentProtection pssh').first.content

# Here you would need to implement the logic to derive the key from the pssh
# For now, we'll assume you have a method to get the key
binding.irb
key = get_decryption_key(pssh)

output_file = input_file.sub(/\.\w+$/, '_dec.mp4')
Expand Down Expand Up @@ -166,9 +175,15 @@ def get_manifests(details) = details['video']['hydra:member'].flat_map { |vh| vh
def read_m3u8(url) = M3u8::Playlist.read(URI.open(url).read)
def read_mpd_manifest(url) = Nokogiri::XML(URI.open(url).read)
def sanitize_string(string) = string.to_url(force_downcase: false).gsub('-', '_')
def full_path(sub_path) = TrainPortal.download_directory(File.join('videos', sub_path))
def tmp_dir = TrainPortal.download_directory('tmp')

def full_path(sub_path, extension = nil)
result = TrainPortal.download_directory(File.join('videos', sub_path))
return result if extension.nil?

Dir.glob("#{result}*.#{extension}").first
end

def download_mp4decrypt(destination_path)
puts 'downloading mp4decrypt'
# Download and unpack Bento4 SDK
Expand All @@ -188,7 +203,8 @@ def download_mp4decrypt(destination_path)
zip_path = File.join(tmp_dir, 'bento4.zip')

File.open(zip_path, 'wb') do |file|
file.write(URI.open(bento_url).open(open_timeout: READ_TIMEOUT, read_timeout: READ_TIMEOUT).read)
uri = URI.parse(bento_url).open(open_timeout: READ_TIMEOUT, read_timeout: READ_TIMEOUT)
file.write(uri.read)
end

Zip::File.open(zip_path) do |zip_file|
Expand Down