Set operations for psresources working

Author: adityapatwardhan

src/dsc/psresourceget.ps1

@@ -12,7 +12,6 @@ param(
     [Parameter(ValueFromPipeline)]
     $stdinput
 )
-
 function Write-Trace {
     param(
         [string]$message,
@@ -23,7 +22,21 @@ function Write-Trace {
         $level.ToLower() = $message
     } | ConvertTo-Json -Compress
 
-    $host.ui.WriteErrorLine($trace)
+    if ($level -eq 'Error') {
+        $host.ui.WriteErrorLine($trace)
+    }
+    elseif ($level -eq 'Warning') {
+        $host.ui.WriteWarningLine($trace)
+    }
+    elseif ($level -eq 'Verbose') {
+        $host.ui.WriteVerboseLine($trace)
+    }
+    elseif ($level -eq 'Debug') {
+        $host.ui.WriteDebugLine($trace)
+    }
+    else {
+        $host.ui.WriteInformation($trace)
+    }
 }
 
 # catch any un-caught exception and write it to the error stream
@@ -32,10 +45,6 @@ trap {
     exit 1
 }
 
-function GetAllPSResources {
-    $resources = Get-PSResource
-}
-
 function GetOperation {
     param(
         [string]$ResourceType
@@ -55,13 +64,13 @@ function GetOperation {
             return $ret
         }
 
-        'repositories' { return 'Get-PSRepository' }
-        'psresource' { return 'Get-PSResource' }
+        'repositories' { throw [System.NotImplementedException]::new("Get operation is not implemented for Repositories resource.") }
+        'psresource' { throw [System.NotImplementedException]::new("Get operation is not implemented for PSResource resource.") }
         'psresources' {
-
-            $allPSResources =  if ($inputObj.scope) {
+            $allPSResources = if ($inputObj.scope) {
                 Get-PSResource -Scope $inputObj.Scope
-            } else {
+            }
+            else {
                 Get-PSResource
             }
 
@@ -97,7 +106,8 @@ function GetOperation {
             }
 
             PopulatePSResourcesObjectByRepository -resourcesExist $resourcesExist -inputResources $inputObj.resources -repositoryName $inputObj.repositoryName -scope $inputObj.Scope
-         }
+        }
+
         default { throw "Unknown ResourceType: $ResourceType" }
     }
 }
@@ -112,16 +122,89 @@ function ExportOperation {
             }
         }
 
-        'repositories' { return 'Get-PSRepository' }
-        'psresource' { return 'Get-PSResource' }
+        'repositories' { throw [System.NotImplementedException]::new("Get operation is not implemented for Repositories resource.") }
+        'psresource' { throw [System.NotImplementedException]::new("Get operation is not implemented for PSResource resource.") }
         'psresources' {
             $allPSResources = Get-PSResource
             PopulatePSResourcesObject -allPSResources $allPSResources
-         }
+        }
         default { throw "Unknown ResourceType: $ResourceType" }
     }
 }
 
+function SetPSResources {
+    param(
+        $inputObj
+    )
+
+    $repositoryName = $inputObj.repositoryName
+    $scope = $inputObj.scope
+
+    if (-not $scope) {
+        $scope = 'CurrentUser'
+    }
+
+    $resourcesToUninstall = @()
+    $resourcesToInstall = [System.Collections.Generic.Dictionary[string, psobject]]::new()
+
+    Add-Type -AssemblyName "$PSScriptRoot/dependencies/NuGet.Versioning.dll"
+
+    $inputObj.resources | ForEach-Object {
+        $resource = $_
+        $name = $resource.name
+        $version = $resource.version
+
+        $getSplat = @{
+            Name   = $name
+            Scope  = $scope
+            ErrorAction = 'SilentlyContinue'
+        }
+
+        $existingResources = if ($repositoryName) {
+            Get-PSResource @getSplat | Where-Object { $_.Repository -eq $repositoryName }
+        }
+        else {
+            Get-PSResource @getSplat
+        }
+
+        # uninstall all resources that do not satisfy the version range and install the ones that do
+        $existingResources | ForEach-Object {
+            $versionRange = [NuGet.Versioning.VersionRange]::Parse($version)
+            $resourceVersion = [NuGet.Versioning.NuGetVersion]::Parse($_.Version.ToString())
+            if (-not $versionRange.Satisfies($resourceVersion)) {
+                if ($resource._exists) {
+                    #$resourcesToInstall += $resource
+                    $key = $resource.Name.ToLowerInvariant() + '-' + $resource.Version.ToLowerInvariant()
+                    if (-not $resourcesToInstall.ContainsKey($key)) {
+                        $resourcesToInstall[$key] = $resource
+                    }
+                }
+
+                $resourcesToUninstall += $_
+            }
+            else {
+                if (-not $resource._exists) {
+                    $resourcesToUninstall += $_
+                }
+            }
+        }
+    }
+
+    if ($resourcesToUninstall.Count -gt 0) {
+        Write-Trace -message "Uninstalling resources: $($resourcesToUninstall | ForEach-Object { "$($_.Name) - $($_.Version)" })" -Level Verbose
+        $resourcesToUninstall | ForEach-Object {
+            Uninstall-PSResource -Name $_.Name -Scope $scope -ErrorAction Stop
+        }
+    }
+
+    if ($resourcesToInstall.Count -gt 0) {
+        Write-Trace -message "Installing resources: $($resourcesToInstall.Values | ForEach-Object { " $($_.Name) -- $($_.Version) " })" -Level Verbose
+        $resourcesToInstall.Values | ForEach-Object {
+            Install-PSResource -Name $_.Name -Version $_.Version -Scope $scope -Repository $repositoryName -ErrorAction Stop
+        }
+    }
+}
+
 function SetOperation {
     param(
         [string]$ResourceType
@@ -165,13 +248,12 @@ function SetOperation {
             return GetOperation -ResourceType $ResourceType
         }
 
-        'repositories' { return 'Set-PSRepository' }
-        'psresource' { return 'Set-PSResource' }
-        'psresources' { return 'Set-PSResource' }
+        'repositories' { throw [System.NotImplementedException]::new("Get operation is not implemented for Repositories resource.") }
+        'psresource' { throw [System.NotImplementedException]::new("Get operation is not implemented for PSResource resource.") }
+        'psresources' { return SetPSResources -inputObj $inputObj }
         default { throw "Unknown ResourceType: $ResourceType" }
     }
 }
-
 function FilterPSResourcesByRepository {
     param (
         $allPSResources,
@@ -247,15 +329,15 @@ function PopulatePSResourcesObject {
 
         $resources = $_.Group | ForEach-Object {
             [pscustomobject]@{
-                name        = $_.Name
-                version     = $_.Version.ToString()
-                _exists     = $true
+                name    = $_.Name
+                version = $_.Version.ToString()
+                _exists = $true
             }
         }
 
         $resourcesObj = [pscustomobject]@{
             repositoryName = $repoName
-            resources  = $resources
+            resources      = $resources
         }
 
         $resourcesObj | ConvertTo-Json -Compress
@@ -297,8 +379,8 @@ function PopulateRepositoryObject {
 }
 
 switch ($Operation.ToLower()) {
-    'get'  { return (GetOperation -ResourceType $ResourceType) }
-    'set'  { return (SetOperation -ResourceType $ResourceType) }
+    'get' { return (GetOperation -ResourceType $ResourceType) }
+    'set' { return (SetOperation -ResourceType $ResourceType) }
     'export' { return (ExportOperation -ResourceType $ResourceType) }
     default { throw "Unknown Operation: $Operation" }
 }