diamond-mechanize 2.1

data/lib/mechanize/form/button.rb

@@ -0,0 +1,6 @@
+##
+# A Submit button in a Form
+
+class Mechanize::Form::Button < Mechanize::Form::Field
+end
+

data/lib/mechanize/form/check_box.rb

@@ -0,0 +1,12 @@
+##
+# This class represents a check box found in a Form.  To activate the CheckBox
+# in the Form, set the checked method to true.
+
+class Mechanize::Form::CheckBox < Mechanize::Form::RadioButton
+
+  def query_value
+    [[@name, @value || "on"]]
+  end
+
+end
+

data/lib/mechanize/form/field.rb

@@ -0,0 +1,54 @@
+##
+# This class represents a field in a form.  It handles the following input
+# tags found in a form:
+#
+# * text
+# * password
+# * hidden
+# * int
+# * textarea
+# * keygen
+#
+# To set the value of a field, just use the value method:
+#
+#   field.value = "foo"
+
+class Mechanize::Form::Field
+  attr_accessor :name, :value, :node, :type
+
+  def initialize node, value = node['value']
+    @node = node
+    @name = Mechanize::Util.html_unescape(node['name'])
+    @value = if value.is_a? String
+               Mechanize::Util.html_unescape(value)
+             else
+               value
+             end
+
+    @type = node['type']
+  end
+
+  def query_value
+    [[@name, @value || '']]
+  end
+
+  def <=> other
+    return 0 if self == other
+    return 1 if Hash === node
+    return -1 if Hash === other.node
+    node <=> other.node
+  end
+
+  # This method is a shortcut to get field's DOM id.
+  # Common usage: form.field_with(:dom_id => "foo")
+  def dom_id
+    node['id']
+  end
+
+  # This method is a shortcut to get field's DOM id.
+  # Common usage: form.field_with(:dom_class => "foo")
+  def dom_class
+    node['class']
+  end
+end
+

data/lib/mechanize/form/file_upload.rb

@@ -0,0 +1,21 @@
+# This class represents a file upload field found in a form.  To use this
+# class, set FileUpload#file_data= to the data of the file you want to upload
+# and FileUpload#mime_type= to the appropriate mime type of the file.
+#
+# See the example in EXAMPLES
+
+class Mechanize::Form::FileUpload < Mechanize::Form::Field
+  attr_accessor :file_name # File name
+  attr_accessor :mime_type # Mime Type (Optional)
+
+  alias :file_data :value
+  alias :file_data= :value=
+
+  def initialize node, file_name
+    @file_name = Mechanize::Util.html_unescape(file_name)
+    @file_data = nil
+    @node      = node
+    super(node, @file_data)
+  end
+end
+

data/lib/mechanize/form/hidden.rb

@@ -0,0 +1,3 @@
+class Mechanize::Form::Hidden < Mechanize::Form::Field
+end
+

data/lib/mechanize/form/image_button.rb

@@ -0,0 +1,19 @@
+##
+# This class represents an image button in a form.  Use the x and y methods to
+# set the x and y positions for where the mouse "clicked".
+
+class Mechanize::Form::ImageButton < Mechanize::Form::Button
+  attr_accessor :x, :y
+
+  def initialize *args
+    @x = nil
+    @y = nil
+    super
+  end
+
+  def query_value
+    [["#{@name}.x", (@x || 0).to_s],
+     ["#{@name}.y", (@y || 0).to_s]]
+  end
+end
+

data/lib/mechanize/form/keygen.rb

@@ -0,0 +1,34 @@
+##
+# This class represents a keygen (public / private key generator) found in a
+# Form. The field will automatically generate a key pair and compute its own
+# value to match the challenge. Call key to access the public/private key
+# pair.
+
+class Mechanize::Form::Keygen < Mechanize::Form::Field
+  # The challenge for this <keygen>.
+  attr_reader :challenge
+
+  # The key associated with this <keygen> tag.
+  attr_reader :key
+
+  def initialize(node, value = nil)
+    super
+    @challenge = node['challenge']
+
+    @spki = OpenSSL::Netscape::SPKI.new
+    @spki.challenge = @challenge
+
+    @key = nil
+    generate_key if value.nil? || value.empty?
+  end
+
+  # Generates a key pair and sets the field's value.
+  def generate_key(key_size = 2048)
+    # Spec at http://dev.w3.org/html5/spec/Overview.html#the-keygen-element
+    @key = OpenSSL::PKey::RSA.new key_size
+    @spki.public_key = @key.public_key
+    @spki.sign @key, OpenSSL::Digest::MD5.new
+    self.value = @spki.to_pem
+  end
+end
+

data/lib/mechanize/form/multi_select_list.rb

@@ -0,0 +1,94 @@
+##
+# This class represents a select list where multiple values can be selected.
+# MultiSelectList#value= accepts an array, and those values are used as
+# values for the select list.  For example, to select multiple values,
+# simply do this:
+#
+#   list.value = ['one', 'two']
+#
+# Single values are still supported, so these two are the same:
+#
+#   list.value = ['one']
+#   list.value = 'one'
+
+class Mechanize::Form::MultiSelectList < Mechanize::Form::Field
+
+  extend Mechanize::ElementMatcher
+
+  attr_accessor :options
+
+  def initialize node
+    value = []
+    @options = []
+
+    # parse
+    node.search('option').each do |n|
+      @options << Mechanize::Form::Option.new(n, self)
+    end
+
+    super node, value
+  end
+
+  ##
+  # :method: option_with
+  #
+  # Find one option on this select list with +criteria+
+  #
+  # Example:
+  #
+  #   select_list.option_with(:value => '1').value = 'foo'
+
+  ##
+  # :method: options_with
+  #
+  # Find all options on this select list with +criteria+
+  #
+  # Example:
+  #
+  #   select_list.options_with(:value => /1|2/).each do |field|
+  #     field.value = '20'
+  #   end
+
+  elements_with :option
+
+  def query_value
+    value ? value.map { |v| [name, v] } : ''
+  end
+
+  # Select no options
+  def select_none
+    @value = []
+    options.each { |o| o.untick }
+  end
+
+  # Select all options
+  def select_all
+    @value = []
+    options.each { |o| o.tick }
+  end
+
+  # Get a list of all selected options
+  def selected_options
+    @options.find_all { |o| o.selected? }
+  end
+
+  def value=(values)
+    select_none
+    [values].flatten.each do |value|
+      option = options.find { |o| o.value == value }
+      if option.nil?
+        @value.push(value)
+      else
+        option.select
+      end
+    end
+  end
+
+  def value
+    value = []
+    value.concat @value
+    value.concat selected_options.map { |o| o.value }
+    value
+  end
+
+end

data/lib/mechanize/form/option.rb

@@ -0,0 +1,50 @@
+##
+# This class contains an option found within SelectList.  A SelectList can
+# have many Option classes associated with it.  An option can be selected by
+# calling Option#tick, or Option#click.
+#
+# To select the first option in a list:
+#
+#   select_list.first.tick
+
+class Mechanize::Form::Option
+  attr_reader :value, :selected, :text, :select_list
+
+  alias :to_s :value
+  alias :selected? :selected
+
+  def initialize(node, select_list)
+    @text     = node.inner_text
+    @value    = Mechanize::Util.html_unescape(node['value'] || node.inner_text)
+    @selected = node.has_attribute? 'selected'
+    @select_list = select_list # The select list this option belongs to
+  end
+
+  # Select this option
+  def select
+    unselect_peers
+    @selected = true
+  end
+
+  # Unselect this option
+  def unselect
+    @selected = false
+  end
+
+  alias :tick   :select
+  alias :untick :unselect
+
+  # Toggle the selection value of this option
+  def click
+    unselect_peers
+    @selected = !@selected
+  end
+
+  private
+  def unselect_peers
+    return unless Mechanize::Form::SelectList === @select_list
+
+    @select_list.select_none
+  end
+end
+

data/lib/mechanize/form/radio_button.rb

@@ -0,0 +1,55 @@
+##
+# This class represents a radio button found in a Form.  To activate the
+# RadioButton in the Form, set the checked method to true.
+
+class Mechanize::Form::RadioButton < Mechanize::Form::Field
+  attr_accessor :checked
+
+  def initialize node, form
+    @checked = !!node['checked']
+    @form    = form
+    super(node)
+  end
+
+  def check
+    uncheck_peers
+    @checked = true
+  end
+
+  alias checked? checked
+
+  def uncheck
+    @checked = false
+  end
+
+  def click
+    checked ? uncheck : check
+  end
+
+  def label
+    (id = self['id']) && @form.page.labels_hash[id] || nil
+  end
+
+  def text
+    label.text rescue nil
+  end
+
+  def [](key)
+    @node[key]
+  end
+
+  def pretty_print_instance_variables # :nodoc:
+    [:@checked, :@name, :@value]
+  end
+
+  private
+
+  def uncheck_peers
+    @form.radiobuttons_with(:name => name).each do |b|
+      next if b.value == value
+      b.uncheck
+    end
+  end
+
+end
+

data/lib/mechanize/form/reset.rb

@@ -0,0 +1,3 @@
+class Mechanize::Form::Reset < Mechanize::Form::Button
+end
+

data/lib/mechanize/form/select_list.rb

@@ -0,0 +1,44 @@
+# This class represents a select list or drop down box in a Form.  Set the
+# value for the list by calling SelectList#value=.  SelectList contains a list
+# of Option that were found.  After finding the correct option, set the select
+# lists value to the option value:
+#
+#   selectlist.value = selectlist.options.first.value
+#
+# Options can also be selected by "clicking" or selecting them.  See Option
+class Mechanize::Form::SelectList < Mechanize::Form::MultiSelectList
+
+  def initialize node
+    super
+    if selected_options.length > 1
+      selected_options.reverse[1..selected_options.length].each do |o|
+        o.unselect
+      end
+    end
+  end
+
+  def value
+    value = super
+    if value.length > 0
+      value.last
+    elsif @options.length > 0
+      @options.first.value
+    else
+      nil
+    end
+  end
+
+  def value=(new)
+    if new != new.to_s and new.respond_to? :first
+      super([new.first])
+    else
+      super([new.to_s])
+    end
+  end
+
+  def query_value
+    value ? [[name, value]] : nil
+  end
+
+end
+

data/lib/mechanize/form/submit.rb

@@ -0,0 +1,3 @@
+class Mechanize::Form::Submit < Mechanize::Form::Button
+end
+

data/lib/mechanize/form/text.rb

@@ -0,0 +1,3 @@
+class Mechanize::Form::Text < Mechanize::Form::Field
+end
+

data/lib/mechanize/form/textarea.rb

@@ -0,0 +1,3 @@
+class Mechanize::Form::Textarea < Mechanize::Form::Field
+end
+

data/lib/mechanize/headers.rb

@@ -0,0 +1,23 @@
+class Mechanize::Headers < Hash
+  def [](key)
+    super(key.downcase)
+  end
+
+  def []=(key, value)
+    super(key.downcase, value)
+  end
+
+  def key?(key)
+    super(key.downcase)
+  end
+
+  def canonical_each
+    block_given? or return enum_for(__method__)
+    each { |key, value|
+      key = key.capitalize
+      key.gsub!(/-([a-z])/) { "-#{$1.upcase}" }
+      yield [key, value]
+    }
+  end
+end
+