Setup communincator service
This commit is contained in:
sophia
Paul Hinze
parent
b6dfff9679
commit
6903f9bf09
@ -98,6 +98,7 @@ module VagrantPlugins
|
|||||||
|
|
||||||
[Service::InternalService, Service::ProviderService, Service::GuestService,
|
[Service::InternalService, Service::ProviderService, Service::GuestService,
|
||||||
Service::HostService, Service::CommandService, Service::SyncedFolderService,
|
Service::HostService, Service::CommandService, Service::SyncedFolderService,
|
||||||
|
Service::CommunicatorService,
|
||||||
Broker::Streamer].each do |service_klass|
|
Broker::Streamer].each do |service_klass|
|
||||||
service = service_klass.new(broker: broker)
|
service = service_klass.new(broker: broker)
|
||||||
s.handle(service)
|
s.handle(service)
|
||||||
|
|||||||
@ -3,6 +3,7 @@ module VagrantPlugins
|
|||||||
module Service
|
module Service
|
||||||
autoload :CapabilityPlatformService, Vagrant.source_root.join("plugins/commands/serve/service/capability_platform_service").to_s
|
autoload :CapabilityPlatformService, Vagrant.source_root.join("plugins/commands/serve/service/capability_platform_service").to_s
|
||||||
autoload :CommandService, Vagrant.source_root.join("plugins/commands/serve/service/command_service").to_s
|
autoload :CommandService, Vagrant.source_root.join("plugins/commands/serve/service/command_service").to_s
|
||||||
|
autoload :CommunicatorService, Vagrant.source_root.join("plugins/commands/serve/service/communicator_service").to_s
|
||||||
autoload :GuestService, Vagrant.source_root.join("plugins/commands/serve/service/guest_service").to_s
|
autoload :GuestService, Vagrant.source_root.join("plugins/commands/serve/service/guest_service").to_s
|
||||||
autoload :HostService, Vagrant.source_root.join("plugins/commands/serve/service/host_service").to_s
|
autoload :HostService, Vagrant.source_root.join("plugins/commands/serve/service/host_service").to_s
|
||||||
autoload :InternalService, Vagrant.source_root.join("plugins/commands/serve/service/internal_service").to_s
|
autoload :InternalService, Vagrant.source_root.join("plugins/commands/serve/service/internal_service").to_s
|
||||||
|
|||||||
404
plugins/commands/serve/service/communicator_service.rb
Normal file
404
plugins/commands/serve/service/communicator_service.rb
Normal file
@ -0,0 +1,404 @@
|
|||||||
|
require "google/protobuf/well_known_types"
|
||||||
|
|
||||||
|
module VagrantPlugins
|
||||||
|
module CommandServe
|
||||||
|
module Service
|
||||||
|
class CommunicatorService < Hashicorp::Vagrant::Sdk::CommunicatorService::Service
|
||||||
|
include Util::ServiceInfo
|
||||||
|
|
||||||
|
prepend Util::HasMapper
|
||||||
|
prepend Util::HasBroker
|
||||||
|
prepend Util::ExceptionLogger
|
||||||
|
prepend Util::HasLogger
|
||||||
|
|
||||||
|
def initialize(*args, **opts, &block)
|
||||||
|
super()
|
||||||
|
end
|
||||||
|
|
||||||
|
def ready_spec(*_)
|
||||||
|
logger.debug("generating ready spec")
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "ready_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.ReadyResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def ready(req, ctx)
|
||||||
|
logger.debug("Checking if ready")
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
target = mapper.funcspec_map(req)
|
||||||
|
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
ready = plugin.new(machine).ready?
|
||||||
|
logger.debug("is ready: #{ready}")
|
||||||
|
SDK::Communicator::ReadyResp.new(
|
||||||
|
ready: ready
|
||||||
|
)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def wait_for_ready_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "ready_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.TimeDuration",
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.ReadyResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def wait_for_ready(req, ctx)
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
target, wait_duration = mapper.funcspec_map(req)
|
||||||
|
logger.debug("Got target #{target}")
|
||||||
|
logger.debug("Got duration #{wait_duration}")
|
||||||
|
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
begin
|
||||||
|
ready = plugin.new(machine).wait_for_ready(wait_duration)
|
||||||
|
rescue => err
|
||||||
|
logger.error(err)
|
||||||
|
logger.debug("#{err.class}: #{err}\n#{err.backtrace.join("\n")}")
|
||||||
|
raise
|
||||||
|
end
|
||||||
|
logger.debug("ready? #{ready}")
|
||||||
|
SDK::Communicator::ReadyResp.new(
|
||||||
|
ready: ready
|
||||||
|
)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def download_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "download_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Path",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Path",
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.FileTransferResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def download(req, ctx)
|
||||||
|
logger.debug("Uploading")
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
|
||||||
|
target, from, to = mapper.funcspec_map(req)
|
||||||
|
logger.debug("Got target #{target}")
|
||||||
|
logger.debug("Got from #{from}")
|
||||||
|
logger.debug("Got to #{to}")
|
||||||
|
|
||||||
|
logger.info("mapping received arguments to guest machine")
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
communicator = plugin.new(machine)
|
||||||
|
logger.debug("communicator: #{communicator}")
|
||||||
|
|
||||||
|
communicator.download(from, to)
|
||||||
|
|
||||||
|
SDK::Communicator::FileTransferResp.new()
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def upload_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "upload_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Path",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Path",
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.FileTransferResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def upload(req, ctx)
|
||||||
|
logger.debug("Uploading")
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
|
||||||
|
target, from, to = mapper.funcspec_map(req)
|
||||||
|
logger.debug("Got target #{target}")
|
||||||
|
logger.debug("Got from #{from}")
|
||||||
|
logger.debug("Got to #{to}")
|
||||||
|
|
||||||
|
logger.info("mapping received arguments to guest machine")
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
communicator = plugin.new(machine)
|
||||||
|
logger.debug("communicator: #{communicator}")
|
||||||
|
|
||||||
|
communicator.upload(from, to)
|
||||||
|
|
||||||
|
SDK::Communicator::FileTransferResp.new()
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def execute_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "execute_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Command",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "", # TODO: get opts
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.ExecuteResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def execute(req, ctx)
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
|
||||||
|
target, cmd, opts = mapper.funcspec_map(req)
|
||||||
|
logger.debug("Got machine #{target}")
|
||||||
|
logger.debug("Got opts #{opts}")
|
||||||
|
logger.debug("Got cmd #{cmd}")
|
||||||
|
|
||||||
|
logger.info("mapping received arguments to guest machine")
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
communicator = plugin.new(machine)
|
||||||
|
logger.debug("communicator: #{communicator}")
|
||||||
|
|
||||||
|
exit_code = communicator.execute(cmd, opts)
|
||||||
|
logger.debug("command exit code: #{exit_code}")
|
||||||
|
|
||||||
|
SDK::Communicator::ExecuteResp.new(
|
||||||
|
exit_code: exit_code
|
||||||
|
)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def privileged_execute_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "privileged_execute_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Command",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "", # TODO: get opts
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.ExecuteResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def privileged_execute(req, ctx)
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
|
||||||
|
target, cmd, opts = mapper.funcspec_map(req)
|
||||||
|
logger.debug("Got machine #{target}")
|
||||||
|
logger.debug("Got opts #{opts}")
|
||||||
|
logger.debug("Got cmd #{cmd}")
|
||||||
|
|
||||||
|
logger.info("mapping received arguments to guest machine")
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
communicator = plugin.new(machine)
|
||||||
|
logger.debug("communicator: #{communicator}")
|
||||||
|
|
||||||
|
exit_code = communicator.sudo(cmd, opts)
|
||||||
|
logger.debug("command exit code: #{exit_code}")
|
||||||
|
|
||||||
|
SDK::Communicator::ExecuteResp.new(
|
||||||
|
exit_code: exit_code
|
||||||
|
)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def test_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "test_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Command",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "", # TODO: get opts
|
||||||
|
name: "",
|
||||||
|
)
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.TestResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def test(req, ctx)
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
|
||||||
|
target, cmd, opts = mapper.funcspec_map(req)
|
||||||
|
logger.debug("Got machine #{target}")
|
||||||
|
logger.debug("Got opts #{opts}")
|
||||||
|
logger.debug("Got cmd #{cmd}")
|
||||||
|
|
||||||
|
logger.info("mapping received arguments to guest machine")
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
communicator = plugin.new(machine)
|
||||||
|
logger.debug("communicator: #{communicator}")
|
||||||
|
|
||||||
|
valid = communicator.test(cmd, opts)
|
||||||
|
logger.debug("command is valid?: #{valid}")
|
||||||
|
|
||||||
|
SDK::Communicator::TestResp.new(
|
||||||
|
valid: valid
|
||||||
|
)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def reset_spec(*_)
|
||||||
|
SDK::FuncSpec.new(
|
||||||
|
name: "reset_spec",
|
||||||
|
args: [
|
||||||
|
SDK::FuncSpec::Value.new(
|
||||||
|
type: "hashicorp.vagrant.sdk.Args.Target.Machine",
|
||||||
|
name: "",
|
||||||
|
),
|
||||||
|
],
|
||||||
|
result: [
|
||||||
|
type: "hashicorp.vagrant.sdk.Communicator.ResetResp",
|
||||||
|
name: "",
|
||||||
|
]
|
||||||
|
)
|
||||||
|
end
|
||||||
|
|
||||||
|
def reset(req, ctx)
|
||||||
|
with_info(ctx) do |info|
|
||||||
|
plugin_name = info.plugin_name
|
||||||
|
logger.debug("Got plugin #{plugin_name}")
|
||||||
|
target = mapper.funcspec_map(req)
|
||||||
|
|
||||||
|
machine = mapper.map(target, to: Vagrant::Machine)
|
||||||
|
logger.debug("Got machine #{machine}")
|
||||||
|
|
||||||
|
plugin = Vagrant.plugin("2").manager.communicators[plugin_name.to_s.to_sym]
|
||||||
|
logger.debug("Got plugin #{plugin}")
|
||||||
|
|
||||||
|
communicator = plugin.new(machine)
|
||||||
|
logger.debug("communicator: #{communicator}")
|
||||||
|
|
||||||
|
communicator.reset()
|
||||||
|
|
||||||
|
SDK::Communicator::ResetResp.new()
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
Loading…
x
Reference in New Issue
Block a user