moviesort 0.1.0

data/.gitignore

@@ -0,0 +1 @@
+pkg/

data/Rakefile

@@ -0,0 +1,12 @@
+require 'jeweler'
+
+Jeweler::Tasks.new do |gemspec|
+  gemspec.name = "moviesort"
+  gemspec.summary = "Small command-line utility for sorting downloaded TV shows and movies"
+  gemspec.email = "luke@lukeredpath.co.uk"
+  gemspec.authors = ["Luke Redpath"]
+end
+
+Jeweler::GemcutterTasks.new
+
+task :release => "gemcutter:release"

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/moviesort

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+lib_path = File.expand_path(File.join(File.dirname(__FILE__), *%w[.. lib]))
+$LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+
+require 'moviesort/command_line'
+Moviesort::CommandLine.new.run

data/features/sorting_tv_shows.feature

@@ -0,0 +1,22 @@
+Feature: Sorting a folder of TV show episodes
+  In order to keep my TV shows organised
+  As a person trying to keep their media server in order
+  I want to be able to sort my downloaded TV episodes into folders
+  
+  Scenario: First episode of a new TV show
+    Given a folder "/movies/new" containing the file "The Sopranos - S01E01 - Pilot.avi"
+    When I run Moviesort with the options "--source /movies/new --target /movies/tv"
+    Then the movie should be moved to "/movies/tv/The Sopranos/Season 1/The.Sopranos.S01E01.avi"
+    
+  Scenario: First episode of a new season of an existing TV show
+    Given a folder "/movies/new" containing the file "The Sopranos - S06E01 - Members Only.avi"
+    And a folder "/movies/tv/The Sopranos" containing the folder "Season 1"
+    When I run Moviesort with the options "--source /movies/new --target /movies/tv"
+    Then the movie should be moved to "/movies/tv/The Sopranos/Season 6/The.Sopranos.S06E01.avi"
+    And the folder "/movies/tv/The Sopranos/Season 1" should exist
+
+  Scenario: Another episode of an existing season of an existing TV show
+    Given a folder "/movies/new" containing the file "The Sopranos - S01E02 - 46 Long.avi"
+    And a folder "/movies/tv/The Sopranos/Season 1" containing the file "The.Sopranos.S01E01.avi"
+    When I run Moviesort with the options "--source /movies/new --target /movies/tv"
+    Then the movie should be moved to "/movies/tv/The Sopranos/Season 1/The.Sopranos.S01E02.avi"

data/features/step_definitions/sorting_tv_shows_steps.rb

@@ -0,0 +1,28 @@
+require 'moviesort/command_line'
+
+### givens
+  
+Given /^a folder "([^\"]*)" containing the file "([^\"]*)"$/ do |parent, file|
+  FileUtils.mkdir_p(parent)
+  FileUtils.touch(File.join(parent, file))
+end
+
+Given /^a folder "([^\"]*)" containing the folder "([^\"]*)"$/ do |parent, folder|
+  FileUtils.mkdir_p(File.join(parent, folder))
+end
+
+## whens
+
+When /^I run Moviesort with the options "([^\"]*)"$/ do |options|
+  Moviesort::CommandLine.parse(options).run
+end
+
+## thens
+
+Then /^the movie should be moved to "([^\"]*)"$/ do |path|
+  assert File.exist?(path)
+end
+
+Then /^the folder "([^\"]*)" should exist$/ do |dir|
+  assert File.directory?(dir)
+end

data/features/support/env.rb

@@ -0,0 +1,15 @@
+require 'test/unit/assertions'
+require 'fakefs/safe'
+
+World(Test::Unit::Assertions)
+
+Before do
+  FakeFS.activate!
+end
+
+After do
+  FakeFS::FileSystem.clear
+  FakeFS.deactivate!
+end
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), *%w[.. .. lib]))

data/lib/moviesort/command_line.rb

@@ -0,0 +1,30 @@
+require 'trollop'
+require 'moviesort/sorter'
+
+module Moviesort
+  class CommandLine
+    def self.parse(options)
+      new(options.split)
+    end
+    
+    def initialize(argv = ARGV)
+      @options = Trollop.options(argv) do
+        banner <<-EOS.gsub(/^ {8}/, '')
+        Moviesort is a script to sort and file away your TV shows and movies
+
+        Usage: moviesort [options]
+        EOS
+
+        opt :source, "Source directory", :default => '.'
+        opt :target, "Target directory", :default => '.'
+      end
+    end
+    
+    def run
+      sorter = Sorter::TV.new(@options[:target])
+      Dir[File.join(@options[:source], '*.avi')].each do |file|
+        sorter.sort_file(file)
+      end
+    end
+  end
+end

data/lib/moviesort/sorter.rb

@@ -0,0 +1,18 @@
+require 'moviesort/tv_show'
+
+module Moviesort
+  module Sorter
+    class TV
+      def initialize(output_dir)
+        @output_dir = output_dir
+      end
+      
+      def sort_file(path)
+        tv_show = Moviesort::TVShow.parse(File.basename(path))
+        target_path = File.join(@output_dir, tv_show.target_path)
+        FileUtils.mkdir_p(target_path)
+        FileUtils.mv(path, File.join(target_path, tv_show.target_filename))
+      end
+    end
+  end
+end

data/lib/moviesort/tv_show.rb

@@ -0,0 +1,53 @@
+require 'pathname'
+
+module Moviesort
+  class TVShow
+    attr_reader :show, :season, :episode
+    
+    def initialize(filename, show, season, episode)
+      @show = show
+      @season = season
+      @episode = episode
+      @extension = File.extname(filename)
+    end
+    
+    def self.parse(filename)
+      parser = Parser.new(filename)
+      new(filename, *parser.parse)
+    end
+    
+    def target_path
+      Pathname.new(File.join(show, "Season #{season}"))
+    end
+    
+    def target_filename
+      "#{@show.gsub(/\W/, '.')}.S%.2iE%.2i#{@extension}" % [season, episode]
+    end
+    
+    private
+    
+    class Parser
+      def initialize(filename)
+        @filename = sanitize(filename)
+      end
+      
+      def parse
+        patterns = [
+          /(.*)\sS(\d+)E(\d+)/i, 
+          /(.*)\s(\d{1,2})x(\d+)/,
+          /(.*)\s(\d{1})(\d{2})/
+        ]
+        match = patterns.map { |regex| @filename.match(regex) }.compact.first
+        extract_data_from_pattern_match(match)
+      end
+
+      def extract_data_from_pattern_match(match)
+        [match[1], *match[2..3].map { |s| s.to_i }]
+      end
+
+      def sanitize(filename)
+        filename.gsub(/\./, ' ').gsub(/-/, ' ').squeeze(' ')
+      end
+    end
+  end
+end

data/test/sorter_test.rb

@@ -0,0 +1,25 @@
+require 'test_helper'
+require 'moviesort/sorter'
+
+class SorterTest < Test::Unit::TestCase
+  
+  def setup
+    @output_path   = "/output"
+    @download_path = "/downloads"
+    FileUtils.mkpath(@output_path)
+    FileUtils.mkpath(@download_path)
+  end
+
+  context "TV show sorter" do
+    setup do
+      @sorter = Moviesort::Sorter::TV.new(@output_path)
+    end
+
+    should "move a given TV show to its target path under the root path" do
+      FileUtils.touch("/downloads/The Sopranos - S02E04.avi")
+      @sorter.sort_file("/downloads/The Sopranos - S02E04.avi")
+      assert File.exist?("/output/The Sopranos/Season 2/The.Sopranos.S02E04.avi")
+    end
+  end  
+  
+end

data/test/test_helper.rb

@@ -0,0 +1,4 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), *%w[.. lib]))
+require 'test/unit'
+require 'shoulda'
+require 'fakefs'

data/test/tv_show_test.rb

@@ -0,0 +1,50 @@
+require 'test_helper'
+require 'moviesort/tv_show'
+
+class TvShowTest < Test::Unit::TestCase
+  
+  context "A TV show" do
+    setup do
+      @tv_show = Moviesort::TVShow.new("The.Sopranos.102.avi", "The Sopranos", season = 1, episode = 2)
+    end
+  
+    should "have a target path based on the show name and season" do
+      assert_equal Pathname.new("The Sopranos/Season 1"), @tv_show.target_path
+    end
+    
+    should "have a filename based on the show name and season with the original file extension" do
+      assert_equal "The.Sopranos.S01E02.avi", @tv_show.target_filename
+    end
+  end
+  
+  context "When parsing" do
+    should "handle file in the format SHOW NAME - SXXEXX.xxx" do
+      @tv_show = Moviesort::TVShow.parse("The Sopranos - S01E02.avi")
+      assert_equal "The Sopranos", @tv_show.show
+      assert_equal 1, @tv_show.season
+      assert_equal 2, @tv_show.episode
+    end
+    
+    should "handle file in the format SHOW NAME - sXXeXX.xxx" do
+      @tv_show = Moviesort::TVShow.parse("The Sopranos - s01e02.avi")
+      assert_equal "The Sopranos", @tv_show.show
+      assert_equal 1, @tv_show.season
+      assert_equal 2, @tv_show.episode
+    end
+    
+    should "handle file in the format SHOW NAME - SxE.xxx" do
+      @tv_show = Moviesort::TVShow.parse("The Sopranos - 1x02.avi")
+      assert_equal "The Sopranos", @tv_show.show
+      assert_equal 1, @tv_show.season
+      assert_equal 2, @tv_show.episode
+    end
+    
+    should "handle file in the format SHOW.NAME.XYY.xxx" do
+      @tv_show = Moviesort::TVShow.parse("The.Sopranos.102.avi")
+      assert_equal "The Sopranos", @tv_show.show
+      assert_equal 1, @tv_show.season
+      assert_equal 2, @tv_show.episode                  
+    end
+  end
+  
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification 
+name: moviesort
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Luke Redpath
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-14 00:00:00 +00:00
+default_executable: moviesort
+dependencies: []
+
+description: 
+email: luke@lukeredpath.co.uk
+executables: 
+- moviesort
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Rakefile
+- VERSION
+- bin/moviesort
+- features/sorting_tv_shows.feature
+- features/step_definitions/sorting_tv_shows_steps.rb
+- features/support/env.rb
+- lib/moviesort/command_line.rb
+- lib/moviesort/sorter.rb
+- lib/moviesort/tv_show.rb
+- test/sorter_test.rb
+- test/test_helper.rb
+- test/tv_show_test.rb
+has_rdoc: true
+homepage: 
+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: Small command-line utility for sorting downloaded TV shows and movies
+test_files: 
+- test/sorter_test.rb
+- test/test_helper.rb
+- test/tv_show_test.rb