ps.ipam/classlib/Cmdlets/NewAddressCmdlet.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.New, "Address", DefaultParameterSetName = "BySubnetId")]
[OutputType(typeof(Address))]
public class NewAddressCmdlet : PSCmdlet
{
    [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 = "BySubnetObject")]
    [ValidateNotNullOrEmpty]
    public Subnetwork? SubnetObject { get; set; }

    [Parameter(
        Mandatory = true,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 1)]
    [ValidateNotNullOrEmpty]
    public string IP { get; set; } = string.Empty;

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 2)]
    public SwitchParameter Gateway { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 3)]
    public string? Description { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 4)]
    public string? Hostname { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 5)]
    public string? MAC { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 6)]
    public string? Owner { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 7)]
    public int? TagId { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 8)]
    public SwitchParameter PTRIgnore { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 7)]
    public int? PTRId { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 10)]
    public string? Note { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 11)]
    public SwitchParameter ExcludePing { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 12)]
    public int? DeviceId { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 13)]
    public string? Port { get; set; }

    [Parameter(
        Mandatory = false,
        ValueFromPipeline = true,
        ValueFromPipelineByPropertyName = true,
        Position = 14)]
    public object? CustomFields { get; set; }

    protected override void ProcessRecord()
    {
        try
        {
            int actualSubnetId;
            if (ParameterSetName == "BySubnetObject")
            {
                actualSubnetId = SubnetObject!.Id;
            }
            else
            {
                actualSubnetId = SubnetId!.Value;
            }

            var body = new Dictionary<string, object>
            {
                { "subnetId", actualSubnetId },
                { "ip", IP }
            };

            if (Gateway.IsPresent)
                body["is_gateway"] = "1";
            if (!string.IsNullOrEmpty(Description))
                body["description"] = Description;
            if (!string.IsNullOrEmpty(Hostname))
                body["hostname"] = Hostname;
            if (!string.IsNullOrEmpty(MAC))
                body["mac"] = MAC;
            if (!string.IsNullOrEmpty(Owner))
                body["owner"] = Owner;
            if (TagId.HasValue)
                body["tag"] = TagId.Value;
            if (PTRIgnore.IsPresent)
                body["PTRignore"] = "1";
            if (PTRId.HasValue)
                body["PTR"] = PTRId.Value;
            if (!string.IsNullOrEmpty(Note))
                body["note"] = Note;
            if (ExcludePing.IsPresent)
                body["excludePing"] = "1";
            if (DeviceId.HasValue)
                body["deviceId"] = DeviceId.Value;
            if (!string.IsNullOrEmpty(Port))
                body["port"] = Port;

            if (CustomFields != null)
            {
                var customDict = ConvertCustomFields(CustomFields);
                foreach (var kvp in customDict)
                {
                    body[kvp.Key] = kvp.Value;
                }
            }

            var result = RequestHelper.InvokeRequest("POST", controllers.addresses, null, null, body, null)
                .GetAwaiter().GetResult();

            if (result != null)
            {
                var address = RequestHelper.InvokeRequest("GET", controllers.addresses, types.Address, null, null, new[] { "search", IP })
                    .GetAwaiter().GetResult();
                if (address != null)
                {
                    WriteObject(address);
                }
            }
        }
        catch (Exception ex)
        {
            WriteError(new ErrorRecord(ex, "NewAddressError", ErrorCategory.InvalidOperation, null));
        }
    }

    private Dictionary<string, object> ConvertCustomFields(object customFields)
    {
        var dict = new Dictionary<string, object>();
        if (customFields is PSObject psobj)
        {
            foreach (var prop in psobj.Properties)
            {
                dict[prop.Name] = prop.Value ?? new object();
            }
        }
        else if (customFields is Dictionary<string, object> dictObj)
        {
            return dictObj;
        }
        return dict;
    }
}