narf 0.5.1

data/lib/web/unit/image.rb

@@ -0,0 +1,27 @@
+=begin
+= Web::Unit::Image
+  Copyright(C) 2001 yuichi TAKAHASHI
+  $Id: image.rb,v 1.1.1.1 2003/01/21 11:50:40 yuichi Exp $
+=end
+
+require 'net/http'
+
+module Web
+  module Unit #:nodoc: all
+
+  class Image < HtmlElem
+
+    attr_reader :src, :alt, :width, :height, :border
+
+    def initialize( ah )
+      super( 'img', ah )
+      @src = ah["src"]
+      @alt = ah["alt"]
+      @width = ah["width"]
+      @height = ah["height"]
+      @border = ah["border"]
+    end
+
+  end
+end
+end

data/lib/web/unit/jscript.rb

@@ -0,0 +1,32 @@
+=begin
+= Web::Unit::JSciriptOpenObject
+  Copyright(C) 2001 yuichi TAKAHASHI
+  $Id: jscript.rb,v 1.1.1.1 2003/01/21 11:50:40 yuichi Exp $
+=end
+
+module Web
+  module Unit #:nodoc: all
+
+  class JSciriptOpenObject < HtmlElem
+
+    attr_reader :url, :file, :target
+
+    def initialize( ah )
+      super( 'open', ah )
+      @url = ah['url']
+      @file = ah['file']
+      @target = ah['target']
+    end
+
+=begin
+--- JSciriptOpenObject#read
+      return a Response of getting of '@url'
+=end
+
+    def read
+      Response::new.init_http( @url, "GET" ) if @url != nil
+    end
+  end
+  end
+
+end

data/lib/web/unit/link.rb

@@ -0,0 +1,35 @@
+=begin
+= Web::Unit::Link
+  Copyright(C) 2001 yuichi TAKAHASHI
+  $Id: link.rb,v 1.1.1.1 2003/01/21 11:50:40 yuichi Exp $
+=end
+
+require 'net/http'
+
+module Web
+  module Unit #:nodoc: all
+
+  class Link < HtmlElem
+
+    attr_reader :href, :target, :onClick
+
+    def initialize( ah )
+      super( 'a', ah )
+      @href = ah["href"]
+      @target = ah["target"]
+      @onClick = ah["onclick"]
+    end
+
+=begin
+--- Link#read
+      return a Response of getting of '@href'
+=end
+
+    def read
+      Response::new.init_http( @href, "GET" ) if @href != nil
+    end
+
+  end
+
+end
+end

data/lib/web/unit/params.rb

@@ -0,0 +1,323 @@
+=begin
+= Web::Unit::Param/...
+  Copyright(C) 2001 yuichi TAKAHASHI, Narushima Hironori.
+  $Id: params.rb,v 1.1.1.1 2003/01/21 11:50:40 yuichi Exp $
+=end
+
+module Web
+  module Unit #:nodoc: all
+
+  class Param < HtmlElem
+
+    attr_reader :name, :type, :relations
+    attr_accessor :stat, :value
+
+    def initialize( tag, ah )
+      super
+      @type = ah["type"] ? ah["type"] : 'text'
+      @value = ah["value"]
+      @relations = []
+      @stat = ah.key?( 'checked' ) ? 'on' : 'off'
+    end
+
+    def update( p )
+      @relations << p
+    end
+
+    def query_data
+      value = @value ? CGI::escape( @value ) : ''
+      @name ? ( CGI::escape( @name ) + '=' + value ) : nil
+    end
+
+    def multipart_query_data( boundary)
+      value = @value ? @value : ''
+      
+      if (@name)
+        name = CGI::escape(@name)
+          "--#{boundary}\n" +
+          %!Content-Disposition: form-data; name="#{name}"\n! +
+          "\n" +
+          value + "\n"
+      else
+        nil
+      end
+    end
+  end
+
+  class Input < Param
+
+    def initialize( ah )
+      super( 'input', ah )
+    end
+    
+    def on
+      @stat = 'on'
+    end
+
+    def off
+      @stat = 'off'
+    end
+
+    def inspect
+      a = super + "[stat:#{@stat}]"
+    end
+
+  end
+
+  class InputRadio < Input
+
+    def on
+      @stat = 'on'
+      @relations.each do |r|
+        r.off
+      end
+    end
+
+    def query_data
+      value = @value ? CGI::escape( @value ) : ''
+      (@stat == 'on' and @name) ? CGI::escape(@name) + '=' + value : nil
+    end
+
+    def multipart_query_data( boundary)
+      if @stat == 'on'
+        super(boundary)
+      else
+        nil
+      end
+    end
+
+    def value=( v )
+      self.stat = self.value == v ? 'on' : 'off'
+      @relations.each do |e|
+        e.stat = e.value == v ? 'on' : 'off'
+      end
+    end
+  end
+
+  class InputCheckbox < Input
+    def query_data
+      value = @value ? CGI::escape( @value ) : 'on'
+      (@stat == 'on' and @name) ? @name + '=' + value : nil
+    end
+
+    def multipart_query_data( boundary)
+      if @stat == 'on'
+        super( boundary)
+      else
+        nil
+      end
+    end
+  end
+
+  class InputSubmit < Input
+    def query_data
+      value = CGI::escape( @value ? @value : 'Submit Query' )
+      @name ? CGI::escape(@name) + '=' + value : nil
+    end
+  end
+
+  class InputImage < Input
+    attr_writer :x, :y
+
+    def query_data
+      @name + '.x=' + @x.to_s + '&' + @name + '.y=' + @y.to_s
+    end
+    
+    def multipart_query_data( boundary)
+      "--#{boundary}\n" +
+      %!Content-Disposition: form-data; name="#{@name}.x"\n! +
+      "\n" +
+      @x.to_s + "\n" +
+      "--#{boundary}\n" +
+      %!Content-Disposition: form-data; name="#{@name}.y"\n! +
+      "\n" +
+      @y.to_s + "\n"
+    end
+  end
+
+  class InputReset < Input
+
+    def query_data
+      nil
+    end
+
+    def multipart_query_data( boundary)
+      nil
+    end
+    
+  end
+
+  class Select < Param
+
+    attr_reader :options
+
+    def initialize( ah )
+      super( 'select', ah )
+      @options = []
+      @size = ah["size"]
+      @type = 'select'
+    end
+
+    def add_option( option )
+      @options.push( option )
+    end
+
+    def end_option
+      if @size == nil && @options.reject { |o| o.stat == 'off' } == []
+        @options[0].select
+      end
+    end
+
+    def select( arr )
+      @options.each do |o|
+        o.deselect
+      end
+      arr.each do |e|
+        @options[e].select if e.kind_of?( Integer )
+        if e.kind_of?( String )
+          @options.each do |o|
+            o.select if o.value == e || o.data == e
+          end
+        end
+      end
+    end
+
+    def query_data
+      a = []
+      @options.each do |o|
+        a << o.query_data( @name )
+      end
+      s = a.compact.join('&')
+      s == '' ? nil : s
+    end
+
+    def multipart_query_data( boundary)
+      lines = ''
+      @options.each do |o|
+        data = o.multipart_query_data(boundary, @name)
+        lines << data if data
+      end
+      
+      lines
+    end
+
+    def value
+      arr = []
+      @options.each do |o|
+        arr << o.value if o.stat == 'on'
+      end
+      case arr.size
+        when 0 then nil
+        when 1 then arr[0]
+        else arr
+      end
+    end
+
+    def value= ( v )
+      @options.each do |o|
+        if o.value == v
+          o.select
+        else
+          o.deselect
+        end
+      end
+    end
+  end
+
+  class SelectOption < HtmlElem
+    attr_reader :stat
+
+    def initialize( ah )
+      super( 'option', ah )
+      @value = ah["value"]
+      @stat = ah.key?( 'selected' ) ? 'on' : 'off'
+      @type = 'option'
+    end
+
+    def value
+      @value ? @value : @data
+    end
+
+    def select
+      @stat = 'on'
+    end
+
+    def deselect
+      @stat = 'off'
+    end
+
+    def query_data( name )
+      @value = @data unless @value
+      value = @value ? CGI::escape( @value ) : ''
+      (@stat == 'on' and name) ? name + '=' + value : nil
+    end
+    
+    def multipart_query_data( boundary, name)
+      @value = @data unless @value
+      value = @value ? @value : ''
+      
+      if @stat == 'on' and name
+        "--#{boundary}\n" +
+        %!Content-Disposition: form-data; name="#{CGI::escape(name)}"\n! +
+        "\n" +
+        value + "\n"
+      else
+        nil
+      end
+    end
+    
+  end
+
+  class Textarea < Param
+    def initialize( ah )
+      super( 'textarea', ah )
+      @type = 'textarea'
+    end
+
+    def value
+      @value ? @value : @data
+    end
+
+    def query_data
+      val = @value ? @value : @data
+      value = val ? CGI::escape( val ) : ''
+      @name ? ( CGI::escape( @name ) + '=' + value ) : nil
+    end
+                          
+    def multipart_query_data( boundary)
+      @value = @data unless @value
+      value = @value ? @value : ''
+
+      if @name
+        "--#{boundary}\n" +
+        %!Content-Disposition: form-data; name="#{CGI::escape(@name)}"\n! +
+        "\n" +
+        value + "\n"
+      else
+        nil
+      end
+    end
+    
+  end
+  
+  class InputFile < Input
+    def query_data
+      nil
+    end
+
+    def multipart_query_data( boundary)
+      if @name and @value
+        "--#{boundary}\n" +
+        %!Content-Disposition: form-data; name="#{File.basename(@name)}"; filename="#{File.basename(@value)}"\n! +
+        "Content-Type: text/plain\n" +
+        "\n" +
+        IO.readlines(@value).join('') + "\n"
+      else
+        nil
+      end
+    end
+  
+  end
+
+end
+end

data/lib/web/unit/parser.rb

@@ -0,0 +1,226 @@
+=begin
+= Web::Unit::Parser
+  Copyright(C) 2001 yuichi TAKAHASHI, Narushima Hironori.
+  $Id: parser.rb,v 1.1.1.1 2003/01/21 11:50:40 yuichi Exp $
+  Part of Web::Unit::Parser is variant of html-parser
+=end
+
+require 'web/unit/htmlrepair'
+require 'web/unit/sgml-parser'
+require 'web/unit/html-parser'
+require 'web/unit/formatter'
+
+module Web
+  module Unit #:nodoc: all
+    
+    class Parser < Web::Unit::SGMLParser
+
+      include Utils
+
+      @@ignoretags = []
+      SingleTags = %w( meta br hr base link ) # input, frame, img
+
+      def initialize()
+        super( NullFormatter.new )
+        @elem_stack = [ HtmlElem::new( nil, nil ) ]
+        @form_stack = []
+        @table_stack = []
+        self.ignore @@ignoretags
+      end
+
+      def feed( response )
+        @response = response
+        begin
+          super( HTMLSplit.new( @response.body ).repair.to_s )
+        rescue
+          raise BadHtmlTags, $!.message + @elem_stack.inspect
+        end
+        @elem_stack.pop.children[0]
+      end
+
+      def handle_data(data)
+        @elem_stack.last.append data.squeeze( " \n" ).strip
+      end
+
+      def starttag( elem )
+        @elem_stack.last.append elem
+        @elem_stack.push elem
+      end
+
+      def endtag( tag )
+        elem = @elem_stack.pop
+        if tag != elem.tag
+          puts "'#{tag}'(wait for '#{elem.tag.to_s}')" if $DEBUG
+          raise BadHtmlTags, "'#{tag}'(wait for '#{elem.tag.to_s}')"
+        end
+        elem
+      end
+
+      def unknown_starttag( tag, attrs )
+        elem = HtmlElem::new( tag, attrs_to_hash( attrs ) )
+        starttag( elem )
+      end
+
+      def unknown_endtag( tag )
+        endtag( tag )
+      end
+
+      # special start and end ( a, form, table, ... )
+
+      def start_a( attrs )
+        ah = attrs_to_hash attrs
+        unless ah["href"] =~ /^#/
+          ah["href"] = complete_url( ah["href"], @response.url )
+          ah["href"].gsub!( "&amp;", "&" )
+        end
+        elem = Link::new( ah )
+        @response.add_link( elem )
+        starttag( elem )
+      end
+
+      def start_form( attrs )
+        ah = attrs_to_hash attrs
+        ah["action"] = complete_url( ah["action"], @response.url )
+        ah["action"].gsub!( "&amp;", "&" )
+        elem = Form::new( ah )
+        @response.add_form elem
+        @form_stack.push elem
+        starttag( elem )
+      end
+      def end_form
+        endtag( 'form' )
+        @form_stack.pop
+      end
+
+      def start_select( attrs )
+        ah = attrs_to_hash attrs
+        elem = Select::new( ah )
+        @form_stack.last.add_param elem
+        starttag( elem )
+      end
+      def end_select
+        endtag( 'select' )
+        @form_stack.last.parameters.last.end_option
+      end
+
+      def start_option( attrs )
+        ah = attrs_to_hash attrs
+        elem = SelectOption::new( ah )
+        @form_stack.last.parameters.last.add_option elem
+        starttag( elem )
+      end
+
+      def start_textarea( attrs )
+        ah = attrs_to_hash attrs
+        elem = Textarea::new( ah )
+        @form_stack.last.add_param elem
+        starttag( elem )
+      end
+
+      def start_table( attrs )
+        ah = attrs_to_hash attrs
+        elem = Table::new( ah )
+        @response.add_table elem
+        @table_stack.push elem
+        starttag( elem )
+      end
+      def end_table
+        endtag( 'table' )
+        @table_stack.pop
+      end
+
+      def start_tr( attrs )
+        elem = TableRow::new( attrs_to_hash(attrs) )
+        starttag( elem )
+        @table_stack.last.add_row( attrs_to_hash(attrs) )
+      end
+
+      def start_th( attrs )
+        elem = TableCell::new( 'th', attrs_to_hash( attrs ) )
+        @table_stack.last.add_cell( elem )
+        starttag( elem )
+      end
+
+      def start_td( attrs )
+        elem = TableCell::new( 'td', attrs_to_hash( attrs ) )
+        @table_stack.last.add_cell( elem )
+        starttag( elem )
+      end
+
+      # do ( start only )
+      def dotag( tag, attrs )
+        elem = HtmlElem::new( tag, attrs_to_hash( attrs ) )
+        @elem_stack.last.children.push elem
+      end
+
+      SingleTags.each do |t|
+        eval "def do_#{t}( attrs ); dotag( '#{t}', attrs ); end"
+      end
+
+      # special do ( input, frame, ... )
+
+      def do_input( attrs )
+        ah = attrs_to_hash attrs
+        case ah['type']
+        when 'submit'
+          elem = InputSubmit::new( ah )
+        when 'reset'
+          elem = InputReset::new( ah )
+        when 'checkbox'
+          elem = InputCheckbox::new( ah )
+        when 'radio'
+          elem = InputRadio::new( ah )
+        when 'image'
+          elem = InputImage::new( ah )
+        when 'file'
+          elem = InputFile::new( ah )
+        else # text
+          elem = Input::new( ah )
+        end
+        
+        @form_stack.last.add_param elem if @form_stack.last
+        @elem_stack.last.children.push elem
+      end
+
+      def do_frame( attrs )
+        ah = attrs_to_hash attrs
+        ah['src'] = complete_url( ah['src'], @response.url )
+        elem = Frame::new( ah )
+        @response.add_frame( elem )
+      end
+
+      def do_img( attrs )
+        ah = attrs_to_hash attrs
+        elem = Image::new( ah )
+        @response.add_image( elem )
+        @elem_stack.last.children.push elem
+      end
+
+      # ignore tag
+      def ignore( arr )
+        arr.each do |t|
+          self.instance_eval( "def start_#{t}( a ); end" )
+          self.instance_eval( "def end_#{t}; end" )
+          self.instance_eval( "def do_#{t}( a ); end" )
+        end
+      end
+
+      def self::ignore( arr )
+        @@ignoretags = arr
+      end
+
+      # utils
+
+      def attrs_to_hash( attrs )
+        h = {}
+        for a, v in attrs
+          v = v.split('"')[1]
+          h[a] = v ? v : ''
+        end
+        h
+      end
+
+    end
+
+  end
+end