# 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
