#!/usr/bin/env pwsh
# -----------------------------------------------------------------------------------
# Linux-safe vCenter + Posh-ACME Script (Production Hardened)
# - Uses HttpClient with AutomaticDecompression disabled
# - Fault-tolerant ACME certificate request and upload
# - Handles PowerDNS plugin, vCenter session, certificate apply, service restart
# -----------------------------------------------------------------------------------

. /opt/idssys/nodemgmt/conf/powerwall/settings.ps1

# ----------------------------
# Global variables for troubleshooting
# ----------------------------
$global:helpme = $null
$global:responseBody = $null

# ----------------------------
# Error handler (robust across platforms)
# ----------------------------
function Show-Failure {
    param([System.Management.Automation.ErrorRecord]$ErrorRecord)

    $global:responseBody = "<No response body available>"

    try {
        $ex = $ErrorRecord.Exception

        if ($ex.Response -is [System.Net.Http.HttpResponseMessage]) {
            $resp = $ex.Response
            try {
                $global:responseBody = $resp.Content.ReadAsStringAsync().GetAwaiter().GetResult()
            }
            catch {
                $global:responseBody = "<Failed to read HTTP content: $($_.Exception.Message)>"
            }
        }
        elseif ($ex.Response -is [System.Net.WebResponse]) {
            try {
                $stream = $ex.Response.GetResponseStream()
                if ($stream) {
                    $reader = [System.IO.StreamReader]::new($stream)
                    $global:responseBody = $reader.ReadToEnd()
                }
                else {
                    $global:responseBody = "<WebResponse had no stream>"
                }
            }
            catch {
                $global:responseBody = "<Failed to read WebResponse: $($_.Exception.Message)>"
            }
        }
        else {
            $global:responseBody = $ex.Message
        }
    }
    catch {
        $global:responseBody = "<Error extracting failure detail: $($_.Exception.Message)>"
    }

    $global:helpme = $global:responseBody

    Write-Host "----------------------------------------" -ForegroundColor Red
    Write-Host "Status: A system exception was caught." -ForegroundColor Red
    Write-Host $global:responseBody -ForegroundColor Red
    Write-Host "The request/response body has been saved to `$global:helpme" -ForegroundColor Red
    Write-Host "----------------------------------------" -ForegroundColor Red
}

# ----------------------------
# HttpClient wrapper (TLS1.2, skip cert check, no decompression)
# ----------------------------
function Invoke-SafeRestMethod {
    param(
        [Parameter(Mandatory=$true)][string]$Uri,
        [string]$Method = 'GET',
        [hashtable]$Headers = @{},
        $Body = $null,
        [switch]$AsJson,
        [int]$TimeoutSec = 60
    )

    try {
        $handler = [System.Net.Http.HttpClientHandler]::new()
        $handler.AutomaticDecompression = [System.Net.DecompressionMethods]::None
        $handler.ServerCertificateCustomValidationCallback = { $true }

        $client = [System.Net.Http.HttpClient]::new($handler)
        $client.Timeout = [System.TimeSpan]::FromSeconds($TimeoutSec)

        foreach ($k in $Headers.Keys) {
            $client.DefaultRequestHeaders.Remove($k) | Out-Null
            $client.DefaultRequestHeaders.Add($k, $Headers[$k])
        }

        if ($Body -ne $null) {
            if ($AsJson) {
                $jsonBody = $Body | ConvertTo-Json -Depth 12 -Compress
                $content = [System.Net.Http.StringContent]::new($jsonBody, [System.Text.Encoding]::UTF8, 'application/json')
            }
            else {
                $content = [System.Net.Http.StringContent]::new($Body)
            }
        }
        else { $content = $null }

        $method = [System.Net.Http.HttpMethod]::$Method
        $request = [System.Net.Http.HttpRequestMessage]::new($method, $Uri)
        if ($content) { $request.Content = $content }

        $response = $client.SendAsync($request).GetAwaiter().GetResult()
        $respBody = $null
        if ($response.Content -ne $null) {
            $respBody = $response.Content.ReadAsStringAsync().GetAwaiter().GetResult()
        }

        if ($response.IsSuccessStatusCode) {
            if ($respBody -and $respBody.Trim().Length -gt 0) {
                try { return $respBody | ConvertFrom-Json }
                catch { return $respBody }
            }
            else { return $respBody }
        }
        else {
            throw [System.Net.Http.HttpRequestException]::new("HTTP $($response.StatusCode): $($response.ReasonPhrase)", $null, $response)
        }
    }
    catch {
        Show-Failure -ErrorRecord $_
        return $null
    }
    finally {
        $client.Dispose()
        $handler.Dispose()
    }
}

# ----------------------------
# Variables
# ----------------------------
$vCenterURL   = $VCENTERHOST
$CommonName   = $VCENTERHOST
$EmailContact = $ACMEEMAIL

[PSCredential]$Credential = New-Object System.Management.Automation.PSCredential -ArgumentList $VCENTERUSER, (ConvertTo-SecureString $VCENTERPASS -AsPlainText -Force)

$pArgs = @{
    PowerDNSApiHost     = $WDNSHOST
    PowerDNSApiKey      = $PDNSAPI
    PowerDNSUseTLS      = $true
    PowerDNSPort        = 443
    PowerDNSServerName  = 'localhost'
}

# ----------------------------
# Ensure Posh-ACME Module
# ----------------------------
if (-not (Get-Module -ListAvailable -Name Posh-ACME)) {
    Install-Module -Name Posh-ACME -Force -Confirm:$false -Scope AllUsers
}
Import-Module Posh-ACME -ErrorAction Stop

# ----------------------------
# vCenter API Session
# ----------------------------
$loginUri = "https://$vCenterURL/rest/com/vmware/cis/session"
Write-Host "Connecting to vCenter at $vCenterURL ..." -ForegroundColor Cyan

try {
    $sessionResponse = Invoke-SafeRestMethod -Uri $loginUri -Method Post -Headers @{}
    $sessionToken = $sessionResponse.value
    if (-not $sessionToken) {
        # fallback: session token may be in header
        $httpClient = [System.Net.Http.HttpClient]::new()
        $req = [System.Net.Http.HttpRequestMessage]::new([System.Net.Http.HttpMethod]::Post, $loginUri)
        $resp = $httpClient.SendAsync($req).GetAwaiter().GetResult()
        if ($resp.Headers.Contains("vmware-api-session-id")) {
            $sessionToken = $resp.Headers.GetValues("vmware-api-session-id") | Select-Object -First 1
        }
        $httpClient.Dispose()
    }
}
catch {
    Show-Failure -ErrorRecord $_
}

if (-not $sessionToken) { Show-Failure -ErrorRecord ([pscustomobject]@{ Exception = [System.Exception] "Unable to obtain vCenter session token" }) }
Write-Host "Connected to vCenter API. Session established." -ForegroundColor Green

$headers = @{ 'vmware-api-session-id' = $sessionToken }

# ----------------------------
# Retrieve VM List (optional)
# ----------------------------
try {
    $vmList = Invoke-SafeRestMethod -Uri "https://$vCenterURL/rest/vcenter/vm" -Headers $headers
    if ($vmList.value) {
        Write-Host "Retrieved VM list:" -ForegroundColor Cyan
        $vmList.value | ForEach-Object { Write-Host " - $($_.name)" }
    }
}
catch {
    Write-Host "Unable to retrieve VM list, continuing..." -ForegroundColor Yellow
}

# ----------------------------
# Fault-tolerant ACME certificate request
# ----------------------------
$certName = "vcenter-cert"
$certSuccess = $false
try {
    Write-Host "Requesting certificate via Posh-ACME..." -ForegroundColor Cyan
    New-PACertificate -Domain $CommonName -DnsPlugin PowerDNS -PluginArgs $pArgs -Contact $EmailContact -AcceptTOS -Verbose -Force
    $certSuccess = $true
}
catch {
    Write-Host "ACME certificate request failed: $($_.Exception.Message)" -ForegroundColor Yellow
    $global:helpme = $_.Exception.Message
}

# ----------------------------
# Collect certificate paths
# ----------------------------
$paAccount = Get-PAAccount
$certFolder = $paAccount.CertFolder
$certPath  = Join-Path $certFolder "$certName\cert.pem"
$keyPath   = Join-Path $certFolder "$certName\privkey.pem"
$chainPath = Join-Path $certFolder "$certName\chain.pem"

foreach ($f in @($certPath, $keyPath, $chainPath)) {
    if (-not (Test-Path $f)) {
        Write-Host "Certificate file missing: $f" -ForegroundColor Yellow
        $certSuccess = $false
    }
}

# ----------------------------
# Upload certificate to vCenter (if ACME succeeded)
# ----------------------------
if ($certSuccess) {
    try {
        Write-Host "Uploading certificate to vCenter..." -ForegroundColor Cyan
        $body = @{
            cert  = Get-Content -Path $certPath -Raw
            key   = Get-Content -Path $keyPath -Raw
            chain = Get-Content -Path $chainPath -Raw
        }
        Invoke-SafeRestMethod -Uri "https://$vCenterURL/rest/vcenter/certificate-management/vcenter/tls" -Method Post -Headers $headers -Body $body -AsJson
        Write-Host "Certificate uploaded successfully." -ForegroundColor Green
    }
    catch {
        Write-Host "Certificate upload failed: $($_.Exception.Message)" -ForegroundColor Yellow
        $global:helpme = $_.Exception.Message
    }

    # Apply certificate
    try {
        Invoke-SafeRestMethod -Uri "https://$vCenterURL/rest/vcenter/certificate-management/vcenter/tls?action=apply" -Method Post -Headers $headers
        Write-Host "TLS certificate applied." -ForegroundColor Green
    }
    catch {
        Write-Host "Failed to apply certificate: $($_.Exception.Message)" -ForegroundColor Yellow
        $global:helpme = $_.Exception.Message
    }
}
else {
    Write-Host "Skipping certificate upload/apply due to previous errors." -ForegroundColor Yellow
}

# ----------------------------
# Restart vCenter vpxd service
# ----------------------------
try {
    Invoke-SafeRestMethod -Uri "https://$vCenterURL/rest/appliance/system/services/vpxd?action=restart" -Method Post -Headers $headers
    Write-Host "vCenter vpxd service restart requested." -ForegroundColor Yellow
}
catch {
    Write-Host "Failed to restart vpxd service: $($_.Exception.Message)" -ForegroundColor Yellow
    $global:helpme = $_.Exception.Message
}

Write-Host "Script completed. Check `$global:helpme for any error details." -ForegroundColor Green