autonzb 0.5.4

data/.gitignore

@@ -0,0 +1,25 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC
+autonzb
+pkg
+manifest
+tasks

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,44 @@
+# AutoNZB
+  
+Ruby tool to automatically download x264 HD nzb movies files from newzleech.com
+Currently only english and french subtitles are supported, but it's super easy to add your own language.
+  
+## Install
+
+    sudo gem install pirate-autonzb --source http://gems.github.com
+
+## Usage
+
+In your terminal:
+
+    autonzb -d /path/of/download/nzb/directory
+    
+Will download new x264 HD movies nzb from newzleech.com, with imdb score >= 7.0, year >= 1950 and nzb age <= 160 days
+
+    autonzb -d /path/of/download/nzb/directory -backup /path/of/backuped/already/downloaded/nzb
+    
+-backup option it's very useful to prevent useless re-download of nzb already downloaded
+
+    autonzb -d /path/of/download/nzb/directory -movies /path/with/already/downloaded/movies -age 1 -imdb 7.5 -year 1980 -srt fr,en
+    
+Will download only new nzb of the day with imdb score >= 7.5, year >= 1980 and subtitles french or english (and unknown srt).
+The -movies setting prevents already owned movies to be re-downloaded (only if the owned movie is 'better' than the new release),
+movies folder need to follow the name convention (especially {imdb_id})
+
+more details with:
+
+    autonzb -h
+    
+## Folder Name Convention
+
+AutoNZB use (and needs if you want to use -movies options) specific folders name for your movies:
+
+    name of the movie (year) tag(s) format source sound encoding lang {imdb_id} [srt(s)]
+    
+    Burn After Reading (2008) PROPER 1080p BluRay DTS x264 {tt0887883} [fr,en]
+    Le Fabuleux Destin d'Amelie Poulain (2001) 720p BluRay x264 FRENCH {tt0211915} [en]
+    ...
+    
+## Copyright
+
+Copyright (c) 2009 Pirate. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,51 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "autonzb"
+    gem.summary = "Automatically download x264 HD nzb movies files."
+    gem.description = "Ruby tool to automatically download x264 HD nzb movies files from newzleech.com & nzbs.org"
+    gem.email = "pirate.2061@gmail.com"
+    gem.homepage = "http://github.com/pirate/autonzb"
+    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.ignore_pattern = ["tmp/*", "script/*"]
+    gem.add_dependency "hpricot"
+    gem.add_dependency "optiflag"
+    gem.add_dependency "rubyzip"
+    gem.add_dependency "htmlentities"
+    gem.default_executable = "bin/autonzb"
+  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 = "test-gem #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.5.4

data/autonzb.gemspec

@@ -0,0 +1,73 @@
+# 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{autonzb}
+  s.version = "0.5.4"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Pirate"]
+  s.date = %q{2009-11-01}
+  s.default_executable = %q{bin/autonzb}
+  s.description = %q{Ruby tool to automatically download x264 HD nzb movies files from newzleech.com & nzbs.org}
+  s.email = %q{pirate.2061@gmail.com}
+  s.executables = ["autonzb"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.markdown"
+  ]
+  s.files = [
+    ".gitignore",
+     "README.markdown",
+     "Rakefile",
+     "VERSION",
+     "asset/failure.png",
+     "autonzb.gemspec",
+     "bin/autonzb",
+     "lib/imdb.rb",
+     "lib/inspector.rb",
+     "lib/movie.rb",
+     "lib/sites/newzleech/nfo.rb",
+     "lib/sites/newzleech/nzb.rb",
+     "lib/sites/nzbs/mechanize_hack.rb",
+     "lib/sites/nzbs/nfo.rb",
+     "lib/sites/nzbs/nzb.rb"
+  ]
+  s.homepage = %q{http://github.com/pirate/autonzb}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Automatically download x264 HD nzb movies files.}
+  s.test_files = [
+    "spec/autonzb_spec.rb",
+     "spec/spec_helper.rb"
+  ]
+
+  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"])
+      s.add_runtime_dependency(%q<optiflag>, [">= 0"])
+      s.add_runtime_dependency(%q<rubyzip>, [">= 0"])
+      s.add_runtime_dependency(%q<htmlentities>, [">= 0"])
+    else
+      s.add_dependency(%q<rspec>, [">= 1.2.9"])
+      s.add_dependency(%q<hpricot>, [">= 0"])
+      s.add_dependency(%q<optiflag>, [">= 0"])
+      s.add_dependency(%q<rubyzip>, [">= 0"])
+      s.add_dependency(%q<htmlentities>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<rspec>, [">= 1.2.9"])
+    s.add_dependency(%q<hpricot>, [">= 0"])
+    s.add_dependency(%q<optiflag>, [">= 0"])
+    s.add_dependency(%q<rubyzip>, [">= 0"])
+    s.add_dependency(%q<htmlentities>, [">= 0"])
+  end
+end
+

data/lib/imdb.rb

@@ -0,0 +1,80 @@
+require 'cgi'
+require 'hpricot'
+require 'open-uri'
+require 'htmlentities'
+
+class IMDB
+  
+  attr_accessor :link, :doc, :id
+  
+  def initialize(name, year = nil, link = nil)
+    # $stdout.print 'i'
+    @try = 3
+    @name, @year, @link = name, year, link
+    @coder = HTMLEntities.new
+    set_doc
+    set_id
+  end
+
+  def score
+    if @doc && score_text = @doc.search("div.meta b").first
+      score_text.inner_html.match(/(.*)\/10/)[1].to_f
+    else
+      0
+    end
+  end
+  
+  def year
+    @doc ? @doc.search("title").inner_html.match(/\s\(([0-9]{4})/)[1].to_i : @year.to_i
+  end
+  
+  def name
+    $KCODE = 'utf-8'
+    @doc ? @coder.decode(@doc.search("title").inner_html.match(/(.*)\s\(/u)[1]) : @name
+  end
+    
+private
+  
+  def set_doc
+    if @link
+      @doc = Hpricot(open(@link.gsub(/\/\s*$/,'')))
+      @id = @link.match(/tt[0-9]+/)[0]
+    else
+      query = "#{@name} (#{@year})"
+      search_url = "http://www.imdb.com/find?q=#{CGI::escape(query)}"
+      doc = Hpricot(open(search_url))
+      case doc.search("title").inner_html
+      when "IMDb Title Search", "IMDb Search" # search result page
+        if !doc.search("b[text()*='Media from'] a").empty?        
+          imdb_id = doc.search("b[text()*='Media from'] a").first[:href]
+          movie_url = "http://www.imdb.com#{imdb_id}"
+        elsif !doc.search("td[@valign='top'] a[@href^='/title/tt']").empty?
+          imdb_id = doc.search("td[@valign='top'] a[@href^='/title/tt']").first[:href]
+          movie_url = "http://www.imdb.com#{imdb_id}"
+        else
+          movie_url = nil
+        end
+        if movie_url
+          @doc = Hpricot(open(movie_url))
+          @id = movie_url.match(/tt[0-9]+/)[0]
+        end
+      else # direct in movie page
+        @doc = doc
+      end
+    end
+  rescue
+    if @try > 0
+      @try -= 1
+      $stdout.print '*'
+      sleep 2
+      set_doc #retry
+    else
+      @doc = nil
+    end
+  end
+  
+  def set_id
+    @id ||= doc.search("a[@href*='/title/tt']").first[:href].match(/tt[0-9]+/)[0] if doc
+  end
+    
+end

data/lib/inspector.rb

@@ -0,0 +1,215 @@
+require File.join(File.dirname(__FILE__), 'movie')
+require File.join(File.dirname(__FILE__), 'sites', 'newzleech', 'nzb')
+require File.join(File.dirname(__FILE__), 'sites', 'nzbs', 'nzb')
+
+class Inspector
+  
+  attr_accessor :download_path, :backup_path, :movies, :nzbs
+  
+  def initialize(download_path, options = {})
+    @download_path = download_path.gsub(/\/$/,'')
+    @paths = options[:movie_paths] ? options[:movie_paths].split(',').map { |p| p.gsub(/\/$/,'') } : []
+    @options = options
+    @options[:srt] = @options[:srt] ? (@options[:srt].split(',') - ["unknown"] + ['unknown']).uniq : nil
+    @options[:imdb_score] = @options[:imdb_score] ? @options[:imdb_score].to_f : 7.0
+    @options[:year] = @options[:year] ? @options[:year].to_i : 1950
+    
+    @movies = []
+    initialize_movies
+    
+    if @options[:backup_path]
+      @backup_path = @options[:backup_path].gsub(/\/$/,'') 
+      @nzbs = []
+      initialize_nzbs
+      @movies = @nzbs + @movies
+      load_age_from_yaml
+    end
+    
+    $stdout.print "Movie criteria: imdb score >= #{@options[:imdb_score]}, year >= #{@options[:year]}#{" and srt [#{@options[:srt].join(',')}]" if @options[:srt]}\n"
+  end
+  
+  def search_and_download
+    if @options[:login] && @options[:pass]
+      site = Nzbs::NZB.new(@options)
+      parse_nzbs(site)
+    end
+    site = Newzleech::NZB.new(@options)
+    parse_nzbs(site)
+    
+    if backup_path
+      keep_only_best_nzbs
+      set_age_on_yaml
+    end
+  end
+
+  def parse_nzbs(site)
+    site.nzbs.each do |movie|
+      $stdout.print "#{movie.dirname}, imdb score: #{movie.score}, age: #{movie.age} day(s)\n"
+      if need?(movie)
+        $stdout.print " => DOWNLOAD: #{movie.name} (#{movie.year})\n"
+        download(site, movie)
+      end
+    end
+    $stdout.print "No nzb found, maybe change -age or -page setting\n" if site.nzbs.empty?
+  end
+
+  def download(site, movie)
+    Tempfile.open("movie.nzb") do |tempfile|
+      tempfile.write(site.download(movie)) # download nzb
+      tempfile.close
+      File.copy(tempfile.path, "#{backup_path}/#{movie.dirname}.nzb") if backup_path
+      File.move(tempfile.path, "#{download_path}/#{movie.dirname}.nzb")
+    end
+    @movies << movie
+  end
+  
+  def need?(movie, movies = @movies, not_validate = false, log = true)
+    if not_validate || valid?(movie)
+      $stdout.print " => movie has required criteria " if log
+      if m = movies.detect { |m| m == movie }
+        $stdout.print "but is already owned " if log
+        if srt_score(movie) > srt_score(m)
+          $stdout.print "but new movie has better subtitle: [#{movie.srt.join(',')}]\n" if log
+          true
+        elsif srt_score(movie) == srt_score(m)
+          if format_score(movie) > format_score(m)
+            $stdout.print "but new movie has better format: #{movie.format}\n" if log
+            true
+          elsif format_score(movie) == format_score(m)
+            if source_score(movie) > source_score(m)
+              $stdout.print "but new movie has better source: #{movie.source}\n" if log
+              true
+            elsif source_score(movie) == source_score(m)
+              if sound_score(movie) > sound_score(m)
+                $stdout.print "but new movie has better sound: #{movie.sound}\n" if log
+                true
+              else
+                $stdout.print "with same srt, format, source and sound\n" if log
+                false
+              end
+            else
+              $stdout.print "with same srt, format and better source: #{m.source}\n" if log
+              false
+            end
+          else
+            $stdout.print "with same srt and better format: #{m.format}\n" if log
+            false
+          end
+        else
+          $stdout.print "with better subtitle: [#{m.srt.join(',')}]\n" if log
+          false
+        end
+      else
+        $stdout.print "and is not already owned\n" if log
+        true
+      end
+    else
+      $stdout.print " => movie doesn't have required criteria\n" if log
+      false
+    end
+  end
+  
+  def self.growl(title, msg, pri = 0)
+    system("/usr/local/bin/growlnotify -w -n autonzb --image #{File.dirname(__FILE__) + "/../asset/failure.png"} -p #{pri} -m #{msg.inspect} #{title} &") 
+  end
+  
+private
+
+  def initialize_movies
+    @paths.each do |path|
+      old_movies_size = @movies.size
+      base_dir = clean_dir(Dir.new(path))
+      base_dir.each do |movie|
+        movie_path = "#{path}/#{movie}"
+        @movies << Movie.new(movie, :path => movie_path) if File.directory?(movie_path)
+      end
+      $stdout.print "Found #{@movies.size - old_movies_size} movie(s) in #{path}\n"
+    end
+  end
+  
+  def initialize_nzbs
+    base_dir = clean_dir(Dir.new(@backup_path))
+    base_dir.each do |nzb|
+      nzb_path = "#{@backup_path}/#{nzb}"
+      @nzbs << Movie.new(nzb, :path => nzb_path) if File.extname(nzb_path) == '.nzb'
+    end
+    $stdout.print "Found #{@nzbs.size} backuped nzb(s) in #{@backup_path}\n"
+  rescue => e
+    $stdout.print "Problem with backup nzb: #{e}\n"
+  end
+
+  def clean_dir(dir)
+    dir.select { |e| !["..", ".", ".DS_Store", ".com.apple.timemachine.supported", "Icon\r"].include?(e) }
+  end
+  
+  def valid?(movie)
+    (@options[:srt].nil? || (((@options[:srt] - movie.srt).size < @options[:srt].size)) || movie.lang == 'FRENCH') &&
+    movie.year != nil && movie.year >= @options[:year] &&
+    movie.score != nil && movie.score >= @options[:imdb_score]
+  end
+  
+  def srt_score(movie)
+    if @options[:srt]
+      srts = @options[:srt].reverse
+      movie.srt.inject(-1) do |score, srt|
+        s = (i = srts.index(srt)) ? i : -1
+        score = s if s > score
+        score
+      end
+    else
+      0
+    end
+  end
+  
+  def format_score(movie)
+    case movie.format
+    when '1080p'; 2
+    when '720p';  1
+    else;         0
+    end
+  end
+  
+  def source_score(movie)
+    case movie.format
+    when 'BluRay'; 4
+    when 'HDDVD';  3
+    when 'HDTV';   2
+    when 'DVD';    1
+    else;          0
+    end
+  end
+  
+  def sound_score(movie)
+    movie.format == 'DTS' ? 1 : 0    
+  end
+  
+  def keep_only_best_nzbs
+    size = 0
+    @nzbs.each do |nzb|
+      nzbs = @nzbs.select { |item| item.path != nzb.path }
+      unless need?(nzb, nzbs, true, false)
+        File.delete(nzb.path)
+        size += 1
+      end
+    end
+    if size > 0
+      $stdout.print "#########################################################################\n"
+      $stdout.print "Deleted #{size} useless backuped nzb(s) (keep only the best nzb by movie)\n"
+    end
+  end
+  
+  def load_age_from_yaml
+    yml_file = "#{backup_path}/options.yml"
+    yml_options = File.exist?(yml_file) ? YAML.load_file(yml_file) : {}
+    if yml_age = yml_options['age']
+      age = sprintf("%.1f", ((Time.now.tv_sec - yml_age.to_i) / 60.0 / 60.0 / 24.0)).to_f
+      @options[:age] = age if age > 0
+    end
+  end
+  
+  def set_age_on_yaml
+    yml_file = "#{backup_path}/options.yml"
+    File.open(yml_file, "w") { |f| f.write(({ 'age' => Time.now.tv_sec }).to_yaml) }
+  end
+
+end

data/lib/movie.rb

@@ -0,0 +1,209 @@
+require File.join(File.dirname(__FILE__), 'imdb')
+
+class Movie
+  include Comparable
+  
+  attr_accessor :path, :name, :format, :source, :sound, :encoding, :year, :srt, :lang, :score, :tags, :age,
+                :imdb_link, :imdb_id, :nzb_link, :nfo
+  
+  def initialize(raw_name, attributes = {})
+    @raw_name = raw_name.gsub(/\.|\_/,' ')
+    attributes.each { |k,v| self.send("#{k}=", v) }
+    @srt, @tags = [], []
+    
+    # for info set with nzbs title
+    if name && year && score
+      @info_get_from_title = true
+    end
+    
+    set_name
+    set_year
+    set_format
+    set_source
+    set_sound
+    set_srt
+    set_lang
+    set_encoding
+    set_tags
+    
+    set_imdb_id
+    set_data_from_imdb unless path || @info_get_from_title
+  end
+    
+  def score
+    @score ||= imdb.score
+  end
+  
+  def dirname
+    "#{name} (#{year}) #{tags.join(' ')} #{format} #{source} #{sound} #{encoding} #{lang} {#{imdb_id}} [#{srt.join(',')}]".gsub(/\s+|\//,' ')
+  end
+
+  def <=>(other_movie)
+    if imdb_id && other_movie.imdb_id
+      imdb_id <=> other_movie.imdb_id
+    else
+      "#{name} #{year}".downcase <=> "#{other_movie.name} #{other_movie.year}".downcase
+    end
+  end
+      
+private
+  
+  def imdb
+    @imdb ||= IMDB.new(name, year, imdb_link)
+  end
+    
+  def imdb_link
+    @imdb_link || (imdb_id && "http://imdb.com/title/#{imdb_id}")
+  end
+  
+  def set_imdb_id
+    if imdb_link
+      @imdb_id = imdb_link.match(/tt[0-9]+/)[0]
+    elsif matched = @raw_name.match(/\{(tt[0-9]+)\}/)
+      @imdb_id = matched[1]
+    elsif imdb
+      @imdb_id = imdb.id
+      add_imdb_id_to_file if path
+    else
+      nil
+    end
+  end
+  
+  def set_name
+    if name.nil?
+      raw_name = @raw_name.gsub(/\(|\)|\[|\]|\{|\}|\//, ' ')
+      if matched = raw_name.match(/(.*)(19[0-9]{2}|20[0-9]{2})[^p]/)
+        @name = matched[1]
+      elsif matched = raw_name.match(/(.*)1080p/i)
+        @name = matched[1]
+      elsif matched = raw_name.match(/(.*)720p/i)
+        @name = matched[1]
+      else
+        @name = ''
+      end
+      @name.gsub!(/REPACK|LIMITED|UNRATED|PROPER|REPOST|Directors\sCut/iu,'')
+      @name.gsub!(/^\s+|\s+$/u,'')
+    end
+  end
+  
+  def set_year
+    if year.nil?
+      if matched = @raw_name.match(/19[0-9]{2}|20[0-9]{2}/)
+        @year = matched[0].to_i
+      end
+    end
+  end
+  
+  def set_format
+    @format = case @raw_name
+    when /1080p/i
+      '1080p'
+    when /720p/i
+      '720p'
+    end
+  end
+  
+  def set_encoding
+    @encoding = case @raw_name
+    when /x264/i
+      'x264'
+    when /VC1/i
+      'VC1'
+    when /PS3/i
+      'PS3'
+    when /divx/i
+      'DIVX'
+    when /xvid/i
+      'XVID'
+    end
+  end
+  
+  def set_source
+    @source = case @raw_name
+    when /Blu\s?Ray|Blu-Ray|BDRip/i
+      'BluRay'
+    when /HDDVD/i
+      'HDDVD'
+    when /HDTV/i
+      'HDTV'
+    when /DVD/i
+      'DVD'
+    end
+  end
+  
+  def set_sound
+    @sound = case @raw_name
+    when /DTS/i
+      'DTS'
+    when /AC3/i
+      'AC3'
+    end
+  end
+  
+  def set_srt
+    if nfo
+      @srt = nfo.srt
+    elsif matched = @raw_name.match(/\[(.*)\]/)
+      srts = matched[1].split(',')
+      if srts.empty?
+        @srt << 'none'
+      else
+        srts.each { |srt| @srt << srt }
+      end
+    else
+      @srt << 'unknown'
+    end
+  end
+  
+  def set_lang
+    @lang = case @raw_name
+    when /FRENCH/i
+      'FRENCH'
+    when /GERMAN/i
+      'GERMAN'
+    when /DANISH/i
+      'DANISH'
+    when /NORDIC/i
+      'NORDIC'
+    end
+  end
+  
+  def set_tags
+    @tags << 'REPACK'        if @raw_name =~ /REPACK/i
+    @tags << 'LIMITED'       if @raw_name =~ /LIMITED/i
+    @tags << 'UNRATED'       if @raw_name =~ /UNRATED/i
+    @tags << 'PROPER'        if @raw_name =~ /PROPER/i
+    @tags << 'REPOST'        if @raw_name =~ /REPOST/i
+    @tags << 'OUTDATED'      if @raw_name =~ /OUTDATED/i
+    @tags << 'Directors Cut' if @raw_name =~ /Directors\sCut|DirCut/i
+  end
+  
+  def set_data_from_imdb
+    @name = imdb.name
+    imdb_year = imdb.year
+    if !imdb_year.nil? && imdb_year != 0
+      @year = imdb_year
+    end
+  end
+  
+  def add_imdb_id_to_file
+    dir_name = File.dirname(path)
+    if File.directory?(path)
+      base_name = File.basename(path)
+      ext_name = ''
+    else
+      ext_name = path.include?('.nzb') ? '.nzb' : File.extname(path)
+      base_name = File.basename(path, ext_name)
+    end
+    if matched = base_name.match(/^(.*)\s(\[{1}.*\]{1})$/)
+      base_name_without_srts = matched[1]
+      srts = matched[2]
+      new_base_name = "#{base_name_without_srts} {#{imdb_id}} #{srts}#{ext_name}"
+    else
+      new_base_name = "#{base_name} {#{imdb_id}}#{ext_name}"
+    end
+    File.rename(path, "#{dir_name}/#{new_base_name}")
+    $stdout.print "Added {#{imdb_id}} (imdb id) => #{dir_name}/#{new_base_name}\n"
+  end
+  
+end

data/lib/sites/newzleech/nfo.rb

@@ -0,0 +1,35 @@
+require 'open-uri'
+
+module Newzleech
+  class Nfo
+  
+    attr_accessor :srt, :imdb_link
+  
+    def initialize(url)
+      @nfo = open(url.gsub(/\/$/,'')).read
+      @srt = []
+    
+      parse_nfo
+      @srt.uniq!
+    end
+  
+  private
+  
+    # searching for subtitles languages info & imdb link
+    def parse_nfo
+      @nfo.split(/\n/).each do |line|
+        case line
+        when /subtitle|sub/i
+          @srt << 'fr' if line =~ /fr|fre|french/i
+          @srt << 'en' if line =~ /en|eng|english/i
+          # TODO add more language parsing here
+          @srt << 'none' if line =~ /none/i
+        when /imdb\.com\/title\//
+          @imdb_link = (matched = line.match(/imdb.com\/title\/(tt[0-9]+)/)) && "http://imdb.com/title/#{matched[1]}"
+        end
+      end
+      @srt << 'unknown' if @srt.empty?
+    end
+  
+  end
+end

data/lib/sites/newzleech/nzb.rb

@@ -0,0 +1,100 @@
+require 'open-uri'
+require 'hpricot'
+require 'zip/zip'
+
+require File.join(File.dirname(__FILE__), 'nfo')
+
+module Newzleech
+  class NZB
+  
+    URL = 'http://www.newzleech.com'
+    DAYS_BY_PAGE = 1.7
+    
+    attr_accessor :nzbs
+    
+    def initialize(options = {})
+      @options = options
+      @options[:age] ||= 160
+      @options[:pages] ||= ((@options[:age].to_f + DAYS_BY_PAGE) / DAYS_BY_PAGE).ceil
+    
+      @nzb_urls = []
+      @nzbs = []
+    
+      (1..(@options[:pages].to_i)).each do |page|
+        @nzb_urls << "#{URL}?group=143&minage=&age=160&min=4000&max=max&q=&m=search&adv=1&offset=#{(page.to_i - 1) * 60}"
+      end
+    
+      parse_newzleech
+    end
+    
+    def download(movie)
+      open(movie.nzb_link).read
+    end
+  
+  private
+    
+    def parse_newzleech
+      $stdout.print "Parsing #{URL} for new x264 HD nzb movies, with age <= #{@options[:age]}, in last #{@options[:pages]} page(s)\n"
+      @nzb_urls.each do |url|
+        doc = Hpricot(open(url))
+        doc.search("table.contentt").each do |table|
+          raw_name = raw_name(table)
+          if nfo_link = nfo_link(table)
+            nfo = Nfo.new(nfo_link)
+            imdb_link = nfo.imdb_link
+          else
+            imdb_link = imdb_link(table)
+          end
+          if raw_name && (imdb_link || (raw_name = find_clean_nzb_name(raw_name))) && raw_name.include?('x264')
+            age = parse_age(table.search("td.age").first.inner_html) # get age of the nzb
+            if age <= @options[:age].to_f
+              raw_name = clean_raw_name(raw_name)
+              movie = Movie.new(raw_name, :nfo => nfo, :imdb_link => imdb_link, :nzb_link => nzb_link(table), :age => age)
+              @nzbs << movie
+            
+              $stdout.print '.'
+              $stdout.flush
+            end
+          end
+        end
+      end
+      $stdout.print "\n"
+    end
+  
+    def clean_raw_name(raw_name)
+      (find_clean_nzb_name(raw_name) || raw_name).strip
+    end
+  
+    def find_clean_nzb_name(raw_name)
+      if matched = raw_name.match(/^\[[0-9]*\]-\[.*\]-\[(.*)\]-/)
+        matched[1]
+      end
+    end
+  
+    def raw_name(table)
+      (a = table.search("td.subject a[@href^='?p=']").first) && a.inner_html
+    end
+  
+    def nfo_link(table)
+      (nfo = table.search("td.subject a[@href^='nfo.php?id=']").first) && "#{URL}/#{nfo[:href]}"
+    end
+  
+    def imdb_link(table)
+      (imdb = table.search("td.subject a[@href^='http://anonym.to/?http://www.imdb.com']").first) && imdb[:href].match(/\?(.*)/)[1]
+    end
+  
+    def nzb_link(table)
+      (nzb = table.search("td.get a[@href^='?m=gen&dl=1']").first) && "#{URL}/#{nzb[:href]}"
+    end
+          
+    def parse_age(string)
+      case string
+      when /h/i
+        ((string.to_f / 24) * 10).to_i / 10.0
+      when /d/i
+        string.to_f
+      end
+    end
+    
+  end
+end

data/lib/sites/nzbs/mechanize_hack.rb

@@ -0,0 +1,41 @@
+module WWW
+  class Mechanize
+    class Chain
+      class ResponseReader
+        include WWW::Handler
+
+        def initialize(response)
+          @response = response
+        end
+
+        def handle(ctx, params)
+          params[:response] = @response
+          body = StringIO.new
+          total = 0
+          @response.read_body { |part|
+            total += part.length
+            body.write(part)
+            Mechanize.log.debug("Read #{total} bytes") if Mechanize.log
+          }
+          body.rewind
+
+          res_klass = Net::HTTPResponse::CODE_TO_OBJ[@response.code.to_s]
+          raise ResponseCodeError.new(@response) unless res_klass
+
+          # Net::HTTP ignores EOFError if Content-length is given, so we emulate it here.
+          # unless res_klass <= Net::HTTPRedirection
+          #   raise EOFError if (!params[:request].is_a?(Net::HTTP::Head)) && @response.content_length() && @response.content_length() != total
+          # end
+  
+          @response.each_header { |k,v|
+            Mechanize.log.debug("response-header: #{ k } => #{ v }")
+          } if Mechanize.log
+
+          params[:response_body] = body
+          params[:res_klass] = res_klass
+          super
+        end
+      end
+    end
+  end
+end

data/lib/sites/nzbs/nfo.rb

@@ -0,0 +1,36 @@
+require 'open-uri'
+
+module Nzbs
+  class Nfo
+  
+    attr_accessor :srt, :imdb_link
+  
+    def initialize(agent, url)
+      @page = agent.get(url.gsub(/\/$/,''))
+      @srt = []
+    
+      parse_nfo
+      @srt.uniq!
+    end
+  
+  private
+  
+    # searching for subtitles languages info & imdb link
+    def parse_nfo
+      pre = @page.search('pre.nfo')
+      pre.to_s.split(/\n/).each do |line|
+        case line
+        when /subtitle|sub/i
+          @srt << 'fr' if line =~ /fr|fre|french/i
+          @srt << 'en' if line =~ /en|eng|english/i
+          # TODO add more language parsing here
+          @srt << 'none' if line =~ /none/i
+        when /imdb\.com\/title\//
+          @imdb_link = (matched = line.match(/imdb.com\/title\/(tt[0-9]+)/)) && "http://imdb.com/title/#{matched[1]}"
+        end
+      end
+      @srt << 'unknown' if @srt.empty?
+    end
+  
+  end
+end

data/lib/sites/nzbs/nzb.rb

@@ -0,0 +1,120 @@
+require 'open-uri'
+require 'hpricot'
+require 'zip/zip'
+require 'mechanize'
+
+require File.join(File.dirname(__FILE__), 'nfo')
+require File.join(File.dirname(__FILE__), 'mechanize_hack')
+
+
+module Nzbs
+  class NZB
+  
+    URL = 'http://www.nzbs.org'
+    DAYS_BY_PAGE = 5 
+  
+    attr_accessor :nzbs, :agent
+  
+    def initialize(options = {})
+      @options = options
+      @options[:age] ||= DAYS_BY_PAGE
+      @options[:pages] ||= ((@options[:age].to_f + DAYS_BY_PAGE) / DAYS_BY_PAGE).ceil
+    
+      @nzb_urls = []
+      @nzbs = []
+    
+      (1..(@options[:pages].to_i)).each do |page|
+        @nzb_urls << "#{URL}/index.php?action=browse&catid=4&page=#{(page)}"
+      end
+      
+      login
+      parse_nzbs
+    end
+    
+    def download(movie)
+      @agent.get_file(movie.nzb_link)
+    end
+    
+  private
+  
+    def login
+      @agent = WWW::Mechanize.new
+      page = @agent.get "#{URL}/index.php?action=browse&catid=4"
+
+      form = page.forms[1]
+      form.username = @options[:login]
+      form.password = @options[:pass]
+
+      @agent.submit form
+    end
+    
+    def parse_nzbs
+      $stdout.print "Parsing #{URL} for new x264 HD nzb movies, with age <= #{@options[:age]}, in last #{@options[:pages]} page(s)\n"
+      @nzb_urls.each do |url|
+        page = @agent.get url
+        page.search("#nzbtable tr").each do |tr|
+          if raw_name = raw_name(tr)
+            if (age = age(tr)) <= @options[:age].to_f
+              if nfo_link = nfo_link(tr)
+                @nfo = Nfo.new(@agent, nfo_link(tr))
+              end
+              
+              name, year, score = get_info_from_imdb_title(tr)
+              movie = Movie.new(raw_name, :name => name,
+                                          :year => year,
+                                          :score => score,
+                                          :imdb_link => imdb_link(tr),
+                                          :nzb_link => nzb_link(tr, page),
+                                          :age => age,
+                                          :nfo => @nfo)
+              @nzbs << movie
+            
+              $stdout.print '.'
+              $stdout.flush
+            end
+          end
+        end
+      end
+      $stdout.print "\n"
+    end
+    
+    def raw_name(tr)
+      (a = tr.search("a.nzb").first) && a.inner_html
+    end
+    
+    def age(tr)
+      age_text = (td = tr.search("td[@align='right']").first) && td.inner_html
+      if age_text && age_text.include?("days ago")
+        @last_age = age_text.to_f
+      else
+        $stdout.print "AGE NOT FOUND"
+      end
+      @last_age
+    end
+  
+    def nfo_link(tr)
+      (a = tr.search("small a[@href$='#nfo']").first) && a[:href]
+    end
+    
+    def get_info_from_imdb_title(tr)
+      title = (a = tr.search("small a.viewimdb").first) && a[:title]
+      if title
+        if matched = title.match(/^(.*)\s\(([0-9]{4}).*\s\((.*)\/10\)/)
+          name = matched[1]
+          year = matched[2]
+          score = matched[3]
+          return name, year.to_i, score.to_f
+        end
+      end
+    end
+  
+    def imdb_link(tr)
+      (a = tr.search("small a.viewimdb").first) && a[:href].match(/^http:\/\/anonym.to\/\?(.*)/)[1]
+    end
+  
+    def nzb_link(tr, page)
+      (a = tr.search("a.dlnzb").first) && "#{URL}/#{a[:href]}"
+    end
+    
+  end
+end

data/spec/autonzb_spec.rb

@@ -0,0 +1,7 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe "AutoNZB" do
+  it "fails" do
+    fail "hey buddy, you should probably rename this file and start specing for real"
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'test-gem'
+require 'spec'
+require 'spec/autorun'
+
+Spec::Runner.configure do |config|
+  
+end

metadata

@@ -0,0 +1,121 @@
+--- !ruby/object:Gem::Specification 
+name: autonzb
+version: !ruby/object:Gem::Version 
+  version: 0.5.4
+platform: ruby
+authors: 
+- Pirate
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-01 00:00:00 +01:00
+default_executable: bin/autonzb
+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"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: optiflag
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rubyzip
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: htmlentities
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: Ruby tool to automatically download x264 HD nzb movies files from newzleech.com & nzbs.org
+email: pirate.2061@gmail.com
+executables: 
+- autonzb
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.markdown
+files: 
+- .gitignore
+- README.markdown
+- Rakefile
+- VERSION
+- asset/failure.png
+- autonzb.gemspec
+- bin/autonzb
+- lib/imdb.rb
+- lib/inspector.rb
+- lib/movie.rb
+- lib/sites/newzleech/nfo.rb
+- lib/sites/newzleech/nzb.rb
+- lib/sites/nzbs/mechanize_hack.rb
+- lib/sites/nzbs/nfo.rb
+- lib/sites/nzbs/nzb.rb
+- LICENSE
+has_rdoc: true
+homepage: http://github.com/pirate/autonzb
+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 x264 HD nzb movies files.
+test_files: 
+- spec/autonzb_spec.rb
+- spec/spec_helper.rb