lib/inspector.rb in pirate-autonzb-0.2.2 vs lib/inspector.rb in pirate-autonzb-0.2.3

@@ -1,20 +1,27 @@
 require File.join(File.dirname(__FILE__), 'movie')
 
 class Inspector
-    
+  
+  attr_accessor :backup
+  
   def initialize(paths, options = {})
     @paths = paths.split(',').map { |p| p.gsub!(/\/$/,'') }
     @options = options
     @options[:srt] = @options[:srt] ? @options[:srt].split(',') : ['none']
     @options[:imdb_score] = @options[:imdb_score] ? @options[:imdb_score].to_f : 7.0
     @options[:year] = @options[:year] ? @options[:year].to_i : 1950
     
+    if @options[:backup]
+      @backup = @options[:backup].gsub(/\/$/,'')
+      @paths << @backup
+    end
+    
     @movies = []
     initialize_movies
     
-    $stdout.print "Movie criteria: imdb_score >= #{@options[:imdb_score]}, year >= #{@options[:year]} and srt [#{@options[:srt].join(',')}]\n"
+    $stdout.print "Movie criteria: imdb score >= #{@options[:imdb_score]}, year >= #{@options[:year]} and srt [#{@options[:srt].join(',')}]\n"
   end
   
   def need?(movie)
     if valid?(movie)
       $stdout.print " => movie has required criteria "
@@ -26,15 +33,23 @@
         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"
             true
           elsif format_score(movie) == format_score(m)
-            if sound_score(movie) > sound_score(m)
-              $stdout.print "but new movie has better sound: #{movie.sound}\n"
+            if source_score(movie) > source_score(m)
+              $stdout.print "but new movie has better source: #{movie.source}\n"
               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"
+                true
+              else
+                $stdout.print "with same srt, format, source and sound\n"
+                false
+              end
             else
-              $stdout.print "with same srt, format and sound\n"
+              $stdout.print "with same srt, format and better source: #{m.source}\n"
               false
             end
           else
             $stdout.print "with same srt and better format: #{m.format}\n"
             false
@@ -60,13 +75,17 @@
     @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) if File.directory?(movie_path)
+        @movies << Movie.new(movie) if File.directory?(movie_path) || File.extname(movie_path) == '.nzb'
       end
-      $stdout.print "Inspected #{@movies.size - old_movies_size} movie(s) in #{path}\n"
+      if path == @backup
+        $stdout.print "Found #{@movies.size - old_movies_size} backuped nzb(s) in #{path}\n"
+      else
+        $stdout.print "Found #{@movies.size - old_movies_size} movie(s) in #{path}\n"
+      end
     end
   end
 
   def clean_dir(dir)
     dir.select { |e| !["..", ".", ".DS_Store"].include?(e) }
@@ -90,9 +109,19 @@
   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    
\ No newline at end of file