Merge branch 'enhancements'

terrcin · Dec 1, 2008 · 86f9209 · 86f9209
2 parents f9b6735 + 781b98d
commit 86f9209
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/lib/xmpp4r/pubsub/children/items.rb b/lib/xmpp4r/pubsub/children/items.rb
@@ -10,6 +10,7 @@ module PubSub
     # Items
     # a collection of Items
     class Items < XMPPElement
+      include Enumerable
       name_xmlns 'items', NS_PUBSUB
 
       def node
@@ -30,6 +31,14 @@ def max_items
       def max_items=(mymaxitems)
         attributes['max_items'] = mymaxitems
       end
+
+      def items
+        get_elements("item")
+      end
+
+      def each(&block)
+        items.each(&block)
+      end
     end
 
     ##

diff --git a/lib/xmpp4r/pubsub/children/publish.rb b/lib/xmpp4r/pubsub/children/publish.rb
@@ -13,7 +13,21 @@ module PubSub
     # A <publish> XMPP element, see example 1 in
     # http://www.xmpp.org/extensions/xep-0060.html#intro-howitworks
     class Publish < XMPPElement
+      include Enumerable
       name_xmlns 'publish', NS_PUBSUB
+
+      ##
+      # support for enumerating <item> elements
+      def each(&block)
+        items.each(&block)
+      end
+
+      ##
+      # return child <item> elements
+      def items
+        get_elements("item")
+      end
+
       ##
       # return the node for this publication
       def node