77 lines
2.2 KiB
Ruby
77 lines
2.2 KiB
Ruby
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 [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 [MachineIndex::Entry]
|
|
# @return [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<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
|