93 lines
1.7 KiB
Go
93 lines
1.7 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type Config struct {
|
|
LogLevel string `yaml:"log_level"`
|
|
Auth bool `yaml:"auth"`
|
|
AuthSecret string `yaml:"auth_secret"`
|
|
Port int `yaml:"port"`
|
|
UStreamer UStreamer `yaml:"ustreamer"`
|
|
FFmpeg FFmpeg `yaml:"ffmpeg"`
|
|
WebRtc WebRtc `yaml:"webrtc"`
|
|
Stream Stream `yaml:"stream"`
|
|
|
|
NanoKVMUISupport bool `yaml:"nano_kvm_ui_support"`
|
|
}
|
|
|
|
type WebRtc struct {
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
}
|
|
|
|
type ExtProcess struct {
|
|
Path string `yaml:"path"`
|
|
Args string `yaml:"args"`
|
|
}
|
|
|
|
type Stream struct {
|
|
Source string `yaml:"source"` // mjpeg or h264
|
|
}
|
|
|
|
const (
|
|
StreamSourceMjpeg = "mjpeg"
|
|
StreamSourceH264 = "h264"
|
|
)
|
|
|
|
type FFmpeg struct {
|
|
ExtProcess
|
|
FPS int `yaml:"fps"`
|
|
Bitrate int `yaml:"bitrate"`
|
|
}
|
|
|
|
func (f FFmpeg) FormatArgs() []string {
|
|
return strings.Split(fmt.Sprintf(f.Args, f.Bitrate*1000, f.FPS), " ")
|
|
}
|
|
|
|
type UStreamer struct {
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
ExtProcess
|
|
}
|
|
|
|
var c Config
|
|
|
|
func Get() Config {
|
|
return c
|
|
}
|
|
|
|
func Init() {
|
|
c = Config{
|
|
LogLevel: "debug",
|
|
Auth: false,
|
|
Port: 8080,
|
|
UStreamer: UStreamer{
|
|
Host: "0.0.0.0",
|
|
Port: 8888,
|
|
ExtProcess: ExtProcess{
|
|
Path: "/app/ustreamer",
|
|
Args: "--host 0.0.0.0 --port 8888 -m BGR24 -f 60",
|
|
},
|
|
},
|
|
FFmpeg: FFmpeg{
|
|
ExtProcess: ExtProcess{
|
|
Path: "/app/ffmpeg",
|
|
Args: "-re -i /dev/video0 -c:v h264_rkmpp -b:v %d -framerate %d -bsf:v h264_mp4toannexb -g 10 -f rtp rtp://127.0.0.1:5004?pkt_size=1200",
|
|
},
|
|
FPS: 60,
|
|
Bitrate: 6000,
|
|
},
|
|
WebRtc: WebRtc{
|
|
Host: "127.0.0.1",
|
|
Port: 5004,
|
|
},
|
|
Stream: Stream{
|
|
Source: StreamSourceMjpeg,
|
|
},
|
|
NanoKVMUISupport: true,
|
|
}
|
|
}
|