DscResources/ChocolateyPackage/ChocolateyPackage.psm1
|
function Get-TargetResource { [CmdletBinding()] [OutputType([System.Collections.Hashtable])] param ( [parameter(Mandatory = $true)] [ValidateSet("Present","Absent")] [System.String] $Ensure, [parameter(Mandatory = $true)] [System.String] $Name, [System.String] $Version, [Microsoft.Management.Infrastructure.CimInstance[]] $ChocolateyOptions ) $Env:Path = [Environment]::GetEnvironmentVariable('Path','Machine') Write-Verbose "Converting CIMInstance[] to hashtable" $ChocoOptions = Convert-CimInstancesToHashtable $ChocolateyOptions Import-Module $PSScriptRoot\..\..\Chocolatey.psd1 -Verbose:$False $TestParams = @{ Name = $Name } if ($Version) { $TestParams.Add('Version',$Version) } foreach ($Option in $ChocoOptions.keys) { if (!$TestParams.ContainsKey($Option) -and $option -in (Get-Command Test-ChocolateyPackageIsInstalled).Parameters.keys) { $null = $TestParams.Add($option,$ChocoOptions[$Option]) } } $InstalledPackage = Test-ChocolateyPackageIsInstalled @TestParams $returnValue = @{ Ensure = @('Absent', 'Present')[[int]$InstalledPackage.VersionGreaterOrEqual] Name = $Name Version = $Version } $returnValue } function Set-TargetResource { [CmdletBinding()] param ( [parameter(Mandatory = $true)] [ValidateSet("Present","Absent")] [System.String] $Ensure, [parameter(Mandatory = $true)] [System.String] $Name, [System.String] $Version, [Microsoft.Management.Infrastructure.CimInstance[]] $ChocolateyOptions, [System.Boolean] $UpdateOnly, [System.Management.Automation.PSCredential] $Credential ) Begin { $Env:Path = [Environment]::GetEnvironmentVariable('Path','Machine') Write-Verbose "Converting CIMInstance[] to hashtable" $ChocoOptions = Convert-CimInstancesToHashtable $ChocolateyOptions } Process { Import-Module $PSScriptRoot\..\..\Chocolatey.psd1 -verbose:$False $TestParams = @{ Name = $Name } if ($Version) { $TestParams.Add('Version',$Version) } foreach ($Option in $ChocoOptions.keys) { if (!$TestParams.ContainsKey($Option) -and $option -in (Get-Command Test-ChocolateyPackageIsInstalled).Parameters.keys) { $null = $TestParams.Add($option,$ChocoOptions[$Option]) } } $testResult = Test-ChocolateyPackageIsInstalled @TestParams $ChocoCommand = switch ($Ensure) { 'Present' { if ($testResult.PackagePresent -and !$testResult.VersionGreaterOrEqual) { Get-Command Update-ChocolateyPackage } elseif (!$UpdateOnly) { Get-Command Install-ChocolateyPackage } else { Write-Verbose "Nothing to do: UpdateOnly : $UpdateOnly" return } } 'Absent' { Get-Command Uninstall-ChocolateyPackage } } Write-Verbose "$($ChocoCommand.Name) ``" Write-Verbose "`t -Name $Name)" Write-Verbose "`t -Confirm `$False)" Write-Verbose "`t -NoProgress `$true)" $ChocoCommandParams = @{ Name = $Name Confirm = $False NoProgress = $true } if ($Version -and $Version -ne 'latest') { $ChocoCommandParams.Add('Version',$Version) } if ($Credential) { $ChocoCommandParams.Add('Credential', $Credential) } #Allow merge but no overrides foreach ($ChocoOptionName in $ChocoOptions.Keys) { if ( !$ChocoCommandParams.ContainsKey($ChocoOptionName) -and $ChocoCommand.Parameters.ContainsKey($ChocoOptionName) ) { Write-Verbose "`t -$ChocoOptionName $($ChocoOptions[$ChocoOptionName])" $ChocoCommandParams.Add($ChocoOptionName,$( if ($ChocoOptions[$ChocoOptionName] -in @('True','False')) { [bool]::Parse($ChocoOptions[$ChocoOptionName]) } else { $ChocoOptions[$ChocoOptionName] } )) } } Write-Verbose "Starting the Execution..." &$ChocoCommand @ChocoCommandParams -verbose | Write-Verbose $PostActionResult = Test-ChocolateyPackageIsInstalled @TestParams if ($PostActionResult.PackagePresent -and $PostActionResult.VersionGreaterOrEqual -and $Ensure -eq 'Present') { Write-Verbose -Message "--> Package Successfully Installed" } elseif ((!$PostActionResult.PackagePresent -or !$PostActionResult.VersionGreaterOrEqual) -and $Ensure -eq 'Absent') { Write-Verbose -Message "--> Package Successfully Removed" } else { Throw "Chocolatey Package $($ChocoCommand.verb) Failed" } } } function Test-TargetResource { [CmdletBinding()] [OutputType([System.Boolean])] param ( [parameter(Mandatory = $true)] [ValidateSet("Present","Absent")] [System.String] $Ensure, [parameter(Mandatory = $true)] [System.String] $Name, [System.String] $Version, [Microsoft.Management.Infrastructure.CimInstance[]] $ChocolateyOptions, [System.Boolean] $UpdateOnly, [System.Management.Automation.PSCredential] $Credential ) Write-Verbose "Converting CIMInstance[] to hashtable" $ChocoOptions = Convert-CimInstancesToHashtable $ChocolateyOptions $Env:Path = [Environment]::GetEnvironmentVariable('Path','Machine') Import-Module $PSScriptRoot\..\..\Chocolatey.psd1 -verbose:$False $TestParams = @{ Name = $Name } if ($Credential) { $TestParams.Add('Credential', $Credential) } #Not decided whether Version should be mandatory or not if ($Version) { $TestParams.Add('Version',$Version) } foreach ($Option in $ChocoOptions.keys) { if (!$TestParams.ContainsKey($Option) -and $option -in (Get-Command Test-ChocolateyPackageIsInstalled).Parameters.keys) { $null = $TestParams.Add($option,$ChocoOptions[$Option]) } } $EnsureResultMap = @{ 'Present' = $true 'Absent' = $false } Write-Verbose "Testing whether the Package $Name is Installed" return ($EnsureResultMap[$Ensure] -eq (Test-ChocolateyPackageIsInstalled @TestParams).VersionGreaterOrEqual) } Export-ModuleMember -Function *-TargetResource #As per Dave Wyatt's : https://powershell.org/forums/topic/hashtable-as-parameter-for-custom-dsc-resource/ function Convert-CimInstancesToHashtable([Microsoft.Management.Infrastructure.CimInstance[]] $Pairs) { Write-Verbose ">> Converting CIM Pairs" $hash = @{} foreach ($pair in $Pairs) { $hash[$pair.Key] = $pair.Value } return $hash } |