Alexey Khit
92 lines
1.6 KiB
Go
92 lines
1.6 KiB
Go
package device
|
|
|
|
import (
|
|
"github.com/AlexxIT/go2rtc/cmd/api"
|
|
"github.com/AlexxIT/go2rtc/cmd/app"
|
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
|
"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 []*core.Media
|
|
|
|
func findMedia(kind string, index int) *core.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()
|
|
}
|
|
|
|
var items []api.Stream
|
|
var iv, ia int
|
|
|
|
for _, media := range medias {
|
|
var source string
|
|
switch media.Kind {
|
|
case core.KindVideo:
|
|
source = "ffmpeg:device?video=" + strconv.Itoa(iv)
|
|
iv++
|
|
case core.KindAudio:
|
|
source = "ffmpeg:device?audio=" + strconv.Itoa(ia)
|
|
ia++
|
|
}
|
|
items = append(items, api.Stream{Name: media.ID, URL: source})
|
|
}
|
|
|
|
api.ResponseStreams(w, items)
|
|
}
|