package webrtc

import (
	"strings"
	"testing"

	appcfg "github.com/datarhei/core/v16/restream/app"
)

// TestSplitRTPLegs_TwoLegs feeds the real BuildArgs output through
// the splitter and checks both legs come out with the correct shape.
func TestSplitRTPLegs_TwoLegs(t *testing.T) {
	args := BuildArgs(appcfg.ConfigWebRTC{Enabled: true, VideoPT: 102, AudioPT: 111}, 49200)

	legs := splitRTPLegs(args)
	if len(legs) != 2 {
		t.Fatalf("expected 2 legs, got %d: %+v", len(legs), legs)
	}

	video := legs[0]
	audio := legs[1]

	// Leg 0 is video: address ends with :49200
	if !strings.HasSuffix(video.Address, ":49200?pkt_size=1316") {
		t.Fatalf("video Address unexpected: %q", video.Address)
	}
	// Leg 1 is audio: address ends with :49201
	if !strings.HasSuffix(audio.Address, ":49201?pkt_size=1316") {
		t.Fatalf("audio Address unexpected: %q", audio.Address)
	}

	// Each leg's options start with -map, end with -f rtp.
	if len(video.Options) < 2 || video.Options[0] != "-map" {
		t.Fatalf("video leg should start with -map, got %v", video.Options)
	}
	if video.Options[len(video.Options)-2] != "-f" || video.Options[len(video.Options)-1] != "rtp" {
		t.Fatalf("video leg should end with -f rtp, got %v", video.Options)
	}
	if len(audio.Options) < 2 || audio.Options[0] != "-map" {
		t.Fatalf("audio leg should start with -map, got %v", audio.Options)
	}

	// Neither leg's Options should contain the address itself.
	for _, opt := range video.Options {
		if strings.HasPrefix(opt, "udp://") {
			t.Fatalf("video Options must not contain udp:// address: %v", video.Options)
		}
	}
}

// TestSplitRTPLegs_FallbackOnUnexpectedShape ensures we don't panic
// or drop data if BuildArgs ever changes shape — the splitter returns
// a single leg wrapping everything.
func TestSplitRTPLegs_FallbackOnUnexpectedShape(t *testing.T) {
	// Single -map: shouldn't happen, but don't panic.
	legs := splitRTPLegs([]string{"-map", "0:v:0", "udp://1.2.3.4:5000"})
	if len(legs) != 1 {
		t.Fatalf("expected single fallback leg, got %d", len(legs))
	}
}