suby 0.0.3 → 0.0.4

data/bin/suby

@@ -1,8 +1,30 @@
 #!/usr/bin/env ruby
 
-if ARGV.include? '-h'
-  puts "#{File.basename $0} video"
-  puts "  Find and download subtitles for the given video file"
+require 'optparse'
+
+options = {
+  lang: 'en'
+}
+
+option_parser = OptionParser.new do |opts|
+   opts.banner = "#{File.basename $0} [options] video"
+   opts.separator '  Find and download subtitles for the given video file'
+   opts.separator "\nOptions:"
+
+   opts.on '-l', '--lang LANG = en', 'Lang for subtitles' do |lang|
+    options[:lang] = lang
+   end
+
+   opts.on '-h', '--help', 'Show usage' do
+     puts opts
+     exit
+   end
+end
+
+option_parser.parse!
+
+if ARGV.empty?
+  puts option_parser
   exit
 end
 
@@ -13,7 +35,6 @@ SUB_EXTENSIONS = %w[srt sub]
 archive = '__archive__'
 site = 'www.tvsubtitles.net'
 search_url = '/search.php'
-lang = 'en'
 
 def extract_subs_from_archive(archive)
   case `file #{archive}`
@@ -41,29 +62,29 @@ ARGV.each { |file|
   next(puts "Skipping: #{file}") if SUB_EXTENSIONS.any? { |ext|
     File.exist? File.basename(file, File.extname(file)) + ".#{ext}" }
 
-  unless /^(?<tvshow>.+) (?<season>\d{1,2})x(?<episode>\d{1,2}) - .+\.[a-z]+?$/ =~ file
-    next puts "wrong file format (#{file}). Must be:\n<tvshow> <season>x<episode> - <title>.<ext>"
+  unless /^(?<show>.+) (?<season>\d{1,2})x(?<episode>\d{1,2}) - .+\.[a-z]+?$/ =~ file
+    next puts "wrong file format (#{file}). Must be:\n<show> <season>x<episode> - <title>.<ext>"
   end
   season, episode = [season, episode].map(&:to_i)
   puts "Searching subtitles for #{file}:"
-  puts "Show: #{tvshow}, Season: #{season}, Episode: #{episode}"
+  puts "Show: #{show}, Season: #{season}, Episode: #{episode}"
 
   Net::HTTP.start(site) { |tvsubtitles|
-    # search tvshow
-    tvshow_url = (show_urls[tvshow] ||= begin
+    # search show
+    show_url = (show_urls[show] ||= begin
       post = Net::HTTP::Post.new(search_url)
-      post.form_data = { 'q' => tvshow }
+      post.form_data = { 'q' => show }
       results = Nokogiri tvsubtitles.request(post).body
       url = results.css('ul li div a').first[:href]
 
       raise 'could not find the show' unless /^\/tvshow-(\d+)\.html$/ =~ url
       "/tvshow-#{$1}-#{season}.html"
     end)
-    puts "tvshow url: #{tvshow_url}"
+    puts "show url: #{show_url}"
 
     # search episode
-    tvshow = (show_pages[tvshow] ||= Nokogiri tvsubtitles.get(tvshow_url).body)
-    tr = tvshow.css('div.left_articles table tr').find { |tr|
+    show = (show_pages[show] ||= Nokogiri tvsubtitles.get(show_url).body)
+    tr = show.css('div.left_articles table tr').find { |tr|
       tr.children.find { |td| td.name == 'td' && td.text =~ /\A#{season}x0?#{episode}\z/ }
     }
     episode_url = nil
@@ -74,7 +95,7 @@ ARGV.each { |file|
     }
 
     raise "invalid episode url: #{episode_url}" unless episode_url =~ /^episode-(\d+)\.html$/
-    episode_url = "/episode-#{$1}-#{lang}.html"
+    episode_url = "/episode-#{$1}-#{options[:lang]}.html"
     puts "episode url: #{episode_url}"
 
     # subtitles

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: suby
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-06-15 00:00:00.000000000 Z
+date: 2011-06-19 00:00:00.000000000 Z
 dependencies: []
 description: Find and download subtitles
 email: