diff --git a/TGBeta.ps1 b/TGBeta.ps1 index dba4308..e69de29 100644 --- a/TGBeta.ps1 +++ b/TGBeta.ps1 @@ -1,1100 +0,0 @@ -### Known issues as of January 5 2025 - -### Install-DattoRMM version 25.1.4 is missing functions you'll need version 25.1.5 - -### Missing Function in module, are added in 25.1.5 -### Install-SVSHelpDesk -### Install SVSWatchtower - -### add tweek to set default provider, add to toolkit? -### need to see if there's anything else we could take from Theo script? -### make sure that when using select all option -### Module install first -### Rmm Second - -# --------------------------------------------------------------------------- -# 1) CREATE A GLOBAL LOG CACHE (NEW) -# --------------------------------------------------------------------------- -if (-not $Global:LogCache -or -not ($Global:LogCache -is [System.Collections.ArrayList])) { - $Global:LogCache = New-Object System.Collections.ArrayList -} - - -# Check if the Write-Log function exists -if (-not (Get-Command -Name Write-Log -CommandType Function -ErrorAction SilentlyContinue)) { - # If the Write-Log function doesn't exist, create the Write-LogHelper function - function Write-LogHelper { - param ( - [string]$Message, - [ValidateSet("Info", "Warning", "Error", "Success", "General")] - [string]$Level = "Info", - [string]$TaskCategory = "GeneralTask", # Task Category for the log entry - [switch]$LogToEvent = $false, # Log to Windows Event Log - [string]$EventSource = "SVSMSP_Module", # Event Source - [string]$EventLog = "Application", # Event Log (default: Application) - [int]$CustomEventID # Optional custom Event ID - ) - - # Simplified Event ID mapping - $EventID = switch ($Level) { - "Info" { 1000 } - "Warning" { 2000 } - "Error" { 3000 } - "Success" { 4000 } - "General" { 1000 } - } - - # Icons for each level - $Icon = switch ($Level) { - "Info" { [System.Char]::ConvertFromUtf32(0x1F4CB) } # Information icon - "Warning" { ([char]0x26A0) } # Warning icon - "Error" { ([char]0x274C) } # Error icon - "Success" { ([char]0x2705) } # Success icon - "General" { ([char]0x1F4E6) } # Package icon - } - - # Map levels to colors - $Color = switch ($Level) { - "Info" { "Cyan" } - "Warning" { "Yellow" } - "Error" { "Red" } - "Success" { "Green" } - "General" { "White" } - } - - # Write to the PowerShell console - Write-Host "$Icon [$Level] [$TaskCategory] $Message (Event ID: $EventID)" -ForegroundColor $Color - - # ------------------------------------------------------------------- - # 2) ALSO STORE THE LOG IN OUR GLOBAL LOG CACHE (NEW) - # ------------------------------------------------------------------- - $logEntry = [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($logEntry) - # ------------------------------------------------------------------- - - # Optionally log to the Windows Event Log - if ($LogToEvent) { - $EntryType = switch ($Level) { - "Info" { "Information" } - "Warning" { "Warning" } - "Error" { "Error" } - default { "Information" } - } - - try { - if (-not (Get-EventLog -LogName $EventLog -Source $EventSource -ErrorAction SilentlyContinue)) { - New-EventLog -LogName $EventLog -Source $EventSource -ErrorAction SilentlyContinue - } - $EventMessage = "TaskCategory: $TaskCategory | Message: $Message" - Write-EventLog -LogName $EventLog -Source $EventSource -EntryType $EntryType -EventId $EventID -Message $EventMessage - } - catch { - Write-Host "([char]0x26A0) [Warning] [EventLog] Failed to write to Event Log: $($_.Exception.Message)" -ForegroundColor Yellow - } - } - } - - function Write-LogHybrid { - param ( - [string]$Message, - [ValidateSet("Info", "Warning", "Error", "Success", "General")] - [string]$Level = "Info", - [string]$TaskCategory = "GeneralTask", - [switch]$LogToEvent = $false, - [string]$EventSource = "SVSMSP_Module", - [string]$EventLog = "Application", - [int]$CustomEventID - ) - Write-LogHelper -Message $Message -Level $Level -TaskCategory $TaskCategory ` - -LogToEvent:$LogToEvent -EventSource $EventSource -EventLog $EventLog ` - -CustomEventID $CustomEventID - } -} -else { - # If Write-Log exists, define Write-LogHybrid to use Write-Log - function Write-LogHybrid { - param ( - [string]$Message, - [ValidateSet("Info", "Warning", "Error", "Success", "General")] - [string]$Level = "Info", - [string]$TaskCategory = "GeneralTask", - [switch]$LogToEvent = $false, - [string]$EventSource = "SVSMSP_Module", - [string]$EventLog = "Application", - [int]$CustomEventID - ) - Write-Log -Message $Message -Level $Level -TaskCategory $TaskCategory ` - -LogToEvent:$LogToEvent -EventSource $EventSource -EventLog $EventLog ` - -CustomEventID $CustomEventID - } -} - -# Example usage of Write-LogHybrid -Write-LogHybrid -Message "This is a test log message. Write-logHybrid" -Level "Info" -TaskCategory "TestCategory" -LogToEvent:$true -Write-Log -Message "This is a test log message. Write-log" -Level "Info" -TaskCategory "TestCategory" -LogToEvent:$true - - -function Install-SVSMSP { - param ( - # Cleanup flag - [switch]$Cleanup, - - # Toolkit installation flag - [switch]$InstallToolkit, - - # Module settings - [Parameter(Mandatory = $false)] - [array]$AllModules = @( - @{ ModuleName = "SVS_Toolkit" }, - @{ ModuleName = "SVSMSP" } - ), - - [Parameter(Mandatory = $false)] - [string]$NewModuleName = "SVSMSP", - - # Repository settings - [Parameter(Mandatory = $false)] - [array]$AllRepositories = @( - @{ RepoName = "SVS_Repo" }, - @{ RepoName = "SVS_Toolkit" } - ), - - [Parameter(Mandatory = $false)] - [string]$NewRepositoryName = "SVS_Repo", - - [Parameter(Mandatory = $false)] - [string]$NewRepositoryURL = "http://proget.svstools.ca:8083/nuget/SVS_Repo/", - - # Commands to check - [Parameter(Mandatory = $false)] - [array]$CommandsToCheck = @( - "Install-DattoRMM", - "Install-CyberQP", - "Install-RocketCyber", - "Install-Splashtop", - "Install-ThreatLocker" - ), - - # Log file path - [Parameter(Mandatory = $false)] - [string]$LogFilePath = "$env:temp\svstoolkit.log", - - # DRMM API Settings - [Parameter(Mandatory = $false)] - [string]$ApiUrl = "https://example-api-url.com", - - [Parameter(Mandatory = $false)] - [string]$ApiKey = "YOUR_API_KEY_HERE", - - [Parameter(Mandatory = $false)] - [string]$ApiSecretKey = "YOUR_API_SECRET_HERE" - ) - - function Perform-Cleanup { - Write-LogHybrid -Message "Cleanup mode enabled. Starting cleanup process..." -Level "Info" -LogToEvent -EventID 1502 - - # Step 1: Remove all old modules - Write-LogHybrid -Message "Starting cleanup of old modules..." -Level "Info" -LogToEvent - foreach ($module in $AllModules) { - $ModuleName = $module.ModuleName - if (Get-Module -Name $ModuleName -ListAvailable) { - Write-LogHybrid -Message "Removing module '$ModuleName'..." -Level "Warning" -LogToEvent - try { - Get-Module -Name $ModuleName -ListAvailable | ForEach-Object { - Uninstall-Module -Name $_.Name -AllVersions -Force - } - Write-LogHybrid -Message "Module '$ModuleName' removed successfully." -Level "Success" -LogToEvent - } - catch { - Write-LogHybrid -Message "Failed to remove module '$ModuleName'. Error: $($_.Exception.Message)" -Level "Error" -LogToEvent - } - } - else { - Write-LogHybrid -Message "Module '$ModuleName' not found. Skipping..." -Level "Info" -LogToEvent - } - } - - # Step 2: Remove all old repositories - Write-LogHybrid -Message "Starting cleanup of old repositories..." -Level "Info" -LogToEvent - foreach ($repo in $AllRepositories) { - $RepoName = $repo.RepoName - Write-LogHybrid -Message "Removing repository '$RepoName'..." -Level "Warning" -LogToEvent - if (Get-PSRepository -Name $RepoName -ErrorAction SilentlyContinue) { - try { - Unregister-PSRepository -Name $RepoName -ErrorAction Stop - Write-LogHybrid -Message "Repository '$RepoName' removed successfully." -Level "Success" -LogToEvent - } - catch { - Write-LogHybrid -Message "Failed to remove repository '$RepoName'. Error: $($_.Exception.Message)" -Level "Error" -LogToEvent - } - } - else { - Write-LogHybrid -Message "Repository '$RepoName' does not exist. Skipping removal." -Level "Info" -LogToEvent - } - } - - Write-LogHybrid -Message "Cleanup process completed successfully." -Level "Success" -LogToEvent -EventID 1510 - } - - function Perform-ToolkitInstallation { - # Perform cleanup - Perform-Cleanup - - # Step 1: Set Execution Policy - $localMachineExecutionPolicy = Get-ExecutionPolicy -Scope LocalMachine - if ($localMachineExecutionPolicy -ne "RemoteSigned") { - Write-LogHybrid -Message "Setting execution policy to RemoteSigned..." -Level "Warning" -LogToEvent -EventID 1522 - try { - Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope LocalMachine -Force - Write-LogHybrid -Message "Execution policy set to RemoteSigned successfully." -Level "Success" -LogToEvent -EventID 1513 - } - catch { - Write-LogHybrid -Message "Failed to set execution policy. Error: $_" -Level "Error" -LogToEvent -EventID 1534 - return - } - } - - # Step 2: Ensure NuGet is Installed - if (!(Get-PackageProvider -Name "NuGet" -ErrorAction SilentlyContinue)) { - Write-LogHybrid -Message "NuGet package provider not found. Installing..." -Level "Warning" -LogToEvent -EventID 1520 - try { - Install-PackageProvider -Name "NuGet" -Force -Scope AllUsers -Confirm:$false - Write-LogHybrid -Message "NuGet package provider installed successfully." -Level "Success" -LogToEvent -EventID 1514 - } - catch { - Write-LogHybrid -Message "Failed to install NuGet package provider. Error: $_" -Level "Error" -LogToEvent -EventID 1535 - return - } - } - - # Step 3: Register the new repository - Write-LogHybrid -Message "Registering the new repository '$NewRepositoryName'..." -Level "Info" -LogToEvent - try { - if (!(Get-PSRepository -Name $NewRepositoryName -ErrorAction SilentlyContinue)) { - Register-PSRepository -Name $NewRepositoryName -SourceLocation $NewRepositoryURL -InstallationPolicy Trusted - Write-LogHybrid -Message "Repository '$NewRepositoryName' registered successfully." -Level "Success" -LogToEvent - } - } - catch { - Write-LogHybrid -Message "Failed to register new repository '$NewRepositoryName'. Error: $($_.Exception.Message)" -Level "Error" -LogToEvent - } - - # Step 4: Install the new module - Write-LogHybrid -Message "Installing the new module '$NewModuleName'..." -Level "Info" -LogToEvent - try { - Install-Module -Name $NewModuleName -Repository $NewRepositoryName -Scope AllUsers -Force - Write-LogHybrid -Message "Module '$NewModuleName' installed successfully." -Level "Success" -LogToEvent - } - catch { - Write-LogHybrid -Message "Failed to install new module '$NewModuleName'. Error: $($_.Exception.Message)" -Level "Error" -LogToEvent - } - - Write-LogHybrid -Message "Toolkit installation process completed successfully." -Level "Success" -LogToEvent -EventID 1510 - } - - Write-LogHybrid -Message "Install-SVSMSP function started." -Level "Info" -LogToEvent -EventID 1500 - - if ($Cleanup) { - Perform-Cleanup - return - } - - if ($InstallToolkit) { - Perform-ToolkitInstallation - return - } - - Write-LogHybrid -Message "No specific mode specified. Defaulting to toolkit installation mode..." -Level "Info" -LogToEvent -EventID 1504 - Perform-ToolkitInstallation -} - - # Install-SVSMSP -InstallToolkit - - -# ---------------------------------------------------------------------------------- -# START THE LISTENER -# ---------------------------------------------------------------------------------- -$listener = New-Object System.Net.HttpListener -$listener.Prefixes.Add("http://localhost:8081/") -$listener.Start() - - -function Get-N8nWebhookData { - param ( - [Parameter(Mandatory = $true)] - [string]$AuthHeaderValue - ) - - $url = "https://automate.svstools.ca/webhook/svsmspkit" - $headers = @{ - "SVSMSPKit" = $AuthHeaderValue - } - try { - $response = Invoke-RestMethod -Uri $url -Headers $headers -Method Get - Write-Host "Response received successfully:" -ForegroundColor Green - - $data = $response - - # Map fields - $global:Comment_SVSmodule = $data._Comment_SVSmodule - $global:ModuleName = $data.ModuleName - $global:RepositoryURL = $data.RepositoryURL - $global:OldRepo = $data.OldRepo - $global:NewRepo = $data.NewRepo - $global:CommandsToCheck = $data.CommandsToCheck - $global:LogFilePath = $data.LogFilePath - $global:Comment_DRMM = $data._Comment_DRMM - $global:ApiUrl = $data.ApiUrl - $global:ApiKey = $data.ApiKey - $global:ApiSecretKey = $data.ApiSecretKey - - - - # return $data - } - catch { - Write-Host "Error making the GET request:" -ForegroundColor Red - Write-Host $_.Exception.Message - return $null - } -} - - -# Define the HTML Content with an Off-Boarding Tab -function GetHtmlContent { - @" - - - - - - SVS TaskGate - - - - -
- SVS Logo -
-
-
- - - -
-
- -
-

On-Boarding

-
- - - - - - - - - - - -
- -
- -
- - - -
-
-

Logs will appear here...

-
-
-
-
- - - - -"@ -} - - - -# Save and launch the HTML -Start-Process "msedge.exe" -ArgumentList "--app=http://localhost:8081/" - - -try { - while ($listener.IsListening) { - $context = $listener.GetContext() - $request = $context.Request - $response = $context.Response - - switch ($request.Url.AbsolutePath) { - - "/" { - $htmlContent = GetHtmlContent - $buffer = [System.Text.Encoding]::UTF8.GetBytes($htmlContent) - $response.ContentType = "text/html" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - "/getn8npw" { - if ($request.HttpMethod -eq "POST") { - $bodyStream = New-Object IO.StreamReader $request.InputStream - $body = $bodyStream.ReadToEnd() - $data = ConvertFrom-Json $body - $password = $data.password - - Get-N8nWebhookData -AuthHeaderValue $password - $sites = Install-DattoRMM -ApiUrl $ApiUrl -ApiKey $ApiKey -ApiSecretKey $ApiSecretKey -FetchSitesOnly - if (-not $sites) { - Write-Host "No sites returned. Please check the API." -ForegroundColor Red - $response.StatusCode = 500 - $buffer = [System.Text.Encoding]::UTF8.GetBytes("No sites found") - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - continue - } - $responseData = $sites | ConvertTo-Json - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseData) - $response.ContentType = "application/json" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - } - - "/installrmm" { - if ($request.HttpMethod -eq "POST") { - try { - # Step 1: Read the Request Body - $bodyStream = New-Object IO.StreamReader $request.InputStream - $body = $bodyStream.ReadToEnd() - $requestData = ConvertFrom-Json $body - - # Step 2: Extract Data from the Request - $checkedValues = $requestData.checkedValues - $UID = $requestData.UID - $Name = $requestData.Name - - # Step 3: Validate Input - if (-not $checkedValues -or -not $UID -or -not $Name) { - Write-LogHybrid -Message "Invalid input received. Missing required parameters: UID, Name, or checkbox values." -Level "Error" - $response.StatusCode = 400 - $responseString = "Error: Missing required input parameters." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - return - } - - # Step 4: Build the PowerShell Command Dynamically - $installRMMCommand = "Install-DattoRMM -ApiUrl '$ApiUrl' -ApiKey '$ApiKey' -ApiSecretKey '$ApiSecretKey' -SiteName $Name -SiteUID $UID " - - if ($checkedValues -contains 'inputVar') { - $installRMMCommand += " -PushSiteVars" - } - if ($checkedValues -contains 'rmm') { - $installRMMCommand += " -InstallRMM" - } - if ($checkedValues -contains 'exe') { - $installRMMCommand += " -SaveCopy" - } - - - - # Step 5: Execute the Command - try { - Invoke-Expression $installRMMCommand - $responseString = "RMM installation triggered successfully for $Name." - Write-LogHybrid -Message $responseString -Level "Success" - $response.StatusCode = 200 - } catch { - $responseString = "Error triggering RMM installation: $($_.Exception.Message)" - Write-LogHybrid -Message $responseString -Level "Error" - $response.StatusCode = 500 - } - } catch { - # Log General Errors - $errorString = "An error occurred while processing the /installrmm request: $($_.Exception.Message)" - Write-LogHybrid -Message $errorString -Level "Error" - $response.StatusCode = 500 - $responseString = $errorString - } - - # Step 6: Return the Response - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } else { - # Handle Invalid HTTP Methods - $response.StatusCode = 405 - $response.StatusDescription = "Method Not Allowed" - $responseString = "Error: Only POST requests are allowed." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - } - - - - "/setSVSPowerplan" { - if ($request.HttpMethod -eq "GET") { - Set-SVSPowerPlan - } - $responseString = "Install SVS Powerplan triggered." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - - "/installCyberQP" { - if ($request.HttpMethod -eq "GET") { - try { - Install-CyberQP - $responseString = "Install CyberQP triggered successfully." - $response.StatusCode = 200 - } catch { - $responseString = "Error triggering Install CyberQP: $_" - $response.StatusCode = 500 - } - } else { - $responseString = "Method not allowed. Use GET." - $response.StatusCode = 405 - } - - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - - } - - "/installSplashtop" { - if ($request.HttpMethod -eq "GET") { - Install-Splashtop - } - $responseString = "Install Splashtop triggered." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - "/installRocketCyber" { - if ($request.HttpMethod -eq "GET") { - Install-RocketCyber - } - $responseString = "Install RocketCyber triggered." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - "/installThreatLocker" { - if ($request.HttpMethod -eq "GET") { - Install-ThreatLocker - } - $responseString = "Install ThreatLocker triggered." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - "/installSVSHelpDesk" { - if ($request.HttpMethod -eq "GET") { - Install-SVSHelpDesk - } - $responseString = "Install SVSHelpDesk triggered." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - "/installSVSWatchtower" { - if ($request.HttpMethod -eq "GET") { - Install-SVSWatchtower - } - $responseString = "Install SVSWatchtower triggered." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - - # ---------------------------------------------------------------- - # 4) NEW ROUTE: /getLogs - # Returns $Global:LogCache as JSON for the polling function - # ---------------------------------------------------------------- - "/getLogs" { - if ($request.HttpMethod -eq "GET") { - $jsonLogs = $Global:LogCache | ConvertTo-Json - $logBuffer = [System.Text.Encoding]::UTF8.GetBytes($jsonLogs) - $response.ContentType = "application/json" - $response.ContentLength64 = $logBuffer.Length - $response.OutputStream.Write($logBuffer, 0, $logBuffer.Length) - $response.OutputStream.Close() - } - } - - "/quit" { - if ($request.HttpMethod -eq "GET") { - $responseString = "Server shutting down." - $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseString) - $response.ContentType = "text/plain" - $response.ContentLength64 = $buffer.Length - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - Write-Host $responseString - $listener.stop() - break - } - } - - default { - $response.StatusCode = 404 - $response.StatusDescription = "Not Found" - $buffer = [System.Text.Encoding]::UTF8.GetBytes("404 - Not Found") - $response.OutputStream.Write($buffer, 0, $buffer.Length) - $response.OutputStream.Close() - } - } - } -} -catch { - Write-Host "Error: $($_.Exception.Message)" - -} -finally { - $listener.Stop() - $listener.Close() -}