# Configuration
[int] $schedule = $ENV:SCHEDULE
[string] $qbtHost = $ENV:QBTHOST
[string] $qbtPort = $ENV:QBTPORT
[string] $qbtUser = $ENV:QBTUSER
[string] $qbtPassword = $ENV:QBTPASSWORD
[string] $notificationWebhookURL = $ENV:NOTIFICATIONWEBHOOKURL

# Function to authenticate and get a session
function Get-QBittorrentSession {
    $qbtHeaders = @{
        Referer = "https://$($qbtHost):$($qbtPort)"
    }
    $qbtBody = @{
        username = $qbtUser
        password = $qbtPassword
    }
    $qbtSession = [Microsoft.PowerShell.Commands.WebRequestSession]::new()
    $qbtResponse = Invoke-WebRequest -Uri "https://$($qbtHost):$($qbtPort)/api/v2/auth/login" -WebSession $qbtSession -Method POST -Headers $qbtHeaders -Body $qbtBody
    
    return $qbtSession
}

# Function to get all torrents
function Get-Torrents {
    param (
        [Microsoft.PowerShell.Commands.WebRequestSession]$session
    )
    $qbtResponse = Invoke-WebRequest -Uri "https://$($qbtHost):$($qbtPort)/api/v2/torrents/info" -WebSession $session -Method GET
    return $qbtResponse.Content | ConvertFrom-Json
}

# Function to delete a torrent
function Remove-Torrent {
    param (
        [Microsoft.PowerShell.Commands.WebRequestSession]$session,
        [string]$hash
    )
    $qbtHeaders = @{
        Referer = "https://$($qbtHost):$($qbtPort)"
    }
    $qbtBody = @{
        hashes = $hash
        deleteFiles = 'true'
    }
    $qbtResponse = Invoke-WebRequest -Uri "https://$($qbtHost):$($qbtPort)/api/v2/torrents/delete" -WebSession $session -Method POST -Headers $qbtHeaders -Body $qbtBody

    if ($notificationWebhookURL)
    {
        $headers = @{
            "Content-Type"="application/json"
        }
        $body = @{
            content="Removed torrent with hash $hash"
        }
        $json = $body | ConvertTo-Json
        Invoke-WebRequest -URI $notificationWebhookURL -Method post -Body $json -Headers $headers
    }
}

while ($true)
{
    # Main script
    $session = Get-QBittorrentSession
    $torrents = Get-Torrents -session $session

    foreach ($torrent in $torrents) {
        if ($torrent.state -eq 'uploading' -or $torrent.state -like "*UP") {
            # Check if any of the excluded trackers are in the torrent's tracker list
            if ($torrent.tracker -match "https?://([^/]+)") {
                $trackers = $matches[1]
            } else {
                $trackers = @()
            }
            if (-not ($trackers | Where-Object { $excludedTrackers -contains $_ })) {
                # Delete the torrent
                Write-Host "Would delete: $($torrent.name)"
                Remove-Torrent -session $session -hash $torrent.hash
            }
        }
    }

    # Logout (optional, but good practice)
    # $session.UploadString("$qbittorrentUrl/api/v2/auth/logout", "")
    
    Write-Host "Sleeping for $schedule minutes..."
    Start-Sleep ($schedule * 60)
}