cocoa-xml 0.4.4 → 0.4.5

data/lib/cocoa-xml.rb

@@ -13,7 +13,7 @@ module CocoaXML
   #   object that responds to #read, or #to_str
   # @return [NSXMLDocument] An NSXMLDocument set to interpret source as HTML
   def self.HTML(source)
-    parse source, NSXMLDocumentTidyHTML
+    parse source, :html
   end
 
   # Parse an input XML source
@@ -22,7 +22,15 @@ module CocoaXML
   #   object that responds to #read, or #to_str
   # @return [NSXMLDocument] An NSXMLDocument set to inperpret source as XML
   def self.XML(source)
-    parse source, NSXMLDocumentTidyXML
+    parse source, :xml
+  end
+
+  def self.test_filter(&block)
+    @@test_filter = lambda { |source, type| block.call(source, type) }
+  end
+
+  def self.test_capture(&block)
+    @@test_capture = lambda { |document, source, type| block.call(document, source, type) }
   end
 
   private
@@ -36,13 +44,20 @@ module CocoaXML
     def self.parse(source, type)
       error = Pointer.new :object
 
+      orig_source, orig_type, source, type = source, type, *@@test_filter.call(source, type) if class_variable_defined?(:@@test_filter)
+      type_id = (:xml == type) ? NSXMLDocumentTidyXML : NSXMLDocumentTidyHTML
+
       url = (source.is_a?(NSURL) && source) || NSURL.URLWithString(source.to_str) if source.respond_to?(:to_str)
-      source = source.read if source.respond_to?(:read)
+      unread_source, source = source, source.read if source.respond_to?(:read)
+
+      doc = unless url.nil?
+              ::NSXMLDocument.alloc.initWithContentsOfURL url, options: type_id, error: error
+            else
+              ::NSXMLDocument.alloc.initWithXMLString source, options: type_id, error: error
+            end
+
+      @@test_capture.call(doc, orig_source || unread_source || source, orig_type || type) if class_variable_defined?(:@@test_capture)
 
-      unless url.nil?
-        ::NSXMLDocument.alloc.initWithContentsOfURL url, options: type, error: error
-      else
-        ::NSXMLDocument.alloc.initWithXMLString source, options: type, error: error
-      end
+      doc
     end
 end

data/lib/cocoa-xml/version.rb

@@ -1,3 +1,3 @@
 module CocoaXML
-  Version = '0.4.4'
+  Version = '0.4.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: cocoa-xml
 version: !ruby/object:Gem::Version 
-  version: 0.4.4
+  version: 0.4.5
 platform: ruby
 authors: 
 - Chris Hoffman
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-01-30 00:00:00 -06:00
+date: 2010-02-05 00:00:00 -06:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency