blitzcrank 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: b8fd04be5f9f957818a9d67d0b092123f8273600
-  data.tar.gz: eaef4cae484010e2a3f17b0cdee60d24a35d0a83
+  metadata.gz: b2f16e683446c8c544077387a401edb95935566e
+  data.tar.gz: 148abfdb0c175dbebed9f3146c3c4a4276b84ea6
 SHA512:
-  metadata.gz: 1962d42effbc0298c48235dc381e1c87a33446eeaab2021172244af3ee50c9dd34fa4a2550ea7ca8e88cd2bfb21dda410d75bc2f6dca3b5c03a2d409b3eec420
-  data.tar.gz: ef2a7a00ce83ab74b4ea20686ea42f49f3cb393a8a0f6ea02ddd77d7cc1d201a2c00df8155fdfa78c9afa29968550efb32c5ff7e52244515d35a80abe5bf5da4
+  metadata.gz: abfc5436dd36673a84a00144a8f01dbd1786aae42852f9dd4339a1fe74c18cd429d15586829eb92bd03995c7f133b9814094d0dba1b5eac30d74b241ce679213
+  data.tar.gz: 98f99d4c9da1619b1b926db2fc3486ab35f8427330c3090965ce443231918d0a9cc71f4c0760fd3e2d6516665439fb6da59808c40c446ca22df26d089ad80c54

data/Gemfile

@@ -2,3 +2,9 @@ source 'https://rubygems.org'
 
 # Specify your gem's dependencies in blitzcrank.gemspec
 gemspec
+
+group :development, :test do
+  gem 'guard-rspec', '~> 4.2.7'  #, require: false
+  gem 'rspec', '~> 3.0.0.beta2'
+  gem 'rspec-nc', '0.2.0'
+end

data/Guardfile

@@ -0,0 +1,5 @@
+guard :rspec, cmd: 'bundle exec rspec' do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/blitzcrank/(.+)\.rb$})     { |m| "spec/lib/#{m[1]}_spec.rb" }
+end
+

data/blitzcrank.gemspec

@@ -23,5 +23,6 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency("colorize", "0.6.0")
   spec.add_dependency("imdb", "0.8.0")
+  spec.add_dependency("ruby-configurable", "1.0.2")
 
 end

data/lib/blitzcrank.rb

@@ -1,61 +1,47 @@
 require "blitzcrank/version"
+require File.join(File.dirname(__FILE__), "blitzcrank/video.rb")
+Dir[File.join(File.dirname(__FILE__), "blitzcrank/*.rb")].each {|f| require f }
 require "colorize"
+require "configurable"
+require "fileutils"
 require "yaml"
-require "imdb"
 
 module Blitzcrank
+  include Configurable
 
   # Configuration defaults
-  @config = {
-              :base_tv_dir => "",
-              :base_movie_dir => "",
-              :season_identifier => "Season ",
-              :remote_host => "localhost" ,
-              :remote_user => %x[whoami],
-              :remote_base_dir => "~/",
-              :dry_run => false
-            }
-
-  @valid_config_keys = @config.keys
-
-  @tv_show_regex = /(.*)\.s?(\d{1,2})[ex](\d{2})/i # Supports s01e01, 1x03
-
-  # Configure through hash
-  def self.configure(opts = {})
-    opts.each {|k,v| @config[k.to_sym] = v if @valid_config_keys.include? k.to_sym}
-  end
+  configurable_options :base_tv_dir => "",
+                       :base_movie_dir => "",
+                       :season_identifier => "Season ",
+                       :remote_host => "localhost" ,
+                       :remote_user => %x[whoami],
+                       :remote_base_dir => "~/",
+                       :dry_run => false
 
   def self.configure_with(yaml_path)
     begin
-      config = YAML::load(IO.read(yaml_path))
+      Blitzcrank.config.update(YAML.load(IO.read(yaml_path)))
     rescue Errno::ENOENT
       log(:warning, "YAML configuration file couldn't be found. Using defaults."); return
     rescue Psych::SyntaxError
       log(:warning, "YAML configuration file contains invalid syntax. Using defaults."); return
     end
-
-    configure(config)
-  end
-
-  def self.config
-    @config
   end
 
   def self.write_sample_config(yaml_path)
-    IO.write(yaml_path, @config.to_yaml)
+    IO.write(yaml_path, Blitzcrank.config.to_yaml)
   end
 
-  def self.transfer_file(remote_path, local_dir)
-    if @config[:dry_run]
-      puts "Copying #{remote_path} to #{local_dir}"
-    else
-      system("rsync -avz #{ '--bwlimit=' + @config[:bwlimit] unless @config[:bwlimit].nil? } --progress --rsh='ssh' \"#{@config[:remote_user]}@#{@config[:remote_host]}:#{@config[:remote_base_dir]}#{remote_path.gsub(' ', '\\ ')}\" \"#{local_dir}\"")
+  def self.transfer_file(video)
+    puts "Copying #{video.remote_path} to #{video.local_path}"
+    if !Blitzcrank.config.dry_run
+      Rsync.sync(video)
     end
   end
 
   # get a listing of all remote files that would be considered "videos"
   def self.remote_video_file_list
-      %x[ssh -q #{@config[:remote_user]}@#{@config[:remote_host]} "cd #{@config[:remote_base_dir]} && find . -type f \\( -iname \'*.avi\' -or -iname \'*.mkv\' -or -iname \'*.mp4\' -or -iname \'*.m4v\' -or -iname \'*.divx\' \\)"]
+    %x[ssh -q #{Blitzcrank.config.remote_user}@#{Blitzcrank.config.remote_host} "cd #{Blitzcrank.config.remote_base_dir} && find . -type f \\( -iname \'*.avi\' -or -iname \'*.mkv\' -or -iname \'*.mp4\' -or -iname \'*.m4v\' -or -iname \'*.divx\' \\)"]
   end
 
   def self.file_menu(search_array = nil)
@@ -100,54 +86,14 @@ module Blitzcrank
 
   # any files (hashes) passed into here will be checked against our local TV folders and IMDB to see if it's a movie
   def self.transfer_files(filesToTransfer)
-    Dir.chdir(@config[:base_tv_dir])
+    Dir.chdir(Blitzcrank.config.base_tv_dir)
 
     filesToTransfer.each do |dh|
-      Dir.chdir(@config[:base_tv_dir])
-      full_path = dh[:path]
-      file_name = dh[:name]
-      nice_name = Blitzcrank.nice_tv_name(file_name)
-      directories = Dir.glob(nice_name, File::FNM_CASEFOLD)
-      if directories.count > 0 # see if we already have a directory for this tv show
-        nice_name = directories.first
-        season_dir = "#{Dir.pwd}/#{nice_name}/#{@config[:season_identifier]}#{Blitzcrank.season(file_name)}"
-        Dir.mkdir(season_dir) unless Dir.exists?(season_dir) # make the folder if it doesn't exist
-        Blitzcrank.transfer_file(full_path, season_dir)
-      elsif Blitzcrank.is_movie?(file_name)
-        Blitzcrank.transfer_file(full_path, @config[:base_movie_dir])
+      video = Video.with_path dh[:path]
+      if video.is_tv_show?
+        FileUtils.mkdir_p video.season_path
       end
-    end
-  end
-
-  # pulls the season number froma  file
-  def self.season(file_name)
-    @tv_show_regex.match(file_name)
-    $2.gsub(/\A0+/, '')
-  end
-
-  def self.nice_tv_name(file_name)
-    unless @tv_show_regex.match(file_name).nil?
-      showName = $1
-      wordsInShowName = showName.gsub(/[\._]/, ' ').downcase.split(" ") # strip . and _
-      wordsInShowName.each do |word|
-        if wordsInShowName.index(word) == 0 || /^(in|a|the|and|on)$/i.match(word).nil?
-          word.capitalize!
-        end
-      end
-      wordsInShowName.join(" ").gsub(/\b(us|uk|\d{4})$/i, '(\1)') # adding parens around US/UK marked shows, or shows with years
-    else
-      file_name
-    end
-  end
-
-  def self.is_movie?(file_name)
-    unless /^(.*).(\d{4}|dvdrip)/i.match(file_name).nil?
-      movie_name = $1
-      nice_movie_name = movie_name.gsub('.', ' ').downcase
-      i = Imdb::Search.new(nice_movie_name)
-      i.movies.size > 0
-    else
-      false
+      Blitzcrank.transfer_file(video)
     end
   end
 

data/lib/blitzcrank/copy.rb

@@ -0,0 +1,7 @@
+module Blitzcrank
+  class Copy
+    def self.sync(video)
+      system("cp #{video.remote_path} #{video.local_path}")
+    end
+  end
+end

data/lib/blitzcrank/movie.rb

@@ -0,0 +1,17 @@
+module Blitzcrank
+  class Movie < Video
+
+    def local_path
+      Blitzcrank.config.base_movie_dir + nice_name
+    end
+
+    def is_movie?
+      true
+    end
+
+    def is_tv_show?
+      false
+    end
+
+  end
+end

data/lib/blitzcrank/rsync.rb

@@ -0,0 +1,7 @@
+module Blitzcrank
+  class Rsync < Copy
+    def self.sync(video)
+      system("rsync -avz #{ '--bwlimit=' + Blitzcrank.config.bwlimit if Blitzcrank.config.respond_to?(:bwlimit) } --progress --rsh='ssh' \"#{Blitzcrank.config.remote_user}@#{Blitzcrank.config.remote_host}:#{Blitzcrank.config.remote_base_dir}#{video.remote_path.gsub(' ', '\\ ')}\" \"#{video.local_path}\"")
+    end
+  end
+end

data/lib/blitzcrank/tv_show.rb

@@ -0,0 +1,26 @@
+module Blitzcrank
+  class TVShow < Video
+
+    def season_path
+      Blitzcrank.config.base_tv_dir + nice_name + "/" + Blitzcrank.config.season_identifier + season
+    end
+
+    def local_path
+      season_path + "/" + @file_name
+    end
+
+    def season
+      TV_SHOW_REGEX.match(@file_name)
+      $2.gsub(/\A0+/, '')
+    end
+
+    def is_movie?
+      false
+    end
+
+    def is_tv_show?
+      true
+    end
+
+  end
+end

data/lib/blitzcrank/version.rb

@@ -1,3 +1,3 @@
 module Blitzcrank
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/blitzcrank/video.rb

@@ -0,0 +1,80 @@
+require "imdb"
+module Blitzcrank
+  class Video
+
+    TV_SHOW_REGEX = /(.*)[\. ]s?(\d{1,2})[ex](\d{2})/i # Supports s01e01, 1x03
+    MOVIE_REGEX = /^(.*).(\d{4}|dvdrip)/i
+
+    def self.with_path(file_path)
+      file_name = Video.file_name(file_path)
+      if Video.is_tv_show? file_name
+        return TVShow.new file_path
+      elsif Video.is_movie? file_name
+        return Movie.new file_path
+      end
+      return Video.new file_path
+    end
+
+    def self.file_name(file_path)
+      file_path.split('/').last
+    end
+
+    # returns if this is a TV show or not
+    def self.is_tv_show?(file_name)
+      TV_SHOW_REGEX.match(file_name).nil? == false
+    end
+
+    # movie methods
+    def self.is_movie?(file_name)
+      if Video.is_tv_show?(file_name)
+        return false
+      elsif MOVIE_REGEX.match(file_name).nil? == false
+        movie_name = $1
+        nice_movie_name = movie_name.gsub('.', ' ').downcase
+        i = Imdb::Search.new(nice_movie_name)
+        i.movies.size > 0
+      else
+        false
+      end
+    end
+
+    def initialize(file_path)
+      @file_path = file_path
+      @file_name = Video.file_name file_path
+    end
+
+    def remote_path
+      @file_path
+    end
+
+    def file_name
+      @file_name
+    end
+
+    def nice_name
+      unless TV_SHOW_REGEX.match(@file_name).nil?
+        showName = $1
+        wordsInShowName = showName.gsub(/[\._]/, ' ').downcase.split(" ") # strip . and _
+        wordsInShowName.each do |word|
+          if wordsInShowName.index(word) == 0 || /^(in|a|the|and|on)$/i.match(word).nil?
+            word.capitalize!
+          end
+        end
+        wordsInShowName.join(" ").gsub(/\b(us|uk|\d{4})$/i, '(\1)') # adding parens around US/UK marked shows, or shows with years
+      else
+        @file_name
+      end
+    end
+
+    # returns if this is a TV show or not
+    def is_tv_show?
+      Video.is_tv_show?(@file_name)
+    end
+
+    # movie methods
+    def is_movie?
+      Video.is_movie?(@file_name)
+    end
+
+  end
+end

data/spec/lib/copy_spec.rb

@@ -0,0 +1,15 @@
+require './lib/blitzcrank.rb'
+
+module Blitzcrank
+  describe Copy do
+
+    let(:video) { Video.with_path 'The.League.S03E03.720p.HDTV.x264-IMMERSE.mkv' }
+
+    it "copies video files" do
+      allow(Copy).to receive(:system).and_return(0)
+      Copy.sync(video)
+      expect(Copy).to have_received(:system)
+    end
+
+  end
+end

data/spec/lib/movie_spec.rb

@@ -0,0 +1,27 @@
+require './lib/blitzcrank.rb'
+
+module Blitzcrank
+  describe Movie do
+    let(:movie_file_names) {['Minority.Report.2002.720p.BluRay.DTS.x264-HiDt.mkv',
+                             'Army.Of.Darkness.1992.Bluray.1080p.DTSMA.x264.mkv']}
+    let(:movies) { movie_file_names.map {|f| Movie.new f } }
+
+    before(:all) do
+      Blitzcrank.config.update(base_movie_dir: 'Movies/')
+    end
+
+    it "correctly detects movies" do
+      movies.each do |video|
+        expect(video.is_movie?).to eq(true)
+        expect(video.is_tv_show?).to eq(false)
+      end
+    end
+
+    it "correctly gets the local path for a movie" do
+      movies.each_with_index do |video, i|
+        expect(video.local_path).to eql("Movies/#{movie_file_names[i]}")
+      end
+    end
+
+  end
+end

data/spec/lib/tv_show_spec.rb

@@ -0,0 +1,46 @@
+require './lib/blitzcrank.rb'
+
+module Blitzcrank
+  describe TVShow do
+    let(:tv_file_names) {['The.League.S03E03.720p.HDTV.x264-IMMERSE.mkv',
+                          'doctor_who_2005.8x05.time_heist.720p_hdtv_x264-fov.mkv',
+                          'Doctor Who S04E14 The Next Doctor 720p BluRay DTS5.1 x264-BG.mkv']}
+    let(:tv_show_names) {['The League',
+                          'Doctor Who (2005)',
+                          'Doctor Who']}
+    let(:tv_show_seasons) {['3',
+                            '8',
+                            '4']}
+    let(:tv_shows) { tv_file_names.map {|f| TVShow.new f } }
+
+    before(:all) do
+      Blitzcrank.config.update(base_tv_dir: 'TV/')
+    end
+
+    it "correctly detects TV shows" do
+      tv_shows.each do |video|
+        expect(video.is_tv_show?).to eq(true)
+        expect(video.is_movie?).to eq(false)
+      end
+    end
+
+    it "correctly parses file names into TV show names" do
+      tv_shows.each_with_index do |video, i|
+        expect(video.nice_name).to eql(tv_show_names[i])
+      end
+    end
+
+    it "correctly parses seasons from TV show file names" do
+      tv_shows.each_with_index do |video, i|
+        expect(video.season).to eql(tv_show_seasons[i])
+      end
+    end
+
+    it "correctly gets the local path for a TV show" do
+      tv_shows.each_with_index do |video, i|
+        expect(video.local_path).to eql("TV/Season #{tv_show_seasons[i]}/#{tv_show_names[i]}")
+      end
+    end
+
+  end
+end

data/spec/lib/video_spec.rb

@@ -0,0 +1,16 @@
+require './lib/blitzcrank.rb'
+
+module Blitzcrank
+  describe Video do
+    let(:movie_file_names) {['Minority.Report.2002.720p.BluRay.DTS.x264-HiDt.mkv',
+                             'Army.Of.Darkness.1992.Bluray.1080p.DTSMA.x264.mkv']}
+    let(:movies) { movie_file_names.map {|f| Video.with_path f } }
+
+    it "correctly creates movies" do
+      movies.each do |video|
+        expect(video.class).to eq(Movie)
+      end
+    end
+
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: blitzcrank
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - pyro2927
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-21 00:00:00.000000000 Z
+date: 2014-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -66,6 +66,20 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 0.8.0
+- !ruby/object:Gem::Dependency
+  name: ruby-configurable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.0.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.0.2
 description: Copy down remote files and sort into nice folders
 email:
 - joseph@pintozzi.com
@@ -78,6 +92,7 @@ extra_rdoc_files: []
 files:
 - ".gitignore"
 - Gemfile
+- Guardfile
 - LICENSE
 - LICENSE.txt
 - README.md
@@ -93,7 +108,16 @@ files:
 - img/Rocket_Grab.jpg
 - img/Static_Field.jpg
 - lib/blitzcrank.rb
+- lib/blitzcrank/copy.rb
+- lib/blitzcrank/movie.rb
+- lib/blitzcrank/rsync.rb
+- lib/blitzcrank/tv_show.rb
 - lib/blitzcrank/version.rb
+- lib/blitzcrank/video.rb
+- spec/lib/copy_spec.rb
+- spec/lib/movie_spec.rb
+- spec/lib/tv_show_spec.rb
+- spec/lib/video_spec.rb
 homepage: ''
 licenses:
 - MIT
@@ -118,4 +142,8 @@ rubygems_version: 2.2.2
 signing_key: 
 specification_version: 4
 summary: Uhhh.... torrent management gem?
-test_files: []
+test_files:
+- spec/lib/copy_spec.rb
+- spec/lib/movie_spec.rb
+- spec/lib/tv_show_spec.rb
+- spec/lib/video_spec.rb