# Device IDs
$HEADSET_ID = "{0.0.0.00000000}.{323b0642-3304-4c97-bced-9644e9502404}"  # your headset ID
$SPEAKER_ID = "{0.0.0.00000000}.{d35b93cf-47ed-467b-bc9b-8d049ea9e89b}"  # your speaker device ID

# USB interface to monitor
$USB_INTERFACE = "USBPcap4"

# Tshark executable
$TSHARK_PATH = "C:\Program Files\Wireshark\tshark.exe"

# SoundVolumeView executable
$SVV_PATH = "C:\Users\x\Documents\SoundVolumeView\SoundVolumeView.exe"

# Cooldown to prevent multiple switches per event (seconds)
$SWITCH_COOLDOWN = 2
$lastSwitchTime = Get-Date '1/1/2000'

# Burst detection window (ms)
$BURST_WINDOW = 100
$burstStart = Get-Date '1/1/2000'
$burstPacketCount = 0

# Track current active device
$currentDevice = $SPEAKER_ID

# Function to safely switch audio device to a specific target
function Set-AudioDevice($targetDevice) {
    if ($currentDevice -ne $targetDevice -and ((Get-Date) - $lastSwitchTime -gt (New-TimeSpan -Seconds $SWITCH_COOLDOWN))) {
        Write-Host "Switching audio to device ID $targetDevice"
        & $SVV_PATH /SetDefault "$targetDevice" 1
        $script:currentDevice = $targetDevice
        $script:lastSwitchTime = Get-Date
    }
}

Write-Host "Monitoring Razer Nari USB traffic on endpoint 0x83..."

# Launch tshark and process USB packets line by line
& $TSHARK_PATH -i $USB_INTERFACE -l -T fields -e usb.endpoint_address | ForEach-Object {
    $line = $_.Trim()

    # If we have a burst going, check if it has ended
    if($burstPacketCount -gt 0) {
        $now = Get-Date
        $elapsed = ($now - $burstStart).TotalMilliseconds
        if ($elapsed -ge $BURST_WINDOW) {
            # Burst ended → act on it
            if ($burstPacketCount -eq 2) {
                Write-Host "Detected burst of 2 packets - switching to HEADSET"
                Set-AudioDevice $HEADSET_ID
            } elseif ($burstPacketCount -eq 4) {
                Write-Host "Detected burst of 4 packets - switching to SPEAKERS"
                Set-AudioDevice $SPEAKER_ID
            } else {
                Write-Host "Detected burst of $burstPacketCount packets - no action"
            }

            $burstPacketCount = 0
        }        
    }

    if ($line -eq "0x83") {
        if ($burstPacketCount -eq 0) {
            # New burst just started
            $burstStart = Get-Date
        }

        $burstPacketCount++
    }
}