diff --git a/SVSTaskGate.ps1 b/SVSTaskGate.ps1 new file mode 100644 index 0000000..3cb2fb7 --- /dev/null +++ b/SVSTaskGate.ps1 @@ -0,0 +1,994 @@ +### now that we have a field for user to enter N8N password we need to figure out how we can make the fetch site button refresh the downdown after +### add tweek to set default provider, add to toolkit? +### power settings in tweeks or onboarding? +### make script install the toolkit +### test the write-log in the log box and in the event viewer +### need to see if there's anything else we could take from Theo script? +### make when using select all the datto rmm pus var runs before install-splashtop + + +# 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 for generic tasks/operations + } + + # Map levels to colors + $Color = switch ($Level) { + "Info" { "Cyan" } + "Warning" { "Yellow" } + "Error" { "Red" } + "Success" { "Green" } + "General" { "White" } + } + + # Include Task Category, Icon, and Message in the console output + Write-Host "$Icon [$Level] [$TaskCategory] $Message (Event ID: $EventID)" -ForegroundColor $Color + + # Optionally log to the Windows Event Log + if ($LogToEvent) { + $EntryType = switch ($Level) { + "Info" { "Information" } + "Warning" { "Warning" } + "Error" { "Error" } + default { "Information" } + } + + try { + # Check if the event source exists, create it if necessary + if (-not (Get-EventLog -LogName $EventLog -Source $EventSource -ErrorAction SilentlyContinue)) { + New-EventLog -LogName $EventLog -Source $EventSource -ErrorAction SilentlyContinue + } + + # Include Task Category in the Event Log Message + $EventMessage = "TaskCategory: $TaskCategory | Message: $Message" + + # Write the event log + 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 + } + } +} + + # Define a fallback Write-LogHybrid function that uses Write-LogHelper + 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 + ) + + # Use the helper function to log + 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 + ) + + # Use the existing Write-Log function + 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-log2 does exist" -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" + ) + + # Helper function: Perform Cleanup + 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 + } + + # Helper function: Perform Toolkit Installation + function Perform-ToolkitInstallation { + # Perform cleanup to remove old modules and repositories + 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 + } + + # Main Logic + 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 + ) + + # Define the URL and headers + $url = "https://automate.svstools.ca/webhook/svsmspkit" + $headers = @{ + "SVSMSPKit" = $AuthHeaderValue + } + + # Make the GET request to the N8N webhook + try { + $response = Invoke-RestMethod -Uri $url -Headers $headers -Method Get + Write-Host "Response received successfully:" -ForegroundColor Green + + # Convert the JSON response to a PowerShell object + $data = $response + + # Map each field to a variable (if needed) + $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 + + # Print the values + Write-Output "Module Name: $ModuleName" + Write-Output "Repository URL: $RepositoryURL" + Write-Output "Old Repo: $OldRepo" + Write-Output "New Repo: $NewRepo" + Write-Output "Commands To Check: $($CommandsToCheck -join ', ')" + Write-Output "Log File Path: $LogFilePath" + Write-Output "API URL: $ApiUrl" + Write-Output "API Key: $ApiKey" + Write-Output "API Secret Key: $ApiSecretKey" + + # Return the parsed data object + 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 { + @" + + + + + + + On-Boarding & Off-Boarding + + + + +
+ +
+ +
+

On-Boarding

+
+ + + + + + + + + +
+ +
+ + +
+ + + +
+
+

Logs will appear here...

+
+
+
+
+ + + + +"@ +} + +# Save and launch the HTML +Start-Process "msedge.exe" -ArgumentList "--app=http://localhost:8081/" + + + +try { + while ($listener.IsListening) { + # Wait for an incoming request + $context = $listener.GetContext() + $request = $context.Request + $response = $context.Response + + switch ($request.Url.AbsolutePath) { + + "/" { + # Serve the main HTML + $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") { + # Parse the received JSON + $bodyStream = New-Object IO.StreamReader $request.InputStream + $body = $bodyStream.ReadToEnd() + $data = ConvertFrom-Json $body + $password = $data.password + + Get-N8nWebhookData -AuthHeaderValue $password + + # Fetch the list of Datto RMM client sites + $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 + } + + # Convert the array (or object) in $sites to JSON + $responseData = $sites | ConvertTo-Json + $buffer = [System.Text.Encoding]::UTF8.GetBytes($responseData) + + # Set the response headers + $response.ContentType = "application/json" + $response.ContentLength64 = $buffer.Length + + $response.OutputStream.Write($buffer, 0, $buffer.Length) + $response.OutputStream.Close() + } + } + + "/installrmm" { + if ($request.HttpMethod -eq "POST") { + $bodyStream = New-Object IO.StreamReader $request.InputStream + $body = $bodyStream.ReadToEnd() + $selectedSite = ConvertFrom-Json $body + Invoke-Expression $selectedSite.installRMMCommand + + # Return a simple success response + $responseString = "RMM install 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") { + Install-CyberQP + } + # Return something + $responseString = "Install CyberQP 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() + } + + "/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() + } + + + + + + "/quit" { + if ($request.HttpMethod -eq "GET") { + # Return a "shutting down" message + $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 # exit the while loop + } + } + + default { + # Handle unknown routes + $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() + } + } # end switch + } # end while +} +catch { + Write-Host "Error: $($_.Exception.Message)" +} +finally { + $listener.Stop() + $listener.Close() +} +