f5d778d51a
This reverts commit 1c26a4abb0c9e095b3f6ec9944c4b15f6f1cd064, reversing changes made to 186824a568583d8f6f2a50501d940ed71608fa0b. The changes broke plugin loading when using subcommands so these changes will be reverted until the underlying issue can be investigated and resolved.
124 lines
3.2 KiB
Go
124 lines
3.2 KiB
Go
package serverclient
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/go-plugin"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
|
|
"github.com/hashicorp/vagrant-plugin-sdk/internal-shared/pluginclient"
|
|
"github.com/hashicorp/vagrant-plugin-sdk/proto/vagrant_plugin_sdk"
|
|
"github.com/hashicorp/vagrant/internal/server/proto/ruby_vagrant"
|
|
)
|
|
|
|
type RubyVagrant interface {
|
|
GetPlugins() ([]*ruby_vagrant.Plugin, error)
|
|
ParseVagrantfile(string) (*vagrant_plugin_sdk.Vagrantfile_Vagrantfile, error)
|
|
}
|
|
|
|
// This is the implementation of plugin.GRPCPlugin so we can serve/consume this.
|
|
type RubyVagrantPlugin struct {
|
|
plugin.NetRPCUnsupportedPlugin
|
|
|
|
Impl RubyVagrant
|
|
}
|
|
|
|
type RubyVagrantClient struct {
|
|
broker *plugin.GRPCBroker
|
|
client ruby_vagrant.RubyVagrantClient
|
|
ctx context.Context
|
|
}
|
|
|
|
func RubyVagrantPluginConfig(log hclog.Logger) *plugin.ClientConfig {
|
|
log = log.ResetNamed("vagrant.legacy")
|
|
config := pluginclient.ClientConfig(log)
|
|
config.Logger = log
|
|
config.VersionedPlugins[1]["vagrantrubyruntime"] = &RubyVagrantPlugin{}
|
|
return config
|
|
}
|
|
|
|
// No Go server implementation. Server is provided by the Vagrant Ruby runtime
|
|
func (p *RubyVagrantPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
|
return errors.New("vagrant ruby runtime server not implemented")
|
|
}
|
|
|
|
func (p *RubyVagrantPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
|
|
return RubyVagrantClient{
|
|
broker: broker,
|
|
client: ruby_vagrant.NewRubyVagrantClient(c),
|
|
ctx: ctx,
|
|
}, nil
|
|
}
|
|
|
|
func (r *RubyVagrantClient) GRPCBroker() *plugin.GRPCBroker {
|
|
return r.broker
|
|
}
|
|
|
|
func (r *RubyVagrantClient) GetPlugins() ([]*ruby_vagrant.Plugin, error) {
|
|
plugins, err := r.client.GetPlugins(context.Background(), &emptypb.Empty{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return plugins.Plugins, nil
|
|
}
|
|
|
|
func (r *RubyVagrantClient) ParseVagrantfile(path string) (*vagrant_plugin_sdk.Args_Hash, error) {
|
|
vf, err := r.client.ParseVagrantfile(
|
|
context.Background(),
|
|
&ruby_vagrant.ParseVagrantfileRequest{Path: path},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return vf.Data, nil
|
|
}
|
|
|
|
func (r *RubyVagrantClient) ParseVagrantfileProc(ref *vagrant_plugin_sdk.Args_ProcRef) (*vagrant_plugin_sdk.Args_Hash, error) {
|
|
vf, err := r.client.ParseVagrantfileProc(
|
|
context.Background(),
|
|
&ruby_vagrant.ParseVagrantfileProcRequest{
|
|
Proc: ref,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return vf.Data, nil
|
|
}
|
|
|
|
func (r *RubyVagrantClient) ParseVagrantfileSubvm(subvm *vagrant_plugin_sdk.Config_RawRubyValue) (*vagrant_plugin_sdk.Args_Hash, error) {
|
|
resp, err := r.client.ParseVagrantfileSubvm(
|
|
context.Background(),
|
|
&ruby_vagrant.ParseVagrantfileSubvmRequest{
|
|
Subvm: subvm,
|
|
},
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.Data, nil
|
|
}
|
|
|
|
func (r *RubyVagrantClient) ParseVagrantfileProvider(provider string, subvm *vagrant_plugin_sdk.Config_RawRubyValue) (*vagrant_plugin_sdk.Args_Hash, error) {
|
|
resp, err := r.client.ParseVagrantfileProvider(
|
|
context.Background(),
|
|
&ruby_vagrant.ParseVagrantfileProviderRequest{
|
|
Provider: provider,
|
|
Subvm: subvm,
|
|
},
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return resp.Data, nil
|
|
}
|