summaryrefslogtreecommitdiff
path: root/lib/bitbucket/collection.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/bitbucket/collection.rb')
-rw-r--r--lib/bitbucket/collection.rb21
1 files changed, 21 insertions, 0 deletions
diff --git a/lib/bitbucket/collection.rb b/lib/bitbucket/collection.rb
new file mode 100644
index 00000000000..3a9379ff680
--- /dev/null
+++ b/lib/bitbucket/collection.rb
@@ -0,0 +1,21 @@
+module Bitbucket
+ class Collection < Enumerator
+ def initialize(paginator)
+ super() do |yielder|
+ loop do
+ paginator.items.each { |item| yielder << item }
+ end
+ end
+
+ lazy
+ end
+
+ def method_missing(method, *args)
+ return super unless self.respond_to?(method)
+
+ self.send(method, *args) do |item|
+ block_given? ? yield(item) : item
+ end
+ end
+ end
+end