1  require "rexml/child"
   2  module REXML
   3    module DTD
   4      class EntityDecl < Child
   5        START = "<!ENTITY"
   6        START_RE = /^\s*#{START}/um
   7        PUBLIC = /^\s*#{START}\s+(?:%\s+)?(\w+)\s+PUBLIC\s+((["']).*?\3)\s+((["']).*?\5)\s*>/um
   8        SYSTEM = /^\s*#{START}\s+(?:%\s+)?(\w+)\s+SYSTEM\s+((["']).*?\3)(?:\s+NDATA\s+\w+)?\s*>/um
   9        PLAIN = /^\s*#{START}\s+(\w+)\s+((["']).*?\3)\s*>/um
  10        PERCENT = /^\s*#{START}\s+%\s+(\w+)\s+((["']).*?\3)\s*>/um
  11        # <!ENTITY name SYSTEM "...">
  12        # <!ENTITY name "...">
  13        def initialize src
  14          super()
  15          md = nil
  16          if src.match( PUBLIC )
  17            md = src.match( PUBLIC, true )
  18            @middle = "PUBLIC"
  19            @content = "#{md[2]} #{md[4]}"
  20          elsif src.match( SYSTEM )
  21            md = src.match( SYSTEM, true )
  22            @middle = "SYSTEM"
  23            @content = md[2]
  24          elsif src.match( PLAIN )
  25            md = src.match( PLAIN, true )
  26            @middle = ""
  27            @content = md[2]
  28          elsif src.match( PERCENT )
  29            md = src.match( PERCENT, true )
  30            @middle = ""
  31            @content = md[2]
  32          end
  33          raise ParseException.new("failed Entity match", src) if md.nil?
  34          @name = md[1]
  35        end
  36 
  37        def to_s
  38          rv = "<!ENTITY #@name "
  39          rv << "#@middle " if @middle.size > 0
  40          rv << @content
  41          rv
  42        end
  43 
  44        def write( output, indent )
  45          indent( output, indent )
  46          output << to_s
  47        end
  48 
  49        def EntityDecl.parse_source source, listener
  50          md = source.match( PATTERN_RE, true )
  51          thing = md[0].squeeze(" \t\n\r")
  52          listener.send inspect.downcase, thing 
  53        end
  54      end
  55    end
  56  end