reorganise 0.0.2 → 0.0.3

data/bin/reorganise

@@ -3,15 +3,179 @@
 module Reorganise
 	class Reorganise
 
-		def initialize()
-			puts "Hi"
+		require 'optparse'
+		require 'to_regexp'
+		require 'fileutils'
+
+		attr_accessor :options, :matcher, :extensions, :directory, :series_name
+
+		def initialize(argv)
+		
+			@extensions = [:avi, :mp4, :mkv]		
+			@options = {}
+			@matcher = /(?:S(\d+)x?E(\d+)|(\d{1,2})(\d{2,})|(\d+)x(\d+))/i
+
+			parse_options!(argv)
+			sort!
+
 		end
 
-		def say_hello
-			puts "RAR!"
+		def parse_options!(argv)
+			optparse = OptionParser.new(argv) do |opts|
+				opts.banner = "Usage: rename.rb [options]\n\nCommand options:"
+
+				@options[:regex] = @matcher
+
+				opts.on('-i', '--input-dir DIRECTORY', 'Specifies where to look for episodes to rename.') do |dir_name|
+					@options[:directory] = dir_name
+				end		
+
+				opts.on('-n', '--name NAME', 'What to rename the series to. Title is derived from series if not given.') do |name|
+					@options[:series_name] = sanitize_user_string(name)
+				end
+			
+				@options[:confirm] = false
+				opts.on('-c', '--confirm', 'Confirms the go ahead to rename files. Only pretends to rename files to show the effects if not given.') do |regex|
+					@options[:confirm] = true
+				end
+			
+				opts.on('-r', '--regexp MATCHER', 'Alternative regexp matcher.') do |regex|
+					@options[:regex] = regex.to_regexp
+				end
+
+				@options[:verbose] = false
+				opts.on('-v', '--verbose', 'Verbose output. Automatically selected for pretend runs.') do |regex|
+					@options[:verbose] = true
+				end
+
+				opts.on('-h', '--help', 'Display this to screen.') do
+					puts opts
+					exit
+				end	
+			end
+
+			optparse.parse!
+		end
+
+		def sort!
+			@directory = @options[:directory]
+			@series_name = @options[:series_name]
+		
+			if directory.nil? or !Dir.exists?(directory)
+				puts "ERROR: Directory '#{directory}' does not exist. Run with -h for more information."
+				exit
+			end
+		
+			Dir.chdir(directory)
+			filenames = Dir.glob("**/*.{#{@extensions.join(",")}}")
+
+			puts "------------------------------------------------------------------------"
+			puts "Finding all files with the following extensions: #{@extensions.join(", ")}"
+			puts ""
+			puts "Directory: #{directory}"
+			puts "Found #{filenames.size} files."
+			puts "------------------------------------------------------------------------"
+			puts ""
+			puts ""
+			puts "!!! NOTICE: This is a PRETEND renaming to show what would happen. !!! "
+			puts "To confirm the renaming of files, run the command again with -c. e.g. `reorganise -c -d [directory] -n [series name]"
+			puts ""
+			puts ""
+			puts "Starting to rename files using regex #{@options[:regex]}..."
+			puts ""
+
+			filenames.each do |filename|
+				if File.file?(filename)
+
+					if details = parse_filename(filename)
+				
+							old_directory = File.absolute_path(File.dirname(filename))
+						
+							# Check to see if the files were already organised into season folders
+							if old_directory.split("/").last.downcase == "season #{details[:season].to_i}"
+								target_directory = old_directory
+							else
+								target_directory = File.join(old_directory, "Season #{details[:season].to_i}")
+							end
+						
+							if details[:series_name]
+								renamed_file = "#{details[:series_name]} - S#{details[:season]}E#{details[:episode]}#{details[:ext]}"
+							else
+								renamed_file = "S#{details[:season]}E#{details[:episode]}#{details[:ext]}"
+							end
+						
+							# Perform mv if confimed
+							if @options[:confirm]
+								FileUtils.mkdir_p(target_directory)
+								FileUtils.mv(File.join(old_directory, details[:basename]), File.join(target_directory, renamed_file))
+							end
+				
+							if @options[:verbose] or !@options[:confirm]
+								puts "-: #{File.join(old_directory, details[:basename])}"
+								puts "+: #{File.join(target_directory, renamed_file)}"
+								puts ""
+							end
+					else
+						puts "ERROR: Skipped file \"#{filename}\" due to invalid file format."
+					end
+				else
+					puts "ERROR: Skipped non-file \"#{filename}\"."
+				end			
+
+			end
+		
+			puts ""
+			puts "Finished renaming #{filenames.size} files."
+		
+			if !@options[:confirm]
+				puts ""
+				puts "!!! NOTICE: This was a pretend run, nothing was renamed. !!!"
+				puts "To confirm the renaming, run this command again with -c"
+			end
+		end
+	
+		def parse_filename(filename)
+			basename = File.basename(filename)
+			extension = File.extname(filename)
+		
+			parts = basename.gsub(/[-._]/i, " ").strip.split(" ")
+		
+			position = season = episode = -1
+		
+			parts.each_with_index do |v, k|
+				if v =~ @matcher
+					season = "%02d" % ($1 || $3 || $5).to_i
+					episode = "%02d" % ($2 || $4 || $6).to_i
+					position = k
+				
+					break if position != -1
+				end
+			end
+		
+			if position != -1
+				# No title if episode numbers are first
+				if position == 0
+					title = (@series_name) ? @series_name : nil
+				else
+					title = parts[0..position-1].map{|p| p.capitalize}.join(" ")
+				end
+				details = {:basename => basename, :ext =>  extension, :season => season, :episode => episode}			
+				details[:series_name] = (@series_name.nil?) ? title : @series_name
+				details
+			else
+				false 
+			end
+		end	
+	
+		def sanitize_user_string(str)
+			return false if str.empty?
+			str = str.gsub(/[^\s\w.\-]/, '')
+			str  
 		end
 
 	end
 end
 
-Reorganise::Reorganise.new
+Reorganise::Reorganise.new(ARGV)
+
+

data/lib/reorganise/version.rb

@@ -1,3 +1,3 @@
 module Reorganise
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/reorganise.gemspec

@@ -11,7 +11,7 @@ Gem::Specification.new do |s|
   s.summary     = "Sorts and renames your series."
   s.description = "Sorts and renames your series."
 
-  s.rubyforge_project = "resort"
+  s.rubyforge_project = "reorganise"
 
 
   s.files         = `git ls-files`.split("\n")
@@ -19,6 +19,6 @@ Gem::Specification.new do |s|
   s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
   s.require_paths = ["lib"]
 
-  s.add_runtime_dependency "tvdb_party"
+  s.add_runtime_dependency "to_regexp"
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: reorganise
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -12,8 +12,8 @@ cert_chain: []
 date: 2012-01-08 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: tvdb_party
-  requirement: &76836100 !ruby/object:Gem::Requirement
+  name: to_regexp
+  requirement: &83562250 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,7 +21,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *76836100
+  version_requirements: *83562250
 description: Sorts and renames your series.
 email:
 - justin.cossutti@gmail.com
@@ -56,7 +56,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: resort
+rubyforge_project: reorganise
 rubygems_version: 1.8.10
 signing_key: 
 specification_version: 3