Kelly Thomas Reardon
88 lines
2.8 KiB
PowerShell
88 lines
2.8 KiB
PowerShell
# 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
|
|
[string[]] $excludedTrackers = $ENV:EXCLUDEDTRACKERS -split ','
|
|
|
|
# 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]$name,
|
|
[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 $name with hash $hash"
|
|
}
|
|
$json = $body | ConvertTo-Json
|
|
Invoke-WebRequest -URI $notificationWebhookURL -Method post -Body $json -Headers $headers
|
|
}
|
|
}
|
|
|
|
# Main script
|
|
$session = Get-QBittorrentSession
|
|
while ($true)
|
|
{
|
|
$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 "Delete: $($torrent.name) $($toreent.hash)"
|
|
Remove-Torrent -session $session -hash $torrent.hash -name $torrent.name
|
|
}
|
|
}
|
|
}
|
|
|
|
Write-Host "Sleeping for $schedule minutes..."
|
|
Start-Sleep ($schedule * 60)
|
|
}
|