1  require 'rexml/validation/validationexception'
   2 
   3  module REXML
   4    module Validation
   5      module Validator
   6        NILEVENT = [ nil ]
   7        def reset
   8          @current = @root
   9          @root.reset
  10          @root.previous = true
  11          @attr_stack = []
  12          self
  13        end
  14        def dump
  15          puts @root.inspect
  16        end
  17        def validate( event ) 
  18          #puts "Current: #@current"
  19          #puts "Event: #{event.inspect}"
  20          @attr_stack = [] unless defined? @attr_stack
  21          match = @current.next(event)
  22          raise ValidationException.new( "Validation error.  Expected: "+
  23            @current.expected.join( " or " )+" from #{@current.inspect} "+
  24            " but got #{Event.new( event[0], event[1] ).inspect}" ) unless match
  25          @current = match
  26 
  27          # Check for attributes
  28          case event[0]
  29          when :start_element
  30            #puts "Checking attributes"
  31            @attr_stack << event[2]
  32            begin
  33              sattr = [:start_attribute, nil]
  34              eattr = [:end_attribute]
  35              text = [:text, nil]
  36              k,v = event[2].find { |k,v| 
  37                sattr[1] = k
  38                #puts "Looking for #{sattr.inspect}"
  39                m = @current.next( sattr )
  40                #puts "Got #{m.inspect}"
  41                if m 
  42                  # If the state has text children...
  43                  #puts "Looking for #{eattr.inspect}"
  44                  #puts "Expect #{m.expected}"
  45                  if m.matches?( eattr )
  46                    #puts "Got end"
  47                    @current = m
  48                  else
  49                    #puts "Didn't get end"
  50                    text[1] = v
  51                    #puts "Looking for #{text.inspect}"
  52                    m = m.next( text )
  53                    #puts "Got #{m.inspect}"
  54                    text[1] = nil
  55                    return false unless m
  56                    @current = m if m
  57                  end
  58                  m = @current.next( eattr )
  59                  if m
  60                    @current = m
  61                    true
  62                  else
  63                    false
  64                  end
  65                else
  66                  false
  67                end
  68              }
  69              event[2].delete(k) if k
  70            end while k
  71          when :end_element
  72            attrs = @attr_stack.pop
  73            raise ValidationException.new( "Validation error.  Illegal "+
  74              " attributes: #{attrs.inspect}") if attrs.length > 0
  75          end
  76        end
  77      end
  78 
  79      class Event
  80        def initialize(event_type, event_arg=nil )
  81          @event_type = event_type
  82          @event_arg = event_arg
  83        end
  84 
  85        attr_reader :event_type
  86        attr_accessor :event_arg
  87 
  88        def done?
  89          @done
  90        end
  91 
  92        def single?
  93          return (@event_type != :start_element and @event_type != :start_attribute)
  94        end
  95 
  96        def matches?( event )
  97          #puts "#@event_type =? #{event[0]} && #@event_arg =? #{event[1]} "
  98          return false unless event[0] == @event_type
  99          case event[0]
 100          when nil
 101            return true
 102          when :start_element
 103            return true if event[1] == @event_arg
 104          when :end_element
 105            return true
 106          when :start_attribute
 107            return true if event[1] == @event_arg
 108          when :end_attribute
 109            return true
 110          when :end_document
 111            return true
 112          when :text
 113            return (@event_arg.nil? or @event_arg == event[1])
 114  =begin
 115          when :processing_instruction
 116            false
 117          when :xmldecl
 118            false
 119          when :start_doctype
 120            false
 121          when :end_doctype
 122            false
 123          when :externalentity
 124            false
 125          when :elementdecl
 126            false
 127          when :entity
 128            false
 129          when :attlistdecl
 130            false
 131          when :notationdecl
 132            false
 133          when :end_doctype
 134            false
 135  =end
 136          else
 137            false
 138          end
 139        end
 140 
 141        def ==( other )
 142          return false unless other.kind_of? Event
 143          @event_type == other.event_type and @event_arg == other.event_arg
 144        end
 145 
 146        def to_s
 147          inspect
 148        end
 149 
 150        def inspect
 151          "#{@event_type.inspect}( #@event_arg )"
 152        end
 153      end
 154    end
 155  end