Alex X
71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package probe
|
|
|
|
import (
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
|
)
|
|
|
|
type Probe struct {
|
|
Type string `json:"type,omitempty"`
|
|
RemoteAddr string `json:"remote_addr,omitempty"`
|
|
UserAgent string `json:"user_agent,omitempty"`
|
|
Medias []*core.Media `json:"medias,omitempty"`
|
|
Receivers []*core.Receiver `json:"receivers,omitempty"`
|
|
Senders []*core.Sender `json:"senders,omitempty"`
|
|
}
|
|
|
|
func NewProbe(query url.Values) *Probe {
|
|
c := &Probe{Type: "probe"}
|
|
c.Medias = core.ParseQuery(query)
|
|
|
|
for _, value := range query["microphone"] {
|
|
media := &core.Media{Kind: core.KindAudio, Direction: core.DirectionRecvonly}
|
|
|
|
for _, name := range strings.Split(value, ",") {
|
|
name = strings.ToUpper(name)
|
|
switch name {
|
|
case "", "COPY":
|
|
name = core.CodecAny
|
|
}
|
|
media.Codecs = append(media.Codecs, &core.Codec{Name: name})
|
|
}
|
|
|
|
c.Medias = append(c.Medias, media)
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
func (p *Probe) GetMedias() []*core.Media {
|
|
return p.Medias
|
|
}
|
|
|
|
func (p *Probe) AddTrack(media *core.Media, codec *core.Codec, track *core.Receiver) error {
|
|
sender := core.NewSender(media, codec)
|
|
sender.Bind(track)
|
|
p.Senders = append(p.Senders, sender)
|
|
return nil
|
|
}
|
|
|
|
func (p *Probe) GetTrack(media *core.Media, codec *core.Codec) (*core.Receiver, error) {
|
|
receiver := core.NewReceiver(media, codec)
|
|
p.Receivers = append(p.Receivers, receiver)
|
|
return receiver, nil
|
|
}
|
|
|
|
func (p *Probe) Start() error {
|
|
return nil
|
|
}
|
|
|
|
func (p *Probe) Stop() error {
|
|
for _, receiver := range p.Receivers {
|
|
receiver.Close()
|
|
}
|
|
for _, sender := range p.Senders {
|
|
sender.Close()
|
|
}
|
|
return nil
|
|
}
|