86 lines
3.0 KiB
Ruby
86 lines
3.0 KiB
Ruby
require "vagrant/plugin/v2/plugin"
|
|
require "vagrant/vagrantfile"
|
|
require "vagrant/box_collection"
|
|
require "vagrant/config"
|
|
require "pathname"
|
|
|
|
require 'google/protobuf/well_known_types'
|
|
|
|
module VagrantPlugins
|
|
module CommandServe
|
|
module Service
|
|
class InternalService < Hashicorp::Vagrant::RubyVagrant::Service
|
|
prepend Util::HasBroker
|
|
prepend Util::HasLogger
|
|
prepend Util::ExceptionLogger
|
|
|
|
def get_plugins(req, _unused_call)
|
|
plugins = []
|
|
plugin_manager = Vagrant::Plugin::V2::Plugin.manager
|
|
plugins = [[:commands, :COMMAND],
|
|
[:communicators, :COMMUNICATOR],
|
|
[:guests, :GUEST],
|
|
[:hosts, :HOST],
|
|
[:providers, :PROVIDER],
|
|
[:provisioners, :PROVISIONER],
|
|
[:synced_folders, :SYNCEDFOLDER]].map do |method, const|
|
|
plugin_manager.send(method).map do |k, v|
|
|
Hashicorp::Vagrant::Plugin.new(name: k, type: Hashicorp::Vagrant::Plugin::Type.const_get(const))
|
|
end
|
|
end.flatten
|
|
Hashicorp::Vagrant::GetPluginsResponse.new(
|
|
plugins: plugins
|
|
)
|
|
end
|
|
|
|
def parse_vagrantfile(req, _unused_call)
|
|
path = req.path
|
|
raw = File.read(path)
|
|
|
|
# Load up/parse the vagrantfile
|
|
config_loader = Vagrant::Config::Loader.new(
|
|
Vagrant::Config::VERSIONS, Vagrant::Config::VERSIONS_ORDER)
|
|
config_loader.set(:root, path)
|
|
v = Vagrant::Vagrantfile.new(config_loader, [:root])
|
|
|
|
machine_configs = []
|
|
# Get the config for each machine
|
|
v.machine_names.each do |mach|
|
|
machine_info = v.machine_config(mach, nil, nil, false)
|
|
root_config = machine_info[:config]
|
|
vm_config = root_config.vm
|
|
|
|
plugin_configs = []
|
|
root_config.__internal_state["keys"].each do |name, config|
|
|
# A builtin plugin
|
|
# TODO: find a better way to check the module
|
|
next if config.class.to_s.split("::")[0] == "VagrantPlugins"
|
|
plugin_configs << config.to_proto(name)
|
|
end
|
|
plugin_configs << root_config.ssh.to_proto("ssh")
|
|
plugin_configs << root_config.winrm.to_proto("winrm")
|
|
plugin_configs << root_config.winssh.to_proto("winssh")
|
|
|
|
machine_configs << Hashicorp::Vagrant::Sdk::Vagrantfile::MachineConfig.new(
|
|
name: mach.to_s,
|
|
config_vm: vm_config.to_proto,
|
|
config_vagrant: root_config.vagrant.to_proto(),
|
|
plugin_configs: plugin_configs
|
|
)
|
|
end
|
|
|
|
vagrantfile = Hashicorp::Vagrant::Sdk::Vagrantfile::Vagrantfile.new(
|
|
path: path,
|
|
raw: raw,
|
|
current_version: Vagrant::Config::CURRENT_VERSION,
|
|
machine_configs: machine_configs,
|
|
)
|
|
Hashicorp::Vagrant::ParseVagrantfileResponse.new(
|
|
vagrantfile: vagrantfile
|
|
)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|