Code refactoring after #859

pkg/stdin/client.go

@@ -0,0 +1,41 @@
+package stdin
+
+import (
+	"io"
+	"os/exec"
+
+	"github.com/AlexxIT/go2rtc/pkg/core"
+)
+
+type Client struct {
+	cmd  *exec.Cmd
+	pipe io.WriteCloser
+
+	medias []*core.Media
+	sender *core.Sender
+	send   int
+}
+
+func NewClient(cmd *exec.Cmd) (*Client, error) {
+	pipe, err := PipeCloser(cmd)
+	if err != nil {
+		return nil, err
+	}
+
+	c := &Client{
+		pipe: pipe,
+		cmd:  cmd,
+		medias: []*core.Media{
+			{
+				Kind:      core.KindAudio,
+				Direction: core.DirectionSendonly,
+				Codecs: []*core.Codec{
+					{Name: core.CodecPCMA, ClockRate: 8000},
+					{Name: core.CodecPCM},
+				},
+			},
+		},
+	}
+
+	return c, nil
+}

pkg/stdin/consumer.go

@@ -0,0 +1,52 @@
+package stdin
+
+import (
+	"encoding/json"
+
+	"github.com/AlexxIT/go2rtc/pkg/core"
+	"github.com/pion/rtp"
+)
+
+func (c *Client) GetMedias() []*core.Media {
+	return c.medias
+}
+
+func (c *Client) GetTrack(media *core.Media, codec *core.Codec) (*core.Receiver, error) {
+	return nil, core.ErrCantGetTrack
+}
+
+func (c *Client) AddTrack(media *core.Media, _ *core.Codec, track *core.Receiver) error {
+	if c.sender == nil {
+		c.sender = core.NewSender(media, track.Codec)
+		c.sender.Handler = func(packet *rtp.Packet) {
+			_, _ = c.pipe.Write(packet.Payload)
+			c.send += len(packet.Payload)
+		}
+	}
+
+	c.sender.HandleRTP(track)
+	return nil
+}
+
+func (c *Client) Start() (err error) {
+	return c.cmd.Run()
+}
+
+func (c *Client) Stop() (err error) {
+	if c.sender != nil {
+		c.sender.Close()
+	}
+	return c.pipe.Close()
+}
+
+func (c *Client) MarshalJSON() ([]byte, error) {
+	info := &core.Info{
+		Type:   "Exec active consumer",
+		Medias: c.medias,
+		Send:   c.send,
+	}
+	if c.sender != nil {
+		info.Senders = []*core.Sender{c.sender}
+	}
+	return json.Marshal(info)
+}

pkg/stdin/pipe.go

@@ -0,0 +1,26 @@
+package stdin
+
+import (
+	"errors"
+	"io"
+	"os/exec"
+)
+
+type pipeCloser struct {
+	io.Writer
+	io.Closer
+	cmd *exec.Cmd
+}
+
+func PipeCloser(cmd *exec.Cmd) (io.WriteCloser, error) {
+	stdin, err := cmd.StdinPipe()
+	if err != nil {
+		return nil, err
+	}
+
+	return pipeCloser{stdin, stdin, cmd}, nil
+}
+
+func (p pipeCloser) Close() (err error) {
+	return errors.Join(p.Closer.Close(), p.cmd.Process.Kill(), p.cmd.Wait())
+}