Class: Horseman::Dom::Document

Inherits:
Object
  • Object
show all
Defined in:
lib/horseman/dom/document.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(body) ⇒ Document

Returns a new instance of Document.



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/horseman/dom/document.rb', line 13

def initialize(body)
	@forms = {}
	@scripts = []
	@frames = {}

	@field_types = {
      'text' => :text,
      'checkbox' => :checkbox,
      'hidden' => :hidden,
      'submit' => :submit
    }
    
    @encoding_types = {
      'application/x-www-form-urlencoded' => :url,
      'multipart/form-data' => :multipart
    }
    
	parse! body
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(method, *arguments, &block) ⇒ Object 



89
90
91
92
93
94
95
96
97
98
 
# File 'lib/horseman/dom/document.rb', line 89

def method_missing(method, *arguments, &block)
	case method
	when :[]
		indexer = arguments[0]
		form = @forms.select {|id, form| id.to_s == indexer}.map {|k,v| v}.first
		return form unless form.nil?
	end

  puts "Not implemented in Document: #{method} #{arguments.join(',')}"
end

Instance Attribute Details

#dom ⇒ Object (readonly)

Returns the value of attribute dom.



11
12
13
 
# File 'lib/horseman/dom/document.rb', line 11

def dom
  @dom
end

#forms ⇒ Object (readonly)

Returns the value of attribute forms.



11
12
13
 
# File 'lib/horseman/dom/document.rb', line 11

def forms
  @forms
end

#frames ⇒ Object (readonly)

Returns the value of attribute frames.



11
12
13
 
# File 'lib/horseman/dom/document.rb', line 11

def frames
  @frames
end

#scripts ⇒ Object (readonly)

Returns the value of attribute scripts.



11
12
13
 
# File 'lib/horseman/dom/document.rb', line 11

def scripts
  @scripts
end

Instance Method Details

#parse!(body) ⇒ Object 



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
73
74
75
76
77
78
79
80
81
82
83
 
# File 'lib/horseman/dom/document.rb', line 33

def parse!(body)
	doc = Nokogiri::HTML(body)
    
    doc.css('form').select{|f| f.attr('id')}.each do |f|
      form = Form.new(f.attr('id'), f.attr('name'))
      form.action = f.attr('action') || '/'
      form.encoding = @encoding_types[f.attr('enctype')] || :url
      form.fields = {}
      
      f.css('input').select{|i| i.attr('name')}.each do |i|
        field = FormField.new(i.attr('id'), i.attr('name'))
        field.type = @field_types[i.attr('type')] || :text
        field.value = i.attr('value')

        form.fields[field.name.to_sym] = field 
        form.submit = field if (field.type == :submit)  
      end
      
      @forms[form.id.to_sym] = form
    end

    valid_script_types = ['javascript', 'text/javascript']
    doc.css('script').select{|s| (s.attr('type').nil?) || (valid_script_types.include? s.attr('type')) }.each do |s|
    	script = Script.new
    	if s.attr('src')
    		# TODO -- account for HTTP failures
    		js_src = s.attr('src')
    		begin
    			script.body = open(s.attr('src'))	{|f| f.read.strip}
    		rescue
    			puts "Could not load javascript at #{js_src}"	
    		end
    	else
    		script.body = s.inner_html.strip
    	end

    	@scripts << script
    end

    doc.css('frame').select{|f| f.attr('src') && f.attr('name')}.each do |f|
    	frame_src = f.attr('src')
    	begin
    		frame_body = open(frame_src) {|f| f.read.strip}
    		@frames[f.attr('name').to_sym] = Document.new(frame_body)
    	rescue
    		puts "Could not load frame at #{frame_src}"
    	end
    end

    @dom = doc
end

#respond_to?(method_sym, include_private = false) ⇒ Boolean

Returns:

  • (Boolean) 


85
86
87
 
# File 'lib/horseman/dom/document.rb', line 85

def respond_to?(method_sym, include_private = false)
  true
end