Added assembly name and root namespace to project file; updated session and vlan classes with new properties; modified module manifest to reference DLL and updated exported cmdlets; corrected property name in address class.
This commit is contained in:
231
classlib/Cmdlets/NewSubnetCmdlet.cs
Normal file
231
classlib/Cmdlets/NewSubnetCmdlet.cs
Normal file
@@ -0,0 +1,231 @@
|
||||
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, "Subnet")]
|
||||
[OutputType(typeof(Subnetwork))]
|
||||
public class NewSubnetCmdlet : PSCmdlet
|
||||
{
|
||||
[Parameter(
|
||||
Mandatory = true,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 0)]
|
||||
[ValidatePattern(@"^\d+\.\d+\.\d+\.\d+/\d{1,2}$")]
|
||||
[ValidateNotNullOrEmpty]
|
||||
public string CIDR { get; set; } = string.Empty;
|
||||
|
||||
[Parameter(
|
||||
Mandatory = true,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 1)]
|
||||
[ValidateNotNullOrEmpty]
|
||||
public int SectionId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 2)]
|
||||
public string? Description { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 3)]
|
||||
public int? VlanId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 4)]
|
||||
public int? VrfId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 5)]
|
||||
public int? MasterSubnetId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 6)]
|
||||
public int? NameserverId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 7)]
|
||||
public SwitchParameter ShowName { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 8)]
|
||||
public SwitchParameter DNSRecursive { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 9)]
|
||||
public SwitchParameter DNSRecords { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 10)]
|
||||
public SwitchParameter AllowRequests { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 11)]
|
||||
public int? ScanAgentId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 12)]
|
||||
public SwitchParameter DiscoverSubnet { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 12)]
|
||||
public SwitchParameter IsFull { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 12)]
|
||||
public int? TagId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 13)]
|
||||
[ValidateRange(1, 100)]
|
||||
public int? Threshold { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 14)]
|
||||
public int? LocationId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
Position = 15)]
|
||||
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 }
|
||||
};
|
||||
|
||||
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();
|
||||
|
||||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
WriteError(new ErrorRecord(ex, "NewSubnetError", 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;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user