47 lines
1.0 KiB
Ruby
47 lines
1.0 KiB
Ruby
require 'forwardable'
|
|
|
|
module Vagrant
|
|
# Represents a collection of boxes, providing helpful methods for
|
|
# finding boxes.
|
|
class BoxCollection
|
|
include Enumerable
|
|
extend Forwardable
|
|
def_delegators :@boxes, :length, :each
|
|
|
|
# The directory that the boxes are being searched for.
|
|
attr_reader :directory
|
|
|
|
# Initializes the class to search for boxes in the given directory.
|
|
def initialize(directory)
|
|
@directory = directory
|
|
@boxes = []
|
|
|
|
reload!
|
|
end
|
|
|
|
# Find a box in the collection by the given name. The name must
|
|
# be a string, for now.
|
|
def find(name)
|
|
@boxes.each do |box|
|
|
return box if box.name == name
|
|
end
|
|
|
|
nil
|
|
end
|
|
|
|
# Loads the list of all boxes from the source. This modifies the
|
|
# current array.
|
|
def reload!
|
|
@boxes.clear
|
|
|
|
Dir.open(@directory) do |dir|
|
|
dir.each do |d|
|
|
next if d == "." || d == ".." || !@directory.join(d).directory?
|
|
@boxes << Box.new(d, @directory.join(d))
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|