Alexey Khit
84 lines
1.5 KiB
Go
84 lines
1.5 KiB
Go
package device
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/AlexxIT/go2rtc/cmd/api"
|
|
"github.com/AlexxIT/go2rtc/cmd/app"
|
|
"github.com/AlexxIT/go2rtc/pkg/streamer"
|
|
"github.com/rs/zerolog"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func Init() {
|
|
log = app.GetLogger("exec")
|
|
|
|
api.HandleFunc("/api/devices", handle)
|
|
}
|
|
|
|
func GetInput(src string) (string, error) {
|
|
if medias == nil {
|
|
loadMedias()
|
|
}
|
|
|
|
input := deviceInputPrefix
|
|
|
|
var videoIdx, audioIdx int
|
|
if i := strings.IndexByte(src, '?'); i > 0 {
|
|
query, err := url.ParseQuery(src[i+1:])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
for key, value := range query {
|
|
switch key {
|
|
case "video":
|
|
videoIdx, _ = strconv.Atoi(value[0])
|
|
case "audio":
|
|
audioIdx, _ = strconv.Atoi(value[0])
|
|
case "framerate":
|
|
input += " -framerate " + value[0]
|
|
case "resolution":
|
|
input += " -video_size " + value[0]
|
|
}
|
|
}
|
|
}
|
|
|
|
input += " -i " + deviceInputSuffix(videoIdx, audioIdx)
|
|
|
|
return input, nil
|
|
}
|
|
|
|
var Bin string
|
|
var log zerolog.Logger
|
|
var medias []*streamer.Media
|
|
|
|
func findMedia(kind string, index int) *streamer.Media {
|
|
for _, media := range medias {
|
|
if media.Kind != kind {
|
|
continue
|
|
}
|
|
if index == 0 {
|
|
return media
|
|
}
|
|
index--
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func handle(w http.ResponseWriter, r *http.Request) {
|
|
if medias == nil {
|
|
loadMedias()
|
|
}
|
|
|
|
data, err := json.Marshal(medias)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("[api.ffmpeg]")
|
|
return
|
|
}
|
|
if _, err = w.Write(data); err != nil {
|
|
log.Error().Err(err).Msg("[api.ffmpeg]")
|
|
}
|
|
}
|