$ErrorActionPreference = "Stop"

function Test-IsAdmin {
    $currentIdentity = [Security.Principal.WindowsIdentity]::GetCurrent()
    $principal = New-Object Security.Principal.WindowsPrincipal($currentIdentity)
    return $principal.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
}

if (-not (Test-IsAdmin)) {
    Start-Process -FilePath "powershell.exe" -Verb RunAs -ArgumentList @(
        "-NoProfile",
        "-ExecutionPolicy", "Bypass",
        "-File", "`"$PSCommandPath`""
    )
    exit 0
}

$projectDir = "/home/hyein/project"
$apiPort = 4000
$localUrl = "http://localhost:$apiPort/PTC/"
$preferredLanIp = "172.16.40.36"
$defaultSource = "/home/hyein/project/PTC(2023-2026.02).xlsx"
$sourceConfigPath = "/home/hyein/project/server/ptc_source_path.txt"

function Invoke-WslBash([string]$command) {
    $output = & wsl.exe bash -lc $command 2>&1
    $exitCode = $LASTEXITCODE
    return [PSCustomObject]@{
        Output = @($output)
        ExitCode = $exitCode
    }
}

$currentSourceResult = Invoke-WslBash "cat '$sourceConfigPath' 2>/dev/null || printf '%s\n' '$defaultSource'"
$currentSource = ($currentSourceResult.Output | Select-Object -First 1).Trim()
if ([string]::IsNullOrWhiteSpace($currentSource)) {
    $currentSource = $defaultSource
}

Write-Host "Starting PTC server..."
Write-Host "Source file: $currentSource"

$startResult = Invoke-WslBash "pkill -f '/home/hyein/project/server/ptc_api_server.py' >/dev/null 2>&1 || true; nohup python3 /home/hyein/project/server/ptc_api_server.py >/tmp/ptc_api.log 2>&1 & sleep 3"
if ($startResult.ExitCode -ne 0) {
    Write-Host "Failed to start the server in WSL." -ForegroundColor Red
    Read-Host "Press Enter to exit"
    exit 1
}

$healthResult = Invoke-WslBash "curl -fsS http://127.0.0.1:$apiPort/api/health >/tmp/ptc_api_health.json && curl -fsSI http://127.0.0.1:$apiPort/PTC/ >/tmp/ptc_web_health.txt"
if ($healthResult.ExitCode -ne 0) {
    Write-Host "Local server check failed. Recent server log:" -ForegroundColor Red
    $logResult = Invoke-WslBash "tail -n 80 /tmp/ptc_api.log"
    $logResult.Output | ForEach-Object { Write-Host $_ }
    Read-Host "Press Enter to exit"
    exit 1
}

Start-Process $localUrl
Write-Host ""
Write-Host "Local URL: $localUrl"

$lanIp = Get-NetIPAddress -AddressFamily IPv4 |
    Where-Object {
        $_.IPAddress -notlike '127.*' -and
        $_.IPAddress -notlike '169.254.*' -and
        $_.PrefixOrigin -ne 'WellKnown'
    } |
    Select-Object -ExpandProperty IPAddress -First 1

$wslIpResult = Invoke-WslBash "hostname -I | awk '{print \$1}'"
$wslIp = ($wslIpResult.Output | Select-Object -First 1).Trim()

if (-not [string]::IsNullOrWhiteSpace($lanIp) -and -not [string]::IsNullOrWhiteSpace($wslIp)) {
    Write-Host "Configuring LAN sharing..."
    & netsh interface portproxy delete v4tov4 listenaddress=0.0.0.0 listenport=$apiPort | Out-Null
    & netsh interface portproxy add v4tov4 listenaddress=0.0.0.0 listenport=$apiPort connectaddress=$wslIp connectport=$apiPort | Out-Null
    & netsh advfirewall firewall delete rule name="PTC 4000" | Out-Null
    & netsh advfirewall firewall add rule name="PTC 4000" dir=in action=allow protocol=TCP localport=$apiPort | Out-Null

    $shareIp = if ([string]::IsNullOrWhiteSpace($preferredLanIp)) { $lanIp } else { $preferredLanIp }
    $shareUrl = "http://$shareIp:$apiPort/PTC/"
    Write-Host "LAN URL: $shareUrl"
    Set-Clipboard -Value $shareUrl
    Write-Host "The share URL has been copied to the clipboard."
}
else {
    Write-Host "LAN sharing was skipped because Windows IP or WSL IP could not be detected."
}

Read-Host "Press Enter to close"