outputer_rpc.go

package specraft

import (
	"io"
	"net/rpc"

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

type Outputer interface {
	Output(data io.Reader, opts any) error
}

type OutputerRPC struct {
	client *rpc.Client
}

type OutputerOutputRequest struct {
	Data io.Reader
	Opts any
}

type OutputerOutputResponse struct {
	Err error
}

func (o *OutputerRPC) Output(data io.Reader) error {
	res := &OutputerOutputResponse{}
	req := &OutputerOutputRequest{
		Data: data,
	}
	err := o.client.Call("Plugin.Output", req, res)
	if err != nil {
		return err
	}
	return res.Err
}

type OutputerRPCServer struct {
	Impl Outputer
}

func (o *OutputerRPCServer) Output(req OutputerOutputRequest, res *OutputerOutputResponse) error {
	res.Err = o.Impl.Output(req.Data, req.Opts)
	return nil
}

type OutputerPlugin struct {
	Impl Outputer
}

func (o *OutputerPlugin) Server(*plugin.MuxBroker) (any, error) {
	return &OutputerRPCServer{Impl: o.Impl}, nil
}

func (OutputerPlugin) Client(b *plugin.MuxBroker, c *rpc.Client) (any, error) {
	return &OutputerRPC{client: c}, nil
}