marc 1.0.4 → 1.2.0

data/lib/marc/xml_parsers.rb

@@ -1,152 +1,187 @@
 module MARC
   # Exception class to be thrown when an XML parser
   # encounters an unrecoverable error.
-  class XMLParseError < StandardError; end
+  class XMLParseError < StandardError
+  end
+
+  IND1 = "ind1".freeze
+  IND2 = "ind2".freeze
+  TAG = "tag".freeze
+  CODE = "code".freeze
 
   # The MagicReader will try to use the best available XML Parser at the
-  # time of initialization.  
+  # time of initialization.
   # The order is currently:
   #   * Nokogiri
-  #   * jrexml (JRuby only)
+  #   * libxml-ruby (MRI only) ** DEPRECATED **
+  #   * jstax (JRuby only) ** DEPRECATED **
   #   * rexml
   #
   # With the idea that other parsers could be added as their modules are
   # added.  Realistically, this list should be limited to stream-based
   # parsers.  The magic should be used selectively, however.  After all,
   # one project's definition of 'best' might not apply universally.  It
-  # is arguable which is "best" on JRuby:  Nokogiri or jrexml.  
-  module MagicReader    
+  # is arguable which is "best" on JRuby:  Nokogiri or jrexml.
+  module MagicReader
     def self.extended(receiver)
       magic = MARC::XMLReader.best_available
       case magic
-      when 'nokogiri' then receiver.extend(NokogiriReader)
-      when 'libxml' then receiver.extend(LibXMLReader)
-      when 'jstax' then receiver.extend(JRubySTAXReader)
-      when 'jrexml' then receiver.extend(JREXMLReader)
+      when "nokogiri"
+        receiver.extend(NokogiriReader)
+      when "libxml"
+        warn "libxml support will be removed in version 1.3. Prefer nokogiri instead"
+        receiver.extend(LibXMLReader)
+      when "jstax"
+        warn "jstax support will be removed in version 1.3. Prefer nokogiri instead"
+        receiver.extend(JRubySTAXReader)
+      when "jrexml"
+        warn "jrexml support is broken upstream; falling back to just rexml. Prefer nokogiri instead"
+        receiver.extend(REXMLReader)
       else receiver.extend(REXMLReader)
       end
     end
   end
-  
+
   module GenericPullParser
     # Submodules must include
     #  self.extended()
     #  init()
     #  attributes_to_hash(attributes)
     #  each
-    
+
+    REC_TAG = "record".freeze
+    LEAD_TAG = "leader".freeze
+    CF_TAG = "controlfield".freeze
+    DF_TAG = "datafield".freeze
+    SF_TAG = "subfield".freeze
+
+    def init
+      @record = {record: nil, leader: "", field: nil, subfield: nil}
+      @current_element = nil
+      @ns = "http://www.loc.gov/MARC21/slim"
+    end
 
     # Returns our MARC::Record object to the #each block.
     def yield_record
-      @block.call(@record[:record])       
+      if @record[:record].valid?
+        @block.call(@record[:record])
+      elsif @error_handler
+        @error_handler.call(self, @record[:record], @block)
+      else
+        raise MARC::RecordException, @record[:record]
+      end
+    ensure
       @record[:record] = nil
-    end    
+    end
 
     def start_element_namespace name, attributes = [], prefix = nil, uri = nil, ns = {}
-       attributes = attributes_to_hash(attributes)
-       if uri == @ns
-         case name.downcase
-         when 'record' then @record[:record] = MARC::Record.new
-         when 'leader' then @current_element = :leader
-         when 'controlfield'
-           @current_element=:field
-           @record[:field] = MARC::ControlField.new(attributes["tag"])
-         when 'datafield'
-           @record[:field] = MARC::DataField.new(attributes["tag"], attributes['ind1'], attributes['ind2'])
-         when 'subfield'
-           @current_element=:subfield
-           @record[:subfield] = MARC::Subfield.new(attributes['code'])
-         end
-       end
-     end
-
-
-    def characters text
+      attributes = attributes_to_hash(attributes)
+      if (uri == @ns) || @ignore_namespace
+        case name.downcase
+        when SF_TAG
+          @current_element = :subfield
+          @record[:subfield] = MARC::Subfield.new(attributes[CODE])
+        when DF_TAG
+          @record[:field] = MARC::DataField.new(attributes[TAG], attributes[IND1], attributes[IND2])
+        when CF_TAG
+          @current_element = :field
+          @record[:field] = MARC::ControlField.new(attributes[TAG])
+        when LEAD_TAG then @current_element = :leader
+        when REC_TAG then @record[:record] = MARC::Record.new
+        end
+      end
+    end
+
+    def characters(text)
       case @current_element
-      when :leader then @record[:record].leader = text
-      when :field then @record[:field].value << text
       when :subfield then @record[:subfield].value << text
+      when :field then @record[:field].value << text
+      when :leader then @record[:leader] << text
       end
     end
 
-    def end_element_namespace name, prefix = nil, uri = nil
+    def end_element_namespace(name, prefix = nil, uri = nil)
       @current_element = nil
-      if uri == @ns
+      if (uri == @ns) || @ignore_namespace
         case name.downcase
-        when 'record' then yield_record
-        when /(control|data)field/
-          @record[:record] << @record[:field]
-          @record[:field] = nil
-          @current_element = nil if @current_element == :field          
-        when 'subfield'
+        when SF_TAG
           @record[:field].append(@record[:subfield])
           @record[:subfield] = nil
           @current_element = nil if @current_element == :subfield
+        when DF_TAG, CF_TAG
+          @record[:record] << @record[:field]
+          @record[:field] = nil
+          @current_element = nil if @current_element == :field
+        when REC_TAG then yield_record
+        when LEAD_TAG
+          @record[:record].leader = @record[:leader]
+          @record[:leader] = ""
+          @current_element = nil if @current_element == :leader
         end
       end
-    end  
+    end
   end
-  
 
   # NokogiriReader uses the Nokogiri SAX Parser to quickly read
   # a MARCXML document.  Because dynamically subclassing MARC::XMLReader
   # is a little ugly, we need to recreate all of the SAX event methods
-  # from Nokogiri::XML::SAX::Document here rather than subclassing.    
-  module NokogiriReader 
+  # from Nokogiri::XML::SAX::Document here rather than subclassing.
+  module NokogiriReader
     include GenericPullParser
+
     def self.extended(receiver)
-      require 'nokogiri'
+      require "nokogiri"
       receiver.init
     end
-    
+
     # Sets our instance variables for SAX parsing in Nokogiri and parser
     def init
-      @record = {:record=>nil,:field=>nil,:subfield=>nil}
-      @current_element = nil
-      @ns = "http://www.loc.gov/MARC21/slim"
-      @parser = Nokogiri::XML::SAX::Parser.new(self)         
+      super
+      @parser = Nokogiri::XML::SAX::Parser.new(self)
     end
-    
+
     # Loop through the MARC records in the XML document
-    def each(&block)    
-      unless block_given?
-        return self.enum_for(:each)
-      else
+    def each(&block)
+      if block
         @block = block
         @parser.parse(@handle)
+      else
+        enum_for(:each)
       end
     end
 
     def error(evt)
       raise(XMLParseError, "XML parsing error: #{evt}")
     end
-        
 
-    def method_missing(methName, *args)
-      sax_methods = [:xmldecl, :start_document, :end_document, :start_element,
-        :end_element, :comment, :warning, :error, :cdata_block, :processing_instruction]
-      unless sax_methods.index(methName)
-        raise NoMethodError.new("undefined method '#{methName} for #{self}", 'no_meth')
+    SAX_METHODS = [:xmldecl, :start_document, :end_document, :start_element,
+      :end_element, :comment, :warning, :error, :cdata_block, :processing_instruction]
+
+    def method_missing(method_name, *args)
+      unless SAX_METHODS.include?(method_name)
+        raise NoMethodError.new("undefined method '#{method_name} for #{self}", "no_meth")
       end
     end
-     
-     private 
-
-     def attributes_to_hash(attributes)
-       hash = {}
-       attributes.each do | att |
-         hash[att.localname] = att.value
-       end
-       hash
-     end     
-  end
 
+    def respond_to_missing?(method_name, include_private = false)
+      SAX_METHODS.include?(method_name) || super
+    end
+
+    private
+
+    def attributes_to_hash(attributes)
+      hash = {}
+      attributes.each do |att|
+        hash[att.localname] = att.value
+      end
+      hash
+    end
+  end
 
-  
   # The REXMLReader is the 'default' parser, since we can at least be
   # assured that REXML is probably there.  It uses REXML's PullParser
   # to handle larger document sizes without consuming insane amounts of
-  # memory, but it's still REXML (read: slow), so it's a good idea to 
+  # memory, but it's still REXML (read: slow), so it's a good idea to
   # use an alternative parser if available.  If you don't know the best
   # parser available, you can use the MagicReader or set:
   #
@@ -158,43 +193,44 @@ module MARC
   #
   # or
   #
-  # reader = MARC::XMLReader.new(fh, :parser=>"magic") 
+  # reader = MARC::XMLReader.new(fh, :parser=>"magic")
   # (or the constant)
   #
   # which will cascade down to REXML if nothing better is found.
-  #  
+  #
   module REXMLReader
     def self.extended(receiver)
-      require 'rexml/document'
-      require 'rexml/parsers/pullparser'
+      require "rexml/document"
+      require "rexml/parsers/pullparser"
       receiver.init
     end
-    
+
     # Sets our parser
     def init
       @parser = REXML::Parsers::PullParser.new(@handle)
     end
-    
+
     # Loop through the MARC records in the XML document
     def each
-      unless block_given?
-        return self.enum_for(:each)
-      else
+      if block_given?
         while @parser.has_next?
           event = @parser.pull
-          # if it's the start of a record element 
-          if event.start_element? and strip_ns(event[0]) == 'record'
+          # if it's the start of a record element
+          if event.start_element? && (strip_ns(event[0]) == "record")
             yield build_record
           end
-        end    
+        end
+      else
+        enum_for(:each)
       end
     end
-    
+
     private
+
     def strip_ns(str)
-      return str.sub(/^.*:/, '')
+      str.sub(/^.*:/, "")
     end
-    
+
     # will accept parse events until a record has been built up
     #
     def build_record
@@ -202,15 +238,15 @@ module MARC
       data_field = nil
       control_field = nil
       subfield = nil
-      text = '' 
+      text = ""
       attrs = nil
-      if Module.constants.index('Nokogiri') and @parser.is_a?(Nokogiri::XML::Reader)
+      if Module.constants.index("Nokogiri") && @parser.is_a?(Nokogiri::XML::Reader)
         datafield = nil
         cursor = nil
         open_elements = []
-        @parser.each do | node |
+        @parser.each do |node|
           if node.value? && cursor
-            if cursor.is_a?(Symbol) and cursor == :leader
+            if cursor.is_a?(Symbol) && (cursor == :leader)
               record.leader = node.value
             else
               cursor.value = node.value
@@ -230,195 +266,184 @@ module MARC
           when "controlfield"
             record << datafield if datafield
             datafield = nil
-            control_field = MARC::ControlField.new(node.attribute('tag'))
+            control_field = MARC::ControlField.new(node.attribute("tag"))
             record << control_field
             cursor = control_field
-          when "datafield"  
+          when "datafield"
             record << datafield if datafield
             datafield = nil
-            data_field = MARC::DataField.new(node.attribute('tag'), node.attribute('ind1'), node.attribute('ind2'))
+            data_field = MARC::DataField.new(node.attribute("tag"), node.attribute(IND1), node.attribute(IND2))
             datafield = data_field
           when "subfield"
             raise "No datafield to add to" unless datafield
-            subfield = MARC::Subfield.new(node.attribute('code'))
+            subfield = MARC::Subfield.new(node.attribute(CODE))
             datafield.append(subfield)
             cursor = subfield
           when "record"
             record << datafield if datafield
             return record
-          end          
-          #puts node.name
+          end
         end
-        
+
       else
         while @parser.has_next?
           event = @parser.pull
 
           if event.text?
-            text += REXML::Text::unnormalize(event[0])
+            text += REXML::Text.unnormalize(event[0])
             next
           end
 
           if event.start_element?
-            text = ''
+            text = ""
             attrs = event[1]
             case strip_ns(event[0])
-            when 'controlfield'
-              text = ''
-              control_field = MARC::ControlField.new(attrs['tag'])
-            when 'datafield'
-              text = ''
-              data_field = MARC::DataField.new(attrs['tag'], attrs['ind1'], 
-                attrs['ind2'])
-            when 'subfield'
-              text = ''
-              subfield = MARC::Subfield.new(attrs['code'])
+            when "controlfield"
+              text = ""
+              control_field = MARC::ControlField.new(attrs[TAG])
+            when "datafield"
+              text = ""
+              data_field = MARC::DataField.new(attrs[TAG], attrs[IND1],
+                attrs[IND2])
+            when "subfield"
+              text = ""
+              subfield = MARC::Subfield.new(attrs[CODE])
             end
           end
 
           if event.end_element?
             case strip_ns(event[0])
-            when 'leader'
+            when "leader"
               record.leader = text
-            when 'record'
+            when "record"
               return record
-            when 'controlfield'
+            when "controlfield"
               control_field.value = text
               record.append(control_field)
-            when 'datafield'
+            when "datafield"
               record.append(data_field)
-            when 'subfield'
+            when "subfield"
               subfield.value = text
               data_field.append(subfield)
             end
           end
         end
       end
-    end        
-  end
-  
-  # The JREXMLReader is really just here to set the load order for
-  # injecting the Java pull parser.  
-  module JREXMLReader
-
-    def self.extended(receiver)
-      require 'rexml/document'
-      require 'rexml/parsers/pullparser'
-      require 'jrexml'
-      receiver.extend(REXMLReader)
     end
   end
-  
-  
-  
-  
+
   unless defined? JRUBY_VERSION
     module LibXMLReader
+      def self.extended(receiver)
+        require "xml"
+        receiver.init
+      end
 
-    def self.extended(receiver)
-      require 'xml'
-      receiver.init
-    end
-
-    def init
-      @ns = "http://www.loc.gov/MARC21/slim"
-      @parser = XML::Reader.io(@handle)
-    end
+      def init
+        @ns = "http://www.loc.gov/MARC21/slim"
+        @parser = XML::Reader.io(@handle)
+      end
 
-    def each
-      unless block_given?
-        return self.enum_for(:each)
-      else
-        while (@parser.read) do
-         if @parser.local_name == 'record' && @parser.namespace_uri == @ns
-           yield build_record
-         end
-        end # while
+      def each
+        if block_given?
+          while @parser.read
+            if @parser.local_name == "record" && @parser.namespace_uri == @ns
+              yield build_record
+            end
+          end # while
+        else
+          enum_for(:each)
+        end
       end
-    end # each
 
-    def build_record
-      r = MARC::Record.new()
-        until (@parser.local_name == 'record' and @parser.node_type == XML::Reader::TYPE_END_ELEMENT) do
+      # each
+
+      def build_record
+        r = MARC::Record.new
+        until (@parser.local_name == "record") && (@parser.node_type == XML::Reader::TYPE_END_ELEMENT)
           @parser.read
           next if @parser.node_type == XML::Reader::TYPE_END_ELEMENT
           case @parser.local_name
-          when 'leader'
-           @parser.read
-           r.leader = @parser.value
-          when 'controlfield'
-           tag = @parser['tag']
-           @parser.read
-           r << MARC::ControlField.new(tag, @parser.value)
-          when 'datafield'
-           data = MARC::DataField.new(@parser['tag'], @parser['ind1'], @parser['ind2'])
-           while (@parser.read and !(@parser.local_name == 'datafield' and @parser.node_type == XML::Reader::TYPE_END_ELEMENT)) do
-             next if @parser.node_type == XML::Reader::TYPE_END_ELEMENT
-             case @parser.local_name
-             when 'subfield'
-               code =@parser['code']
-               @parser.read
-               data.append(MARC::Subfield.new(code, @parser.value))
-             end
-           end
-           r << data
+          when "leader"
+            @parser.read
+            r.leader = @parser.value
+          when "controlfield"
+            tag = @parser[TAG]
+            @parser.read
+            r << MARC::ControlField.new(tag, @parser.value)
+          when "datafield"
+            data = MARC::DataField.new(@parser[TAG], @parser[IND1], @parser[IND2])
+            while @parser.read && !((@parser.local_name == "datafield") && (@parser.node_type == XML::Reader::TYPE_END_ELEMENT))
+              next if @parser.node_type == XML::Reader::TYPE_END_ELEMENT
+              case @parser.local_name
+              when "subfield"
+                code = @parser[CODE]
+                @parser.read
+                data.append(MARC::Subfield.new(code, @parser.value))
+              end
+            end
+            r << data
 
           end # case
-        end #until
-      return r
+        end # until
+        r
+      end
     end
   end
-end
 
   # The JrubySTAXReader uses native java calls to parse the incoming stream
   # of marc-xml. It includes most of the work from GenericPullParser
 
   if defined? JRUBY_VERSION
+    # *DEPRECATED*: JRubySTAXReader is deprecated and will be removed in a
+    # future version of ruby-marc. Please use NokogiriReader
+    # instead.
     module JRubySTAXReader
       include GenericPullParser
+
       def self.extended(receiver)
-        require 'java' # may only be neccesary in jruby 1.6
+        require "java" # may only be neccesary in jruby 1.6
         receiver.init
       end
 
       def init
-        @record = {:record=>nil,:field=>nil,:subfield=>nil}
-        @current_element = nil
-        @ns = "http://www.loc.gov/MARC21/slim"
+        warn "JRubySTAXReader is deprecated and will be removed in a future version of ruby-marc."
+
+        super
         @factory = javax.xml.stream.XMLInputFactory.newInstance
         @parser = @factory.createXMLStreamReader(@handle.to_inputstream)
       end
 
       # Loop through the MARC records in the XML document
-      def each(&block)  
-        unless block_given?
-          return self.enum_for(:each)
-        else
+      def each(&block)
+        if block
           @block = block
           parser_dispatch
+        else
+          enum_for(:each)
         end
       end
 
       def parser_dispatch
-        while event = @parser.next and event != javax.xml.stream.XMLStreamConstants.END_DOCUMENT do
+        while (event = @parser.next) && (event != javax.xml.stream.XMLStreamConstants::END_DOCUMENT)
           case event
-            when javax.xml.stream.XMLStreamConstants.START_ELEMENT
-              start_element_namespace(@parser.getLocalName, [], nil,  @parser.getNamespaceURI, nil)
-            when javax.xml.stream.XMLStreamConstants.END_ELEMENT
-              end_element_namespace(@parser.getLocalName, @parser.getPrefix, @parser.getNamespaceURI)
-            when javax.xml.stream.XMLStreamConstants.CHARACTERS
-              characters(@parser.getText)
+          when javax.xml.stream.XMLStreamConstants::START_ELEMENT
+            start_element_namespace(@parser.getLocalName, [], nil, @parser.getNamespaceURI, nil)
+          when javax.xml.stream.XMLStreamConstants::END_ELEMENT
+            end_element_namespace(@parser.getLocalName, @parser.getPrefix, @parser.getNamespaceURI)
+          when javax.xml.stream.XMLStreamConstants::CHARACTERS
+            characters(@parser.getText)
           end
         end
       end
 
       def attributes_to_hash(attributes)
         hash = {}
-        @parser.getAttributeCount.times do | i |
+        @parser.getAttributeCount.times do |i|
           hash[@parser.getAttributeName(i).getLocalPart] = @parser.getAttributeValue(i)
         end
         hash
-      end     
+      end
     end # end of module
-  end # end of if jruby  
+  end # end of if jruby
 end