65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"rkkvm/config"
|
|
"rkkvm/http/hw/rtc"
|
|
"rkkvm/http/hw/stream"
|
|
"rkkvm/http/route"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func main() {
|
|
config.Init()
|
|
cfg := config.Get()
|
|
lvl, err := log.ParseLevel(cfg.LogLevel)
|
|
if err == nil {
|
|
log.SetLevel(lvl)
|
|
} else {
|
|
log.Println("Failed to parse log level, use default level: info")
|
|
}
|
|
|
|
if cfg.Stream.Source == config.StreamSourceMjpeg {
|
|
ustreamer := stream.Init(cfg.UStreamer.Path, strings.Split(cfg.UStreamer.Args, " "))
|
|
ustreamer.Start()
|
|
//go ustreamer.Watch()
|
|
} else if cfg.Stream.Source == config.StreamSourceH264 {
|
|
ffmpeg := stream.InitFFmpeg(cfg.FFmpeg.Path, cfg.FFmpeg.FormatArgs())
|
|
ffmpeg.Start()
|
|
//go ffmpeg.Watch()
|
|
} else {
|
|
log.Fatalf("unsupported stream source type: %v", cfg.Stream.Source)
|
|
}
|
|
|
|
webrtc, err := rtc.Init(cfg.WebRtc.Host, cfg.WebRtc.Port)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
go webrtc.Read()
|
|
go webrtc.Listen()
|
|
|
|
r := gin.Default()
|
|
r.Use(gin.Recovery())
|
|
r.GET("/ping", func(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"message": "pong",
|
|
})
|
|
})
|
|
route.Api(r)
|
|
route.Static(r)
|
|
|
|
if cfg.NanoKVMUISupport {
|
|
r.POST("/api/auth/login", func(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"token": "disabled",
|
|
})
|
|
})
|
|
}
|
|
|
|
r.Run(fmt.Sprintf(":%d", cfg.Port))
|
|
}
|