81 lines
2.1 KiB
Ruby
81 lines
2.1 KiB
Ruby
require 'set'
|
|
|
|
module VagrantPlugins
|
|
module DockerProvisioner
|
|
class Config < Vagrant.plugin("2", :config)
|
|
attr_reader :images
|
|
attr_accessor :version
|
|
|
|
def initialize
|
|
@images = Set.new
|
|
@version = UNSET_VALUE
|
|
|
|
@__build_images = []
|
|
@__containers = Hash.new { |h, k| h[k] = {} }
|
|
end
|
|
|
|
# Accessor for internal state.
|
|
def build_images
|
|
@__build_images
|
|
end
|
|
|
|
# Accessor for the internal state.
|
|
def containers
|
|
@__containers
|
|
end
|
|
|
|
# Defines an image to build using `docker build` within the machine.
|
|
#
|
|
# @param [String] path Path to the Dockerfile to pass to
|
|
# `docker build`.
|
|
def build_image(path, **opts)
|
|
@__build_images << [path, opts]
|
|
end
|
|
|
|
def images=(images)
|
|
@images = Set.new(images)
|
|
end
|
|
|
|
def pull_images(*images)
|
|
@images += images.map(&:to_s)
|
|
end
|
|
|
|
def run(name, **options)
|
|
@__containers[name.to_s] = options.dup
|
|
end
|
|
|
|
def merge(other)
|
|
super.tap do |result|
|
|
result.pull_images(*(other.images + self.images))
|
|
|
|
build_images = @__build_images.dup
|
|
build_images += other.build_images
|
|
result.instance_variable_set(:@__build_images, build_images)
|
|
|
|
containers = {}
|
|
@__containers.each do |name, params|
|
|
containers[name] = params.dup
|
|
end
|
|
other.containers.each do |name, params|
|
|
containers[name] = @__containers[name].merge(params)
|
|
end
|
|
|
|
result.instance_variable_set(:@__containers, containers)
|
|
end
|
|
end
|
|
|
|
def finalize!
|
|
@version = "latest" if @version == UNSET_VALUE
|
|
@version = @version.to_sym
|
|
|
|
@__containers.each do |name, params|
|
|
params[:image] ||= name
|
|
params[:auto_assign_name] = true if !params.key?(:auto_assign_name)
|
|
params[:daemonize] = true if !params.key?(:daemonize)
|
|
params[:restart] = "always" if !params.key?(:restart)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|