module PortReservation
  extend Chef::DSL::DataQuery

  class << self
    def for(type)
      reservations[type]
    end

    def reservations
      @reservations ||= data_bag_item('ports', 'reservations')['mapping']
    end
  end
end

module PortReservation
  extend Chef::DSL::DataQuery

  class MissingReservation < StandardError; end

  class << self
    def for(type)
      reservations[type].tap do |reservation|
        raise MissingReservation.new("No port reservation found for #{type}") if reservation.nil?
      end
    end

    def reservations
      @reservations ||= data_bag_item('ports', 'reservations')['mapping']
    end
  end
end

Raise if no reservation is configured in data bag
Raise if no reservation is configured in data bag
