80 lines
2.3 KiB
Ruby
80 lines
2.3 KiB
Ruby
# Copyright (c) HashiCorp, Inc.
|
|
# SPDX-License-Identifier: BUSL-1.1
|
|
|
|
module VagrantPlugins
|
|
module CommandServe
|
|
class Client
|
|
class TargetIndex < Client
|
|
# @param [String]
|
|
# @return [Boolean] true if delete is successful
|
|
def delete(ident)
|
|
logger.debug("deleting machine with id #{ident} from index")
|
|
client.delete(
|
|
SDK::TargetIndex::TargetIdentifier.new(
|
|
id: ident
|
|
)
|
|
)
|
|
true
|
|
end
|
|
|
|
# @param [String]
|
|
# @return [Vagrant::MachineIndex::Entry]
|
|
def get(ident)
|
|
logger.debug("getting machine with id #{ident} from index")
|
|
begin
|
|
resp = client.get(
|
|
SDK::TargetIndex::TargetIdentifier.new(
|
|
id: ident
|
|
)
|
|
)
|
|
machine = Target.load(resp, broker: broker).to_machine
|
|
Vagrant::MachineIndex::Entry.load(machine)
|
|
rescue GRPC::NotFound
|
|
nil
|
|
end
|
|
end
|
|
|
|
# @param [String]
|
|
# @return [Boolean]
|
|
def include?(ident)
|
|
logger.debug("checking for machine with id #{ident} in index")
|
|
client.includes(
|
|
SDK::TargetIndex::TargetIdentifier.new(
|
|
id: ident
|
|
)
|
|
).exists
|
|
end
|
|
|
|
# @param [Vagrant::MachineIndex::Entry]
|
|
# @return [Vagrant::MachineIndex::Entry]
|
|
def set(entry)
|
|
logger.debug("setting machine #{entry} in index")
|
|
if entry.id.to_s.empty?
|
|
raise ArgumentError,
|
|
"Entry id must be set"
|
|
end
|
|
resp = client.get(
|
|
SDK::TargetIndex::TargetIdentifier.new(
|
|
id: ident
|
|
)
|
|
)
|
|
machine = Target.load(resp, broker: broker).to_machine
|
|
machine.set_name(entry.name)
|
|
machine.set_state(entry.full_state)
|
|
Vagrant::MachineIndex::Entry.load(machine)
|
|
end
|
|
|
|
# Get all targets
|
|
# @return [Array<Vagrant::MachineIndex::Entry>]
|
|
def all
|
|
logger.debug("getting all machines")
|
|
client.all(Empty.new).targets.map do |t_ref|
|
|
machine = Target.load(t_ref, broker: broker).to_machine
|
|
Vagrant::MachineIndex::Entry.load(machine)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|