download_service.rb 670 字节
Newer Older
J
Jared Szechy 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
module Projects
  class DownloadService < BaseService
    WHITELIST = [
      /^[^.]+\.fogbugz.com$/
    ]

    def initialize(project, url)
      @project, @url = project, url
    end

    def execute
      return nil unless valid_url?(@url)

      uploader = FileUploader.new(@project)
      uploader.download!(@url)
      uploader.store!

D
Douwe Maan 已提交
18
      uploader.to_h
J
Jared Szechy 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
    end

    private

    def valid_url?(url)
      url && http?(url) && valid_domain?(url)
    end

    def http?(url)
      url =~ /\A#{URI::regexp(['http', 'https'])}\z/
    end

    def valid_domain?(url)
      host = URI.parse(url).host
      WHITELIST.any? { |entry| entry === host }
    end
  end
end