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)
}