init windowshost

computer_use_demo/windowshost/vm/win11setup/setupscripts/setup-tools.psm1

@@ -0,0 +1,197 @@
+function Get-Tools {
+    param(
+        [string]$toolsConfigJson
+    )
+
+    # Convert the JSON string to a PowerShell object
+    $toolsList = $toolsConfigJson | ConvertFrom-Json
+
+    return $toolsList
+}
+
+function Get-ToolDetails {
+    param(
+        $toolsList,
+        [string]$toolName
+    )
+
+    # Check if the program exists in the JSON data
+    if ($toolsList.PSObject.Properties.Name -contains $toolName) {
+        # Return the program details as a PowerShell object
+        return $toolsList.$toolName
+    } else {
+        # Handle the case where the program is not found
+        Write-Host "Program '$toolName' not found in the list."
+        return $null
+    }
+}
+
+function Invoke-DownloadFileFromAvailableMirrors {
+    param (
+        [string[]]$mirrorUrls,
+        [string]$outfile
+    )
+    foreach ($url in $mirrorUrls) {
+        try {
+            $result = Invoke-DownloadFile -url $url -outfile $outfile
+            if ($result -eq $true) {
+                Write-Host "Downloaded using $url"
+                return $true
+            }
+        } catch {
+            Write-Host "Error downloading from $url. Please check and update the mirrors."
+        }
+    }
+
+    Write-Host "Downloading from the provided mirrors failed. Please check and update the mirrors."
+    return $false
+}
+
+function Invoke-DownloadFile {
+    param (
+        [string]$url,
+        [string]$outfile
+    )
+    # Makes download faster by disabling progress bar
+    $ProgressPreference = "SilentlyContinue"
+
+    $retryCount = 0
+    $maxRetries = 3
+    $sleepSeconds = 2
+    $maxSleepSeconds = 10
+    $userAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.36"
+
+    # Ensure directory exists
+    $directory = Split-Path -Path $outfile -Parent
+    if (-Not (Test-Path -Path $directory)) {
+        Write-Host "Creating directory $directory..."
+        New-Item -Path $directory -ItemType Directory -Force | Out-Null
+    }
+
+    while ($retryCount -lt $maxRetries) {
+        try {
+            Invoke-RestMethod -Uri $url -OutFile $outfile -Headers @{"User-Agent" = $userAgent}
+            Write-Host "Download successful, file saved to: $outfile"
+            break
+        } catch {
+            $retryCount++
+            Write-Host "Attempt $retryCount of $maxRetries failed. Error: $($_.Exception.Message)"
+            Start-Sleep -Seconds $sleepSeconds
+            $sleepSeconds = [Math]::Min($sleepSeconds * 2, $maxSleepSeconds)  # Exponential backoff with a cap
+        }
+    }
+
+    if ($retryCount -eq $maxRetries) {
+        Write-Host "Failed to download the file after $maxRetries attempts."
+        return $false
+    }
+
+    return $true
+}
+
+function Add-ToEnvPath {
+    param (
+        [string]$NewPath
+    )
+
+    # Get the current PATH environment variable
+    $envPath = [Environment]::GetEnvironmentVariable("PATH", "Machine")
+
+    # Append the new path to the existing PATH
+    $newPath = "$envPath;$NewPath"
+
+    # Set the updated PATH environment variable
+    [Environment]::SetEnvironmentVariable("PATH", $newPath, "Machine")
+    
+    # Fetch updates from the shell
+    $env:PATH += ";${newPath}"
+}
+
+function Register-LogonTask {
+    param(
+
+        [parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = "Name of the scheduled task")]
+        [string]
+        $TaskName,
+
+        [parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = "Path to the .py script")]
+        [string]
+        $ScriptPath,
+        
+        [parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, HelpMessage = "Arguments to the .py script")]
+        [string]
+        $Arguments = "",
+
+        [parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, HelpMessage = "Local Account username")]
+        [string]
+        $LocalUser,
+
+        [parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, HelpMessage = "Local Account password")]
+        [string]
+        $LocalPassword,
+
+        [parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, HelpMessage = "Whether to execute the command as SYSTEM")]
+        [switch]
+        $AsSystem = $false,
+
+        [parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, HelpMessage = "logging file")]
+        [string]
+        $LogFilePath
+    )
+    
+    $scriptDirectory = Split-Path $ScriptPath
+    
+    $taskActionArgument = "-ExecutionPolicy Bypass -windowstyle hidden -Command `"try { . '$ScriptPath' $Arguments } catch { Write `$_.Exception.Message | Out-File $($TaskName)_Log.txt } finally { } `""    
+    $taskAction = New-ScheduledTaskAction -Execute "$PSHome\powershell.exe" -Argument $taskActionArgument -WorkingDirectory $scriptDirectory
+    
+    $params = @{
+        Force    = $True
+        Action   = $taskAction
+        RunLevel = "Highest"
+        TaskName = $TaskName
+    }
+
+    $taskTrigger = New-ScheduledTaskTrigger -AtLogOn
+    $params.Add("Trigger", $taskTrigger)
+
+    if ($AsSystem) {
+        $params.Add("User", "NT AUTHORITY\SYSTEM")
+    }
+    else {
+        $params.Add("User", $LocalUser)
+        if ($LocalPassword) {
+            $params.Add("Password", $LocalPassword)
+        }
+    }
+
+    Write-Host "Registering scheduled task '$TaskName' to run 'powershell.exe $taskActionArgument'..."
+    Register-ScheduledTask @params
+}
+
+# Function to attempt pip install and handle failures
+function Install-PythonPackages {
+    param (
+        [string]$Package = "",
+        [string]$Arguments = "",
+        [string]$RequirementsPath = ""
+    )
+    $RetryCount = 3
+    $currentAttempt = 0
+    while ($currentAttempt -lt $RetryCount) {
+        if (-not [string]::IsNullOrWhiteSpace($RequirementsPath)) {
+            & python -m pip install --no-cache-dir -r $RequirementsPath $Arguments
+        } else {
+            & python -m pip install --no-cache-dir $Package $Arguments
+        }
+        if ($LASTEXITCODE -eq 0) {
+            Write-Host "Installation successful."
+            return
+        } else {
+            Write-Host "Attempt $($currentAttempt + 1) failed. Retrying..."
+            Start-Sleep -Seconds 10
+            $currentAttempt++
+        }
+    }
+    Write-Error "Failed to install after $RetryCount attempts."
+    exit
+}