using System;

using System.Collections.Generic;

using System.Linq;

using System.Management;

using System.Text.RegularExpressions;



namespace TrueLib

{

    /// <summary>

    /// Static wrapper for windows system devices.

    /// </summary>

    // TODO: on release BACK TO INTERNAL!!!

    static class SystemDevices

    {

        // WMI class names

        public const string Win32_LogicalDisk = "Win32_LogicalDisk";

        public const string Win32_DiskDrive = "Win32_DiskDrive";

        public const string Win32_DiskPartition = "Win32_DiskPartition";

        public const string Win32_USBControllerDevice = "Win32_USBControllerDevice";



        private static ManagementClass logicalDisks = new ManagementClass(Win32_LogicalDisk);

        private static ManagementClass diskDrives = new ManagementClass(Win32_DiskDrive);

        private static ManagementClass diskPartitions = new ManagementClass(Win32_DiskPartition);

        private static ManagementClass usbControllerDevices = new ManagementClass(Win32_USBControllerDevice);



        /// <summary>

        /// Checks if logical device is online (=plugged in and mounted with drive letter).

        /// </summary>

        /// <param name="letter">Drive letter of the device.</param>

        /// <returns>Returns true if device with given drive letter is online, else false.</returns>

        public static bool IsLogicalDiskOnline(String letter, int type = 2)

        {

            var ldQuery =

                from ManagementObject ldisk in logicalDisks.GetInstances()

                where ldisk["Name"].ToString() == letter &&

                int.Parse(ldisk["DriveType"].ToString()) == type

                select ldisk;



            if (ldQuery.Count() > 0)

                return true;

            else

                return false;

        }



        /// <summary>

        /// Gets a device with given caption and signature.

        /// </summary>

        /// <param name="caption">Caption (title, name) of the device.</param>

        /// <param name="signature">Signature (10 digit unsigned integer) of the device. May be 0 in some cases.</param>

        /// <returns>Returns the device with the given parameters.</returns>

        public static ManagementObject GetDiskDriveBySignature(String caption, uint signature)

        {

            var diskQuery =

                from ManagementObject disk in diskDrives.GetInstances()

                where (string)disk["Caption"] == caption

                && (uint)disk["Signature"] == signature

                select disk;



            if (diskQuery.Count() > 0)

                return diskQuery.First();

            else

                return null;

        }



        /// <summary>

        /// Gets a partition with a given drive signature and partition index.

        /// </summary>

        /// <param name="caption">The disk drive caption.</param>

        /// <param name="signature">The disk drive signature.</param>

        /// <param name="partitionNr">The zero-based partition number.</param>

        /// <returns>Returns a found partition, else null.</returns>

        public static ManagementObject GetPartitionBySignature(String caption, uint signature, uint partitionNr)

        {

            var partQiery =

                from ManagementObject disk in diskDrives.GetInstances()

                where (string)disk["Caption"] == caption

                && (uint)disk["Signature"] == signature

                from ManagementObject partition in disk.GetRelated(Win32_DiskPartition)

                where (uint)partition["Index"] == partitionNr

                select partition;



            if (partQiery.Count() > 0)

                return partQiery.First();

            else

                return null;

        }



        /// <summary>

        /// Checks if a partition with given DeviceID is removable.

        /// </summary>

        /// <param name="deviceId">The disk partition DeviceID.</param>

        /// <returns>Returns true if partition is removable, else false.</returns>

        public static bool IsRemovablePartition(string deviceId)

        {

            var ldiskQuery =

                from ManagementObject partition in diskPartitions.GetInstances()

                where (string)partition["DeviceID"] == deviceId

                from ManagementObject ldisk in partition.GetRelated(Win32_LogicalDisk)

                where (uint)ldisk["DriveType"] == 2

                select ldisk;



            if (ldiskQuery.Count() > 0)

                return true;

            else

                return false;

        }



        /// <summary>

        /// Checks if a given disk is online.

        /// </summary>

        /// <param name="caption">The caption of the disk drive.</param>

        /// <param name="signature">The signature of the disk drive.</param>

        /// <returns>True or false.</returns>

        public static bool IsDiskOnline(string caption, uint signature)

        {

            var query =

                from ManagementObject disk in diskDrives.GetInstances()

                where (string)disk["Caption"] == caption

                && (uint)disk["Signature"] == signature

                select disk;



            return (query.Count() > 0);

        }



        /// <summary>

        /// Checks if a given disk drive partition is online.

        /// </summary>

        /// <param name="caption">The caption of the disk drive.</param>

        /// <param name="signature">The signature of the disk drive.</param>

        /// <param name="partitionIndex">The zero-based partition index.</param>

        /// <returns>Returns true if partition is online, else false.</returns>

        public static bool IsPartitionOnline(string caption, uint signature, uint partitionIndex)

        {

            var partitionQuery =

                from ManagementObject disk in diskDrives.GetInstances()

                where (string)disk["Caption"] == caption

                && (uint)disk["Signature"] == signature

                from ManagementObject partition in disk.GetRelated(Win32_DiskPartition)

                where (uint)partition["Index"] == partitionIndex

                select partition;



            return (partitionQuery.Count() > 0);

        }



        /// <summary>

        /// Contains all disks of the system.

        /// </summary>

        public static ManagementObjectCollection DiskDrives

        {

            get { return diskDrives.GetInstances(); }

        }



        /// <summary>

        /// Gets a partition by its disk index and partition index.

        /// </summary>

        /// <param name="DiskIndex">Disk index (zero based).</param>

        /// <param name="Index">Partition index (zero based).</param>

        /// <returns>Returns a found partition or null.</returns>

        public static ManagementObject GetPartitionByIndex(uint DiskIndex, uint Index)

        {

            try

            {

                var partQuery =

                    from ManagementObject partition in DiskPartitions

                    where (uint)partition["DiskIndex"] == DiskIndex

                    && (uint)partition["Index"] == Index

                    select partition;



                if (partQuery.Count() > 0)

                    return partQuery.First();

                else

                    return null;

            }

            catch { return null; }

        }



        /// <summary>

        /// Contains all partitions of the system.

        /// </summary>

        public static ManagementObjectCollection DiskPartitions

        {

            get { return diskPartitions.GetInstances(); }

        }



        /// <summary>

        /// Contains all logical disks of the system.

        /// </summary>

        public static ManagementObjectCollection LogicalDisks

        {

            get { return logicalDisks.GetInstances(); }

        }



        /// <summary>

        /// Contains all usb controller devices.

        /// </summary>

        public static ManagementObjectCollection USBControllerDevices

        {

            get { return usbControllerDevices.GetInstances(); }

        }



        /// <summary>

        /// Gets the drive letter of a logical disk identified by its caption, signature and partition number.

        /// </summary>

        /// <param name="caption">Caption (title, name) of the disk.</param>

        /// <param name="signature">Signature (10 digit unsigned integer) of the disk.</param>

        /// <param name="partitionNr">Partition number (zero based).</param>

        /// <returns>Returns the drive letter of the logical disk.</returns>

        public static String GetDriveLetterBySignature(String caption, uint signature, uint partitionNr)

        {

            var devQuery =

                from ManagementObject disk in DiskDrives

                where (string)disk["Caption"] == caption

                && (uint)disk["Signature"] == signature

                from ManagementObject partition in disk.GetRelated(SystemDevices.Win32_DiskPartition)

                where (uint)partition["Index"] == partitionNr

                from ManagementObject ldisk in partition.GetRelated(SystemDevices.Win32_LogicalDisk)

                select (string)ldisk["Name"];



            if (devQuery.Count() > 0)

                return (string)devQuery.First();

            else

                return null;

        }



        /// <summary>

        /// Gets a logical disk by a drive letter an disk type.

        /// </summary>

        /// <param name="letter">Drive letter of the disk.</param>

        /// <param name="drive_type">Optional drive type (3 = local, 2 = removable).</param>

        /// <returns>Returns the disk or null.</returns>

        public static ManagementObject GetLogicalDisk(String letter, int drive_type = 3)

        {

            var ldQuery =

                from ManagementObject ldisk in LogicalDisks

                where int.Parse(ldisk["DriveType"].ToString()) == drive_type

                && ldisk["Name"].ToString().StartsWith(letter)

                select ldisk;



            if (ldQuery.Count() > 0)

                return ldQuery.First();

            else

                return null;

        }



        /// <summary>

        /// Gets the disk caption, signature and partition number by the drive letter.

        /// </summary>

        /// <param name="letter">Drive letter of the logical disk.</param>

        /// <param name="caption">Caption of the disk.</param>

        /// <param name="signature">Signature of the disk.</param>

        /// <param name="partitionNr">Partition number (zero based).</param>

        public static void GetDiskSignatureFromLetter(String letter,

            ref string caption, ref uint signature, ref uint partitionNr)

        {

            var ldiskQuery =

                from ManagementObject ldisk in LogicalDisks

                where ldisk["Name"].ToString().StartsWith(letter)

                select ldisk;



            var partQuery =

                from ManagementObject part in ldiskQuery.First().GetRelated(SystemDevices.Win32_DiskPartition)

                select part;



            if (partQuery.Count() > 0)

                partitionNr = (uint)partQuery.First()["Index"];



            var diskQuery =

                from ManagementObject disk in partQuery.First().GetRelated(Win32_DiskDrive)

                select disk;



            if (diskQuery.Count() > 0)

            {

                ManagementObject disk_drive = diskQuery.First();



                caption = (string)disk_drive["Caption"];

                signature = (uint)disk_drive["Signature"];

            }

        }



        /// <summary>

        /// For debugging: returns identity information about all attached disks.

        /// </summary>

        /// <returns>A list of disk information.</returns>

        public static List<string> GetLocalDiskPartitions()

        {

            List<string> values = new List<string>();



            foreach (ManagementObject disk in DiskDrives)

            {

                foreach (ManagementObject partition in disk.GetRelated(Win32_DiskPartition))

                {

                    values.Add(string.Format("Disk caption({0}), signature({1}); Partition index({2})",

                        disk["Caption"], disk["Signature"], partition["Index"]));

                }

            }



            return values;

        }



        /// <summary>

        /// Converts the device id into a TrueCrypt compatible path.

        /// </summary>

        /// <param name="deviceid">Device ID</param>

        /// <returns>Returns a string with the device path.</returns>

        public static String GetTCCompatibleName(String deviceid)

        {

            return Regex.Replace(deviceid, "Disk #([0-9]*), Partition #([0-9]*)", ReplaceEvaluator);

        }



        /// <summary>

        /// Contains logic to build a TC compatible device path.

        /// </summary>

        /// <param name="m"></param>

        /// <returns></returns>

        private static string ReplaceEvaluator(Match m)

        {

            return @"\Device\Harddisk" +

                (int.Parse(m.Groups[1].Value)).ToString() +

                @"\Partition" +

                (int.Parse(m.Groups[2].Value) + 1).ToString();

        }



        /// <summary>

        /// Returns a TC compatible path to an encrypted hard disk.

        /// </summary>

        /// <param name="index"></param>

        /// <returns></returns>

        public static string GetTCCompatibleDiskPath(uint index)

        {

            return string.Format(@"\Device\Harddisk{0}\Partition0", index.ToString());

        }

    }

}