undertexter 0.0.7 → 0.1.0

data/Gemfile.lock

@@ -1,7 +1,9 @@
 PATH
   remote: .
   specs:
-    undertexter (0.0.6)
+    undertexter (0.1.0)
+      levenshteinish
+      mimer_plus
       nokogiri
       rest-client
 
@@ -9,7 +11,11 @@ GEM
   remote: http://rubygems.org/
   specs:
     diff-lcs (1.1.2)
+    hintable_levenshtein (0.0.3)
+    levenshteinish (0.0.1)
+      hintable_levenshtein
     mime-types (1.16)
+    mimer_plus (0.0.4)
     nokogiri (1.4.4)
     rest-client (1.6.1)
       mime-types (>= 1.16)
@@ -26,5 +32,9 @@ PLATFORMS
   ruby
 
 DEPENDENCIES
+  levenshteinish
+  mimer_plus
+  nokogiri
+  rest-client
   rspec
   undertexter!

data/README.markdown

@@ -33,6 +33,39 @@ You can also provide an language option.
     $ Undertexter.find("tt0840361", :language => :swedish).count
     => 8
 
+Download the subtitle to disk
+
+    $ Undertexter.find("tt0840361").first.download!
+    => "/tmp/The.Town.2010.EXTENDED.480p.BRRip.XviD-NYDIC.rar"
+    $ File.exists?("/tmp/The.Town.2010.EXTENDED.480p.BRRip.XviD-NYDIC.rar")
+    => true
+    
+You can also specify a destination folder to download the file, both relative and absolute
+    
+    $ Undertexter.find("tt0840361").first.download!(:to => /some/dir)
+    => "/some/dir/The.Town.2010.EXTENDED.480p.BRRip.XviD-NYDIC.rar"
+    
+    $ Dir.pwd
+    => /Users/linus/Downloads
+    $ Undertexter.find("tt0840361").first.download!(:to => 'my_dir')
+    => "/Users/linus/Downloads/my_dir/The.Town.2010.EXTENDED.480p.BRRip.XviD-NYDIC.rar"
+    
+Find the right subtitle based on the release name of the movie
+    
+    $ Undertexter.find("tt0840361").based_on("The Town EXTENDED 2010 480p BRRip XviD AC3 FLAWL3SS")
+    => #<Subtitle:0x00000101b739d0 @cds=1, @downloads=1644, @title="The.Town.EXTENDED.2010.480p.BRRip.XviD.AC3-FLAWL3SS", @details="http://www.undertexter.se/?p=undertext&id=23752", @movie_title="The Town", @language=:swedish>
+    
+Specify how sensitive the `based_on` method should be, where `0` is a perfect match and `1` is dont care. Default is 0.4
+    
+    $ Undertexter.find("tt0840361").based_on("The Town EXTENDED 2010 480p BRRip XviD AC3 FLAWL3SS", limit: 0)
+    => nil
+    
+    $ Undertexter.find("tt0840361").based_on("The Town EXTENDED 2010 480p BRRip XviD AC3 FLAWL3SS", limit: 0.4)
+    => #<Subtitle:0x00000101b8d808 @cds=1, @downloads=1644, @title="The.Town.EXTENDED.2010.480p.BRRip.XviD.AC3-FLAWL3SS", @details="http://www.undertexter.se/?p=undertext&id=23752", @movie_title="The Town", @language=:swedish>
+    
+    $ Undertexter.find("tt0840361").based_on("The.Town.EXTENDED.2010.480p.BRRip.XviD.AC3-FLAWL3SS", limit: 0.0)
+    => #<Subtitle:0x00000101b8d718 @cds=1, @downloads=1644, @title="The.Town.EXTENDED.2010.480p.BRRip.XviD.AC3-FLAWL3SS", @details="http://www.undertexter.se/?p=undertext&id=23752", @movie_title="The Town", @language=:swedish>
+    
 If no language option is being passed to find, it will fall back to swedish
 
 What is being returned from the find method?

data/lib/subtitle.rb

@@ -1,3 +1,6 @@
+require 'fileutils'
+require 'mimer_plus'
+
 class Subtitle
   attr_accessor :details, :downloads, :cds, :title, :movie_title, :url
   
@@ -9,8 +12,54 @@ class Subtitle
     @language == :english ? "http://eng.undertexter.se/subtitle.php?id=#{id}" : "http://www.undertexter.se/utext.php?id=#{id}"
   end
   
+  # Downloading the file and saves it disk
+  def download!(args = {})
+    if args[:to].nil?
+      dir = "/tmp"
+      file_name = "#{dir}/#{generate_file_name}"
+    else
+      dir = generate_custom_file_path(args)
+      file_name = "#{dir}/#{generate_file_name}"
+    end
+    
+    data = RestClient.get(self.url, :timeout => 10) rescue nil
+    file = File.new(file_name, 'w')
+    file.write(data)
+    file.close
+    
+    type = Mimer.identify(file_name)
+    
+    if type.zip?
+      file_ending = ".zip"
+    elsif type.rar?
+      file_ending = ".rar"
+    else
+      file_ending = ""
+    end
+    
+    new_file_name = "#{dir}/#{title.gsub(/\s+/, '.')}#{file_ending}"
+    
+    # Changing the name on the file
+    FileUtils.mv(file_name, new_file_name)
+    
+    # I like return :)
+    return new_file_name
+  end
+  
   private
     def id
       @details.match(/id=(\d+)/)[1]
     end
+    
+    def generate_file_name
+      (0...30).map{65.+(rand(25)).chr}.join.downcase
+    end
+    
+    def generate_custom_file_path(args)
+      # If the path is relative
+      args[:to] = File.expand_path(args[:to]) unless args[:to].match(/^\//)
+
+      # Makes sure that every directory structure looks the same
+      Dir.new(args[:to]).path
+    end
 end

data/lib/undertexter.rb

@@ -4,6 +4,7 @@ require 'rest-client'
 require 'subtitle'
 require 'nokogiri'
 require 'iconv'
+require 'undertexter/array'
 
 class Undertexter
   attr_accessor :raw_data, :base_details, :subtitles

data/lib/undertexter/array.rb

@@ -0,0 +1,13 @@
+require 'levenshteinish'
+
+class Array
+  def based_on(string, args = {})    
+    self.any? ? self.map do|s|
+      [Levenshtein.distance(s.title, string, args[:limit] || 0.4), s]
+    end.reject do |value| 
+      value.first.nil?
+    end.sort_by do |value|
+      value.first
+    end.map(&:last).first : nil
+  end
+end

data/spec/array_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe Array do
+  before(:each) do
+    @subtitles = Undertexter.find("tt0840361")
+  end
+  
+  it "should have a based_on method" do
+    lambda{
+      @subtitles.based_on("some args")
+    }.should_not raise_error(NoMethodError)
+  end
+  
+  it "should return the right one" do
+    @subtitles.based_on("The Town EXTENDED 2010 480p BRRip XviD AC3 FLAWL3SS").details.should match(/id=23752/)
+  end
+  
+  it "should not return anything if the limit is set to low" do
+    @subtitles.based_on("The Town EXTENDED 2010 480p BRRip XviD AC3 FLAWL3SS", :limit => 0).should be_nil
+  end
+  
+  it "should not return the right movie if to litle info i passed" do
+    @subtitles.based_on("The Town").should be_nil
+  end
+  
+  it "should return the right instance" do
+    @subtitles.based_on("The Town EXTENDED 2010 480p BRRip XviD AC3 FLAWL3SS").should be_instance_of(Subtitle)
+  end
+  
+  it "should return nil if trying to fetch an non existing imdb id" do
+    Undertexter.find("tt123456789").based_on("some random argument").should be_nil
+  end
+end

data/spec/spec_helper.rb

@@ -1,5 +1,6 @@
 require 'rspec'
 require "#{File.dirname(__FILE__)}/../lib/undertexter"
+require "#{File.dirname(__FILE__)}/../lib/subtitle"
 
 RSpec.configure do |config|
   config.mock_with :rspec

data/spec/subtitle_spec.rb

@@ -0,0 +1,55 @@
+require 'spec_helper'
+describe Subtitle do
+  before(:each) do
+    @title = "127 Hours 2010 DVDSCR XViD-MC8"
+    @new_file = "/tmp/#{@title.gsub(/\s+/, '.')}.zip"
+    
+    @subtitle = Subtitle.new({
+      :details => "http://www.undertexter.se/?p=undertext&id=23984",
+      :downloads => 100,
+      :cds => 1,
+      :title => @title,
+      :movie_title => "127 Hours (127 Timmar)",
+      :language => :swedish
+    })
+  end
+  
+  it "should be an instance of subtitle" do
+    @subtitle.should be_instance_of(Subtitle)
+  end
+  
+  it "should have the right accessors" do
+    @subtitle.details.should eq("http://www.undertexter.se/?p=undertext&id=23984")
+    @subtitle.downloads.should eq(100)
+    @subtitle.cds.should eq(1)
+    @subtitle.title.should eq("127 Hours 2010 DVDSCR XViD-MC8")
+    @subtitle.movie_title.should eq("127 Hours (127 Timmar)")
+    @subtitle.url.should eq("http://www.undertexter.se/utext.php?id=23984")
+  end
+  
+  it "should be able to download the subtitle" do
+    @subtitle.download!.should eq(@new_file)
+  end
+  
+  it "should have created an existing file" do
+    FileUtils.rm(@new_file)
+    @subtitle.download!
+    File.exists?(@new_file).should be_true
+  end
+  
+  it "should be able to download the file to a specific absolute directory" do
+    %x{rm -r /tmp/new_dir; mkdir /tmp/new_dir}
+    @subtitle.download!(:to => '/tmp/new_dir')
+    %x{ls /tmp/new_dir}.should_not be_empty
+  end
+  
+  it "should be able to download the file to a specific relative directory" do
+    folder = "#{Dir.pwd}/spec/data"
+    
+    @subtitle.download!(:to => 'spec/data')
+    
+    %x{ls #{folder}}.should_not be_empty
+    
+    %x{cd #{folder} && rm 127.Hours.2010.DVDSCR.XViD-MC8.zip}
+  end
+end

data/undertexter.gemspec

@@ -3,7 +3,7 @@ $:.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = "undertexter"
-  s.version     = "0.0.7"
+  s.version     = "0.1.0"
   s.platform    = Gem::Platform::RUBY
   s.authors     = ["Linus Oleander"]
   s.email       = ["linus@oleander.nu"]
@@ -20,5 +20,7 @@ Gem::Specification.new do |s|
   
   s.add_dependency('rest-client')
   s.add_dependency('nokogiri')
+  s.add_dependency('mimer_plus')
+  s.add_dependency('levenshteinish')
   s.add_development_dependency('rspec')
 end

metadata

@@ -1,13 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: undertexter
 version: !ruby/object:Gem::Version 
-  hash: 17
-  prerelease: 
+  prerelease: false
   segments: 
   - 0
+  - 1
   - 0
-  - 7
-  version: 0.0.7
+  version: 0.1.0
 platform: ruby
 authors: 
 - Linus Oleander
@@ -26,7 +25,6 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 3
         segments: 
         - 0
         version: "0"
@@ -40,26 +38,50 @@ dependencies:
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 3
         segments: 
         - 0
         version: "0"
   type: :runtime
   version_requirements: *id002
 - !ruby/object:Gem::Dependency 
-  name: rspec
+  name: mimer_plus
   prerelease: false
   requirement: &id003 !ruby/object:Gem::Requirement 
     none: false
     requirements: 
     - - ">="
       - !ruby/object:Gem::Version 
-        hash: 3
         segments: 
         - 0
         version: "0"
-  type: :development
+  type: :runtime
   version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: levenshteinish
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id005
 description: A subtitle search client to search for swedish subtitles on undertexter.se
 email: 
 - linus@oleander.nu
@@ -78,7 +100,10 @@ files:
 - Rakefile
 - lib/subtitle.rb
 - lib/undertexter.rb
+- lib/undertexter/array.rb
+- spec/array_spec.rb
 - spec/spec_helper.rb
+- spec/subtitle_spec.rb
 - spec/undertexter_spec.rb
 - undertexter.gemspec
 has_rdoc: true
@@ -95,7 +120,6 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 3
       segments: 
       - 0
       version: "0"
@@ -104,17 +128,18 @@ required_rubygems_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 3
       segments: 
       - 0
       version: "0"
 requirements: []
 
 rubyforge_project: undertexter
-rubygems_version: 1.4.2
+rubygems_version: 1.3.7
 signing_key: 
 specification_version: 3
 summary: A subtitle search client for undertexter.se
 test_files: 
+- spec/array_spec.rb
 - spec/spec_helper.rb
+- spec/subtitle_spec.rb
 - spec/undertexter_spec.rb