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:
186
classlib/Cmdlets/SetAddressCmdlet.cs
Normal file
186
classlib/Cmdlets/SetAddressCmdlet.cs
Normal file
@@ -0,0 +1,186 @@
|
||||
namespace PS.IPAM.Cmdlets;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Management.Automation;
|
||||
using PS.IPAM;
|
||||
using PS.IPAM.Helpers;
|
||||
|
||||
[Cmdlet(VerbsCommon.Set, "Address", DefaultParameterSetName = "ById")]
|
||||
[OutputType(typeof(Address))]
|
||||
public class SetAddressCmdlet : 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 = "ByAddressObject")]
|
||||
[ValidateNotNullOrEmpty]
|
||||
public Address? AddressObject { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Position = 1,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
ParameterSetName = "ById")]
|
||||
public bool? Gateway { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Position = 2,
|
||||
ValueFromPipeline = true,
|
||||
ValueFromPipelineByPropertyName = true,
|
||||
ParameterSetName = "ById")]
|
||||
public string? Description { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 3)]
|
||||
public string? Hostname { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 4)]
|
||||
public string? MAC { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 5)]
|
||||
public string? Owner { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 6)]
|
||||
public int? TagId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 7)]
|
||||
public bool? PTRIgnore { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 8)]
|
||||
public int? PTRId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 9)]
|
||||
public string? Note { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 10)]
|
||||
public bool? ExcludePing { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 11)]
|
||||
public int? DeviceId { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false,
|
||||
Position = 12)]
|
||||
public string? Port { get; set; }
|
||||
|
||||
[Parameter(
|
||||
Mandatory = false)]
|
||||
public object? CustomFields { get; set; }
|
||||
|
||||
protected override void ProcessRecord()
|
||||
{
|
||||
try
|
||||
{
|
||||
int addressId;
|
||||
if (ParameterSetName == "ById")
|
||||
{
|
||||
addressId = Id!.Value;
|
||||
}
|
||||
else
|
||||
{
|
||||
addressId = AddressObject!.Id;
|
||||
}
|
||||
|
||||
var identifiers = new List<string> { addressId.ToString() };
|
||||
var body = new Dictionary<string, object>();
|
||||
|
||||
if (Gateway.HasValue)
|
||||
body["is_gateway"] = Gateway.Value;
|
||||
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.HasValue)
|
||||
body["PTRignore"] = PTRIgnore.Value;
|
||||
if (PTRId.HasValue)
|
||||
body["PTR"] = PTRId.Value;
|
||||
if (!string.IsNullOrEmpty(Note))
|
||||
body["note"] = Note;
|
||||
if (ExcludePing.HasValue)
|
||||
body["excludePing"] = ExcludePing.Value;
|
||||
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;
|
||||
}
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
RequestHelper.InvokeRequest("PATCH", controllers.addresses, null, null, body, identifiers.ToArray())
|
||||
.GetAwaiter().GetResult();
|
||||
}
|
||||
finally
|
||||
{
|
||||
var address = RequestHelper.InvokeRequest("GET", controllers.addresses, types.Address, null, null, identifiers.ToArray())
|
||||
.GetAwaiter().GetResult();
|
||||
if (address != null)
|
||||
{
|
||||
WriteObject(address);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
WriteError(new ErrorRecord(ex, "SetAddressError", 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