initial commit

config/config.go

@@ -0,0 +1,92 @@
+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,
+	}
+}