2022-04-25 12:24:43 -05:00

90 lines
2.6 KiB
Ruby

module VagrantPlugins
module CommandServe
class Mappers
# Build a machine client from a FuncSpec value
class MachineFromSpec < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: SDK::FuncSpec::Value) { |arg|
arg.type == "hashicorp.vagrant.sdk.Args.Machine" &&
!arg&.value&.value.nil?
}
i << Input.new(type: Broker)
end
super(inputs: inputs, output: Client::Machine, func: method(:converter))
end
def converter(proto, broker)
Client::Machine.load(proto.value.value, broker: broker)
end
end
# Build a machine client from a proto instance
class MachineFromProto < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: SDK::Args::Target::Machine)
i << Input.new(type: Broker)
end
super(inputs: inputs, output: Client::Machine, func: method(:converter))
end
def converter(proto, broker)
Client::Machine.load(proto, broker: broker)
end
end
# Build a machine client from a serialized proto string
class MachineFromString < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: String)
i << Input.new(type: Broker)
end
super(inputs: inputs, output: Client::Machine, func: method(:converter))
end
def converter(proto, broker)
Client::Machine.load(proto, broker: broker)
end
end
# Build a machine from a target
class MachineFromTarget < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: Vagrant::Environment)
i << Input.new(type: Client::Target)
end
super(inputs: inputs, output: Vagrant::Machine, func: method(:converter))
end
def converter(env, target)
env.machine(
target.name.to_sym,
target.provider_name.to_sym,
)
end
end
# Build a machine from a machine client
class MachineFromMachineClient < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: Vagrant::Environment)
i << Input.new(type: Client::Machine)
end
super(inputs: inputs, output: Vagrant::Machine, func: method(:converter))
end
def converter(env, machine)
env.machine(
machine.name.to_sym,
machine.provider_name.to_sym,
)
end
end
end
end
end