94 lines
2.6 KiB
Ruby
94 lines
2.6 KiB
Ruby
# Copyright (c) HashiCorp, Inc.
|
|
# SPDX-License-Identifier: BUSL-1.1
|
|
|
|
module VagrantPlugins
|
|
module CommandServe
|
|
module Service
|
|
class ProvisionerService < ProtoService(SDK::ProvisionerService::Service)
|
|
|
|
def cleanup(req, ctx)
|
|
with_plugin(ctx, :provisioners, broker: broker) do |plugin|
|
|
machine, plugin_config = _process_args(req)
|
|
provisioner = load_provisioner(plugin, machine, plugin_config)
|
|
provisioner.cleanup
|
|
|
|
Empty.new
|
|
end
|
|
end
|
|
|
|
def cleanup_spec(*_)
|
|
funcspec(
|
|
args: [
|
|
SDK::Args::Target::Machine,
|
|
SDK::Args::ConfigData,
|
|
]
|
|
)
|
|
end
|
|
|
|
def configure(req, ctx)
|
|
with_plugin(ctx, :provisioners, broker: broker) do |plugin|
|
|
machine, plugin_config = _process_args(req)
|
|
provisioner = load_provisioner(plugin, machine, plugin_config)
|
|
provisioner.configure(machine.config)
|
|
|
|
Empty.new
|
|
end
|
|
end
|
|
|
|
def configure_spec(*_)
|
|
funcspec(
|
|
args: [
|
|
SDK::Args::Target::Machine,
|
|
SDK::Args::ConfigData,
|
|
]
|
|
)
|
|
end
|
|
|
|
def provision(req, ctx)
|
|
with_plugin(ctx, :provisioners, broker: broker) do |plugin|
|
|
machine, plugin_config = _process_args(req)
|
|
provisioner = load_provisioner(plugin, machine, plugin_config)
|
|
provisioner.provision
|
|
|
|
Empty.new
|
|
end
|
|
end
|
|
|
|
def provision_spec(*_)
|
|
funcspec(
|
|
args: [
|
|
SDK::Args::Target::Machine,
|
|
SDK::Args::ConfigData,
|
|
]
|
|
)
|
|
end
|
|
|
|
# All of the methods here take the same args, so they have the same
|
|
# args processing which can be shared
|
|
#
|
|
# @param req the request
|
|
# @return [Array(Vagrant::Machine, Vagrant::Plugin::V2::Config)]
|
|
def _process_args(req)
|
|
machine, config = mapper.funcspec_map(
|
|
req, mapper, broker, expect: [
|
|
Vagrant::Machine, Vagrant::Plugin::V2::Config
|
|
]
|
|
)
|
|
return machine, config
|
|
end
|
|
|
|
def load_provisioner(klass, machine, config)
|
|
ident = config.instance_variable_get(:@_vagrant_config_identifier)
|
|
if ident
|
|
key = cache.key(klass, machine, ident)
|
|
return cache.get(key) if cache.registered?(key)
|
|
end
|
|
klass.new(machine, config).tap do |i|
|
|
cache.register(key, i) if key
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|