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;
    }
}