140 lines
4.5 KiB
PowerShell
140 lines
4.5 KiB
PowerShell
param(
|
|
[string]$BaseUrl = "http://localhost:8880",
|
|
[string]$AccountId = "douyin_company",
|
|
[int]$WorkerIntervalSeconds = 30,
|
|
[switch]$RemoteBackend
|
|
)
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
$Root = Resolve-Path (Join-Path $PSScriptRoot "..")
|
|
Set-Location $Root
|
|
|
|
function Write-Step([string]$Message) {
|
|
Write-Host ""
|
|
Write-Host "==> $Message" -ForegroundColor Cyan
|
|
}
|
|
|
|
function Require-Command([string]$Name, [string]$InstallHint) {
|
|
if (-not (Get-Command $Name -ErrorAction SilentlyContinue)) {
|
|
throw "$Name is not installed or not in PATH. $InstallHint"
|
|
}
|
|
}
|
|
|
|
function Test-BackendHealth([string]$Url) {
|
|
try {
|
|
$response = Invoke-WebRequest -UseBasicParsing -Uri "$Url/actuator/health" -TimeoutSec 3
|
|
return $response.StatusCode -ge 200 -and $response.StatusCode -lt 300
|
|
} catch {
|
|
return $false
|
|
}
|
|
}
|
|
|
|
function Find-BackendJar {
|
|
$jars = Get-ChildItem -Path (Join-Path $Root "target") -Filter "agent-foundation-*.jar" -ErrorAction SilentlyContinue |
|
|
Where-Object { $_.Name -notlike "*.original" } |
|
|
Sort-Object LastWriteTime -Descending
|
|
if ($jars.Count -eq 0) {
|
|
return $null
|
|
}
|
|
return $jars[0].FullName
|
|
}
|
|
|
|
function Test-BrowserInstalled {
|
|
$paths = @(
|
|
"$env:ProgramFiles\Microsoft\Edge\Application\msedge.exe",
|
|
"${env:ProgramFiles(x86)}\Microsoft\Edge\Application\msedge.exe",
|
|
"$env:ProgramFiles\Google\Chrome\Application\chrome.exe",
|
|
"${env:ProgramFiles(x86)}\Google\Chrome\Application\chrome.exe"
|
|
)
|
|
foreach ($path in $paths) {
|
|
if ($path -and (Test-Path $path)) {
|
|
return $true
|
|
}
|
|
}
|
|
return $false
|
|
}
|
|
|
|
function Test-LocalBaseUrl([string]$Url) {
|
|
return $Url -match "^https?://(localhost|127\.0\.0\.1)(:\d+)?/?$"
|
|
}
|
|
|
|
Write-Step "Checking local runtime"
|
|
Require-Command "node" "Install Node.js LTS from https://nodejs.org/"
|
|
Require-Command "npm.cmd" "Install Node.js LTS from https://nodejs.org/"
|
|
if (-not $RemoteBackend) {
|
|
Require-Command "java" "Install JDK 17+ or JRE 17+."
|
|
}
|
|
if (-not (Test-BrowserInstalled)) {
|
|
throw "Microsoft Edge or Google Chrome is required."
|
|
}
|
|
|
|
Write-Step "Installing Node dependencies if needed"
|
|
if (-not (Test-Path (Join-Path $Root "node_modules\playwright"))) {
|
|
& npm.cmd install
|
|
if ($LASTEXITCODE -ne 0) {
|
|
throw "npm install failed."
|
|
}
|
|
} else {
|
|
Write-Host "Node dependencies already installed."
|
|
}
|
|
|
|
if ($RemoteBackend -or -not (Test-LocalBaseUrl $BaseUrl)) {
|
|
Write-Step "Checking remote backend"
|
|
if (Test-BackendHealth $BaseUrl) {
|
|
Write-Host "Remote backend is reachable: $BaseUrl"
|
|
} else {
|
|
Write-Host "Remote backend health check did not return OK. Continuing because /actuator/health may be disabled."
|
|
Write-Host "Worker will connect to: $BaseUrl"
|
|
}
|
|
} else {
|
|
Write-Step "Starting backend service if needed"
|
|
if (Test-BackendHealth $BaseUrl) {
|
|
Write-Host "Backend is already running: $BaseUrl"
|
|
} else {
|
|
$jar = Find-BackendJar
|
|
if (-not $jar) {
|
|
throw "Backend jar not found under target\. Build the Spring Boot jar first."
|
|
}
|
|
$backendCommand = "Set-Location '$Root'; java -jar '$jar'"
|
|
Start-Process powershell.exe -ArgumentList @("-NoExit", "-ExecutionPolicy", "Bypass", "-Command", $backendCommand) -WindowStyle Normal
|
|
Write-Host "Waiting for backend health..."
|
|
$ready = $false
|
|
for ($i = 0; $i -lt 40; $i++) {
|
|
Start-Sleep -Seconds 2
|
|
if (Test-BackendHealth $BaseUrl) {
|
|
$ready = $true
|
|
break
|
|
}
|
|
}
|
|
if (-not $ready) {
|
|
throw "Backend did not become healthy at $BaseUrl. Check the backend window."
|
|
}
|
|
}
|
|
}
|
|
|
|
Write-Step "Checking Douyin login"
|
|
$env:MEDIA_BASE_URL = $BaseUrl
|
|
$env:DOUYIN_ACCOUNT_ID = $AccountId
|
|
$checkOutput = & npm.cmd run douyin:check -- $AccountId 2>&1
|
|
if ($LASTEXITCODE -ne 0) {
|
|
Write-Host $checkOutput
|
|
Write-Host ""
|
|
Write-Host "Douyin login is required. A browser will open. Login, then press Enter in that window."
|
|
& npm.cmd run douyin:login -- $AccountId
|
|
if ($LASTEXITCODE -ne 0) {
|
|
throw "Douyin login failed."
|
|
}
|
|
} else {
|
|
Write-Host "Douyin login profile is ready."
|
|
}
|
|
|
|
Write-Step "Starting Douyin worker loop"
|
|
$workerCommand = "Set-Location '$Root'; `$env:MEDIA_BASE_URL='$BaseUrl'; `$env:DOUYIN_ACCOUNT_ID='$AccountId'; `$env:MEDIA_WORKER_INTERVAL_MS='$($WorkerIntervalSeconds * 1000)'; npm.cmd run douyin:worker:loop"
|
|
Start-Process powershell.exe -ArgumentList @("-NoExit", "-ExecutionPolicy", "Bypass", "-Command", $workerCommand) -WindowStyle Normal
|
|
|
|
Write-Host ""
|
|
Write-Host "Local client is ready."
|
|
Write-Host "Backend: $BaseUrl"
|
|
Write-Host "Account: $AccountId"
|
|
Write-Host "Worker polling interval: $WorkerIntervalSeconds seconds"
|