Contents

require 'typhoeus'
require 'pathname'

module PRSS
  class Downloader
    def initialize(links)
      @links = links
      @hydra = Typhoeus::Hydra.new
    end

    def download_to(output_path)
      output = Pathname.new(output_path).expand_path
      raise 'output path is not directory' unless output.directory?

      @files = []
      @links.each do |link|
        request = Typhoeus::Request.new(link, :follow_location => true)

        request.on_complete do |response|
          @files << save_file(response, output)
        end

        @hydra.queue request
      end

      @hydra.run

      @files
    end

    def save_file response, output
      filename = response.headers_hash['Content-Disposition'][/filename="(.+)"$/ ,1]
      file = output.join(filename)

      open(file, 'wb') do |file|
        file.write(response.body)
      end

      filename
    end

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
prss-0.0.2	lib/prss/downloader.rb