lib/rexml/dtd/entitydecl.rb

   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