DscResources/ChocolateySource/ChocolateySource.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
    )
    <#
        ,
 
        [parameter()]
        [System.String]
        $Source,
 
        [parameter()]
        [System.Boolean]
        $Disabled,
 
        [parameter()]
        [System.Boolean]
        $ByPassProxy,
 
        [parameter()]
        [System.Boolean]
        $SelfService,
 
        [parameter()]
        [System.Int]
        $priority,
 
        [parameter()]
        [System.String]
        $username
    #>

    $Env:Path = [Environment]::GetEnvironmentVariable('Path','Machine')
    
    Import-Module $PSScriptRoot\..\..\Chocolatey.psd1 -verbose:$False

    $ChocoSourceParams = @{
        Name = $Name
    }
    switch ($PSBoundParameters.keys) {
        'Source'      {$ChocoSourceParams.add('Source',$Source)}
        'disabled'    {$ChocoSourceParams.Add('disabled',$Disabled)}
        'bypassproxy' {$ChocoSourceParams.add('bypassproxy',$bypassproxy)}
        'selfservice' {$ChocoSourceParams.add('selfservice',$selfservice)}
        'priority'    {$ChocoSourceParams.add('priority',$priority)}
    }

    if (!($SourceConfigured = Get-ChocolateySource @ChocoSourceParams) ) {
        Write-verbose ("Source $Name not found with configuration `r`n{0}" -f ($ChocoSourceParams|Format-list))
    }
    else {
        Write-Verbose "Source $Name has an exact match."
    }

    return @{
        Source      = $SourceConfigured.Source
        disabled    = $SourceConfigured.disabled
        bypassproxy = $sourceconfigured.bypassproxy
        selfservice = $SourceConfigured.selfservice
        priority    = $SourceConfigured.priority
    }
}


function Set-TargetResource
{
    [CmdletBinding()]
    param
    (
        [parameter(Mandatory = $true)]
        [ValidateSet("Present","Absent")]
        [System.String]
        $Ensure,

        [parameter(Mandatory = $true)]
        [System.String]
        $Name,

        [System.String]
        $Source,

        [System.Int16]
        $Priority,

        [System.Boolean]
        $Disabled,

        [System.Boolean]
        $BypassProxy,

        [System.Boolean]
        $SelfService,

        [System.Management.Automation.PSCredential]
        $Credential
    )

    $Env:Path = [Environment]::GetEnvironmentVariable('Path','Machine')
    
    Import-Module $PSScriptRoot\..\..\Chocolatey.psd1 -verbose:$False

    $ChocoSourceParams = @{
        Name = $Name
    }
    switch ($PSBoundParameters.keys) {
        'Source'      {$ChocoSourceParams.add('Source',$Source)}
        'disabled'    {$ChocoSourceParams.Add('disabled',$Disabled)}
        'bypassproxy' {$ChocoSourceParams.add('bypassproxy',$bypassproxy)}
        'selfservice' {$ChocoSourceParams.add('selfservice',$selfservice)}
        'priority'    {$ChocoSourceParams.add('priority', $priority)}
        'Credential'  {$ChocoSourceParams.add('Credential',$Credential)}
    }

    switch ($Ensure) {
        'Present' {
            #If we only ensure a Source is Enabled/Disabled by Name, without other arguments
            if (!$ChocoSourceParams.keys.Where{$_ -notin @('Name','Disabled')} ) {
                if($Disabled) {
                    Disable-ChocolateySource -Name $Name -NoProgress
                }
                else {
                    Enable-ChocolateySource -Name $Name -NoProgress
                }
            }
            else {
                #If we provide more information, we're given an expected state
                Register-ChocolateySource @ChocoSourceParams -noProgress
            }
        }
        'Absent' { Unregister-ChocolateySource @ChocoSourceParams -noProgress }
    }
}

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]
        $Source,

        [System.Int16]
        $Priority,

        [System.Boolean]
        $Disabled,

        [System.Boolean]
        $BypassProxy,

        [System.Boolean]
        $SelfService,

        [System.Management.Automation.PSCredential]
        $Credential
    )
    
    $Env:Path = [Environment]::GetEnvironmentVariable('Path','Machine')
    Import-Module $PSScriptRoot\..\..\Chocolatey.psd1 -verbose:$False

    $ChocoSourceParams = @{
        Name = $Name
    }
    switch ($PSBoundParameters.keys) {
        'Source'      {$ChocoSourceParams.add('Source',$Source)}
        'disabled'    {$ChocoSourceParams.Add('disabled',$Disabled)}
        'bypassproxy' {$ChocoSourceParams.add('bypassproxy',$bypassproxy)}
        'selfservice' {$ChocoSourceParams.add('selfservice',$selfservice)}
        'priority'    {$ChocoSourceParams.add('priority', $priority)}
        'Credential'  {$ChocoSourceParams.add('Credential',$Credential)}
    }

    $EnsureResultMap = @{
        'Present'=$true
        'Absent'=$false
    }
    
    #Test-Chocolatey source return false when source name does not exist
    # the differences of compare-object when not an exact match with properties

    if(($SourceDifferences = Test-ChocolateySource @ChocoSourceParams) -or $SourceDifferences -eq $false) {
        if( ($SourceDifferences -eq $False) -and 
            $Disabled -and  
            $Ensure -eq 'Present' -and
            !($ChocoSourceParams.Keys.where{$_ -notin @('Name','Disabled') } ) 
        ) {
            # 1. The Source does not exist
            # 2. We want to ensure that
            # 3. it's only disabled
            # 4. we don't provide more data to register a source as disabled
            # so it does not need disabling (edge case)
            $SourceDoesMatch = $True
        }
        else {
            $SourceDoesMatch = $false
        }
        
        # If the source does not match, it's a good thing when Ensure = 'Absent'
        Write-Verbose "Differences = `r`n$($SourceDifferences | Out-String)"
    }
    else {
        Write-Verbose "The sources match exactly with the provided parameters."
        $SourceDoesMatch = $true
    }

    return ($EnsureResultMap[$Ensure] -eq $SourceDoesMatch )
}


Export-ModuleMember -Function *-TargetResource