ps.ipam/classlib/Cmdlets/GetAddressCmdlet.cs

namespace PS.IPAM.Cmdlets;
using System;
using System.Collections.Generic;
using System.Management.Automation;
using System.Net;
using PS.IPAM;
using PS.IPAM.Helpers;

[Cmdlet(VerbsCommon.Get, "Address", DefaultParameterSetName = "ByID")]
[OutputType(typeof(Address))]
public class GetAddressCmdlet : PSCmdlet
{
    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByID")]
    [ValidateNotNullOrEmpty]
    public int Id { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByIP")]
    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 1,
        ParameterSetName = "BySubnetId")]
    [ValidateNotNullOrEmpty]
    public IPAddress? IP { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByHostName")]
    [ValidateNotNullOrEmpty]
    public string? HostName { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByHostBase")]
    [ValidateNotNullOrEmpty]
    public string? HostBase { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByTag")]
    [ValidateNotNullOrEmpty]
    public int? TagId { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "BySubnetId")]
    [ValidateNotNullOrEmpty]
    public int? SubnetId { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "BySubnetCIDR")]
    [ValidatePattern(@"^\d+\.\d+\.\d+\.\d+/\d{1,2}$")]
    [ValidateNotNullOrEmpty]
    public string? SubnetCIDR { get; set; }

    protected override void ProcessRecord()
    {
        try
        {
            var controller = controllers.addresses;
            var identifiers = new List<string>();

            switch (ParameterSetName)
            {
                case "ByID":
                    identifiers.Add(Id.ToString());
                    break;
                case "ByIP":
                    identifiers.Add("search");
                    identifiers.Add(IP!.ToString());
                    break;
                case "ByHostName":
                    identifiers.Add("search_hostname");
                    identifiers.Add(HostName!);
                    break;
                case "ByHostBase":
                    identifiers.Add("search_hostbase");
                    identifiers.Add(HostBase!);
                    break;
                case "ByTag":
                    identifiers.Add("tags");
                    identifiers.Add(TagId!.Value.ToString());
                    identifiers.Add("addresses");
                    break;
                case "BySubnetId":
                    if (IP != null)
                    {
                        identifiers.Add(IP.ToString());
                        identifiers.Add(SubnetId!.Value.ToString());
                    }
                    else
                    {
                        controller = controllers.subnets;
                        identifiers.Add(SubnetId!.Value.ToString());
                        identifiers.Add("addresses");
                    }
                    break;
                case "BySubnetCIDR":
                    controller = controllers.subnets;
                    var subnet = RequestHelper.InvokeRequest("GET", controllers.subnets, types.Subnetwork, null, null, new[] { "cidr", SubnetCIDR! })
                        .GetAwaiter().GetResult();
                    if (subnet == null)
                    {
                        throw new Exception("Cannot find subnet!");
                    }
                    var subnetObj = subnet as Subnetwork;
                    identifiers.Add(subnetObj!.Id.ToString());
                    identifiers.Add("addresses");
                    break;
            }

            var result = RequestHelper.InvokeRequest("GET", controller, types.Address, null, null, identifiers.ToArray())
                .GetAwaiter().GetResult();

            if (result != null)
            {
                if (result is System.Collections.IEnumerable enumerable && !(result is string))
                {
                    foreach (var item in enumerable)
                    {
                        WriteObject(item);
                    }
                }
                else
                {
                    WriteObject(result);
                }
            }
        }
        catch (Exception ex)
        {
            WriteError(new ErrorRecord(ex, "GetAddressError", ErrorCategory.InvalidOperation, null));
        }
    }
}