# ==============================================================================
# Script Name: 8x8_Work_Updater.ps1
# Description: Downloads 8x8, uses ServiceUI (from internal Gitea) to display 
#              a 1-hour countdown GUI directly on the active user's desktop, 
#              silently installs 8x8, and relaunches it.
# ==============================================================================

# Ensure script is running elevated
if (-not ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)) {
    Write-Error "This script must be executed with Administrator privileges."
    exit 1
}

Add-Type -AssemblyName System.Net.Http

$workDir       = "C:\ProgramData\JCT600_8x8Update"
$flagFile      = Join-Path $workDir "8x8_Update_Proceed.flag"
$guiScript     = Join-Path $workDir "Show-8x8UserGui.ps1"
$serviceUiPath = Join-Path $workDir "ServiceUI.exe"
$installerPath = Join-Path $workDir "8x8_Work_Installer.msi"

if (-not (Test-Path $workDir)) {
    New-Item -Path $workDir -ItemType Directory -Force | Out-Null
}

# Grant standard users permissions to the directory
$acl = Get-Acl $workDir
$rule = New-Object System.Security.AccessControl.FileSystemAccessRule("BUILTIN\Users", "FullControl", "ContainerInherit,ObjectInherit", "None", "Allow")
$acl.AddAccessRule($rule)
Set-Acl $workDir $acl

if (Test-Path $flagFile) { Remove-Item $flagFile -Force }

try {
    # --------------------------------------------------------------------------
    # Step 1: Download 8x8 Installer & ServiceUI from internal Gitea
    # --------------------------------------------------------------------------
    $installerUrl = "https://work-desktop-assets.8x8.com/prod-publish/ga/work-64-msi-v8.35.2-6.msi"
    $serviceUiUrl = "https://gitea.ostroveni.com/JCT/8x8-Work-updater/raw/branch/main/ServiceUI.exe"

    Write-Host "[1/5] Downloading 8x8 Work installer and helper tools..." -ForegroundColor Cyan

    try {
        # Download ServiceUI from internal Gitea
        Invoke-WebRequest -Uri $serviceUiUrl -OutFile $serviceUiPath -UseBasicParsing
        
        # Download 8x8 Installer with stream progress bar
        $httpClient = New-Object System.Net.Http.HttpClient
        $response = $httpClient.GetAsync($installerUrl, [System.Net.Http.HttpCompletionOption]::ResponseHeadersRead).Result
        if (-not $response.IsSuccessStatusCode) { throw "HTTP Error $($response.StatusCode)" }

        $totalBytes = $response.Content.Headers.ContentLength
        $inputStream = $response.Content.ReadAsStreamAsync().Result
        $outputStream = [System.IO.File]::Create($installerPath)
        $buffer = New-Object byte[] 8192
        $totalBytesRead = 0; $bytesRead = 0

        while (($bytesRead = $inputStream.Read($buffer, 0, $buffer.Length)) -gt 0) {
            $outputStream.Write($buffer, 0, $bytesRead)
            $totalBytesRead += $bytesRead

            if ($totalBytes -gt 0) {
                $percent = [math]::Round(($totalBytesRead / $totalBytes) * 100)
                $downloadedMB = [math]::Round($totalBytesRead / 1MB, 2)
                $totalMB = [math]::Round($totalBytes / 1MB, 2)
                $barLength = 30
                $filledLength = [math]::Round(($percent / 100) * $barLength)
                $bar = ("#" * $filledLength) + ("-" * ($barLength - $filledLength))

                Write-Host "`rDownloading: [$bar] $percent% ($downloadedMB MB / $totalMB MB)" -NoNewline -ForegroundColor Yellow
            }
        }
        $outputStream.Close(); $inputStream.Close(); $httpClient.Dispose()
        Write-Host "`nDownload complete: $installerPath" -ForegroundColor Green
    } catch {
        Write-Error "`nFailed to download installer or tools. Error: $_"
        exit 1
    }

    # --------------------------------------------------------------------------
    # Step 2: Write User-Session GUI Script (BOM-Free UTF-8)
    # --------------------------------------------------------------------------
    Write-Host "[2/5] Generating user-session GUI dialog script..." -ForegroundColor Cyan

    $guiContent = @'
Add-Type -AssemblyName System.Windows.Forms
Add-Type -AssemblyName System.Drawing

$signature = @"
[DllImport("user32.dll", SetLastError = true)]
public static extern bool SetWindowPos(IntPtr hWnd, IntPtr hWndInsertAfter, int X, int Y, int cx, int cy, uint uFlags);
"@
$Win32SetWindowPos = Add-Type -memberDefinition $signature -Name "Win32SetWindowPos" -namespace Win32Functions -passThru

$flagPath = "C:\ProgramData\JCT600_8x8Update\8x8_Update_Proceed.flag"

$form = New-Object System.Windows.Forms.Form
$form.Text = "JCT600 IT Support — 8x8 Work Update"
$form.Size = New-Object System.Drawing.Size(440, 250)
$form.StartPosition = 'CenterScreen'
$form.FormBorderStyle = 'FixedDialog'
$form.MaximizeBox = $false
$form.MinimizeBox = $false
$form.ControlBox = $false
$form.TopMost = $true

$script:allowClose = $false
$form.Add_FormClosing({
    param($sender, $e)
    if (-not $script:allowClose) { $e.Cancel = $true }
})

$labelHeader = New-Object System.Windows.Forms.Label
$labelHeader.Text = "JCT600 — IT SUPPORT"
$labelHeader.Location = New-Object System.Drawing.Point(20, 15)
$labelHeader.Size = New-Object System.Drawing.Size(380, 25)
$labelHeader.Font = New-Object System.Drawing.Font("Segoe UI", 11, [System.Drawing.FontStyle]::Bold)
$labelHeader.ForeColor = [System.Drawing.Color]::Navy
$labelHeader.TextAlign = 'TopCenter'
$form.Controls.Add($labelHeader)

$labelMsg = New-Object System.Windows.Forms.Label
$labelMsg.Text = "An automatic update for 8x8 Work has been scheduled by IT Support.`nPlease save active work or click 'Update Now' to begin."
$labelMsg.Location = New-Object System.Drawing.Point(20, 45)
$labelMsg.Size = New-Object System.Drawing.Size(380, 45)
$labelMsg.Font = New-Object System.Drawing.Font("Segoe UI", 9)
$labelMsg.TextAlign = 'TopCenter'
$form.Controls.Add($labelMsg)

$labelTimer = New-Object System.Windows.Forms.Label
$labelTimer.Location = New-Object System.Drawing.Point(20, 95)
$labelTimer.Size = New-Object System.Drawing.Size(380, 40)
$labelTimer.Font = New-Object System.Drawing.Font("Segoe UI", 20, [System.Drawing.FontStyle]::Bold)
$labelTimer.ForeColor = [System.Drawing.Color]::DarkRed
$labelTimer.TextAlign = 'MiddleCenter'
$form.Controls.Add($labelTimer)

$btnUpdateNow = New-Object System.Windows.Forms.Button
$btnUpdateNow.Text = "Update Now"
$btnUpdateNow.Font = New-Object System.Drawing.Font("Segoe UI", 9, [System.Drawing.FontStyle]::Bold)
$btnUpdateNow.Size = New-Object System.Drawing.Size(140, 35)
$btnUpdateNow.Location = New-Object System.Drawing.Point(140, 150)
$btnUpdateNow.UseVisualStyleBackColor = $true

$btnUpdateNow.Add_Click({
    $timer.Stop()
    $script:allowClose = $true
    New-Item -Path $flagPath -ItemType File -Force | Out-Null
    $form.Close()
})
$form.Controls.Add($btnUpdateNow)

$form.Add_Shown({
    [Win32Functions.Win32SetWindowPos]::SetWindowPos($form.Handle, [IntPtr] -1, 0, 0, 0, 0, 0x0003) | Out-Null
    $form.Activate()
})

$timer = New-Object System.Windows.Forms.Timer
$timer.Interval = 1000
$script:remaining = 3600 # 1 Hour

$timer.Add_Tick({
    if ($script:remaining -gt 0) {
        $timeSpan = [TimeSpan]::FromSeconds($script:remaining)
        $labelTimer.Text = $timeSpan.ToString("hh\:mm\:ss")
        $script:remaining--
        [Win32Functions.Win32SetWindowPos]::SetWindowPos($form.Handle, [IntPtr] -1, 0, 0, 0, 0, 0x0003) | Out-Null
    } else {
        $timer.Stop()
        $script:allowClose = $true
        New-Item -Path $flagPath -ItemType File -Force | Out-Null
        $form.Close()
    }
})

$initialSpan = [TimeSpan]::FromSeconds($script:remaining)
$labelTimer.Text = $initialSpan.ToString("hh\:mm\:ss")

$timer.Start()
$form.ShowDialog() | Out-Null
$timer.Dispose()
$form.Dispose()
'@

    $utf8NoBom = New-Object System.Text.UTF8Encoding($false)
    [System.IO.File]::WriteAllText($guiScript, $guiContent, $utf8NoBom)

    # --------------------------------------------------------------------------
    # Step 3: Launch GUI via ServiceUI Directly into Active User's Session
    # --------------------------------------------------------------------------
    Write-Host "[3/5] Launching interactive dialog on active user desktop via ServiceUI..." -ForegroundColor Cyan

    $explorerProc = Get-Process -Name explorer -ErrorAction SilentlyContinue | Select-Object -First 1

    if (-not $explorerProc) {
        Write-Warning "No active desktop session detected. Proceeding directly to update."
    } else {
        $serviceUiArgs = "-process:explorer.exe powershell.exe -ExecutionPolicy Bypass -WindowStyle Hidden -File `"$guiScript`""
        Start-Process -FilePath $serviceUiPath -ArgumentList $serviceUiArgs

        Write-Host "Dialog displayed on user screen. Waiting for response or 1-hour expiration..." -ForegroundColor Yellow
        while (-not (Test-Path $flagFile)) {
            Start-Sleep -Seconds 2
        }
    }

    # --------------------------------------------------------------------------
    # Step 4: Close Running 8x8 Processes & Install Silently
    # --------------------------------------------------------------------------
    Write-Host "[4/5] Stopping 8x8 processes and installing silently..." -ForegroundColor Cyan

    $targetProcesses = Get-Process | Where-Object { 
        $_.ProcessName -like "*8x8*" -or $_.MainWindowTitle -like "*8x8*" 
    }

    if ($targetProcesses) {
        foreach ($proc in $targetProcesses) {
            Stop-Process -Id $proc.Id -Force -ErrorAction SilentlyContinue
        }
        Start-Sleep -Seconds 3
    }

    $msiArgs = "/i `"$installerPath`" /quiet /norestart"
    $installJob = Start-Process msiexec.exe -ArgumentList $msiArgs -Wait -PassThru

    if ($installJob.ExitCode -ne 0 -and $installJob.ExitCode -ne 3010) {
        Write-Error "Installation failed with exit code $($installJob.ExitCode)."
        exit $installJob.ExitCode
    }

    Write-Host "Installation completed successfully." -ForegroundColor Green

    # --------------------------------------------------------------------------
    # Step 5: Launch 8x8 Work in User Session
    # --------------------------------------------------------------------------
    Write-Host "[5/5] Launching 8x8 Work in user session..." -ForegroundColor Cyan

    $possiblePaths = @(
        "${env:ProgramFiles}\8x8 Inc\8x8 Work\8x8 Work.exe",
        "${env:ProgramFiles(x86)}\8x8 Inc\8x8 Work\8x8 Work.exe",
        "${env:ProgramFiles}\8x8 Work\8x8 Work.exe",
        "$env:LocalAppData\Programs\8x8 Work\8x8 Work.exe"
    )

    $exePath = $possiblePaths | Where-Object { Test-Path $_ } | Select-Object -First 1

    if ($exePath) {
        if ($explorerProc) {
            $launchArgs = "-process:explorer.exe `"$exePath`""
            Start-Process -FilePath $serviceUiPath -ArgumentList $launchArgs
            Write-Host "8x8 Work launched successfully in user context." -ForegroundColor Green
        } else {
            Start-Process -FilePath $exePath
        }
    } else {
        Write-Warning "Could not locate 8x8 Work executable."
    }

} finally {
    Write-Host "Cleaning up temporary files..." -ForegroundColor Gray
    if (Test-Path $workDir) {
        Remove-Item $workDir -Recurse -Force -ErrorAction SilentlyContinue
    }
}