### 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 ### checkbox not working ### Threatlocker ### setsvspowerplan ### 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() }