autosub 0.4.0

data/.gitignore

@@ -0,0 +1,27 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC
+pkg
+doc
+tasks
+test
+rep
+fixtures

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Pirate
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.markdown

@@ -0,0 +1,38 @@
+# AutoSub
+  
+Ruby tool to automatically download subtitles (srt) inside your TV Shows folder.
+Currently only english and french subtitles are supported.
+  
+## Install
+
+    sudo gem install pirate-autosub --source http://gems.github.com
+
+## Usage
+
+In your terminal:
+
+    autosub -p /path/of/your/TV Shows/folder -lang fr,en
+
+more details with:
+
+    autosub -h
+    
+## Folder Name Convention
+
+AutoNZB use (and needs) specific folders/episodes name for your TV Shows:
+  
+    + TV Shows
+      + Dexter (TV Show's name needs to be 'clean')
+        + Season 01
+          - Dexter S01E01.mkv (.mkv for 720p HD episode)
+          - Dexter 1x02.avi (.avi for SD episode)
+          - S01E03.mkv (TV Show's name in episode file's name not needed)
+          - S01E03.fr.srt (.fr.srt used to define subtitle's language)
+    + Fringe
+      - (season folder not needed)
+      - Fringe S01E01.mkv
+      - 1x01.avi
+  
+## Copyright
+
+Copyright (c) 2009 Pirate

data/Rakefile

@@ -0,0 +1,54 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "autosub"
+    gem.summary = "Automatically download srt for your TV Shows"
+    gem.description = "Ruby tool to automatically download subtitles (srt) inside your TV Shows folder"
+    gem.email = "pirate.2061@gmail.com"
+    gem.homepage = "http://github.com/pirate/autosub"
+    gem.authors = ["Pirate"]
+    gem.add_development_dependency "rspec", ">= 1.2.9"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+    gem.add_dependency "hpricot", ">= 0.8.1"
+    gem.add_dependency "httparty", ">= 0.4.5"
+    gem.add_dependency "optiflag", ">= 0.7"
+    gem.add_dependency "rubyzip", ">= 0.9.1"
+    gem.add_dependency "mechanize", ">= 0.9.3"
+    gem.add_dependency "simple-rss", ">= 1.2.2"
+    gem.default_executable = "bin/autosub"
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "autosub #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+Dir["#{File.dirname(__FILE__)}/tasks/*.rake"].sort.each { |ext| load ext }

data/VERSION

@@ -0,0 +1 @@
+0.4.0

data/autosub.gemspec

@@ -0,0 +1,75 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{autosub}
+  s.version = "0.4.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Pirate"]
+  s.date = %q{2009-11-06}
+  s.default_executable = %q{bin/autosub}
+  s.description = %q{Ruby tool to automatically download subtitles (srt) inside your TV Shows folder}
+  s.email = %q{pirate.2061@gmail.com}
+  s.executables = ["autosub"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.markdown"
+  ]
+  s.files = [
+    ".gitignore",
+     "LICENSE",
+     "README.markdown",
+     "Rakefile",
+     "VERSION",
+     "asset/failure.png",
+     "asset/srt.png",
+     "autosub.gemspec",
+     "bin/autosub",
+     "lib/episode.rb",
+     "lib/inspector.rb",
+     "lib/sites/betaseries.rb",
+     "lib/sites/podnapisi.rb",
+     "lib/sites/seriessub.rb",
+     "lib/sites/tvsubtitle.rb"
+  ]
+  s.homepage = %q{http://github.com/pirate/autosub}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Automatically download srt for your TV Shows}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_runtime_dependency(%q<hpricot>, [">= 0.8.1"])
+      s.add_runtime_dependency(%q<httparty>, [">= 0.4.5"])
+      s.add_runtime_dependency(%q<optiflag>, [">= 0.7"])
+      s.add_runtime_dependency(%q<rubyzip>, [">= 0.9.1"])
+      s.add_runtime_dependency(%q<mechanize>, [">= 0.9.3"])
+      s.add_runtime_dependency(%q<simple-rss>, [">= 1.2.2"])
+    else
+      s.add_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_dependency(%q<hpricot>, [">= 0.8.1"])
+      s.add_dependency(%q<httparty>, [">= 0.4.5"])
+      s.add_dependency(%q<optiflag>, [">= 0.7"])
+      s.add_dependency(%q<rubyzip>, [">= 0.9.1"])
+      s.add_dependency(%q<mechanize>, [">= 0.9.3"])
+      s.add_dependency(%q<simple-rss>, [">= 1.2.2"])
+    end
+  else
+    s.add_dependency(%q<rspec>, [">= 1.2.9"])
+    s.add_dependency(%q<hpricot>, [">= 0.8.1"])
+    s.add_dependency(%q<httparty>, [">= 0.4.5"])
+    s.add_dependency(%q<optiflag>, [">= 0.7"])
+    s.add_dependency(%q<rubyzip>, [">= 0.9.1"])
+    s.add_dependency(%q<mechanize>, [">= 0.9.3"])
+    s.add_dependency(%q<simple-rss>, [">= 1.2.2"])
+  end
+end
+

data/bin/autosub

@@ -0,0 +1,26 @@
+#!/usr/bin/env ruby
+
+require 'optiflag'
+require File.join(File.dirname(__FILE__), '..', 'lib', 'inspector')
+
+module AutoSub extend OptiFlagSet
+  flag "p" do
+    description "Directory path with many TV Shows directories"
+  end
+  optional_flag "lang" do
+    description "Language(s) you want for your subtitles, default: fr,en
+                  (Order is important to define if a srt is needed)"
+  end
+  optional_flag "log" do
+    description "path for log file"
+  end
+
+  and_process!
+end
+
+begin
+  inspector = Inspector.new(ARGV.flags.p, :lang => ARGV.flags.lang, :log_path => ARGV.flags.log)
+rescue => e
+  p e.to_s
+  Inspector.growl("AutoSub Error!", 'look into the console log', 'failure.png')
+end

data/lib/episode.rb

@@ -0,0 +1,89 @@
+class Episode
+  
+  attr_accessor :path, :name, :ext, :format, :srt, :tv_show, :season, :number
+  
+  def initialize(tv_show, path)
+    @path = path
+    @ext  = File.extname(@path)
+    @name = File.basename(path, @ext)
+    @tv_show = tv_show.gsub(/\s\([0-9]{4}\)/,'').chomp(' ')
+    @srt = []
+    
+    search_existing_srt
+    define_format
+    define_season_and_episode_number
+  end
+  
+  def filename
+    @name + @ext
+  end
+  
+  def episode_name
+    "#{@tv_show} S#{@season < 10 ? "0#{@season}" : @season}E#{@number < 10 ? "0#{@number}" : @number}"
+  end
+  
+  def episode_name_with_format    
+    episode_name + "#{format == 'hd' ? ' 720p' : ''}"
+  end
+  
+  def need_srt?(langs)
+    need = -1
+    langs = langs.reverse
+    langs.each do |lang|
+      need = langs.index(lang) if @srt.include?(lang) && need < langs.index(lang)
+    end
+    need != langs.size - 1
+  end
+  
+  def srt_filename(lang)
+    "#{File.dirname(path)}/#{name}.#{lang}.srt"
+  end
+  
+  def self.valid?(filename)
+    filename.match(/.*(([0-9]{1,2}X[0-9]{1,2})|(S[0-9][0-9]E[0-9][0-9])).*((\.avi)|(\.mkv))$/i)
+  end
+  
+  def good_format?(filename)
+    case @format
+    when 'hd'
+      filename.include?('720p') || filename.include?('720')
+    when 'sd'
+      !filename.include?('720p')
+    end
+  end
+  
+private
+  
+  def search_existing_srt
+    dir = Dir.new(File.dirname(@path))
+    dir.select { |el| File.extname(el) == '.srt' && el.include?(@name) && !el.include?('._') }.each do |srt|
+      case srt
+      when /\.(french|fr|fre|fra)\.srt/i
+        @srt << 'fr'
+      when /\.(english|en|eng)\.srt/i
+        @srt << 'en'
+      end
+    end
+  end
+  
+  def define_format
+    case @ext
+    when '.mkv'
+      @format = 'hd'
+    when '.avi'
+      @format = 'sd'
+    end
+  end
+  
+  def define_season_and_episode_number
+    case @name
+    when /S[0-9][0-9]E[0-9][0-9]/i # S02E03
+      @season = @name.match(/.*S([0-9]{1,2})E.*$/i)[1].to_i
+      @number = @name.match(/.*E([0-9]{1,2}).*$/i)[1].to_i
+    when /[0-9]{1,2}X[0-9]{1,2}/i  # 2X03
+      @season = @name.match(/.*([0-9]{1,2})X[0-9]{1,2}.*$/i)[1].to_i
+      @number = @name.match(/.*[0-9]{1,2}X([0-9]{1,2}).*$/i)[1].to_i
+    end
+  end
+  
+end

data/lib/inspector.rb

@@ -0,0 +1,104 @@
+require File.join(File.dirname(__FILE__), '..', 'lib', 'episode')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'betaseries')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'seriessub')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'tvsubtitle')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'podnapisi')
+require 'logger'
+require "pathname"
+
+class Inspector
+  
+  attr_accessor :tv_shows, :episodes, :langs, :log
+  
+  def initialize(path , options = {})
+    @path = Pathname.new(path).realpath.to_s
+    @langs = options[:lang] ? options[:lang].split(',') : ['fr','en']
+    
+    # Initialize Logger
+    @log = Logger.new(options[:log_path] || "#{@path}/_autosub_.log" )
+    @log.datetime_format = "%Y-%m-%d %H:%M:%S "
+    log.info "Begin searching for '#{langs.join(',')}' subtitles..."
+    
+    $stdout.print "--- Searching for '#{langs.join(',')}' subtitles ---\n"
+    @tv_shows = [] # {:name => ..., :path => ...}
+    initialize_tv_shows
+    $stdout.print "Found #{@tv_shows.size} TV Show(s) in #{@path}\n"
+    
+    @episodes = []
+    initialize_episodes
+    # keep only episodes who needs srt
+    $stdout.print "Found #{@episodes.select { |e| e.need_srt?(@langs) }.size} episode(s) who need(s) srt\n"
+    
+    # Find new srt
+    begin
+      BetaSeries.new(self)
+    rescue => e
+      log.fatal "Problem with BetaSeries: #{e}"
+    end
+    # begin
+    #   TVSubtitle.new(self)
+    # rescue => e
+    #   log.fatal "Problem with TVSubtitle: #{e}"
+    # end
+    # begin  
+    #   SeriesSub.new(self)
+    # rescue => e
+    #   log.fatal "Problem with SeriesSub: #{e}"
+    # end
+    # begin
+    #   Podnapisi.new(self)
+    # rescue => e
+    #   log.fatal "Problem with Podnapisi: #{e}"
+    # end
+  end
+  
+  def growl_episode(episode, lang)
+    Inspector.growl episode.episode_name_with_format, "#{lang_name(lang)} srt added", "srt.png"
+  end
+  
+  def self.growl(title, msg, icon, pri = 0)
+    system("/usr/local/bin/growlnotify -w -n autosub --image #{File.dirname(__FILE__) + "/../asset/#{icon}"} -p #{pri} -m #{msg.inspect} #{title} &") 
+  end
+  
+private
+  
+  def initialize_tv_shows
+    # path is a dir with many tv_shows dir
+    base_dir = clean_dir(Dir.new(@path))
+    base_dir.each do |tv_show_name|
+      @tv_shows << { :name => tv_show_name, :path => "#{@path}/#{tv_show_name}" }
+    end
+  end
+  
+  def initialize_episodes
+    @tv_shows.each do |tv_show|
+      search_episodes(tv_show[:name], tv_show[:path])
+    end
+  end
+  
+  def search_episodes(tv_show_name, path)
+    elements = clean_dir(Dir.new(path))
+    elements.each do |el|
+      el_path = "#{path}/#{el}"
+      if File.directory?(el_path)
+        search_episodes(tv_show_name, el_path)
+      elsif Episode.valid?(el)
+        episodes << Episode.new(tv_show_name, el_path)
+      end
+    end
+  end
+  
+  def clean_dir(dir)
+    dir.select { |e| !["..", ".", ".DS_Store", ".com.apple.timemachine.supported", "Icon\r", "autosub.log", "_autosub_.log"].include?(e) }
+  end
+
+  def lang_name(lang)
+    case lang
+    when 'en'
+      'English'
+    when 'fr'
+      'French'
+    end
+  end
+  
+end

data/lib/sites/betaseries.rb

@@ -0,0 +1,140 @@
+require 'hpricot'
+require 'open-uri'
+require 'zip/zip'
+
+# http://www.betaseries.com/
+class BetaSeries
+  
+  def initialize(inspector)
+    @inspector = inspector
+    episodes = @inspector.episodes.select { |e| e.need_srt?(@inspector.langs) }
+
+    $stdout.print "------------------------------------------\n"
+    $stdout.print "Start srt searching from www.betaseries.com\n"
+    $stdout.print "------------------------------------------\n"
+    episodes.each do |e|
+      @inspector.langs.each do |l|
+        unless e.srt.include?(l) 
+          $stdout.print "#{e.episode_name_with_format} [#{l}]: "
+          if get_srt(e, l)
+            inspector.log.info "#{e.episode_name_with_format} [#{l}] // BetaSeries // #{@srt_name}"
+            $stdout.print "FOUND\n"
+            e.srt << l
+            @inspector.growl_episode(e, l)
+            break
+          else
+            $stdout.print "NOT FOUND\n"
+          end
+        end
+      end
+    end
+  end
+    
+private
+  
+  def get_srt(episode, lang)
+    found = false
+    agent = WWW::Mechanize.new
+    doc = Hpricot(open(url(episode)))
+    doc.search("div.episode").each do |el|
+      title = el.search("div.titreep").inner_text
+      if good_episode?(episode, title)
+        el.search("ul.srt li a") do |srt_link|
+          title = srt_link.inner_text
+          if !found && good_language?(title, lang) && good_format?(episode, title) && no_tag?(title)
+            Tempfile.open("tv.zip") do |tempfile|
+              tempfile.write(agent.get_file(srt_link[:href])) # download the zip
+              tempfile.close # tempfile need to be closed for unzip it
+              Zip::ZipFile.open(tempfile.path) do |zip_file|
+                if zip_file.size == 1
+                  file = zip_file.first
+                else
+                  zip_file.each do |f|
+                    if f.size > 5000 && good_episode?(episode, f.name) && good_language?(f.name, lang) && episode.good_format?(f.name) && no_tag?(f.name)
+                      file = f
+                    end
+                  end
+                end
+                if file
+                  @srt_name = file.name
+                  zip_file.extract(file, episode.srt_filename(lang))
+                  found = true
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  rescue => e
+    @inspector.log.fatal "BetaSeries // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
+    found
+  end
+
+  
+  def url(episode)
+    "http://www.betaseries.com/ajax/planning_serie.php?url=#{tv_show(episode)}&saison=#{episode.season}"
+  end
+  
+  def tv_show(episode)
+    case episode.tv_show
+    when "Terminator The Sarah Connor Chronicles"
+      "thesarahconnorchronicles"
+    when "CSI New York"
+      "csiny"
+    when "The IT Crowd"
+      "itcrowd"
+    else
+      episode.tv_show
+    end.downcase.gsub(/\s/, '')
+  end
+  
+  def lang_version(lang)
+    case lang
+    when 'fr'
+      "VF"
+    when 'en'
+      "VO"
+    end
+  end
+  
+  def num(episode)
+    "#{episode.season}#{episode.number < 10 ? "0#{episode.number}" : episode.number}"
+  end
+  
+  def good_episode?(episode, name)
+    num2 = "S#{episode.season < 10 ? "0#{episode.season}" : episode.season}E#{episode.number < 10 ? "0#{episode.number}" : episode.number}"
+    name.include?(num(episode)) || name.include?(num2)
+  end
+  
+  def good_language?(name, lang)
+    case lang
+    when 'fr'
+      name.include?("VF") || name.include?("fr")
+    when 'en'
+      name.include?("VO") || name.include?("en")
+    end
+  end
+  
+  def good_format?(episode, name)
+    if name.include?("VO") || name.include?("VF")
+      true
+    else
+      episode.good_format?(name)
+    end
+  end
+  
+  def no_tag?(filename)
+    case filename
+    when /no.?tag/i
+      true
+    when /tag/i
+      false
+    else
+      true
+    end
+  end
+  
+end

data/lib/sites/podnapisi.rb

@@ -0,0 +1,101 @@
+require 'hpricot'
+require 'open-uri'
+require 'zip/zip'
+require 'cgi'
+
+# http://www.podnapisi.net/ | http://www.sub-titles.net/
+class Podnapisi
+  
+  def initialize(inspector)
+    @inspector = inspector
+    episodes = @inspector.episodes.select { |e| e.need_srt?(@inspector.langs) }
+
+    $stdout.print "------------------------------------------\n"
+    $stdout.print "Start srt searching from www.podnapisi.net\n"
+    $stdout.print "------------------------------------------\n"
+    episodes.each do |e|
+      @inspector.langs.each do |l|
+        unless e.srt.include?(l) 
+          $stdout.print "#{e.episode_name_with_format} [#{l}]: "
+          if get_srt(e, l)
+            @inspector.log.info "#{e.episode_name_with_format} [#{l}] // Podnapisi // #{@srt_name}"
+            $stdout.print "FOUND\n"
+            e.srt << l
+            @inspector.growl_episode(e, l)
+            break
+          else
+            $stdout.print "NOT FOUND\n"
+          end
+        end
+      end
+    end
+  end
+    
+private
+  
+  def get_srt(episode, lang)
+    found = false
+    doc = Hpricot(open(url(episode, lang)))
+    doc.search("table.list tr").each do |el|
+      if !found && have_srt_format?(el) && !hearing_impaired?(el) && url = open(download_url(el))
+        Tempfile.open("tv.zip") do |tempfile|
+          tempfile.write(open(url).read) # download the zip
+          tempfile.close # tempfile need to be closed for unzip it
+          Zip::ZipFile.open(tempfile.path) do |zip_file|
+            zip_file.each do |f|
+              if f.size > 5000
+                @srt_name = f.name
+                zip_file.extract(f, episode.srt_filename(lang))
+                found = true
+              end
+            end
+          end
+        end
+      end
+    end
+  rescue => e
+    @inspector.log.fatal "Podnapisi // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
+    found
+  end
+
+  def url(episode, lang)
+    "http://simple.podnapisi.net/ppodnapisi/search?tbsl=3&asdp=1&sK=#{tv_show(episode)}&sJ=#{lang_num(lang)}&sO=desc&sS=time&submit=Search&sTS=#{episode.season}&sTE=#{episode.number}&sY=&sR=#{format(episode)}&sT=1"
+  end
+  
+  def tv_show(episode)
+    CGI.escape(episode.tv_show)
+  end
+  
+  def format(episode)
+    case episode.format
+    when 'hd'
+      "720"
+    when 'sd'
+      ""
+    end
+  end
+  
+  def lang_num(lang)
+    case lang
+    when 'fr'
+      8
+    when 'en'
+      2
+    end
+  end
+  
+  def have_srt_format?(el)
+    !el.search("td[text()*='SubRip']").empty?
+  end
+  
+  def hearing_impaired?(el)
+    !el.search("img[@title='Subtitle is for hearing impaired']").empty?
+  end
+  
+  def download_url(el)
+    "http://simple.podnapisi.net#{el.search("a[text()*='download subtitle']").first[:href]}"
+  end
+  
+end

data/lib/sites/seriessub.rb

@@ -0,0 +1,122 @@
+require 'hpricot'
+require 'open-uri'
+require 'zip/zip'
+
+# http://www.seriessub.com/
+class SeriesSub
+  
+  def initialize(inspector)
+    @inspector = inspector
+    episodes = @inspector.episodes.select { |e| e.need_srt?(@inspector.langs) }
+
+    $stdout.print "------------------------------------------\n"
+    $stdout.print "Start srt searching from www.seriessub.com\n"
+    $stdout.print "------------------------------------------\n"
+    episodes.each do |e|
+      @inspector.langs.each do |l|
+        unless e.srt.include?(l) 
+          $stdout.print "#{e.episode_name_with_format} [#{l}]: "
+          if get_srt(e, l)
+            inspector.log.info "#{e.episode_name_with_format} [#{l}] // SeriesSub // #{@srt_name}"
+            $stdout.print "FOUND\n"
+            e.srt << l
+            @inspector.growl_episode(e, l)
+            break
+          else
+            $stdout.print "NOT FOUND\n"
+          end
+        end
+      end
+    end
+  end
+    
+private
+  
+  def get_srt(episode, lang)
+    found = false
+    doc = Hpricot(open(url(episode)))
+    doc.search("td.gst_fichier a").each do |el|
+      text = el.inner_text.gsub(/(\_|\.)/, ' ')
+      if good_episode?(episode, text) && text.include?(lang_version(lang))
+        if text.include?('zip')
+          Tempfile.open("tv.zip") do |tempfile|
+            tempfile.write(open(el[:href]).read) # download the zip
+            tempfile.close # tempfile need to be closed for unzip it
+            Zip::ZipFile.open(tempfile.path) do |zip_file|
+              zip_file.each do |f|
+                if f.size > 5000 && good_episode?(episode, f.name) && f.name.include?(lang_version(lang)) && episode.good_format?(f.name) && no_tag?(f.name)
+                  @srt_name = f.name
+                  zip_file.extract(f, episode.srt_filename(lang))
+                  found = true
+                end
+              end
+            end
+          end
+        elsif episode.good_format?(text) && no_tag?(text) # just a single srt file
+          @srt_name = text
+          Tempfile.open("tv.srt") do |tempfile|
+            tempfile.write(open(el[:href]).read) # download the zip
+            tempfile.close
+            File.move(tempfile.path, episode.srt_filename(lang))
+            found = true
+          end
+        end
+      end
+    end
+  rescue => e
+    @inspector.log.fatal "SeriesSub // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
+    found
+  end
+
+  
+  def url(episode)
+    "http://www.seriessub.com/sous-titres/#{tv_show(episode)}/#{season(episode)}"
+  end
+  
+  def tv_show(episode)
+    case episode.tv_show
+    when "Terminator The Sarah Connor Chronicles"
+      "The Sarah Connor Chronicles"
+    when "The IT Crowd"
+      "IT Crowd"
+    else
+      episode.tv_show
+    end.downcase.gsub(/\s/, '_')
+  end
+  
+  def season(episode)
+    "saison_#{episode.season}"
+  end
+  
+  def lang_version(lang)
+    case lang
+    when 'fr'
+      "VF"
+    when 'en'
+      "VO"
+    end
+  end
+  
+  def num(episode)
+    "#{episode.season}#{episode.number < 10 ? "0#{episode.number}" : episode.number}"
+  end
+  
+  def good_episode?(episode, name)
+    num2 = "S#{episode.season < 10 ? "0#{episode.season}" : episode.season}E#{episode.number < 10 ? "0#{episode.number}" : episode.number}"
+    name.include?(num(episode)) || name.include?(num2)
+  end
+  
+  def no_tag?(filename)
+    case filename
+    when /notag/i
+      true
+    when /tag/i
+      false
+    else
+      true
+    end
+  end
+  
+end

data/lib/sites/tvsubtitle.rb

@@ -0,0 +1,81 @@
+require 'simple-rss'
+require 'open-uri'
+require 'zip/zip'
+require 'cgi'
+require 'mechanize'
+
+# http://www.tvsubtitles.net/
+class TVSubtitle
+  
+  def initialize(inspector)
+    @inspector = inspector
+    episodes = @inspector.episodes.select { |e| e.need_srt?(@inspector.langs) }
+    
+    $stdout.print "--------------------------------------------\n"
+    $stdout.print "Searching for srt in www.tvsubtitles.net RSS\n"
+    $stdout.print "--------------------------------------------\n"
+    @inspector.langs.each do |l|
+      rss(l).items.each do |item|
+        episodes.each do |e|
+          unless e.srt.include?(l)
+            if episode_title(e, l) == item_title(item)
+              download_srt(item, e, l)
+              inspector.log.info "#{e.episode_name_with_format} [#{l}] // TVSubtitle // #{@srt_name}"
+              $stdout.print "FOUND: #{e.episode_name_with_format} [#{l}]\n"
+              e.srt << l
+              @inspector.growl_episode(e, l)
+            end
+          end
+        end
+      end
+    end
+  end
+  
+private
+  
+  def download_srt(item, episode, lang)
+    found = false
+    agent = WWW::Mechanize.new
+    id = item_id(item)
+    Tempfile.open("srt") do |tempfile|
+      agent.get("http://www.tvsubtitles.net/subtitle-#{id}.html") # set cookie
+      tempfile.write(agent.get_file("http://www.tvsubtitles.net/download-#{id}.html")) # download the srt
+      tempfile.close
+      Zip::ZipFile.open(tempfile.path) do |zip_file|
+        zip_file.each do |f|
+          if f.size > 5000
+            @srt_name = f.name
+            zip_file.extract(f, episode.srt_filename(lang))
+            found = true
+          end
+        end
+      end
+    end
+  rescue => e
+    @inspector.log.fatal "TVSubtitle // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
+    found
+  end
+  
+  def rss(lang)
+    SimpleRSS.parse open("http://fr.tvsubtitles.net/rss#{lang}.xml") 
+  end
+  
+  def num(episode)
+    "#{episode.season}X#{episode.number < 10 ? "0#{episode.number}" : episode.number}"
+  end
+  
+  def episode_title(episode, lang)
+    "#{episode.tv_show} #{num(episode)} #{lang} #{episode.format}".downcase
+  end
+
+  def item_title(item)
+    "#{item.title} #{item.description.include?('720p') ? 'hd' : 'sd'}".downcase
+  end
+  
+  def item_id(item)
+    item.link.match(/([0-9]+)/)[1]
+  end
+  
+end

metadata

@@ -0,0 +1,139 @@
+--- !ruby/object:Gem::Specification 
+name: autosub
+version: !ruby/object:Gem::Version 
+  version: 0.4.0
+platform: ruby
+authors: 
+- Pirate
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-06 00:00:00 +01:00
+default_executable: bin/autosub
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.2.9
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hpricot
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.8.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: httparty
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.4.5
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: optiflag
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0.7"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rubyzip
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.1
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: mechanize
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.9.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: simple-rss
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 1.2.2
+    version: 
+description: Ruby tool to automatically download subtitles (srt) inside your TV Shows folder
+email: pirate.2061@gmail.com
+executables: 
+- autosub
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.markdown
+files: 
+- .gitignore
+- LICENSE
+- README.markdown
+- Rakefile
+- VERSION
+- asset/failure.png
+- asset/srt.png
+- autosub.gemspec
+- bin/autosub
+- lib/episode.rb
+- lib/inspector.rb
+- lib/sites/betaseries.rb
+- lib/sites/podnapisi.rb
+- lib/sites/seriessub.rb
+- lib/sites/tvsubtitle.rb
+has_rdoc: true
+homepage: http://github.com/pirate/autosub
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Automatically download srt for your TV Shows
+test_files: []
+