Class: Mechanize::Chain::URIResolver

Inherits:
Object
  • Object
show all
Includes:
Handler
Defined in:
lib/mechanize/chain/uri_resolver.rb

Instance Attribute Summary

Attributes included from Handler

#chain

Instance Method Summary collapse

Constructor Details

#initialize(scheme_handlers) ⇒ URIResolver

Returns a new instance of URIResolver.



6
7
8
 
# File 'lib/mechanize/chain/uri_resolver.rb', line 6

def initialize(scheme_handlers)
  @scheme_handlers = scheme_handlers
end

Instance Method Details

#handle(ctx, params) ⇒ Object

Raises:

  • (ArgumentError) 


10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/mechanize/chain/uri_resolver.rb', line 10

def handle(ctx, params)
  raise ArgumentError.new('uri must be specified') unless params[:uri]
  params[:uri] = params[:uri].dup if params[:uri].is_a?(URI)
  uri     = params[:uri]
  referer = params[:referer]
  unless uri.is_a?(URI)
    uri = uri.to_s.strip.gsub(/[^#{0.chr}-#{126.chr}]/) { |match|
      if RUBY_VERSION >= "1.9.0"
        CGI.escape(match)
      else
        sprintf('%%%X', match.unpack($KCODE == 'UTF8' ? 'U' : 'c')[0])
      end
    }

    escaped_uri = Util.html_unescape(
                                     uri.split(/(?:%[0-9A-Fa-f]{2})+|#/).zip(
                                                                             uri.scan(/(?:%[0-9A-Fa-f]{2})+|#/)
                                                                             ).map { |x,y|
                                       "#{URI.escape(x)}#{y}"
                                     }.join('')
                                     )

    begin
      uri = URI.parse(escaped_uri)
    rescue
      uri = URI.parse(URI.escape(escaped_uri))
    end

  end
  uri = @scheme_handlers[
                         uri.relative? ? 'relative' : uri.scheme.downcase
                        ].call(uri, params[:referer])

  if params[:referer] && params[:referer].uri
    if uri.path.length == 0 && uri.relative?
      uri.path = params[:referer].uri.path
    end
  end

  uri.path = '/' if uri.path.length == 0

  if uri.relative?
    raise 'need absolute URL' unless referer && referer.uri
    base = nil
    if referer.respond_to?(:bases) && referer.parser
      base = referer.bases.last
    end

    uri = ((base && base.uri && base.uri.absolute?) ?
           base.uri :
           referer.uri) + uri
    uri = referer.uri + uri
    # Strip initial "/.." bits from the path
    uri.path.sub!(/^(\/\.\.)+(?=\/)/, '')
  end

  unless ['http', 'https', 'file'].include?(uri.scheme.downcase)
    raise "unsupported scheme: #{uri.scheme}"
  end
  params[:uri] = uri

  super
end