functions/namedLocations/Register-TmfNamedLocation.ps1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
function Register-TmfNamedLocation
{
    [CmdletBinding(DefaultParameterSetName = 'IPRanges')]
    Param (
        [Parameter(Mandatory = $true)]
        [string] $displayName,
        [string[]] $oldNames,

        [Parameter(Mandatory = $true)]
        [ValidateSet('countryNamedLocation', 'ipNamedLocation')]
        [string] $type = "ipNamedLocation",
        
        [Parameter(Mandatory = $true, ParameterSetName = "IPRanges")]
        [object[]] $ipRanges,
        [Parameter(ParameterSetName = "IPRanges")]
        [bool] $isTrusted = $false,
        [Parameter(Mandatory = $true, ParameterSetName = "Country")]
        [string[]] $countriesAndRegions,
        [Parameter(ParameterSetName = "Country")]
        [bool] $includeUnknownCountriesAndRegions = $false,

        [bool] $present = $true,        

        [string] $sourceConfig = "<Custom>",

        [System.Management.Automation.PSCmdlet]
        $Cmdlet = $PSCmdlet
    )
    
    begin
    {
        $resourceName = "namedLocations"
        if (!$script:desiredConfiguration[$resourceName]) {
            $script:desiredConfiguration[$resourceName] = @()
        }

        if ($script:desiredConfiguration[$resourceName].displayName -contains $displayName) {            
            $alreadyLoaded = $script:desiredConfiguration[$resourceName] | Where-Object {$_.displayName -eq $displayName}
        }

        try {
            if (
                ($type -eq "ipNamedLocation" -and -not $PSBoundParameters.ContainsKey('ipRanges')) -or
                ($type -ne "ipNamedLocation" -and $PSBoundParameters.ContainsKey('ipRanges'))
            ) { throw "If you want to define a ipNamedLocation, you need to provide ipRanges and set the type to ipNamedLocation." }
            
            if (
                ($type -eq "countryNamedLocation" -and -not $PSBoundParameters.ContainsKey('countriesAndRegions')) -or
                ($type -ne "countryNamedLocation" -and $PSBoundParameters.ContainsKey('countriesAndRegions'))
            ) { throw "If you want to define a countryNamedLocation, you need to provide countriesAndRegions and set the type to countryNamedLocation." }
        }
        catch {
            Write-PSFMessage -Level Error -String 'TMF.Register.PropertySetNotPossible' -StringValues $displayName, "Named Location" -Tag "failed" -ErrorRecord $_ -FunctionName $Cmdlet.CommandRuntime            
            $cmdlet.ThrowTerminatingError($_)
        }
    }
    process
    {
        if (Test-PSFFunctionInterrupt) { return }                

        $object = [PSCustomObject] @{
            displayName = $displayName
            present = $present
            sourceConfig = $sourceConfig
        }

        if ($PSBoundParameters.ContainsKey("oldNames")) {
            Add-Member -InputObject $object -MemberType NoteProperty -Name "oldNames" -Value @($oldNames | ForEach-Object {Resolve-String $_})
        }

        switch ($type) {
            "ipNamedLocation" {
                Add-Member -InputObject $object -MemberType NoteProperty -Name "@odata.type" -Value "#microsoft.graph.ipNamedLocation"
                Add-Member -InputObject $object -MemberType NoteProperty -Name "ipRanges" -Value @($ipRanges)
                Add-Member -InputObject $object -MemberType NoteProperty -Name "isTrusted" -Value $isTrusted
            }
            "countryNamedLocation" {
                Add-Member -InputObject $object -MemberType NoteProperty -Name "@odata.type" -Value "#microsoft.graph.countryNamedLocation"
                Add-Member -InputObject $object -MemberType NoteProperty -Name "countriesAndRegions" -Value @($countriesAndRegions)
                Add-Member -InputObject $object -MemberType NoteProperty -Name "includeUnknownCountriesAndRegions" -Value $includeUnknownCountriesAndRegions
            }
        }        
        Add-Member -InputObject $object -MemberType ScriptMethod -Name Properties -Value { ($this | Get-Member -MemberType NoteProperty).Name }

        if ($alreadyLoaded) {
            $script:desiredConfiguration[$resourceName][$script:desiredConfiguration[$resourceName].IndexOf($alreadyLoaded)] = $object
        }
        else {
            $script:desiredConfiguration[$resourceName] += $object
        }        
    }
}