Refactor IPAM model classes to use records for Address, Subnetwork, Vlan, Vrf, Section, Tag, Domain, Nameserver, and Session; enhance documentation and implement value equality for records.
This commit is contained in:
@@ -1,20 +1,23 @@
|
||||
namespace PS.IPAM.Cmdlets;
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Management.Automation;
|
||||
using System.Net;
|
||||
using PS.IPAM;
|
||||
using PS.IPAM.Helpers;
|
||||
|
||||
/// <summary>
|
||||
/// Creates a new subnet in phpIPAM.
|
||||
/// </summary>
|
||||
[Cmdlet(VerbsCommon.New, "Subnet")]
|
||||
[OutputType(typeof(Subnetwork))]
|
||||
public class NewSubnetCmdlet : PSCmdlet
|
||||
public class NewSubnetCmdlet : BaseCmdlet
|
||||
{
|
||||
[Parameter(
|
||||
Mandatory = true,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 0)]
|
||||
Position = 0,
|
||||
HelpMessage = "The subnet in CIDR notation (e.g., 192.168.1.0/24).")]
|
||||
[ValidatePattern(@"^\d+\.\d+\.\d+\.\d+/\d{1,2}$")]
|
||||
[ValidateNotNullOrEmpty]
|
||||
public string CIDR { get; set; } = string.Empty;
|
||||
@@ -23,187 +26,79 @@ public class NewSubnetCmdlet : PSCmdlet
|
||||
Mandatory = true,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 1)]
|
||||
Position = 1,
|
||||
HelpMessage = "The section ID to create the subnet in.")]
|
||||
[ValidateNotNullOrEmpty]
|
||||
public int SectionId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 2)]
|
||||
[Parameter(Mandatory = false, Position = 2, HelpMessage = "Description for the subnet.")]
|
||||
public string? Description { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 3)]
|
||||
[Parameter(Mandatory = false, Position = 3, HelpMessage = "VLAN ID to associate.")]
|
||||
public int? VlanId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 4)]
|
||||
[Parameter(Mandatory = false, Position = 4, HelpMessage = "VRF ID to associate.")]
|
||||
public int? VrfId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 5)]
|
||||
[Parameter(Mandatory = false, Position = 5, HelpMessage = "Master subnet ID for hierarchy.")]
|
||||
public int? MasterSubnetId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 6)]
|
||||
[Parameter(Mandatory = false, Position = 6, HelpMessage = "Nameserver ID.")]
|
||||
public int? NameserverId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 7)]
|
||||
[Parameter(Mandatory = false, Position = 7, HelpMessage = "Show subnet name.")]
|
||||
public SwitchParameter ShowName { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 8)]
|
||||
[Parameter(Mandatory = false, Position = 8, HelpMessage = "Enable recursive DNS.")]
|
||||
public SwitchParameter DNSRecursive { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 9)]
|
||||
[Parameter(Mandatory = false, Position = 9, HelpMessage = "Enable DNS records.")]
|
||||
public SwitchParameter DNSRecords { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 10)]
|
||||
[Parameter(Mandatory = false, Position = 10, HelpMessage = "Allow IP requests.")]
|
||||
public SwitchParameter AllowRequests { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 11)]
|
||||
[Parameter(Mandatory = false, Position = 11, HelpMessage = "Scan agent ID.")]
|
||||
public int? ScanAgentId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 12)]
|
||||
[Parameter(Mandatory = false, Position = 12, HelpMessage = "Enable subnet discovery.")]
|
||||
public SwitchParameter DiscoverSubnet { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 12)]
|
||||
[Parameter(Mandatory = false, Position = 13, HelpMessage = "Mark subnet as full.")]
|
||||
public SwitchParameter IsFull { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 12)]
|
||||
[Parameter(Mandatory = false, Position = 14, HelpMessage = "Tag ID for the subnet.")]
|
||||
public int? TagId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 13)]
|
||||
[Parameter(Mandatory = false, Position = 15, HelpMessage = "Usage threshold percentage (1-100).")]
|
||||
[ValidateRange(1, 100)]
|
||||
public int? Threshold { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 14)]
|
||||
[Parameter(Mandatory = false, Position = 16, HelpMessage = "Location ID.")]
|
||||
public int? LocationId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 15)]
|
||||
[Parameter(Mandatory = false, HelpMessage = "Custom fields as a hashtable or PSObject.")]
|
||||
public object? CustomFields { get; set; }
|
||||
|
||||
protected override void ProcessRecord()
|
||||
{
|
||||
try
|
||||
{
|
||||
var parts = CIDR.Split('/');
|
||||
var body = new Dictionary<string, object>
|
||||
{
|
||||
{ "subnet", parts[0] },
|
||||
{ "mask", parts[1] },
|
||||
{ "sectionId", SectionId }
|
||||
};
|
||||
var body = BuildRequestBody();
|
||||
|
||||
if (!string.IsNullOrEmpty(Description))
|
||||
body["description"] = Description;
|
||||
if (VlanId.HasValue)
|
||||
body["vlanId"] = VlanId.Value;
|
||||
if (VrfId.HasValue)
|
||||
body["vrfId"] = VrfId.Value;
|
||||
if (MasterSubnetId.HasValue)
|
||||
body["masterSubnetId"] = MasterSubnetId.Value;
|
||||
if (NameserverId.HasValue)
|
||||
body["nameserverId"] = NameserverId.Value;
|
||||
if (ShowName.IsPresent)
|
||||
body["showName"] = "1";
|
||||
if (DNSRecursive.IsPresent)
|
||||
body["DNSrecursive"] = "1";
|
||||
if (DNSRecords.IsPresent)
|
||||
body["DNSrecords"] = "1";
|
||||
if (AllowRequests.IsPresent)
|
||||
body["allowRequests"] = "1";
|
||||
if (ScanAgentId.HasValue)
|
||||
body["scanAgent"] = ScanAgentId.Value;
|
||||
if (DiscoverSubnet.IsPresent)
|
||||
body["discoverSubnet"] = "1";
|
||||
if (IsFull.IsPresent)
|
||||
body["isFull"] = "1";
|
||||
if (TagId.HasValue)
|
||||
body["state"] = TagId.Value;
|
||||
if (Threshold.HasValue)
|
||||
body["threshold"] = Threshold.Value;
|
||||
if (LocationId.HasValue)
|
||||
body["location"] = LocationId.Value;
|
||||
|
||||
if (CustomFields != null)
|
||||
{
|
||||
var customDict = ConvertCustomFields(CustomFields);
|
||||
foreach (var kvp in customDict)
|
||||
{
|
||||
body[kvp.Key] = kvp.Value;
|
||||
}
|
||||
}
|
||||
|
||||
var result = RequestHelper.InvokeRequest("POST", controllers.subnets, null, null, body, null)
|
||||
.GetAwaiter().GetResult();
|
||||
var result = RequestHelper.InvokeRequest(
|
||||
"POST", ApiController.Subnets, null, null, body
|
||||
).GetAwaiter().GetResult();
|
||||
|
||||
if (result != null)
|
||||
{
|
||||
// Get the created subnet
|
||||
var subnet = RequestHelper.InvokeRequest("GET", controllers.subnets, types.Subnetwork, null, null, new[] { "cidr", CIDR })
|
||||
.GetAwaiter().GetResult();
|
||||
if (subnet != null)
|
||||
{
|
||||
WriteObject(subnet);
|
||||
}
|
||||
var subnet = RequestHelper.InvokeRequest(
|
||||
"GET", ApiController.Subnets, ModelType.Subnetwork, null, null,
|
||||
new[] { "cidr", CIDR }
|
||||
).GetAwaiter().GetResult();
|
||||
|
||||
WriteResult(subnet);
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
@@ -212,20 +107,37 @@ public class NewSubnetCmdlet : PSCmdlet
|
||||
}
|
||||
}
|
||||
|
||||
private Dictionary<string, object> ConvertCustomFields(object customFields)
|
||||
private Dictionary<string, object> BuildRequestBody()
|
||||
{
|
||||
var dict = new Dictionary<string, object>();
|
||||
if (customFields is PSObject psobj)
|
||||
var parts = CIDR.Split('/');
|
||||
var body = new Dictionary<string, object>
|
||||
{
|
||||
foreach (var prop in psobj.Properties)
|
||||
{
|
||||
dict[prop.Name] = prop.Value ?? new object();
|
||||
}
|
||||
}
|
||||
else if (customFields is Dictionary<string, object> dictObj)
|
||||
["subnet"] = parts[0],
|
||||
["mask"] = parts[1],
|
||||
["sectionId"] = SectionId
|
||||
};
|
||||
|
||||
if (!string.IsNullOrEmpty(Description)) body["description"] = Description;
|
||||
if (VlanId.HasValue) body["vlanId"] = VlanId.Value;
|
||||
if (VrfId.HasValue) body["vrfId"] = VrfId.Value;
|
||||
if (MasterSubnetId.HasValue) body["masterSubnetId"] = MasterSubnetId.Value;
|
||||
if (NameserverId.HasValue) body["nameserverId"] = NameserverId.Value;
|
||||
if (ShowName.IsPresent) body["showName"] = "1";
|
||||
if (DNSRecursive.IsPresent) body["DNSrecursive"] = "1";
|
||||
if (DNSRecords.IsPresent) body["DNSrecords"] = "1";
|
||||
if (AllowRequests.IsPresent) body["allowRequests"] = "1";
|
||||
if (ScanAgentId.HasValue) body["scanAgent"] = ScanAgentId.Value;
|
||||
if (DiscoverSubnet.IsPresent) body["discoverSubnet"] = "1";
|
||||
if (IsFull.IsPresent) body["isFull"] = "1";
|
||||
if (TagId.HasValue) body["state"] = TagId.Value;
|
||||
if (Threshold.HasValue) body["threshold"] = Threshold.Value;
|
||||
if (LocationId.HasValue) body["location"] = LocationId.Value;
|
||||
|
||||
foreach (var kvp in ConvertCustomFields(CustomFields))
|
||||
{
|
||||
return dictObj;
|
||||
body[kvp.Key] = kvp.Value;
|
||||
}
|
||||
return dict;
|
||||
|
||||
return body;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user