package webrtc

import (
	"github.com/AlexxIT/go2rtc/cmd/api"
	"github.com/AlexxIT/go2rtc/pkg/webrtc"
	"github.com/pion/sdp/v3"
	"strconv"
	"strings"
)

type Address struct {
	Host string
	Port int
}

var addresses []Address

func AddCandidate(address string) {
	var port int

	// try to get port from address string
	if i := strings.LastIndexByte(address, ':'); i > 0 {
		if v, _ := strconv.Atoi(address[i+1:]); v != 0 {
			address = address[:i]
			port = v
		}
	}

	// use default WebRTC port
	if port == 0 {
		port, _ = strconv.Atoi(Port)
	}

	addresses = append(addresses, Address{Host: address, Port: port})
}

func GetCandidates() (candidates []string) {
	for _, address := range addresses {
		// using stun server for receive public IP-address
		if address.Host == "stun" {
			ip, err := webrtc.GetCachedPublicIP()
			if err != nil {
				continue
			}
			// this is a copy, original host unchanged
			address.Host = ip.String()
		}

		candidates = append(
			candidates,
			webrtc.CandidateHostUDP(address.Host, address.Port),
			webrtc.CandidateHostTCPPassive(address.Host, address.Port),
		)
	}

	return
}

func asyncCandidates(tr *api.Transport) {
	for _, candidate := range GetCandidates() {
		log.Trace().Str("candidate", candidate).Msg("[webrtc] config")
		tr.Write(&api.Message{Type: "webrtc/candidate", Value: candidate})
	}
}

func syncCanditates(answer string) (string, error) {
	if len(addresses) == 0 {
		return answer, nil
	}

	sd := &sdp.SessionDescription{}
	if err := sd.Unmarshal([]byte(answer)); err != nil {
		return "", err
	}

	md := sd.MediaDescriptions[0]

	_, end := md.Attribute("end-of-candidates")
	if end {
		md.Attributes = md.Attributes[:len(md.Attributes)-1]
	}

	for _, candidate := range GetCandidates() {
		md.WithPropertyAttribute(candidate)
	}

	if end {
		md.WithPropertyAttribute("end-of-candidates")
	}

	data, err := sd.Marshal()
	if err != nil {
		return "", err
	}

	return string(data), nil
}

func candidateHandler(tr *api.Transport, msg *api.Message) error {
	if tr.Consumer == nil {
		return nil
	}
	if conn := tr.Consumer.(*webrtc.Server); conn != nil {
		candidate := msg.String()
		log.Trace().Str("candidate", candidate).Msg("[webrtc] remote")
		conn.AddCandidate(candidate)
	}
	return nil
}