nakajima-elementor 0.0.5 → 0.0.6

data/lib/elementor/element_set.rb

@@ -1,7 +1,40 @@
 module Elementor
+
+  module NodeSet
+    def self.included(base)
+      base.class_eval do
+        attr_accessor :node_set
+      end
+    end
+
+    def initialize(document, array)
+      @node_set = Nokogiri::XML::NodeSet.new(document, array)
+    end
+
+    def length
+      node_set.length
+    end
+    alias_method :size, :length
+
+    def select(&block)
+      @node_set.select &block
+    end
+
+    def replace(array)
+      @node_set = Nokogiri::XML::NodeSet.new(@node_set.document, array)
+    end
+
+    def search(selector)
+      @node_set.search(selector)
+    end
+  end
+
   # ElementSet objects wrap a Nokogiri #search result and
   # add additional functionality such as chained filtering.
-  class ElementSet < Array
+  class ElementSet
+
+    include NodeSet
+
     attr_accessor :result, :selector
     
     # A simple filter for selecting only elements with content
@@ -34,7 +67,7 @@ module Elementor
     alias_method :attrs, :with_attrs
 
     def method_missing(sym, *args, &block)
-      result.respond_to?(sym) ? result.send(sym, doc, *args) : super
+      result.respond_to?(sym) ? result.send(sym, self, *args) : super
     end
     
     def respond_to?(sym)
@@ -51,4 +84,4 @@ module Elementor
       replace(select(&block)) ; return self
     end
   end
-end
+end

data/lib/elementor/result.rb

@@ -81,7 +81,7 @@ module Elementor
     end
     
     def make_element_set(name, selector, *filters, &block)
-      set = ElementSet.new scope(filters).search(selector)
+      set = ElementSet.new(doc, scope(filters).search(selector))
       set.result = self
       set.selector = selector
       set = filters.empty? ? set : filters.inject(set) { |result, fn| fn[result] }
@@ -101,4 +101,4 @@ module Elementor
       @content ||= context.send(opts[:from] || :body)
     end
   end
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: nakajima-elementor
 version: !ruby/object:Gem::Version 
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors: 
 - Pat Nakajima