syelle/Logo
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Update SVSTaskGate.ps1

Browse Source
This commit is contained in:
Stephan Yelle
2025-01-31 16:39:05 -05:00
parent 194c4afcb9
commit c4a5d42462
1 changed files with 195 additions and 123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

312
SVSTaskGate.ps1
View File

@@ -8,10 +8,6 @@
### need to fix path in the uninstall-DattoEDR -
####### ❌ [Error] [GeneralTask] Uninstallation command 'C:\Program Files\Infocyte\Agent\agent.exe' not found. (Event ID: 3000) - bad path
### need to have the fetch button check if Install-DattoRMM function exist if not run the build in helpder function to fetch sites
### need to move the tweaks to the tweeks tab
# ---------------------------------------------------------------------------
# 1) CREATE A GLOBAL LOG CACHE (NEW)
@@ -222,12 +218,22 @@ function Install-DattoRMM-Helper {
#region SVS Module
### SVS Module Overview:
# - This section includes the Install-SVSMSP function, which handles:
# 1. Cleanup of old modules and repositories.
# 2. Installation of new modules and repositories.
# 3. Setting execution policies and prerequisites.
# - TODO:
# 1. Validate all parameters for completeness and add error messages for missing inputs.
# 2. Refactor cleanup and installation steps for modularity and reusability.
function Install-SVSMSP {
param (
# Cleanup flag
# Cleanup flag: Removes old modules and repositories if enabled.
[switch]$Cleanup,
# Toolkit installation flag
# Toolkit installation flag: Installs required modules and repositories.
[switch]$InstallToolkit,
# Module settings
@@ -265,21 +271,15 @@ function Install-SVSMSP {
),
# Log file path
# Log file path for tracking installation steps
[Parameter(Mandatory = $false)]
[string]$LogFilePath = "$env:SVSMSP\svstoolkit.log",
[string]$LogFilePath = "$env:SVSMSP\svstoolkit.log"
# DRMM API Settings
[Parameter(Mandatory = $false)]
[string]$ApiUrl,
[Parameter(Mandatory = $false)]
[string]$ApiKey,
[Parameter(Mandatory = $false)]
[string]$ApiSecretKey
)
### Function: Perform-Cleanup
# - Removes all old modules and repositories.
# - Logs each step for debugging purposes.
function Perform-Cleanup {
Write-LogHybrid -Message "Cleanup mode enabled. Starting cleanup process..." -Level "Info" -LogToEvent
@@ -326,11 +326,15 @@ function Install-SVSMSP {
Write-LogHybrid -Message "Cleanup process completed successfully." -Level "Success" -LogToEvent
}
### Function: Perform-ToolkitInstallation
# - Handles the installation of new repositories and modules.
# - Ensures execution policies are correctly set.
function Perform-ToolkitInstallation {
# Perform cleanup
# Step 1: Cleanup old modules and repositories
Perform-Cleanup
# Step 1: Set Execution Policy
# Step 2: Set Execution Policy
$localMachineExecutionPolicy = Get-ExecutionPolicy -Scope LocalMachine
if ($localMachineExecutionPolicy -ne "RemoteSigned") {
Write-LogHybrid -Message "Setting execution policy to RemoteSigned..." -Level "Warning" -LogToEvent
@@ -344,20 +348,10 @@ function Install-SVSMSP {
}
}
# 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
# try {
Install-PackageProvider -Name "NuGet" -Force -Scope AllUsers -Confirm:$false
# Write-LogHybrid -Message "NuGet package provider installed successfully." -Level "Success" -LogToEvent
# }
# catch {
# Write-LogHybrid -Message "Failed to install NuGet package provider. Error: $_" -Level "Error" -LogToEvent
# return
# }
#}
# Step 3: Ensure NuGet is Installed
Install-PackageProvider -Name "NuGet" -Force -Scope AllUsers -Confirm:$false
# Step 3: Register the new repository
# Step 4: Register the new repository
Write-LogHybrid -Message "Registering the new repository '$NewRepositoryName'..." -Level "Info" -LogToEvent
try {
if (!(Get-PSRepository -Name $NewRepositoryName -ErrorAction SilentlyContinue)) {
@@ -369,7 +363,7 @@ function Install-SVSMSP {
Write-LogHybrid -Message "Failed to register new repository '$NewRepositoryName'. Error: $($_.Exception.Message)" -Level "Error" -LogToEvent
}
# Step 4: Install the new module
# Step 5: 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
@@ -382,6 +376,8 @@ function Install-SVSMSP {
Write-LogHybrid -Message "Toolkit installation process completed successfully." -Level "Success" -LogToEvent
}
### Function Execution Logic
# - Determines whether to run cleanup, toolkit installation, or default to toolkit installation mode.
Write-LogHybrid -Message "Install-SVSMSP function started." -Level "Info" -LogToEvent
if ($Cleanup) {
@@ -400,11 +396,39 @@ function Install-SVSMSP {
#endregion SVS Module
# ----------------------------------------------------------------------------------
# START THE LISTENER
# START THE HTTP LISTENER
# ----------------------------------------------------------------------------------
$listener = New-Object System.Net.HttpListener
$listener.Prefixes.Add("http://localhost:8081/")
$listener.Start()
# This listener serves as the backend for handling requests from the GUI.
# It supports multiple routes, such as:
# - "/" (Root): Serves the HTML GUI.
# - "/getn8npw": Fetches n8n password and site information.
# - "/installSVSMSPModule": Triggers the installation of SVSMSP modules.
# - "/installrmm": Handles RMM installation with dynamic parameters.
# - Additional routes for tweaks and other tasks.
### Listener Initialization with Error Handling
try {
# Create a new HttpListener object
$listener = New-Object System.Net.HttpListener
# Check if the object was successfully created
if (-not $listener) {
throw "Failed to initialize HttpListener."
}
# Add prefix for the listener
$listener.Prefixes.Add("http://localhost:8081/")
Write-LogHybrid -Message "Listener initialized with prefix http://localhost:8081/" -Level "Info"
# Start the listener
$listener.Start()
Write-LogHybrid -Message "Listener started successfully." -Level "Info"
} catch {
# Log the error and rethrow it for visibility
Write-LogHybrid -Message "Critical error initializing listener: $($_.Exception.Message)" -Level "Error"
throw $_
}
function Get-N8nWebhookData {
@@ -903,73 +927,64 @@ function GetHtmlContent {
<p>Logs will appear here...</p>
</div>
</div>
<script>
<script>
function toggleOnboardCheckboxes(selectedCheckbox) {
const checkboxes = document.querySelectorAll('#onboardTab input[type="checkbox"]');
const dattoRMMCheckbox = document.getElementById('installDattoRMMCheckbox');
const optionsContainer = document.getElementById('dattoRMMOptionsContainer');
const n8nPasswordContainer = document.getElementById('n8nPasswordContainer');
const dattoRMMContainer = document.getElementById('DattoRMMContainer');
// Central function to handle DattoRMM visibility
function toggleDattoRMMVisibility() {
const dattoRMMCheckbox = document.getElementById('installDattoRMMCheckbox');
const optionsContainer = document.getElementById('dattoRMMOptionsContainer');
const n8nPasswordContainer = document.getElementById('n8nPasswordContainer');
const dattoRMMContainer = document.getElementById('DattoRMMContainer');
if (dattoRMMCheckbox.checked) {
optionsContainer.style.display = 'block';
n8nPasswordContainer.style.display = 'block';
dattoRMMContainer.style.display = 'block';
} else {
optionsContainer.style.display = 'none';
n8nPasswordContainer.style.display = 'none';
dattoRMMContainer.style.display = 'none';
}
if (dattoRMMCheckbox.checked) {
optionsContainer.style.display = 'block';
n8nPasswordContainer.style.display = 'block';
dattoRMMContainer.style.display = 'block';
} else {
optionsContainer.style.display = 'none';
n8nPasswordContainer.style.display = 'none';
dattoRMMContainer.style.display = 'none';
}
}
function toggleLeftColumnCheckboxes(selectAllCheckbox) {
const leftCheckboxes = document.querySelectorAll('#leftColumn input[type="checkbox"]:not(#selectAllLeftCheckbox)');
const dattoRMMCheckbox = document.getElementById('installDattoRMMCheckbox');
const optionsContainer = document.getElementById('dattoRMMOptionsContainer');
const n8nPasswordContainer = document.getElementById('n8nPasswordContainer');
const dattoRMMContainer = document.getElementById('DattoRMMContainer');
// Function to update "Select All" checkbox state
function updateSelectAllCheckbox(selectAllId, checkboxGroupSelector) {
const selectAllCheckbox = document.getElementById(selectAllId);
const checkboxes = document.querySelectorAll(checkboxGroupSelector);
// Toggle all checkboxes
leftCheckboxes.forEach(checkbox => {
checkbox.checked = selectAllCheckbox.checked;
});
// Uncheck "Select All" if any checkbox is unchecked
selectAllCheckbox.checked = Array.from(checkboxes).every(checkbox => checkbox.checked);
}
// Check if "Install DattoRMM" is selected
if (dattoRMMCheckbox.checked) {
optionsContainer.style.display = 'block';
n8nPasswordContainer.style.display = 'block';
dattoRMMContainer.style.display = 'block';
} else {
optionsContainer.style.display = 'none';
n8nPasswordContainer.style.display = 'none';
dattoRMMContainer.style.display = 'none';
}
}
// Function to handle "Select All" logic for the left column
function toggleLeftColumnCheckboxes(selectAllCheckbox) {
const leftCheckboxes = document.querySelectorAll('#leftColumn input[type="checkbox"]:not(#selectAllLeftCheckbox)');
function updateSelectAllonboard() {
const selectAllCheckbox = document.getElementById('selectAllOnboardCheckbox');
const checkboxes = document
.getElementById('onboardTab')
.querySelectorAll('input[type="checkbox"]:not(#selectAllOnboardCheckbox)');
// If any checkbox is unchecked, uncheck "Select All"
selectAllCheckbox.checked = Array.from(checkboxes).every(checkbox => checkbox.checked);
}
function gatherSelectedTasks() {
const selectedLeftTasks = Array.from(document.querySelectorAll('.left-checkbox:checked')).map(checkbox => checkbox.name);
const selectedRightTasks = Array.from(document.querySelectorAll('.right-checkbox:checked')).map(checkbox => checkbox.name);
return [...selectedLeftTasks, ...selectedRightTasks];
}
// Attach the updateSelectAllonboard function to all individual checkboxes
document.querySelectorAll('#onboardTab input[type="checkbox"]:not(#selectAllOnboardCheckbox)').forEach(checkbox => {
checkbox.addEventListener('change', updateSelectAllonboard);
// Toggle all checkboxes
leftCheckboxes.forEach(checkbox => {
checkbox.checked = selectAllCheckbox.checked;
});
// Handle DattoRMM visibility
toggleDattoRMMVisibility();
}
// Function to handle checkbox changes in the Onboard tab
function toggleOnboardCheckboxes(selectedCheckbox) {
// Update DattoRMM visibility
toggleDattoRMMVisibility();
// Update "Select All" checkbox state
updateSelectAllCheckbox('selectAllLeftCheckbox', '#onboardTab input[type="checkbox"]:not(#selectAllLeftCheckbox)');
}
// Attach event listeners to dynamically update the "Select All" checkbox
document.querySelectorAll('#onboardTab input[type="checkbox"]:not(#selectAllLeftCheckbox)').forEach(checkbox => {
checkbox.addEventListener('change', () => {
updateSelectAllCheckbox('selectAllLeftCheckbox', '#onboardTab input[type="checkbox"]:not(#selectAllLeftCheckbox)');
});
});
function toggleOffboardCheckboxes(selectAllCheckbox) {
const checkboxes = document.querySelectorAll('#offboardTab input[type="checkbox"]:not(#selectAllOffboardCheckbox)');
checkboxes.forEach(checkbox => {
@@ -1111,18 +1126,36 @@ function GetHtmlContent {
alert("Tweaks applied successfully!");
}
function toggleDattoRMMOptions() {
const checkbox = document.getElementById('installDattoRMMCheckbox');
const dattoRMMCheckbox = document.getElementById('installDattoRMMCheckbox');
const optionsContainer = document.getElementById('dattoRMMOptionsContainer');
const n8nPasswordContainer = document.getElementById('n8nPasswordContainer');
const DattoRMMContainer = document.getElementById('DattoRMMContainer');
const dattoRMMContainer = document.getElementById('DattoRMMContainer');
const subCheckboxes = document.querySelectorAll('#dattoRMMOptionsContainer input[type="checkbox"]');
optionsContainer.style.display = checkbox.checked ? 'block' : 'none';
n8nPasswordContainer.style.display = checkbox.checked ? 'block' : 'none';
DattoRMMContainer.style.display = checkbox.checked ? 'block' : 'none';
// Toggle visibility of sub-options
if (dattoRMMCheckbox.checked) {
optionsContainer.style.display = 'block';
n8nPasswordContainer.style.display = 'block';
dattoRMMContainer.style.display = 'block';
// Automatically check sub-checkboxes
subCheckboxes.forEach(subCheckbox => {
subCheckbox.checked = true;
});
} else {
optionsContainer.style.display = 'none';
n8nPasswordContainer.style.display = 'none';
dattoRMMContainer.style.display = 'none';
// Automatically uncheck sub-checkboxes
subCheckboxes.forEach(subCheckbox => {
subCheckbox.checked = false;
});
}
}
const tabButtons = document.querySelectorAll('.tab-button');
const tabContents = document.querySelectorAll('.tab-content');
const logArea = document.getElementById('logArea');
@@ -1399,12 +1432,23 @@ Start-Process "msedge.exe" -ArgumentList "--app=http://localhost:8081/"
try {
while ($listener.IsListening) {
### Process Incoming Requests
# - `$context`: Contains the request and response objects.
# - `$request`: Represents the HTTP request.
# - `$response`: Represents the HTTP response.
$context = $listener.GetContext()
$request = $context.Request
$response = $context.Response
### Route Handling
# - Routes are matched based on the `AbsolutePath` of the request URL.
# - Each route corresponds to a specific action or task.
switch ($request.Url.AbsolutePath) {
# ----------------------------------------------------------------
# ROOT ROUTE ("/")
# Serves the main HTML GUI to the client.
# ----------------------------------------------------------------
"/" {
$htmlContent = GetHtmlContent
$buffer = [System.Text.Encoding]::UTF8.GetBytes($htmlContent)
@@ -1414,29 +1458,46 @@ try {
$response.OutputStream.Close()
}
# ----------------------------------------------------------------
# ROUTE: /getn8npw
# Fetches the n8n password and retrieves DattoRMM site details.
# ----------------------------------------------------------------
"/getn8npw" {
if ($request.HttpMethod -eq "POST") {
$bodyStream = New-Object IO.StreamReader $request.InputStream
$body = $bodyStream.ReadToEnd()
$data = ConvertFrom-Json $body
$password = $data.password
try {
# Parse the JSON body to extract the password.
$bodyStream = New-Object IO.StreamReader $request.InputStream
$body = $bodyStream.ReadToEnd()
$data = ConvertFrom-Json $body
$password = $data.password
Get-N8nWebhookData -AuthHeaderValue $password
$sites = Install-DattoRMM-Helper -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")
# Call the webhook to fetch site details.
Get-N8nWebhookData -AuthHeaderValue $password
$sites = Install-DattoRMM-Helper -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
}
# Return site details as JSON.
$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()
}catch {
Write-LogHybrid -Message "Error processing /getn8npw: $($_.Exception.Message)" -Level "Error"
$response.StatusCode = 500
$buffer = [System.Text.Encoding]::UTF8.GetBytes("Error: Failed to process the request.")
$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()
}
}
@@ -1914,6 +1975,17 @@ catch {
}
finally {
$listener.Stop()
$listener.Close()
if ($listener -ne $null) {
try {
Write-LogHybrid -Message "Stopping the listener." -Level "Info"
$listener.Stop()
$listener.Close()
Write-LogHybrid -Message "Listener stopped successfully." -Level "Info"
} catch {
Write-LogHybrid -Message "Error stopping the listener: $($_.Exception.Message)" -Level "Error"
}
} else {
Write-LogHybrid -Message "Listener object is null; nothing to stop." -Level "Warning"
}
}