update editintunescript to support other script types

Author: JohnDuprey

Modules/CIPPCore/Public/Entrypoints/HTTP Functions/Endpoint/MEM/Invoke-EditIntuneScript.ps1

@@ -14,35 +14,110 @@ function Invoke-EditIntuneScript {
     $Headers = $Request.Headers
     Write-LogMessage -Headers $Headers -API $APINAME -message 'Accessed this API' -Sev Debug
 
-    $graphUrl = "https://graph.microsoft.com/beta"
-    switch($Request.Method) {
-        "GET" {
-            $parms = @{
-                uri = "$graphUrl/deviceManagement/deviceManagementScripts/$($Request.Query.ScriptId)"
-                tenantid = $Request.Query.TenantFilter
+    $graphUrl = 'https://graph.microsoft.com/beta'
+
+    # Define the endpoint based on script type
+    function Get-ScriptEndpoint {
+        param (
+            [Parameter(Mandatory = $true)]
+            [string]$ScriptType
+        )
+
+        switch ($ScriptType) {
+            'Windows' { return 'deviceManagement/deviceManagementScripts' }
+            'MacOS' { return 'deviceManagement/deviceShellScripts' }
+            'Remediation' { return 'deviceManagement/deviceHealthScripts' }
+            'Linux' { return 'deviceManagement/configurationPolicies' }
+            default { return 'deviceManagement/deviceManagementScripts' }
+        }
+    }
+
+    switch ($Request.Method) {
+        'GET' {
+            # First get the script type by querying the script ID
+            $scriptId = $Request.Query.ScriptId
+            $scriptTypeFound = $false
+
+            # Try each endpoint to find the script
+            foreach ($scriptType in @('Windows', 'MacOS', 'Remediation', 'Linux')) {
+                $endpoint = Get-ScriptEndpoint -ScriptType $scriptType
+                $parms = @{
+                    uri      = "$graphUrl/$endpoint/$scriptId"
+                    tenantid = $Request.Query.TenantFilter
+                }
+
+                try {
+                    $intuneScript = New-GraphGetRequest @parms -ErrorAction Stop
+                    if ($intuneScript) {
+                        $intuneScript | Add-Member -MemberType NoteProperty -Name scriptType -Value $scriptType -Force
+                        $scriptTypeFound = $true
+                        break
+                    }
+                } catch {
+                    # Script not found in this endpoint, try next one
+                    continue
+                }
             }
 
-            $intuneScript = New-GraphGetRequest @parms
-            Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
-                StatusCode = [HttpStatusCode]::OK
-                Body       = $intuneScript
-            })
+            if ($scriptTypeFound) {
+                Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
+                        StatusCode = [HttpStatusCode]::OK
+                        Body       = $intuneScript
+                    })
+            } else {
+                Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
+                        StatusCode = [HttpStatusCode]::NotFound
+                        Body       = "Script with ID $scriptId was not found in any endpoint."
+                    })
+            }
         }
-        "PATCH" {
+        'PATCH' {
+            # Parse the script data to determine type
+            $scriptData = $Request.Body.IntuneScript | ConvertFrom-Json
+            $scriptType = $Request.Body.ScriptType
+
+            if (-not $scriptType) {
+                # Try to determine script type from the request body
+                if ($scriptData.PSObject.Properties.Name -contains '@odata.type') {
+                    switch ($scriptData.'@odata.type') {
+                        '#microsoft.graph.deviceManagementScript' { $scriptType = 'Windows' }
+                        '#microsoft.graph.deviceShellScript' { $scriptType = 'MacOS' }
+                        '#microsoft.graph.deviceHealthScript' { $scriptType = 'Remediation' }
+                        default {
+                            if ($scriptData.platforms -eq 'linux' -and $scriptData.templateReference.templateFamily -eq 'deviceConfigurationScripts') {
+                                $scriptType = 'Linux'
+                            } else {
+                                $scriptType = 'Windows' # Default to Windows if no definitive type found
+                            }
+                        }
+                    }
+                }
+            }
+
+            $endpoint = Get-ScriptEndpoint -ScriptType $scriptType
             $parms = @{
-                uri = "$graphUrl/deviceManagement/deviceManagementScripts/$($Request.Body.ScriptId)"
+                uri      = "$graphUrl/$endpoint/$($Request.Body.ScriptId)"
                 tenantid = $Request.Body.TenantFilter
-                body = $Request.Body.IntuneScript
+                body     = $Request.Body.IntuneScript
+            }
+
+            try {
+                $patchResult = New-GraphPOSTRequest @parms -type 'PATCH'
+                $body = [pscustomobject]@{'Results' = $patchResult }
+                Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
+                        StatusCode = [HttpStatusCode]::OK
+                        Body       = $body
+                    })
+            } catch {
+                $ErrorMessage = Get-CippException -Exception $_
+                Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
+                        StatusCode = [HttpStatusCode]::BadRequest
+                        Body       = "Failed to update script: $($ErrorMessage.NormalizedError)"
+                    })
             }
-            $patchResult = New-GraphPOSTRequest @parms -type "PATCH"
-            $body = [pscustomobject]@{'Results' = $patchResult }
-            Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
-                StatusCode = [HttpStatusCode]::OK
-                Body       = $body
-            })
         }
-        "POST" {
-            Write-Output "Adding script"
+        'POST' {
+            Write-Output 'Adding script'
         }
     }
 }