video-dl 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: aa500ccd48ac4dd53b0f533d9152a2568bea2ea7
-  data.tar.gz: acebbe336a0b629330a88c3dc9da4c6db8f8672a
+  metadata.gz: f5689a2a59b76619ccb4bea029c53f7aec696acd
+  data.tar.gz: 5de666c235a1f91203e4b29aeea253796bf16032
 SHA512:
-  metadata.gz: 408e28e321bcdb2d6ea7532b2e2e9fe2ed82c072722444e7657e73e6b38738433b061247e2df640af6b3b8c109ef00a11b2107641c652c735b418cac7401b4c2
-  data.tar.gz: ee75592ed19318366d46dbfacb93ade891344640fe0d2a34203773edfe9a02e3f91d4e5500e4375818d242cc97ad3ef12ae24da556992d4a3f8ba47bb48f105b
+  metadata.gz: 319e22fba01fd4dd11cdd71745790a19517be7af2f4e859f91c76c22ae8778b32c00010582d5824df29bdb166240a27b95ec55df4f8d5cf8f1eb14b6e8a80d9b
+  data.tar.gz: 9a938fcdc5783d5be6a01d6855ba155404e6b27f3d13dd0ea1496367473539462e6de0901a962d08b57406b6bc1524cbd67d6d700f3b2983bc22c031ccc385ee

data/bin/video-dl

@@ -3,9 +3,9 @@
 require 'rubygems'
 require "bundler/setup"
 require 'commander/import'
-require 'json'
 require 'open-uri'
 require 'nokogiri'
+require 'uri'
 
 program :name, "video-dl"
 program :version, '1.0.0'
@@ -14,16 +14,34 @@ program :description, 'Download video easily'
 default_command :download
 
 command :download do |c|
-    c.syntax = 'video-dl download http://some.url'
-    c.description = 'Download a video'
-    c.option '--bar STRING', String, 'Option bar with a string'
-    c.action do |args, options|
-    	options.default :bar => 'BAR'
-    	# puts "Args: #{args}"
-    	# puts "Option Bar: #{options.bar}"
-    	url = args[0]
-    	download_video(url)
-    end
+  c.syntax = 'video-dl download http://some.url'
+  c.description = 'Download a video'
+  c.option '--bar STRING', String, 'Option bar with a string'
+  c.action do |args, options|
+  	options.default :bar => 'BAR'
+  	# puts "Args: #{args}"
+  	# puts "Option Bar: #{options.bar}"
+  	url = args[0]
+  	download_video(url)
+  end
+end
+
+command :episodes do |c|
+  c.syntax = 'video-dl episodes http://some.url'
+  c.description = 'Download all the episodes of a drama'
+  c.action do |args, options|
+  	url = args[0]
+  	download_all_episodes(url)
+  end
+end
+
+command :html do |c|
+  c.syntax = 'video-dl html http://some.url'
+  c.description = 'Display the html of the url'
+  c.action do |args, options|
+    url = args[0]
+  	display_html(url)
+  end
 end
 
 
@@ -33,25 +51,107 @@ end
 # Video Download
 ########################################################################################
 
+def display_html(url)
+  doc = Nokogiri::HTML(open(url))
+  puts doc
+end
+
 def download_video(url)
-  puts "Downloading #{url}"
+  puts "Downloading from #{url}"
 
+  # Read the url HTML
   doc = Nokogiri::HTML(open(url))
   # puts doc
 
+  # Find the video title
   title = doc.css("h1[class='title']")[0].content
   puts "Title: #{title}"
 
+  # doc.css("source[type='video/mp4']").each { |source|
+  #   puts "Source: #{source['src']}"
+  # }
+
+  sources = doc.css("source[type='video/mp4']")
+  videoUrl = findHighestRes(sources)
+  if videoUrl != nil
+    download(title, videoUrl)
+    return
+  end
+
+  # Otherwise, video is in an iframe
   iframeSrc = doc.css('div#servermobile > iframe')[0]['src']
-  # puts iframeSrc
+  puts "iframe src: #{iframeSrc}"
 
+  # Read the iframe HTML
   doc = Nokogiri::HTML(open(iframeSrc))
   # puts doc
 
-  videoUrl = doc.css("video#my-video > source[label='720']")[0]['src']
-  puts "Video is at #{videoUrl} \n\n"
+  # Finally we have the video url
+  sources = doc.css("source[type='video/mp4']")
+  videoUrl = findHighestRes(sources)
+  if videoUrl != nil
+    download(title, videoUrl)
+    return
+  else
+    puts "Failed to find video url"
+  end
+end
+
 
+# Run youtube-dl shell command
+def download(title, videoUrl)
+  puts "Video is at #{videoUrl}\n\n"
   cmd = "youtube-dl -o #{Shellwords.escape(title)}.mp4 #{Shellwords.escape(videoUrl)}"
   puts "Running #{cmd}"
-  puts system(cmd)
+  system(cmd)
+end
+
+
+# Find the highest resolution source from an array of sources like this:
+# <source src="http://3.bp.blogspot.com/xyz" type="video/mp4" label="720"></source>
+# Return the video url
+def findHighestRes(sources)
+  if sources.length > 0
+    highestResSource = nil
+    sources.each { |source|
+      if highestResSource == nil
+        highestResSource = source
+      else
+        if source['label'].to_i > highestResSource['label'].to_i
+          highestResSource = source
+        end
+      end
+    }
+    # puts "Highest Res Source: #{highestResSource}"
+    videoUrl = highestResSource['src']
+    return videoUrl
+  end
+  return nil
+end
+
+
+def download_all_episodes(url)
+  puts "Downloading all episodes from #{url}"
+
+  uri = URI.parse(url)
+  domain = "#{uri.scheme}://#{uri.host}"
+  # puts "Domain: #{domain}"
+
+  # Read the url HTML
+  doc = Nokogiri::HTML(open(url))
+
+  # Find the episode list
+  episodes = doc.css("select[class='gotolistepisode'] > option")
+  episodeUrls = []
+  puts "Episodes:"
+  episodes.reverse.each { |episode|
+    episodeUrl = "#{domain}#{episode['value']}"
+    puts "\tEpisode URL: #{episodeUrl}"
+    episodeUrls.push(episodeUrl)
+  }
+
+  # Download each
+  episodeUrls.each { |episodeUrl|
+    download_video(episodeUrl)
+  }
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: video-dl
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Junda