download_tv 2.2.1 → 2.2.2

data/lib/download_tv/grabbers/addic7ed.rb

@@ -1,67 +1,62 @@
 module DownloadTV
+  ##
+  # Addic7ed prototype (WIP)
+  class Addic7ed < LinkGrabber
+    def initialize
+      super('http://www.addic7ed.com/search.php?search=%s&Submit=Search')
+    end
 
-	class Addic7ed < LinkGrabber
-		def initialize
-			super("http://www.addic7ed.com/search.php?search=%s&Submit=Search", "+")
-		end
+    def get_subs(show)
+      url = get_url(show)
+      download_file(url)
+    end
 
-		def get_subs(show)
-			url = get_url(show)
-			download_file(url)
-		end
-		
-		def get_url(show)
-			# Change spaces for the separator
-			s = show.gsub(" ", @sep)
+    def get_url(show)
+      # Change spaces for the separator
+      s = show.gsub(' ', @sep)
 
-			# Format the url
-			search = @url % [s]
+      # Format the url
+      search = Format(@url, s)
 
-			agent = Mechanize.new
-			res = agent.get(search)
+      agent = Mechanize.new
+      res = agent.get(search)
 
-			# No redirection means no subtitle found
-			raise NoSubtitlesError if res.uri.to_s == search
+      # No redirection means no subtitle found
+      raise NoSubtitlesError if res.uri.to_s == search
 
-			##########
-			# DO OPENSUBTITLES FIRST (see subtitles.rb)
-			#####
+      ##########
+      # DO OPENSUBTITLES FIRST (see subtitles.rb)
+      #####
 
-			# We now have an URL like:
-			# http://www.addic7ed.com/serie/Mr._Robot/2/3/eps2.1k3rnel-pan1c.ksd
+      # We now have an URL like:
+      # http://www.addic7ed.com/serie/Mr._Robot/2/3/eps2.1k3rnel-pan1c.ksd
 
-			# To find the real links:
-			# see comments at the end of file
+      # To find the real links:
+      # see comments at the end of file
+    end
 
+    def download_file(url)
+      # Url must be like 'http://www.addic7ed.com/updated/1/115337/0'
 
-		end	
+      # ADDIC7ED PROVIDES RSS
 
-		def download_file(url)
-			# Url must be like "http://www.addic7ed.com/updated/1/115337/0"
-
-			# ADDIC7ED PROVIDES RSS
-
-			agent = Mechanize.new
-			page = agent.get(url, [], @url)
-			puts page.save("Hi")
-			
-		end
-		
-	end
+      agent = Mechanize.new
+      page = agent.get(url, [], @url)
+      puts page.save('Hi')
+    end
+  end
 end
-
-
 # subtitles = {}
-# html.css(".tabel95 .newsDate").each do |td|
-# 	if downloads = td.text.match(/\s(\d*)\sDownloads/i)
-# 		done = false
-# 		td.parent.parent.xpath("./tr/td/a[@class='buttonDownload']/@href").each do |link|
-# 			if md = link.value.match(/updated/i)
-# 				subtitles[downloads[1].to_i] = link.value
-# 				done = true
-# 			elsif link.value.match(/original/i) && done == false
-# 				subtitles[downloads[1].to_i] = link.value
-# 				done = true
-# 			end
-# 		end
-# 	end
+# html.css('.tabel95 .newsDate').each do |td|
+#   if downloads = td.text.match(/\s(\d*)\sDownloads/i)
+#     done = false
+#     td.parent.parent.xpath('./tr/td/a[@class='buttonDownload']/@href').each do |link|
+#       if md = link.value.match(/updated/i)
+#         subtitles[downloads[1].to_i] = link.value
+#         done = true
+#       elsif link.value.match(/original/i) && done == false
+#         subtitles[downloads[1].to_i] = link.value
+#         done = true
+#       end
+#     end
+#   end

data/lib/download_tv/grabbers/eztv.rb

@@ -1,30 +1,27 @@
 module DownloadTV
-	class Eztv < LinkGrabber
-		def initialize
-			super("https://eztv.ag/search/%s")
-		end
+  ##
+  # EZTV.ag grabber
+  class Eztv < LinkGrabber
+    def initialize
+      super('https://eztv.ag/search/%s')
+    end
 
-		def get_links(s)
+    def get_links(s)
+      # Format the url
+      search = format(@url, s)
 
-			# Format the url
-			search = @url % [s]
+      data = @agent.get(search).search('a.magnet')
 
-			data = @agent.get(search).search("a.magnet")
+      # Torrent name in data[i].attribute 'title'
+      # 'Suits S04E01 HDTV x264-LOL Torrent: Magnet Link'
 
-			# Torrent name in data[i].attribute "title"
-			# "Suits S04E01 HDTV x264-LOL Torrent: Magnet Link"
+      # EZTV shows 50 latest releases if it can't find the torrent
+      raise NoTorrentsError if data.size == 50
 
-			# EZTV shows 50 latest releases if it can't find the torrent
-			raise NoTorrentsError if data.size == 50
+      names = data.collect { |i| i.attribute('title').text.chomp(' Magnet Link') }
+      links = data.collect { |i| i.attribute('href').text }
 
-			names = data.collect { |i| i.attribute("title").text.chomp(" Magnet Link") }
-			links = data.collect { |i| i.attribute("href").text }
-
-			names.zip(links)
-			
-		end
-		
-		
-	end
-	
+      names.zip(links)
+    end
+  end
 end

data/lib/download_tv/grabbers/kat.rb

@@ -1,34 +1,30 @@
 module DownloadTV
-	class KAT < LinkGrabber
-		def initialize
-			super('https://katcr.co/new/search-torrents.php?search="%s"&sort=seeders&order=desc')
-		end
+  ##
+  # KATcr.co grabber
+  class KAT < LinkGrabber
+    def initialize
+      super('https://katcr.co/new/search-torrents.php?search="%s"&sort=seeders&order=desc')
+    end
 
-		def get_links(s)
+    def get_links(s)
+      # Format the url
+      search = format(@url, s)
 
-			# Format the url
-			search = @url % [s]
+      data = @agent.get(search).links.select { |i| i.href.include? 'torrents-details.php?' }
 
-			data = @agent.get(search).links.select {|i| i.href.include? "torrents-details.php?" }
-			
-			raise NoTorrentsError if data == []
+      raise NoTorrentsError if data == []
 
-			# Remove duplicates
-			data.keep_if { |i| i.text != "" }
+      # Remove duplicates
+      data.keep_if { |i| i.text != '' }
 
-			names = data.collect(&:text)
-			links = []
+      names = data.collect(&:text)
+      links = []
 
-			data.each do |res|
-				page = res.click
-				links << page.search('a.kaGiantButton[title="Magnet link"]').attribute("href").text
-			end
+      data.each do |res|
+        links << res.click.search('a.kaGiantButton[title="Magnet link"]').attribute('href').text
+      end
 
-			names.zip(links)
-			
-		end
-		
-		
-	end
-	
+      names.zip(links)
+    end
+  end
 end

data/lib/download_tv/grabbers/torrentapi.rb

@@ -1,68 +1,62 @@
 module DownloadTV
-
-	class TorrentAPI < LinkGrabber
-
-		attr_accessor :token
-		attr_reader :wait
-
-		def initialize
-			super("https://torrentapi.org/pubapi_v2.php?mode=search&search_string=%s&token=%s&app_id=DownloadTV")
-			@wait = 2.1
-			
-		end
-
-		##
-		# Connects to Torrentapi.org and requests a token.
-		# Returns said token.
-		def get_token
-			page = @agent.get("https://torrentapi.org/pubapi_v2.php?get_token=get_token&app_id=DownloadTV").content
-			
-			obj = JSON.parse(page)
-
-			@token = obj['token']
-
-		end
-
-		def get_links(s)
-			@token ||= get_token
-
-			# Format the url
-			search = @url % [s, @token]
-
-			page = @agent.get(search).content
-			obj = JSON.parse(page)
-
-			if obj["error_code"]==4 # Token expired
-				get_token
-				search = @url % [s, @token]
-				page = @agent.get(search).content
-				obj = JSON.parse(page)
-			end
-
-			while obj["error_code"]==5 # Violate 1req/2s limit
-				# puts "Torrentapi request limit hit. Wait a few seconds..."
-				sleep(@wait) 
-				page = @agent.get(search).content
-				obj = JSON.parse(page)
-
-			end
-
-			raise NoTorrentsError if obj["error"]
-
-			names = obj["torrent_results"].collect {|i| i["filename"]}
-			links = obj["torrent_results"].collect {|i| i["download"]}
-
-			names.zip(links)
-
-		rescue Mechanize::ResponseCodeError
-			# Temporary solution for Cloudflare being obnoxious
-			raise NoTorrentsError
-		end
-
-	end
-
+  ##
+  # TorrentAPI.org grabber
+  class TorrentAPI < LinkGrabber
+    attr_accessor :token
+    attr_reader :wait
+
+    def initialize
+      super('https://torrentapi.org/pubapi_v2.php?mode=search&search_string=%s&token=%s&app_id=DownloadTV')
+      @wait = 2.1
+    end
+
+    ##
+    # Connects to Torrentapi.org and requests a token.
+    # Returns said token.
+    def renew_token
+      page = @agent.get('https://torrentapi.org/pubapi_v2.php?get_token=get_token&app_id=DownloadTV').content
+
+      obj = JSON.parse(page)
+
+      @token = obj['token']
+      # Tokens automaticly expire in 15 minutes.
+      # The api has a 1req/2s limit.
+      # http://torrentapi.org/apidocs_v2.txt
+    end
+
+    def get_links(s)
+      @token ||= renew_token
+
+      # Format the url
+      search = format(@url, s, @token)
+
+      page = @agent.get(search).content
+      obj = JSON.parse(page)
+
+      if obj['error_code'] == 4 # Token expired
+        renew_token
+        search = format(@url, s, @token)
+        page = @agent.get(search).content
+        obj = JSON.parse(page)
+      end
+
+      while obj['error_code'] == 5 # Violate 1req/2s limit
+        # puts 'Torrentapi request limit hit. Wait a few seconds...'
+        sleep(@wait)
+        page = @agent.get(search).content
+        obj = JSON.parse(page)
+
+      end
+
+      raise NoTorrentsError if obj['error']
+
+      names = obj['torrent_results'].collect { |i| i['filename'] }
+      links = obj['torrent_results'].collect { |i| i['download'] }
+
+      names.zip(links)
+      # Temporary solution for Cloudflare being obnoxious
+    rescue Mechanize::ResponseCodeError
+      raise NoTorrentsError
+    end
+  end
 end
-
-# Tokens automaticly expire in 15 minutes.
-# The api has a 1req/2s limit.
-# http://torrentapi.org/apidocs_v2.txt

data/lib/download_tv/grabbers/tpb.rb

@@ -1,34 +1,30 @@
 module DownloadTV
+  ##
+  # ThePirateBay grabber
+  class ThePirateBay < LinkGrabber
+    def initialize(tpb_proxy = 'https://pirateproxy.cc')
+      proxy = tpb_proxy.gsub(%r{/+$}, '') || 'https://thepiratebay.cr'
 
-	class ThePirateBay < LinkGrabber
+      super("#{proxy}/search/%s/0/7/0")
+    end
 
-		def initialize(tpb_proxy = "https://pirateproxy.cc")
-			proxy = tpb_proxy.gsub(/\/+$/, "") || "https://thepiratebay.cr"
+    def get_links(s)
+      # Format the url
+      search = format(@url, s)
 
-			super("#{proxy}/search/%s/0/7/0")
-			
-		end
+      data = @agent.get(search).search('#searchResult tr')
+      # Skip the header
+      data = data.drop 1
 
-		def get_links(s)
+      raise NoTorrentsError if data.empty?
 
-			# Format the url
-			search = @url % [s]
+      # Second cell of each row contains links and name
+      results = data.map { |d| d.search('td')[1] }
 
-			data = @agent.get(search).search("#searchResult tr")
-			# Skip the header
-			data = data.drop 1
+      names = results.collect { |i| i.search('.detName').text.strip }
+      links = results.collect { |i| i.search('a')[1].attribute('href').text }
 
-			raise NoTorrentsError if data.size == 0
-
-			# Second cell of each row contains links and name
-			results = data.map { |d| d.search("td")[1] }
-
-			names = results.collect {|i| i.search(".detName").text.strip }
-			links = results.collect {|i| i.search("a")[1].attribute("href").text }
-
-			names.zip(links)
-
-		end
-
-	end
-end
+      names.zip(links)
+    end
+  end
+end

data/lib/download_tv/linkgrabber.rb

@@ -1,32 +1,26 @@
 module DownloadTV
-
-	class LinkGrabber
-		attr_reader :url
-
-		def initialize(url)
-			@url = url
-			@agent = Mechanize.new
-			
-		end
-
-		def test_connection
-			agent = Mechanize.new
-			agent.read_timeout = 2
-			agent.get(@url)
-		end
-
-		def get_links(_s)
-			raise NotImplementedError
-		end
-		
-	end
-
-	class NoTorrentsError < StandardError
-
-	end
-
-	class NoSubtitlesError < StandardError
-
-	end
-
-end
+  ##
+  # Interface for the grabbers
+  class LinkGrabber
+    attr_reader :url
+
+    def initialize(url)
+      @url = url
+      @agent = Mechanize.new
+    end
+
+    def test_connection
+      agent = Mechanize.new
+      agent.read_timeout = 2
+      agent.get(@url)
+    end
+
+    def get_links(_s)
+      raise NotImplementedError
+    end
+  end
+
+  class NoTorrentsError < StandardError; end
+
+  class NoSubtitlesError < StandardError; end
+end