plugin_cmd.go

package protokit

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"os/exec"
	"path/filepath"

	"github.com/sandwich-go/boost/xpanic"
	"github.com/sandwich-go/boost/xproc"
)

// MustRun pluginPath目前只支持本地文件, 后续加入远程版本支持
func MustRun(pluginPath string, parameter *Parameter, args ...string) {
	pluginPath, err := getCommand(pluginPath)
	xpanic.WhenErrorAsFmtFirst(err, "got err:%w while check plugin")
	bb, err := json.Marshal(parameter)
	xpanic.WhenErrorAsFmtFirst(err, "got err:%w while marshal parameter")
	content, err := xproc.Run(pluginPath,
		xproc.WithArgs(args...),
		xproc.WithWorkingDir(parameter.WorkingDir),
		xproc.WithStdin(bytes.NewBuffer(bb)),
	)
	xpanic.WhenErrorAsFmtFirst(err, "got err:%w while run plugin :%s with args:%s ", pluginPath, args)
	fmt.Println(content)
}

var _ = MustRun

func getCommand(pluginPath string) (string, error) {
	p, err := exec.LookPath(pluginPath)
	if err == nil {
		abs, err := filepath.Abs(p)
		if err != nil {
			return abs, err
		}
		return abs, nil
	}
	defaultErr := errors.New("invalid plugin value " + pluginPath)
	return pluginPath, defaultErr
}