torrentz 0.0.1 → 0.0.2

data/Rakefile

@@ -1,2 +1,20 @@
 #!/usr/bin/env rake
 require "bundler/gem_tasks"
+
+require 'rake'
+require 'rake/testtask'
+
+task :default => :test
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+  t.pattern = "test/**/*_test.rb"
+  t.verbose = true
+end
+Rake::Task['test'].comment = "Run all tests"
+
+Rake::TestTask.new('integration') do |t|
+  t.libs << 'test'
+  t.pattern = "test/integration/*_test.rb"
+  t.verbose = true
+end

data/lib/torrentz/fetch.rb

@@ -1,7 +1,11 @@
+require 'addressable/uri'
+
 module Torrentz
   class Fetch
     include Torrentz::Logger
 
+    attr_accessor :url
+
     def initialize(url)
       @url = url
     end
@@ -24,14 +28,14 @@ module Torrentz
         end
       end
 
-      return nil
+      return torrent_from_torrentz_url
     end
 
     def get_magnet
       candidate_urls.each do |candidate_url|
         case candidate_url
         when /thepiratebay/
-          logger.info "Found Piratebay!"
+          logger.info "Found Piratebay: #{candidate_url}"
           return PirateBay.new(candidate_url).get
         end
       end
@@ -40,7 +44,7 @@ module Torrentz
     end
 
     def torrentz_doc
-      @torrentz_doc ||= Nokogiri::HTML(open(@url))
+      @torrentz_doc ||= Nokogiri::HTML(open(Addressable::URI.encode(url)))
     end
 
     def candidate_urls
@@ -49,6 +53,13 @@ module Torrentz
       end
     end
 
+    ##
+    # Uses torcache to try to fetch a torrent.
+    def torrent_from_torrentz_url
+      torrent_id = url.split("/").last
+      "https://torcache.net/torrent/#{torrent_id}.torrent"
+    end
+
     class Simple
       include Torrentz::Logger
 
@@ -57,7 +68,7 @@ module Torrentz
       end
 
       def doc
-        @doc ||= Nokogiri::HTML(open(@url))
+        @doc ||= Nokogiri::HTML(open(Addressable::URI.encode(@url)))
       end
     end
 

data/lib/torrentz/logger.rb

@@ -5,5 +5,9 @@ module Torrentz
     def logger
       @@logger ||= ::Logger.new($stdout)
     end
+
+    def logger=(logger)
+      @@logger = logger
+    end
   end
 end

data/lib/torrentz/version.rb

@@ -1,3 +1,3 @@
 module Torrentz
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/test/integration/fetch_test.rb

@@ -0,0 +1,28 @@
+require 'test_helper'
+
+describe Torrentz do
+  it 'will find a file that resembles what was queried' do
+    [
+      'Big Bang Theory S05E04 720p',
+      'House S01E01'
+    ].each do |search|
+      result = Torrentz.get(search)
+      logger.info result.inspect
+
+      torrent_info = Typhoeus::Request.get(result[:torrent], :follow_location => true).body.bdecode
+      torrent_name = torrent_info['info']['name']
+
+      size_difference = torrent_name.size - search.size
+      size_difference * -1 if size_difference < 0
+
+      smallest = [torrent_name.size, search.size].min
+      distance = Text::Levenshtein.distance(torrent_name, search)
+      logger.info "Distance of result: #{distance}"
+
+      difference = (distance - size_difference) / smallest.to_f
+      logger.info "Difference of result: #{difference}"
+
+      difference.must_be_within_delta 0, 0.2
+    end
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,15 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'minitest/autorun'
+require 'minitest/spec'
+require 'open-uri'
+require 'text'
+require 'bencode'
+require 'typhoeus'
+require 'torrentz'
+
+begin
+  require 'ruby-debug'
+rescue LoadError; end
+
+include Torrentz::Logger

data/test/torrentz/fetch_test.rb

@@ -0,0 +1,7 @@
+require 'test_helper'
+
+describe Torrentz::Fetch do
+  test 'truth' do
+    assert true
+  end
+end

data/torrentz.gemspec

@@ -17,4 +17,9 @@ Gem::Specification.new do |gem|
 
   gem.add_dependency "rack"
   gem.add_dependency "nokogiri"
+  gem.add_dependency "addressable"
+
+  gem.add_development_dependency "text"
+  gem.add_development_dependency "typhoeus"
+  gem.add_development_dependency "torrent-ruby"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torrentz
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-09 00:00:00.000000000 Z
+date: 2012-03-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack
-  requirement: &70201271885300 !ruby/object:Gem::Requirement
+  requirement: &70148867837020 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70201271885300
+  version_requirements: *70148867837020
 - !ruby/object:Gem::Dependency
   name: nokogiri
-  requirement: &70201271884500 !ruby/object:Gem::Requirement
+  requirement: &70148867836360 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,7 +32,51 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70201271884500
+  version_requirements: *70148867836360
+- !ruby/object:Gem::Dependency
+  name: addressable
+  requirement: &70148867835740 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70148867835740
+- !ruby/object:Gem::Dependency
+  name: text
+  requirement: &70148867835100 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70148867835100
+- !ruby/object:Gem::Dependency
+  name: typhoeus
+  requirement: &70148867834480 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70148867834480
+- !ruby/object:Gem::Dependency
+  name: torrent-ruby
+  requirement: &70148867834060 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70148867834060
 description: torrentz.eu ruby client to search and download torrentz & magnets
 email:
 - mrsimo@gmail.com
@@ -50,6 +94,9 @@ files:
 - lib/torrentz/logger.rb
 - lib/torrentz/search.rb
 - lib/torrentz/version.rb
+- test/integration/fetch_test.rb
+- test/test_helper.rb
+- test/torrentz/fetch_test.rb
 - torrentz.gemspec
 homepage: https://github.com/mrsimo/ruby-torrentz
 licenses: []
@@ -75,4 +122,7 @@ rubygems_version: 1.8.10
 signing_key: 
 specification_version: 3
 summary: torrentz.eu ruby client to search and download torrentz & magnets
-test_files: []
+test_files:
+- test/integration/fetch_test.rb
+- test/test_helper.rb
+- test/torrentz/fetch_test.rb