mirror of
https://github.com/xaionaro-go/streamctl.git
synced 2025-12-24 12:27:57 +08:00
90 lines
1.4 KiB
Go
90 lines
1.4 KiB
Go
package streamtypes
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/xaionaro-go/xsync"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type StreamID string
|
|
type DestinationID string
|
|
type OBSInstanceID string
|
|
|
|
type ServerType int
|
|
|
|
const (
|
|
ServerTypeUndefined = ServerType(iota)
|
|
ServerTypeSRT
|
|
ServerTypeRTSP
|
|
ServerTypeRTMP
|
|
ServerTypeHLS
|
|
ServerTypeWebRTC
|
|
endOfServerType
|
|
)
|
|
|
|
func (t ServerType) String() string {
|
|
switch t {
|
|
case ServerTypeUndefined:
|
|
return "<undefined>"
|
|
case ServerTypeRTMP:
|
|
return "rtmp"
|
|
case ServerTypeRTSP:
|
|
return "rtsp"
|
|
case ServerTypeSRT:
|
|
return "srt"
|
|
case ServerTypeHLS:
|
|
return "hls"
|
|
case ServerTypeWebRTC:
|
|
return "webrtc"
|
|
default:
|
|
return fmt.Sprintf("unknown_type_%d", t)
|
|
}
|
|
}
|
|
|
|
func (t ServerType) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(t.String())
|
|
}
|
|
|
|
func ParseServerType(s string) ServerType {
|
|
for c := ServerTypeUndefined; c < endOfServerType; c++ {
|
|
if c.String() == s {
|
|
return c
|
|
}
|
|
}
|
|
|
|
return ServerTypeUndefined
|
|
}
|
|
|
|
func (t *ServerType) UnmarshalJSON(b []byte) error {
|
|
var s string
|
|
err := json.Unmarshal(b, &s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*t = ParseServerType(s)
|
|
return nil
|
|
}
|
|
|
|
func (t ServerType) MarshalYAML() ([]byte, error) {
|
|
return yaml.Marshal(t.String())
|
|
}
|
|
|
|
func (t *ServerType) UnmarshalYAML(b []byte) error {
|
|
var s string
|
|
err := yaml.Unmarshal(b, &s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*t = ParseServerType(s)
|
|
return nil
|
|
}
|
|
|
|
type OBSState struct {
|
|
xsync.Mutex
|
|
VolumeMeters map[string][][3]float64
|
|
}
|