package pluginextensionv2

import (
	"context"

	"github.com/hashicorp/go-plugin"
	"google.golang.org/grpc"
)

type RendererPlugin interface {
	RendererClient
	SanitizerClient
}

type RendererGRPCPlugin struct {
	plugin.NetRPCUnsupportedPlugin
}

func (p *RendererGRPCPlugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
	return nil
}

func (p *RendererGRPCPlugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (any, error) {
	return &RendererGRPCClient{NewRendererClient(c), NewSanitizerClient(c)}, nil
}

type RendererGRPCClient struct {
	RendererClient
	SanitizerClient
}

func (m *RendererGRPCClient) Sanitize(ctx context.Context, req *SanitizeRequest, opts ...grpc.CallOption) (*SanitizeResponse, error) {
	return m.SanitizerClient.Sanitize(ctx, req, opts...)
}

func (m *RendererGRPCClient) Render(ctx context.Context, req *RenderRequest, opts ...grpc.CallOption) (*RenderResponse, error) {
	return m.RendererClient.Render(ctx, req)
}

var _ RendererClient = &RendererGRPCClient{}
var _ SanitizerClient = &RendererGRPCClient{}
var _ plugin.GRPCPlugin = &RendererGRPCPlugin{}