diff --git a/testTaskGate.ps1 b/testTaskGate.ps1 index 0da32b9..0035b32 100644 --- a/testTaskGate.ps1 +++ b/testTaskGate.ps1 @@ -1,338 +1,172 @@ -###-----------------------------------------------------------------------------### -### SVS TaskGate Launcher -###-----------------------------------------------------------------------------### +# SVS TaskGate: Self-contained PowerShell Task Runner UI (All-in-One Script) -# 1) GLOBAL LOG CACHE -if (-not $Global:LogCache -or -not ($Global:LogCache -is [System.Collections.ArrayList])) { - $Global:LogCache = New-Object System.Collections.ArrayList +# ----- 1. TASK REGISTRY ----- +$TaskRegistry = @{ + Onboarding = @( + @{ Key = "installSVSMSP"; Label = "Install SVSMSP Module"; Function = "Install-SVSMSP" } + @{ Key = "installDattoRMM"; Label = "Install DattoRMM"; Function = "Install-DattoRMM" } + @{ Key = "installCyberQP"; Label = "Install CyberQP"; Function = "Install-CyberQP" } + ) + Offboarding = @( + @{ Key = "uninstallSVSMSP"; Label = "Uninstall SVSMSP Module"; Function = "Uninstall-SVSMSP" } + @{ Key = "uninstallDattoRMM"; Label = "Uninstall DattoRMM"; Function = "Uninstall-DattoRMM" } + @{ Key = "uninstallCyberQP"; Label = "Uninstall CyberQP"; Function = "Uninstall-CyberQP" } + ) + Tweaks = @( + @{ Key = "setPowerPlan"; Label = "Set Power Plan"; Function = "Set-SVSPowerPlan" } + @{ Key = "enableBitLocker"; Label = "Enable BitLocker"; Function = "Enable-BitLocker" } + ) } -#region Write-LogHybrid -if (-not (Get-Command Write-Log -ErrorAction SilentlyContinue)) { - function Write-LogHelper { - param( - [string]$Message, - [ValidateSet("Info","Warning","Error","Success","General")] - [string]$Level = "Info", - [string]$TaskCategory = "GeneralTask", - [switch]$LogToEvent, - [string]$EventSource = "SVSMSP_Module", - [string]$EventLog = "Application" - ) - $EventID = switch($Level){ - "Info" {1000}; "Warning"{2000}; "Error"{3000}; "Success"{4000}; default{1000} - } - $Icon = switch($Level){ - "Info" { [char]0x1F4CB } - "Warning" { [char]0x26A0 } - "Error" { [char]0x274C } - "Success" { [char]0x2705 } - default { [char]0x1F4E6 } - } - $entry = [PSCustomObject]@{ - Timestamp = (Get-Date).ToString("yyyy-MM-dd HH:mm:ss") - Level = $Level - Message = "$Icon [$Level] [$TaskCategory] $Message (Event ID: $EventID)" - } - [void]$Global:LogCache.Add($entry) - if ($LogToEvent) { - try { - if (-not (Get-EventLog -LogName $EventLog -Source $EventSource -ErrorAction SilentlyContinue)) { - New-EventLog -LogName $EventLog -Source $EventSource -ErrorAction SilentlyContinue - } - Write-EventLog -LogName $EventLog -Source $EventSource ` - -EntryType $Level -EventId $EventID -Message $Message - } catch { - Write-Host "⚠ Failed writing to EventLog: $($_.Exception.Message)" -ForegroundColor Yellow - } - } - } - function Write-LogHybrid { param($Message,$Level,$TaskCategory,$LogToEvent) - Write-LogHelper -Message $Message -Level $Level -TaskCategory $TaskCategory -LogToEvent:$LogToEvent - } -} else { - function Write-LogHybrid { param($Message,$Level,$TaskCategory,$LogToEvent) - Write-Log -Message $Message -Level $Level -TaskCategory $TaskCategory -LogToEvent:$LogToEvent +# ----- 2. FUNCTION STUBS (Replace with real logic) ----- +function Install-SVSMSP { Write-Host "Installing SVSMSP Module..." } +function Install-DattoRMM { Write-Host "Installing DattoRMM..." } +function Install-CyberQP { Write-Host "Installing CyberQP..." } +function Uninstall-SVSMSP { Write-Host "Uninstalling SVSMSP Module..." } +function Uninstall-DattoRMM { Write-Host "Uninstalling DattoRMM..." } +function Uninstall-CyberQP { Write-Host "Uninstalling CyberQP..." } +function Set-SVSPowerPlan { Write-Host "Setting SVS Power Plan..." } +function Enable-BitLocker { Write-Host "Enabling BitLocker..." } + +# ----- 3. HTML UI GENERATOR ----- +function Get-TasksHtml($group) { + $html = "" + foreach ($task in $TaskRegistry[$group]) { + $html += "
`n" } + return $html } -Write-LogHybrid -Message "Starting SVS TaskGate" -Level Info -TaskCategory Startup -LogToEvent -#region Helpers for DattoRMM, LastPass, SVS Module… -# (Insert your existing Install-DattoRMM-Helper, LastPass and Install-SVSMSP functions here.) -#endregion - - -#region HTTP Listener Setup -$listener = New-Object System.Net.HttpListener -$listener.Prefixes.Add("http://localhost:8081/") -$listener.Start() -Write-LogHybrid -Message "Listener started on http://localhost:8081/" -Level Info -TaskCategory Listener -#endregion - - -#region HTML UI & CSS -function GetHtmlContent { -@" +$HtmlContent = @" - - SVS TaskGate - - - + + SVS TaskGate + - -
- SVS Logo -
- -
-
- - - +

SVS TaskGate

+
+

Onboarding

+
+ $(Get-TasksHtml 'Onboarding') +
+
- -
- -
-

On-Boarding

-

…your original onboarding controls…

-
- - -
-

Apply Tweaks

-
-
-
- +
+

Offboarding

+
+ $(Get-TasksHtml 'Offboarding')
-
- -
-
- - -
-

Install SVS Apps

-
-

Winget Apps

-
- -
-
-

Browser Extensions

-
- -
-
- -
-
- +
-
- - - + "@ -} -#endregion +# ----- 4. HTTP LISTENER (Self-contained, single-file, no dependencies) ----- +Add-Type -AssemblyName System.Net.HttpListener -# 3) LAUNCH THE UI -Start-Process "msedge.exe" -ArgumentList "--app=http://localhost:8081/" +$listener = New-Object System.Net.HttpListener +$listener.Prefixes.Add("http://localhost:8081/") +$listener.Start() +Write-Host "TaskGate listening at http://localhost:8081/" - -#region 4) ROUTE HANDLING +# Open browser to UI (Edge, fallback Chrome) try { - while ($listener.IsListening) { - $ctx = $listener.GetContext() - $req = $ctx.Request - $res = $ctx.Response + Start-Process "msedge.exe" -ArgumentList "--app=http://localhost:8081/" +} catch { + Start-Process "chrome.exe" "http://localhost:8081/" +} - switch ($req.Url.AbsolutePath) { - "/" { - $html = GetHtmlContent - $bytes = [Text.Encoding]::UTF8.GetBytes($html) - $res.ContentType = "text/html" - $res.ContentLength64 = $bytes.Length - $res.OutputStream.Write($bytes,0,$bytes.Length) +while ($listener.IsListening) { + $context = $listener.GetContext() + $req = $context.Request + $res = $context.Response + + try { + switch ($req.Url.AbsolutePath) { + "/" { + $bytes = [System.Text.Encoding]::UTF8.GetBytes($HtmlContent) + $res.ContentType = "text/html" + $res.ContentLength64 = $bytes.Length + $res.OutputStream.Write($bytes, 0, $bytes.Length) + $res.OutputStream.Close() + } + "/run-tasks" { + if ($req.HttpMethod -ne "POST") { + $res.StatusCode = 405 + $res.OutputStream.Close() + continue + } + $reader = New-Object IO.StreamReader $req.InputStream + $body = $reader.ReadToEnd() | ConvertFrom-Json + $group = $body.group + $selectedKeys = $body.tasks + $executed = @() + foreach ($key in $selectedKeys) { + $task = $TaskRegistry[$group] | Where-Object { $_.Key -eq $key } + if ($task) { + try { + & $task.Function + $executed += $task.Label + } catch { + $executed += "$($task.Label) (ERROR)" + } + } + } + $resp = @{ status = "OK"; executed = $executed } + $bytes = [System.Text.Encoding]::UTF8.GetBytes(($resp | ConvertTo-Json)) + $res.ContentType = "application/json" + $res.ContentLength64 = $bytes.Length + $res.OutputStream.Write($bytes, 0, $bytes.Length) + $res.OutputStream.Close() + } + default { + $res.StatusCode = 404 + $bytes = [System.Text.Encoding]::UTF8.GetBytes("404 Not Found") + $res.OutputStream.Write($bytes, 0, $bytes.Length) + $res.OutputStream.Close() + } + } + } catch { + $res.StatusCode = 500 + $err = "Error: $($_.Exception.Message)" + $bytes = [System.Text.Encoding]::UTF8.GetBytes($err) + $res.OutputStream.Write($bytes, 0, $bytes.Length) $res.OutputStream.Close() - } - - "/runTweaks" { - if ($req.HttpMethod -eq "POST") { - $body = (New-Object IO.StreamReader $req.InputStream).ReadToEnd() | ConvertFrom-Json - Write-LogHybrid -Message "Tweaks: $($body.tweaks -join ', ')" -Level Info -TaskCategory Tweaks - # TODO: call your actual tweak functions here... - $res.StatusCode = 200 - $res.OutputStream.Write(([Text.Encoding]::UTF8.GetBytes("OK")),0,2) - } - } - - "/installSVSApps" { - if ($req.HttpMethod -eq "POST") { - $data = (New-Object IO.StreamReader $req.InputStream).ReadToEnd() | ConvertFrom-Json - Write-LogHybrid -Message "Winget: $($data.winget -join ', ')" -Level Info -TaskCategory "SVSApps" - Write-LogHybrid -Message "Extensions: $($data.extensions -join ', ')" -Level Info -TaskCategory "SVSApps" - # TODO: invoke your winget & extension helpers here... - $res.StatusCode = 200 - $res.OutputStream.Write(([Text.Encoding]::UTF8.GetBytes("OK")),0,2) - } - } - - "/quit" { - $res.StatusCode = 200 - $res.OutputStream.Write(([Text.Encoding]::UTF8.GetBytes("bye")),0,3) - $listener.Stop() - break - } - - default { - $res.StatusCode = 404 - $res.OutputStream.Write(([Text.Encoding]::UTF8.GetBytes("Not Found")),0,9) - } } - } -} -finally { - if ($listener) { - Write-LogHybrid -Message "Stopping listener" -Level Info -TaskCategory Listener - $listener.Stop(); $listener.Close() - } }