#!/usr/bin/env pwsh
# -----------------------------------------------------------------------------------
# vCenter SSL Automation (with Posh-ACME 4.30 and XDG path support)
# Fully idempotent, rate-limit safe, uses existing cert if valid >30 days
# Auto-discovers cert files under ~/.config/Posh-ACME
# -----------------------------------------------------------------------------------

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

# ----------------------------
# LOGGING
# ----------------------------
$LogFile = "/opt/idssys/nodemgmt/logs/vc-ssl.log"
$logDir  = Split-Path -Path $LogFile -Parent
if (-not (Test-Path $logDir)) { New-Item -Path $logDir -ItemType Directory -Force | Out-Null }

function Write-Log {
    param([string]$Level,[string]$Message,[string]$Color="White")
    $ts = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
    $line = "[{0}] {1}: {2}" -f $ts,$Level.ToUpper(),$Message
    Write-Host $line -ForegroundColor $Color
    Add-Content -Path $LogFile -Value $line
}

function Show-Banner {
    param([string]$Text,[string]$Type="INFO")
    switch ($Type.ToUpper()) {
        "SUCCESS" { $color="Green";  $level="SUCCESS" }
        "ERROR"   { $color="Red";    $level="ERROR" }
        "WARN"    { $color="Yellow"; $level="WARN" }
        default   { $color="Cyan";   $level="INFO" }
    }
    $line = "=" * [math]::Max($Text.Length,40)
    Write-Host $line -ForegroundColor $color
    Write-Host "${level}: $Text" -ForegroundColor $color
    Write-Host $line -ForegroundColor $color
    Add-Content -Path $LogFile -Value ("[{0}] {1}: {2}" -f (Get-Date),$level,$Text)
}

# ----------------------------
# ERROR HANDLING
# ----------------------------
$global:helpme = $null
function Show-Failure {
    param([System.Management.Automation.ErrorRecord]$ErrorRecord)
    $global:helpme = $ErrorRecord.Exception.Message
    Show-Banner -Text $ErrorRecord.Exception.Message -Type "ERROR"
    Write-Log -Level "ERROR" -Message $ErrorRecord | Out-Null
    exit 1
}

# ----------------------------
# CERT FINGERPRINT
# ----------------------------
function Get-CertFingerprintFromPem {
    param([string]$PemString)
    try {
        $pemBody = $PemString -replace "`r","" -split "`n" |
            Where-Object {$_ -notmatch "^-----" -and $_ -ne ""}
        $bytes = [Convert]::FromBase64String(($pemBody -join ""))
        $cert = [System.Security.Cryptography.X509Certificates.X509Certificate2]::new($bytes)
        $sha256 = [System.Security.Cryptography.SHA256]::Create()
        $hash = $sha256.ComputeHash($cert.RawData)
        return ($hash | ForEach-Object { $_.ToString("X2") }) -join ""
    }
    catch {
        Write-Log WARN "Fingerprint calculation error: $($_.Exception.Message)" Yellow
        return $null
    }
}

# ----------------------------
# Posh-ACME CERT FOLDER DISCOVERY (XDG FIX)
# ----------------------------
function Find-LocalPACertFolder {
    param([string]$HostName)

    # Correct Linux path for Posh-ACME 4.x
    $paRoot = Join-Path $HOME ".config/Posh-ACME"

    if (-not (Test-Path $paRoot)) {
        Write-Log WARN "Posh-ACME config folder not found: $paRoot" Yellow
        return $null
    }

    # Look for folders named exactly after the hostname
    $folders = Get-ChildItem -Path $paRoot -Recurse -Directory -ErrorAction SilentlyContinue |
               Where-Object { $_.Name -eq $HostName }

    if (-not $folders) {
        Write-Log WARN "No hostname-matched folders found under $paRoot" Yellow
        return $null
    }

    $folder = $folders | Select-Object -First 1
    Write-Log INFO "Using Posh-ACME cert folder: $($folder.FullName)" Gray
    return $folder.FullName
}

# ----------------------------
# LOAD POWERCLI
# ----------------------------
if (-not (Get-Module -ListAvailable VMware.PowerCLI)) {
    Install-Module VMware.PowerCLI -Force -Scope AllUsers
}
Import-Module VMware.PowerCLI -ErrorAction Stop
Set-PowerCLIConfiguration -InvalidCertificateAction Ignore -Confirm:$false | Out-Null

# ----------------------------
# CONNECT TO VCENTER
# ----------------------------
try {
    Write-Log INFO "Connecting to vCenter $VCENTERHOST..." Cyan
    $vCenterConn = Connect-VIServer -Server $VCENTERHOST -User $VCENTERUSER -Password $VCENTERPASS -Force
    Show-Banner "Connected to vCenter." SUCCESS
} catch { Show-Failure $_ }

# ----------------------------
# LOAD POSH-ACME
# ----------------------------
if (-not (Get-Module -ListAvailable Posh-ACME)) {
    Install-Module Posh-ACME -Force -Scope AllUsers
}
Import-Module Posh-ACME -ErrorAction Stop

# ----------------------------
# ACME CERTIFICATE LOGIC
# ----------------------------
$certSuccess = $false
$certPath = $keyPath = $chainPath = $null

$allPACerts = Get-PACertificate -ErrorAction SilentlyContinue
if ($allPACerts) {
    Write-Log INFO "Found $($allPACerts.Count) Posh-ACME cert(s)." Gray
    foreach ($c in $allPACerts) {
        Write-Log INFO "  MainDomain=$($c.MainDomain) SANs=$($c.AllSANs -join ',') Exp=$($c.NotAfter)" Gray
    }
}

# Match certificate for vCenter
$existingPACert = $allPACerts |
    Where-Object { $_.MainDomain -eq $VCENTERHOST -or ($_.AllSANs -contains $VCENTERHOST) } |
    Sort-Object NotAfter -Descending |
    Select-Object -First 1

$renewCert = $true
$skipReason = ""

# RULE: If existing cert valid >30 days → skip ACME
if ($existingPACert) {
    $daysLeft = ($existingPACert.NotAfter - (Get-Date)).TotalDays
    Write-Log INFO "Existing cert expires $($existingPACert.NotAfter) (~$([math]::Round($daysLeft)) days)" Gray
    if ($daysLeft -gt 30) {
        $renewCert = $false
        $skipReason = "Existing certificate valid >30 days."
    }
}

# RULE: Rate-limit safety
if ($renewCert -and $existingPACert) {
    $hoursSince = ((Get-Date) - $existingPACert.Created).TotalHours
    if ($hoursSince -lt 168) {
        $renewCert = $false
        $skipReason = "Rate-limit: cert issued $([math]::Round($hoursSince)) hours ago."
    }
}

# -------------------------------------------------------
# USE EXISTING CERT FILES (XDG-FIXED)
# -------------------------------------------------------
if (-not $renewCert -and $existingPACert) {
    Write-Log INFO "Skipping ACME issuance: $skipReason" Yellow

    $certFolder = Find-LocalPACertFolder -HostName $VCENTERHOST
    if (-not $certFolder) {
        Show-Banner "Could not locate certificate files on disk. Aborting." ERROR
        exit 1
    }

    $certPath  = Join-Path $certFolder "cert.pem"
    $keyPath   = Join-Path $certFolder "privkey.pem"
    $chainPath = Join-Path $certFolder "chain.pem"

    $certSuccess = $true
}

# -------------------------------------------------------
# REQUEST NEW ACME CERT IF NEEDED
# -------------------------------------------------------
if ($renewCert) {
    Write-Log INFO "Requesting new ACME certificate..." Cyan

    $securePDNSAPI = if ($PDNSAPI -is [string]) {
        ConvertTo-SecureString $PDNSAPI -AsPlainText -Force
    } else { $PDNSAPI }

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

    try {
        New-PACertificate -Domain $VCENTERHOST `
            -DnsPlugin PowerDNS `
            -PluginArgs $pArgs `
            -Contact $ACMEEMAIL `
            -AcceptTOS `
            -DnsSleep 15 `
            -Force `
            -Verbose

        # Find new cert folder via XDG
        $certFolder = Find-LocalPACertFolder -HostName $VCENTERHOST
        if (-not $certFolder) { Show-Banner "ACME succeeded but no files found." ERROR; exit 1 }

        $certPath  = Join-Path $certFolder "cert.pem"
        $keyPath   = Join-Path $certFolder "privkey.pem"
        $chainPath = Join-Path $certFolder "chain.pem"
        $certSuccess = $true
        Show-Banner "New ACME certificate created." SUCCESS
    }
    catch {
        $msg = $_.Exception.Message
        Write-Log ERROR "ACME failed: $msg" Red
        $global:helpme = $msg

        if ($msg -like "*too many certificates*") {
            Show-Banner "Rate-limit hit! Falling back to existing cert." WARN

            if ($existingPACert) {
                $certFolder = Find-LocalPACertFolder -HostName $VCENTERHOST
                if ($certFolder) {
                    $certPath  = Join-Path $certFolder "cert.pem"
                    $keyPath   = Join-Path $certFolder "privkey.pem"
                    $chainPath = Join-Path $certFolder "chain.pem"
                    $certSuccess = $true
                } else {
                    Show-Banner "Cannot locate fallback cert files. Aborting." ERROR
                    exit 1
                }
            }
            else { Show-Banner "No fallback cert exists." ERROR; exit 1 }
        }
        else { exit 1 }
    }
}

# -------------------------------------------------------
# VERIFY CERT FILES EXIST
# -------------------------------------------------------
foreach ($f in @($certPath,$keyPath,$chainPath)) {
    if (-not (Test-Path $f)) {
        Write-Log ERROR "Missing certificate file: $f" Red
        exit 1
    }
}

# -------------------------------------------------------
# CHECK VCENTER CURRENT CERT & COMPARE FINGERPRINTS
# -------------------------------------------------------
$sessionHeaders = @{
    'vmware-api-session-id' = $vCenterConn.ExtensionData.Content.SessionManager.SessionId
}

$vcenterCertUri = "https://$VCENTERHOST/rest/vcenter/certificate-management/vcenter/tls"
$updateNeeded = $true

try {
    $vcResp = Invoke-RestMethod -Uri $vcenterCertUri -Method Get -Headers $sessionHeaders -SkipCertificateCheck
    $currentPem = $vcResp.value.cert

    $currentFp = Get-CertFingerprintFromPem $currentPem
    $newFp     = Get-CertFingerprintFromPem (Get-Content -Raw $certPath)

    if ($currentFp -and $newFp -and ($currentFp -eq $newFp)) {
        Show-Banner "vCenter certificate already up-to-date." SUCCESS
        $updateNeeded = $false
    } else {
        Write-Log INFO "Fingerprint mismatch → update required." Yellow
    }
}
catch {
    Write-Log WARN "Could not read vCenter certificate; assuming update needed." Yellow
}

# -------------------------------------------------------
# UPLOAD / APPLY NEW CERT
# -------------------------------------------------------
$restartNeeded = $false

if ($updateNeeded) {
    try {
        $body = @{
            cert  = Get-Content -Raw $certPath
            key   = Get-Content -Raw $keyPath
            chain = Get-Content -Raw $chainPath
        }

        Write-Log INFO "Uploading new cert..." Cyan
        Invoke-RestMethod -Uri $vcenterCertUri -Method Post -Headers $sessionHeaders `
            -ContentType "application/json" -Body ($body | ConvertTo-Json -Compress) -SkipCertificateCheck

        Write-Log INFO "Applying certificate..." Cyan
        Invoke-RestMethod -Uri "$vcenterCertUri?action=apply" -Method Post `
            -Headers $sessionHeaders -SkipCertificateCheck

        Show-Banner "Certificate applied to vCenter." SUCCESS
        $restartNeeded = $true
    }
    catch {
        Show-Banner "Failed to apply certificate: $($_.Exception.Message)" ERROR
        exit 1
    }
}
else {
    Write-Log INFO "Certificate update not required." Green
}

# -------------------------------------------------------
# RESTART VPXD (IF NEEDED)
# -------------------------------------------------------
if ($restartNeeded) {
    $maxRetries = 20
    for ($i=1; $i -le $maxRetries; $i++) {
        try {
            Invoke-RestMethod -Uri "https://$VCENTERHOST/rest/appliance/health/system" `
                -Method Get -SkipCertificateCheck -ErrorAction Stop

            Invoke-RestMethod -Uri "https://$VCENTERHOST/rest/appliance/system/services/vpxd?action=restart" `
                -Method Post -SkipCertificateCheck -ErrorAction Stop

            Show-Banner "vpxd restarted successfully." SUCCESS
            break
        }
        catch {
            Write-Log WARN "vpxd not ready, retrying $i/$maxRetries..." Yellow
            Start-Sleep -Seconds 15
        }
    }
}

Show-Banner "Script complete. Log: $LogFile" INFO