package rpc

import (
	"context"
	"net"
	"os"

	"github.com/puzpuzpuz/xsync/v4"
	"google.golang.org/grpc"
)

type RPC struct {
	grpcServer *grpc.Server
	listener   net.Listener
	execs      *xsync.Map[string, *os.Process]

	UnimplementedAgentServer
}

func New(listener net.Listener) (*RPC, error) {
	rpc := &RPC{
		grpcServer: grpc.NewServer(),
		listener:   listener,
		execs:      xsync.NewMap[string, *os.Process](),
	}

	RegisterAgentServer(rpc.grpcServer, rpc)

	return rpc, nil
}

func (rpc *RPC) Run(ctx context.Context) error {
	go func() {
		<-ctx.Done()

		rpc.grpcServer.Stop()
	}()

	return rpc.grpcServer.Serve(rpc.listener)
}