using System;
using Cosmos.HAL.BlockDevice;

namespace Cosmos.HAL.BlockDevice
{
    public class IDE
    {
        private static PCIDevice xDevice = HAL.PCI.GetDeviceClass(HAL.ClassID.MassStorageController,
                                                                  HAL.SubclassID.IDEInterface);

        internal static void InitDriver()
        {
            if (xDevice != null)
            {
                Console.WriteLine("ATA Primary Master");
                Initialize(Ata.ControllerIdEnum.Primary, Ata.BusPositionEnum.Master);
                //Console.WriteLine("ATA Primary Slave");
                //Initialize(Ata.ControllerIdEnum.Primary, Ata.BusPositionEnum.Slave);
                Console.WriteLine("ATA Secondary Master");
                Initialize(Ata.ControllerIdEnum.Secondary, Ata.BusPositionEnum.Master);
                //Console.WriteLine("ATA Secondary Slave");
                //Initialize(Ata.ControllerIdEnum.Secondary, Ata.BusPositionEnum.Slave);
            }
        }

        private static void Initialize(Ata.ControllerIdEnum aControllerID, Ata.BusPositionEnum aBusPosition)
        {
            var xIO = aControllerID == Ata.ControllerIdEnum.Primary ? Core.Global.BaseIOGroups.ATA1 : Core.Global.BaseIOGroups.ATA2;
            var xATA = new AtaPio(xIO, aControllerID, aBusPosition);
            if (xATA.DriveType == Ata.DriveTypeEnum.Null)
            {
                return;
            }

            if (xATA.DriveType == Ata.DriveTypeEnum.ATA)
            {
                Ata.AtaDebugger.Send("ATA device with speclevel ATA found.");
                BlockDevice.Devices.Add(xATA);

                if (GPT.IsGPTPartition(xATA))
                {
                    var xGPT = new GPT(xATA);

                    Ata.AtaDebugger.Send("Number of GPT partitions found:");
                    Ata.AtaDebugger.SendNumber(xGPT.Partitions.Count);
                    for (int i = 0; i < xGPT.Partitions.Count; i++)
                    {
                        var xPart = xGPT.Partitions[i];
                        if (xPart == null)
                        {
                            Console.WriteLine("Null partition found at idx: " + i);
                        }
                        else
                        {
                            var xPartDevice = new Partition(xATA, xPart.StartSector, xPart.SectorCount);
                            BlockDevice.Devices.Add(xPartDevice);
                            Console.WriteLine("Found partition at idx: " + i);
                        }
                    }
                }
                else
                {
                    var xMbrData = new byte[512];
                    xATA.ReadBlock(0UL, 1U, ref xMbrData);
                    var xMBR = new MBR(xMbrData);

                    if (xMBR.EBRLocation != 0)
                    {
                        //EBR Detected
                        var xEbrData = new byte[512];
                        xATA.ReadBlock(xMBR.EBRLocation, 1U, ref xEbrData);
                        var xEBR = new EBR(xEbrData);

                        for (int i = 0; i < xEBR.Partitions.Count; i++)
                        {
                            //var xPart = xEBR.Partitions[i];
                            //var xPartDevice = new BlockDevice.Partition(xATA, xPart.StartSector, xPart.SectorCount);
                            //BlockDevice.BlockDevice.Devices.Add(xPartDevice);
                        }
                    }

                    // TODO Change this to foreach when foreach is supported
                    Ata.AtaDebugger.Send("Number of MBR partitions found:");
                    Ata.AtaDebugger.SendNumber(xMBR.Partitions.Count);
                    for (int i = 0; i < xMBR.Partitions.Count; i++)
                    {
                        var xPart = xMBR.Partitions[i];
                        if (xPart == null)
                        {
                            Console.WriteLine("Null partition found at idx: " + i);
                        }
                        else
                        {
                            var xPartDevice = new Partition(xATA, xPart.StartSector, xPart.SectorCount);
                            BlockDevice.Devices.Add(xPartDevice);
                            Console.WriteLine("Found partition at idx: " + i);
                        }
                    }
                }
            }
            else if (xATA.DriveType == Ata.DriveTypeEnum.ATAPI)
            {
                Ata.AtaDebugger.Send("ATA device with speclevel ATAPI found.");
                BlockDevice.Devices.Add(xATA);

                var xData = new byte[2048];
                xATA.ReadBlock(1UL, 1U, ref xData);
            }
        }
    }
}