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.Helpers;

/// <summary>
/// Retrieves IP address entries from phpIPAM.
/// </summary>
[Cmdlet(VerbsCommon.Get, "Address", DefaultParameterSetName = "ByID")]
[OutputType(typeof(Address))]
public class GetAddressCmdlet : BaseCmdlet
{
    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByID",
        HelpMessage = "The address ID to retrieve.")]
    [ValidateNotNullOrEmpty]
    public int Id { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByIP",
        HelpMessage = "The IP address to search for.")]
    [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",
        HelpMessage = "The hostname to search for.")]
    [ValidateNotNullOrEmpty]
    public string? HostName { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByHostBase",
        HelpMessage = "The hostname base pattern to search for.")]
    [ValidateNotNullOrEmpty]
    public string? HostBase { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "ByTag",
        HelpMessage = "The tag ID to filter addresses by.")]
    [ValidateNotNullOrEmpty]
    public int? TagId { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 0,
        ParameterSetName = "BySubnetId",
        HelpMessage = "The subnet ID to get addresses from.")]
    [ValidateNotNullOrEmpty]
    public int? SubnetId { get; set; }

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

    protected override void ProcessRecord()
    {
        try
        {
            var controller = ApiController.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 = ApiController.Subnets;
                        identifiers.Add(SubnetId!.Value.ToString());
                        identifiers.Add("addresses");
                    }
                    break;

                case "BySubnetCIDR":
                    controller = ApiController.Subnets;
                    var subnet = RequestHelper.InvokeRequest(
                        "GET", ApiController.Subnets, ModelType.Subnetwork, null, null,
                        new[] { "cidr", SubnetCIDR! }
                    ).GetAwaiter().GetResult() as Subnetwork;

                    if (subnet == null)
                    {
                        throw new ItemNotFoundException($"Subnet '{SubnetCIDR}' not found.");
                    }

                    identifiers.Add(subnet.Id.ToString());
                    identifiers.Add("addresses");
                    break;
            }

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

            WriteResult(result);
        }
        catch (Exception ex)
        {
            WriteError(new ErrorRecord(ex, "GetAddressError", ErrorCategory.InvalidOperation, null));
        }
    }
}