package manage

import (
	"fmt"
	"gitea.cdlsxd.cn/BaseSystem/plugin/shared"
	"github.com/go-playground/validator/v10"
	"github.com/hashicorp/go-plugin"
	"github.com/pkg/errors"
	"os/exec"
	"time"
)

type Config struct {
	Cmd         string        `validate:"required"` // 插件的执行路径
	Tag         string        `validate:"required"` // 插件的tag,对应插件
	Version     uint          `validate:"required"` // 插件的版本
	CookieKey   string        `validate:"required"` // 插件通信的cookie key
	CookieValue string        `validate:"required"` // 插件通信的cookie value
	Timeout     time.Duration // 插件超时时间

	// 供应商配置
	// Config json.RawMessage
}

func (c *Config) Validate() error {
	err := validator.New().Struct(c)
	if err == nil {
		return nil
	}
	var combinedErr error
	for index, err2 := range err.(validator.ValidationErrors) {
		combinedErr = errors.Wrap(err2, fmt.Sprintf("index:%d,field:%s \n", index, err2.Field()))
	}
	return combinedErr
}

func newClient(c *Config) (shared.PluginService, func(), error) {
	pluginClientConfig := &plugin.ClientConfig{
		HandshakeConfig:  shared.HandshakeConfig(c.Version, c.CookieKey, c.CookieValue),
		Cmd:              exec.Command(c.Cmd),
		Plugins:          map[string]plugin.Plugin{c.Tag: &shared.GRPCPlugin{}},
		AllowedProtocols: []plugin.Protocol{plugin.ProtocolGRPC},
	}
	if c.Timeout != 0 {
		pluginClientConfig.StartTimeout = c.Timeout
	}
	client := plugin.NewClient(pluginClientConfig)
	protocol, err := client.Client()
	if err != nil {
		return nil, func() {}, fmt.Errorf("pluginClient.Client err:%v", err)
	}
	cleanup := func() {
		_ = protocol.Close()
		client.Kill()
	}
	raw, err := protocol.Dispense(c.Tag)
	if err != nil {
		return nil, cleanup, fmt.Errorf("protocol.Dispense err:%v", err)
	}
	pluginService, ok := raw.(shared.PluginService)
	if !ok {
		return nil, cleanup, fmt.Errorf("unexpected plugin type: %T", raw)
	}
	return pluginService, cleanup, nil
}