mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package plugins
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os/exec"
|
|
|
|
"github.com/hashicorp/go-hclog"
|
|
"github.com/hashicorp/go-plugin"
|
|
"github.com/synctv-org/synctv/internal/provider"
|
|
"github.com/synctv-org/synctv/internal/provider/providers"
|
|
sysnotify "github.com/synctv-org/synctv/internal/sysNotify"
|
|
providerpb "github.com/synctv-org/synctv/proto/provider"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func InitProviderPlugins(name string, arg []string, Logger hclog.Logger) error {
|
|
client := NewProviderPlugin(name, arg, Logger)
|
|
sysnotify.RegisterSysNotifyTask(0, sysnotify.NewSysNotifyTask("plugin", sysnotify.NotifyTypeEXIT, func() error {
|
|
client.Kill()
|
|
return nil
|
|
}))
|
|
c, err := client.Client()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i, err := c.Dispense("Provider")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
provider, ok := i.(provider.ProviderInterface)
|
|
if !ok {
|
|
return fmt.Errorf("%s not implement ProviderInterface", name)
|
|
}
|
|
providers.RegisterProvider(provider)
|
|
return nil
|
|
}
|
|
|
|
var HandshakeConfig = plugin.HandshakeConfig{
|
|
ProtocolVersion: 1,
|
|
MagicCookieKey: "BASIC_PLUGIN",
|
|
MagicCookieValue: "hello",
|
|
}
|
|
|
|
var pluginMap = map[string]plugin.Plugin{
|
|
"Provider": &ProviderPlugin{},
|
|
}
|
|
|
|
type ProviderPlugin struct {
|
|
plugin.Plugin
|
|
Impl provider.ProviderInterface
|
|
}
|
|
|
|
func (p *ProviderPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
|
providerpb.RegisterOauth2PluginServer(s, &GRPCServer{Impl: p.Impl})
|
|
return nil
|
|
}
|
|
|
|
func (p *ProviderPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (interface{}, error) {
|
|
return &GRPCClient{client: providerpb.NewOauth2PluginClient(c)}, nil
|
|
}
|
|
|
|
func NewProviderPlugin(name string, arg []string, Logger hclog.Logger) *plugin.Client {
|
|
return plugin.NewClient(&plugin.ClientConfig{
|
|
HandshakeConfig: HandshakeConfig,
|
|
Plugins: pluginMap,
|
|
Cmd: exec.Command(name, arg...),
|
|
AllowedProtocols: []plugin.Protocol{
|
|
plugin.ProtocolGRPC},
|
|
Logger: Logger,
|
|
})
|
|
}
|