pluginFlag.go

package subcmds

import (
	"fmt"
	"os/exec"
	"strings"

	"github.com/hashicorp/go-hclog"

	"github.com/yoheimuta/protolint/internal/addon/plugin/shared"

	"github.com/hashicorp/go-plugin"
)

// PluginFlag manages a flag for plugins.
type PluginFlag struct {
	raws []string
}

// String implements flag.Value.
func (f *PluginFlag) String() string {
	return fmt.Sprint(strings.Join(f.raws, ","))
}

// Set implements flag.Value.
func (f *PluginFlag) Set(value string) error {
	f.raws = append(f.raws, value)
	return nil
}

// BuildPlugins builds all plugins.
func (f *PluginFlag) BuildPlugins(verbose bool) ([]shared.RuleSet, error) {
	var plugins []shared.RuleSet

	for _, value := range f.raws {
		level := hclog.Warn
		if verbose {
			level = hclog.Trace
		}
		client := plugin.NewClient(&plugin.ClientConfig{
			HandshakeConfig: shared.Handshake,
			Plugins:         shared.PluginMap,
			Cmd:             exec.Command("sh", "-c", value),
			AllowedProtocols: []plugin.Protocol{
				plugin.ProtocolGRPC,
			},
			Logger: hclog.New(&hclog.LoggerOptions{
				Output: hclog.DefaultOutput,
				Level:  level,
				Name:   "plugin",
			}),
		})

		rpcClient, err := client.Client()
		if err != nil {
			return nil, fmt.Errorf("failed client.Client(), err=%s", err)
		}

		ruleSet, err := rpcClient.Dispense("ruleSet")
		if err != nil {
			return nil, fmt.Errorf("failed Dispense, err=%s", err)
		}
		plugins = append(plugins, ruleSet.(shared.RuleSet))
	}
	return plugins, nil
}