generator_rpc.go

package specraft

import (
	"io"
	"net/rpc"

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

type Generator interface {
	Generate(opts any, patterns ...string) (io.Reader, error)
}

type GeneratorRPC struct {
	client *rpc.Client
}

type GeneratorGenerateRequest struct {
	Patterns []string
	Opts     any
}

type GeneratorGenerateResponse struct {
	Data io.Reader
	Err  error
}

func (g *GeneratorRPC) Generate(patterns ...string) (io.Reader, error) {
	res := &GeneratorGenerateResponse{}
	req := &GeneratorGenerateRequest{
		Patterns: patterns,
	}
	err := g.client.Call("Plugin.Generate", req, res)
	if err != nil {
		return nil, err
	}
	return res.Data, res.Err
}

type GeneratorRPCServer struct {
	Impl Generator
}

func (s *GeneratorRPCServer) Generate(req GeneratorGenerateRequest, res *GeneratorGenerateResponse) error {
	res.Data, res.Err = s.Impl.Generate(req.Opts, req.Patterns...)
	return nil
}

type GeneratorPlugin struct {
	Impl Generator
}

func (g *GeneratorPlugin) Server(*plugin.MuxBroker) (any, error) {
	return &GeneratorRPCServer{Impl: g.Impl}, nil
}

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