summaryrefslogtreecommitdiff
path: root/lib/chef/resource_collection
diff options
context:
space:
mode:
authorSeth Chisamore <schisamo@opscode.com>2012-10-30 10:39:35 -0400
committerSeth Chisamore <schisamo@opscode.com>2012-10-30 10:39:35 -0400
commit24dc69a9a97e82a6e4207de68d6dcc664178249b (patch)
tree19bb289c9f88b4bbab066bc56b95d6d222fd5c35 /lib/chef/resource_collection
parent9348c1c9c80ee757354d624b7dc1b78ebc7605c4 (diff)
downloadchef-24dc69a9a97e82a6e4207de68d6dcc664178249b.tar.gz
[OC-3564] move core Chef to the repo root \o/ \m/
The opscode/chef repository now only contains the core Chef library code used by chef-client, knife and chef-solo!
Diffstat (limited to 'lib/chef/resource_collection')
-rw-r--r--lib/chef/resource_collection/stepable_iterator.rb124
1 files changed, 124 insertions, 0 deletions
diff --git a/lib/chef/resource_collection/stepable_iterator.rb b/lib/chef/resource_collection/stepable_iterator.rb
new file mode 100644
index 0000000000..ec1e244758
--- /dev/null
+++ b/lib/chef/resource_collection/stepable_iterator.rb
@@ -0,0 +1,124 @@
+# Author:: Daniel DeLeo (<dan@kallistec.com>)
+# Copyright:: Copyright (c) 2009 Daniel DeLeo
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+class Chef
+ class ResourceCollection
+ class StepableIterator
+
+ def self.for_collection(new_collection)
+ instance = new(new_collection)
+ instance
+ end
+
+ attr_accessor :collection
+ attr_reader :position
+
+ def initialize(collection=[])
+ @position = 0
+ @paused = false
+ @collection = collection
+ end
+
+ def size
+ collection.size
+ end
+
+ def each(&block)
+ reset_iteration(block)
+ @iterator_type = :element
+ iterate
+ end
+
+ def each_index(&block)
+ reset_iteration(block)
+ @iterator_type = :index
+ iterate
+ end
+
+ def each_with_index(&block)
+ reset_iteration(block)
+ @iterator_type = :element_with_index
+ iterate
+ end
+
+ def paused?
+ @paused
+ end
+
+ def pause
+ @paused = true
+ end
+
+ def resume
+ @paused = false
+ iterate
+ end
+
+ def rewind
+ @position = 0
+ end
+
+ def skip_back(skips=1)
+ @position -= skips
+ end
+
+ def skip_forward(skips=1)
+ @position += skips
+ end
+
+ def step
+ return nil if @position == size
+ call_iterator_block
+ @position += 1
+ end
+
+ def iterate_on(iteration_type, &block)
+ @iterator_type = iteration_type
+ @iterator_block = block
+ end
+
+ private
+
+ def reset_iteration(iterator_block)
+ @iterator_block = iterator_block
+ @position = 0
+ @paused = false
+ end
+
+ def iterate
+ while @position < size && !paused?
+ step
+ end
+ collection
+ end
+
+ def call_iterator_block
+ case @iterator_type
+ when :element
+ @iterator_block.call(collection[@position])
+ when :index
+ @iterator_block.call(@position)
+ when :element_with_index
+ @iterator_block.call(collection[@position], @position)
+ else
+ raise "42error: someone forgot to set @iterator_type, wtf?"
+ end
+ end
+
+ end
+ end
+end