93 lines
2.4 KiB
Ruby

# Copyright (c) HashiCorp, Inc.
# SPDX-License-Identifier: BUSL-1.1
module VagrantPlugins
module CommandServe
class Mappers
class TargetIndexProtoFromSpec < Mapper
def initialize
super(
inputs: [
Input.new(type: SDK::FuncSpec::Value) { |arg|
arg.type == "hashicorp.vagrant.sdk.Args.TargetIndex" &&
!arg&.value&.value.nil?
}
],
output: SDK::Args::TargetIndex,
func: method(:converter)
)
end
def converter(fv)
SDK::Args::TargetIndex.decode(fv.value.value)
end
end
# Build a target index client from a FuncSpec value
class TargetIndexFromSpec < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: SDK::FuncSpec::Value) { |arg|
arg.type == "hashicorp.vagrant.sdk.Args.TargetIndex" &&
!arg&.value&.value.nil?
}
i << Input.new(type: Broker)
end
super(inputs: inputs, output: Client::TargetIndex, func: method(:converter))
end
def converter(proto, broker)
Client::TargetIndex.load(proto.value.value, broker: broker)
end
end
# Build a target index client from a proto instance
class TargetIndexFromProto < Mapper
def initialize
inputs = [].tap do |i|
i << Input.new(type: SDK::Args::TargetIndex)
i << Input.new(type: Broker)
end
super(inputs: inputs, output: Client::TargetIndex, func: method(:converter))
end
def converter(proto, broker)
Client::TargetIndex.load(proto, broker: broker)
end
end
class TargetIndexToProto < Mapper
def initialize
super(
inputs: [
Input.new(type: Client::TargetIndex),
],
output: SDK::Args::TargetIndex,
func: method(:converter)
)
end
def converter(index)
index.to_proto
end
end
class MachineIndexToProto < Mapper
def initialize
super(
inputs: [
Input.new(type: Vagrant::MachineIndex),
],
output: SDK::Args::TargetIndex,
func: method(:converter)
)
end
def converter(index)
index.client.to_proto
end
end
end
end
end