2021-09-27 08:36:28 +00:00
|
|
|
package conf
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"fmt"
|
|
|
|
|
2023-04-01 16:39:12 +00:00
|
|
|
"github.com/bluenviron/gortsplib/v3"
|
2021-09-27 08:36:28 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// SourceProtocol is the sourceProtocol parameter.
|
|
|
|
type SourceProtocol struct {
|
2021-10-22 16:41:10 +00:00
|
|
|
*gortsplib.Transport
|
2021-09-27 08:36:28 +00:00
|
|
|
}
|
|
|
|
|
2022-06-21 11:41:15 +00:00
|
|
|
// MarshalJSON implements json.Marshaler.
|
2021-09-27 08:36:28 +00:00
|
|
|
func (d SourceProtocol) MarshalJSON() ([]byte, error) {
|
|
|
|
var out string
|
|
|
|
|
2021-10-22 16:41:10 +00:00
|
|
|
if d.Transport == nil {
|
2021-09-27 08:36:28 +00:00
|
|
|
out = "automatic"
|
|
|
|
} else {
|
2021-10-22 16:41:10 +00:00
|
|
|
switch *d.Transport {
|
|
|
|
case gortsplib.TransportUDP:
|
2021-09-27 08:36:28 +00:00
|
|
|
out = "udp"
|
|
|
|
|
2021-10-22 16:41:10 +00:00
|
|
|
case gortsplib.TransportUDPMulticast:
|
2021-09-27 08:36:28 +00:00
|
|
|
out = "multicast"
|
|
|
|
|
2023-03-31 14:22:08 +00:00
|
|
|
case gortsplib.TransportTCP:
|
2021-09-27 08:36:28 +00:00
|
|
|
out = "tcp"
|
2023-03-31 14:22:08 +00:00
|
|
|
|
|
|
|
default:
|
|
|
|
return nil, fmt.Errorf("invalid protocol: %v", d.Transport)
|
2021-09-27 08:36:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return json.Marshal(out)
|
|
|
|
}
|
|
|
|
|
2022-06-21 11:41:15 +00:00
|
|
|
// UnmarshalJSON implements json.Unmarshaler.
|
2021-09-27 08:36:28 +00:00
|
|
|
func (d *SourceProtocol) UnmarshalJSON(b []byte) error {
|
|
|
|
var in string
|
|
|
|
if err := json.Unmarshal(b, &in); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
switch in {
|
|
|
|
case "udp":
|
2021-10-22 16:41:10 +00:00
|
|
|
v := gortsplib.TransportUDP
|
|
|
|
d.Transport = &v
|
2021-09-27 08:36:28 +00:00
|
|
|
|
|
|
|
case "multicast":
|
2021-10-22 16:41:10 +00:00
|
|
|
v := gortsplib.TransportUDPMulticast
|
|
|
|
d.Transport = &v
|
2021-09-27 08:36:28 +00:00
|
|
|
|
|
|
|
case "tcp":
|
2021-10-22 16:41:10 +00:00
|
|
|
v := gortsplib.TransportTCP
|
|
|
|
d.Transport = &v
|
2021-09-27 08:36:28 +00:00
|
|
|
|
|
|
|
case "automatic":
|
2023-03-31 14:22:08 +00:00
|
|
|
d.Transport = nil
|
2021-09-27 08:36:28 +00:00
|
|
|
|
|
|
|
default:
|
2021-09-27 08:43:48 +00:00
|
|
|
return fmt.Errorf("invalid protocol '%s'", in)
|
2021-09-27 08:36:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
2021-10-11 09:46:40 +00:00
|
|
|
|
2022-12-15 23:50:47 +00:00
|
|
|
// unmarshalEnv implements envUnmarshaler.
|
2021-10-11 09:46:40 +00:00
|
|
|
func (d *SourceProtocol) unmarshalEnv(s string) error {
|
|
|
|
return d.UnmarshalJSON([]byte(`"` + s + `"`))
|
|
|
|
}
|