my_shows 0.0.3 → 0.0.4

data/bin/my_shows

@@ -3,7 +3,7 @@
 require 'my_shows'
 
 # Output message to $stderr, prefixed with the program name
-def pute(message="")
+def pute(message = '')
   $stderr.puts "#{$0}: #{message}"
 end
 
@@ -12,6 +12,6 @@ begin
 
   MyShows::CLI.start(ARGV)
 rescue Interrupt
-  pute "Quitting..."
+  pute 'Quitting...'
   exit 1
 end

data/lib/my_shows.rb

@@ -3,7 +3,7 @@
 
 require 'my_shows/version'
 
-require "bundler/setup"
+require 'bundler/setup'
 
 require 'netrc'
 require 'fuzzystringmatch'
@@ -29,21 +29,21 @@ module MyShows
         MyShows::Show.client = SidereelClient.new(*self.credentials)
       end
 
-      def lookup_magnet_links shows
+      def lookup_magnet_links(shows)
         shows.map { |show|
           show.episode.torrent_link!
         }.compact
       end
 
-      def enque_to_download links
+      def enque_to_download(links)
         links.each do |link|
-          logger.debug "Enque #{link}"
+          logger.info "Enque #{URI(link).to_s}"
           sleep 5
-          Launchy::Application::General.new.open(["#{URI(link).to_s}"])
+          Launchy::Application::General.new.open([URI(link).to_s])
         end
       end
 
-      def start *args
+      def start(*args)
         print_header
         configure_client
         shows = Show.next_episodes
@@ -58,7 +58,7 @@ module MyShows
       end
 
       def print_episodes episodes
-        puts "Next episodes:"
+        puts 'Next episodes:'
         episodes.each do |show|
           episode = show.episode
           puts "#{episode} [#{episode.magnet_link ? '✓'.colorize(:green) : '✗'.colorize(:red)}]"

data/lib/my_shows/episode.rb

@@ -2,7 +2,7 @@ require 'my_shows/the_pirate_bay_client'
 
 module MyShows
   class Episode
-    @@jarow = FuzzyStringMatch::JaroWinkler.create(:native)
+    @@jarow   = FuzzyStringMatch::JaroWinkler.create(:native)
     @@tracker = ThePirateBayClient.new
 
     attr_accessor :season, :episode, :show, :magnet_link
@@ -14,23 +14,37 @@ module MyShows
     end
 
     def torrent_link!
-      episode_search_query = "#{self.to_s} PublicHD"
+      special_authors = %w(PublicHD DIMENSION eztv \ )
+      sizes           = %w(1080p 720p \ )
+
+      special_authors.each do |author|
+        sizes.each do |size|
+          keyword = [author, size].join(' ')
+          return self.magnet_link if torrent_link_with_ext_keyword(keyword)
+        end
+      end
+    end
+
+    def torrent_link_with_ext_keyword(keyword)
+      episode_search_query = "#{self.to_s} #{keyword}"
       MyShows.logger.info "Looking for '#{episode_search_query}' ..."
 
       begin
         torrents = @@tracker.search(episode_search_query)
-        torrent = torrents.sort_by { |t| @@jarow.getDistance(episode_search_query, t.name) }.reverse.first
+        torrent  = torrents.sort_by do |t|
+          @@jarow.getDistance(episode_search_query, t.name)
+        end.reverse.first
 
         self.magnet_link = torrent && torrent.magnet_link
       rescue => e
-        MyShows.logger.warn "Problem with looking torrent link"
+        MyShows.logger.warn "Problem with looking torrent link for '#{episode_search_query}'"
         MyShows.logger.debug e.message
         nil
       end
     end
 
     def to_s
-      "%s s%02de%02d" % [show.name, season, episode]
+      '%s s%02de%02d' % [show.name, season, episode]
     end
   end
 end

data/lib/my_shows/the_pirate_bay_client.rb

@@ -4,7 +4,7 @@ require 'hashie'
 
 class ThePirateBayClient
   def connection
-    @connection ||= Faraday.new url: 'http://thepiratebay.se' do |conn|
+    @connection ||= Faraday.new url: 'http://thepiratebay.is' do |conn|
       conn.headers[:user_agent] = 'libcurl-agent/1.0'
       conn.request :url_encoded
       conn.response :logger, MyShows.logger
@@ -12,14 +12,18 @@ class ThePirateBayClient
     end
   end
 
-  def search query
+  def search(query)
     response = connection.get URI.escape("/search/#{query}/0/99/200")
-    Nokogiri::HTML(response.body).css('#searchResult > tr td:nth-child(2)').map do |row|
-      Hashie::Mash.new(name: row.at_css('.detName a').content, magnet_link: row.at_css('a[href^=magnet]')['href'])
+    html     = Nokogiri::HTML(response.body)
+    html.css('#searchResult > tr').map do |row|
+      Hashie::Mash.new(name:        row.at_css('.detName a').content,
+                       magnet_link: row.at_css('a[href^=magnet]')['href'],
+                       seeders:     row.css('td')[-2].content.to_i)
     end
   end
 end
 
 if __FILE__ == $0
-  p ThePirateBayClient.new.search('mike and molly')
+  require 'my_shows'
+  p ThePirateBayClient.new.search('The Mentalist s05e20 PublicHD')
 end

data/lib/my_shows/version.rb

@@ -1,3 +1,3 @@
 module MyShows
-  VERSION = "0.0.3"
+  VERSION = '0.0.4'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: my_shows
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-15 00:00:00.000000000 Z
+date: 2013-04-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: hashie
@@ -162,12 +162,18 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+      segments:
+      - 0
+      hash: 3563414062259935485
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+      segments:
+      - 0
+      hash: 3563414062259935485
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.25