// ReSharper disable MemberCanBePrivate.Global
// ReSharper disable UnusedMember.Global

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using HidSharp;
using RGB.NET.Core;
using RGB.NET.Devices.Logitech.Native;
using RGB.NET.Devices.Logitech.HID;
using RGB.NET.HID;

namespace RGB.NET.Devices.Logitech;

/// <inheritdoc />
/// <summary>
/// Represents a device provider responsible for logitech devices.
/// </summary>
public class LogitechDeviceProvider : AbstractRGBDeviceProvider
{
    #region Properties & Fields

    // ReSharper disable once InconsistentNaming
    private static readonly Lock _lock = new();

    private static LogitechDeviceProvider? _instance;
    /// <summary>
    /// Gets the singleton <see cref="LogitechDeviceProvider"/> instance.
    /// </summary>
    public static LogitechDeviceProvider Instance
    {
        get
        {
            lock (_lock)
                return _instance ?? new LogitechDeviceProvider();
        }
    }

    /// <summary>
    /// Gets a modifiable list of paths used to find the native SDK-dlls for x86 applications.
    /// The first match will be used.
    /// </summary>
    public static List<string> PossibleX86NativePaths { get; } = ["x86/LogitechLedEnginesWrapper.dll"];

    /// <summary>
    /// Gets a modifiable list of paths used to find the native SDK-dlls for x64 applications.
    /// The first match will be used.
    /// </summary>
    public static List<string> PossibleX64NativePaths { get; } = ["x64/LogitechLedEnginesWrapper.dll"];

    private LogitechPerDeviceUpdateQueue? _perDeviceUpdateQueue;
    private LogitechPerKeyUpdateQueue? _perKeyUpdateQueue;

    private const int VENDOR_ID = 0x046D;

    /// <summary>
    /// Gets the HID-definitions for wired per-key-devices.
    /// </summary>
    public static HIDLoader<LogitechLedId, int> PerKeyDeviceDefinitions { get; } = new(VENDOR_ID)
    {
        { 0xC32B, RGBDeviceType.Keyboard, "G910", LedMappings.PerKey, 0 },
        { 0xC330, RGBDeviceType.Keyboard, "G410", LedMappings.PerKey, 0 },
        { 0xC331, RGBDeviceType.Keyboard, "G810", LedMappings.PerKey, 0 },
        { 0xC335, RGBDeviceType.Keyboard, "G910v2", LedMappings.PerKey, 0 },
        { 0xC337, RGBDeviceType.Keyboard, "G810", LedMappings.PerKey, 0 },
        { 0xC339, RGBDeviceType.Keyboard, "Pro", LedMappings.PerKey, 0 },
        { 0xC33C, RGBDeviceType.Keyboard, "G513", LedMappings.PerKey, 0 },
        { 0xC33E, RGBDeviceType.Keyboard, "G915", LedMappings.PerKey, 0 },
        { 0xC33F, RGBDeviceType.Keyboard, "G815", LedMappings.PerKey, 0 },
        { 0xC342, RGBDeviceType.Keyboard, "G512", LedMappings.PerKey, 0 },
        { 0xC343, RGBDeviceType.Keyboard, "G915 TKL", LedMappings.PerKey, 0 },
        { 0xC541, RGBDeviceType.Keyboard, "G915", LedMappings.PerKey, 0 },
        { 0xC359, RGBDeviceType.Keyboard, "G915 X", LedMappings.PerKey, 0 },
        { 0xC547, RGBDeviceType.Keyboard, "G915 X TKL", LedMappings.PerKey, 0 },
        { 0xC358, RGBDeviceType.Keyboard, "G515", LedMappings.PerKey, 0 },

        //non-rgb
        { 0xC333, RGBDeviceType.Keyboard, "G610", LedMappings.PerKey, 0 },
        { 0xC338, RGBDeviceType.Keyboard, "G610", LedMappings.PerKey, 0 },
    };

    /// <summary>
    /// Gets the HID-definitions for wireless per-key-devices.
    /// </summary>
    public static LightspeedHIDLoader<LogitechLedId, int> PerKeyWirelessDeviceDefinitions { get; } = new()
    {
        { 0x407C, RGBDeviceType.Keyboard, "G915", LedMappings.PerKey, 0 },
        { 0x408E, RGBDeviceType.Keyboard, "G915 TKL", LedMappings.PerKey, 0 },
        { 0xC232, RGBDeviceType.Keyboard, "G915 TKL", LedMappings.PerKey, 0 },
    };

    /// <summary>
    /// Gets the HID-definitions for wired per-zone-devices.
    /// </summary>
    public static HIDLoader<int, (LogitechDeviceType deviceType, int zones, int zoneOffset)> PerZoneDeviceDefinitions { get; } = new(VENDOR_ID)
    {
        { 0xC336, RGBDeviceType.Keyboard, "G213", LedMappings.ZoneKeyboard, (LogitechDeviceType.Keyboard, 5, 0) },

        { 0xC092, RGBDeviceType.Mouse, "G203 Lightsync", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 1, 0) },
        { 0xC080, RGBDeviceType.Mouse, "G303", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC081, RGBDeviceType.Mouse, "G900", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC082, RGBDeviceType.Mouse, "G403", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC083, RGBDeviceType.Mouse, "G403", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC084, RGBDeviceType.Mouse, "G203", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 1, 0) },
        { 0xC085, RGBDeviceType.Mouse, "G Pro", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 1, 0) },
        { 0xC086, RGBDeviceType.Mouse, "G903", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC087, RGBDeviceType.Mouse, "G703", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC088, RGBDeviceType.Mouse, "G Pro Wireless", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC08B, RGBDeviceType.Mouse, "G502 HERO", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC08C, RGBDeviceType.Mouse, "G Pro Hero", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 1, 1) },
        { 0xC08D, RGBDeviceType.Mouse, "G502 Lightspeed", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC08F, RGBDeviceType.Mouse, "G403 HERO", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC090, RGBDeviceType.Mouse, "G703 Lightspeed", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC091, RGBDeviceType.Mouse, "G903 Lightspeed", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0xC24A, RGBDeviceType.Mouse, "G600", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 1, 0) },
        { 0xC332, RGBDeviceType.Mouse, "G502", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        
        //G633 and G635 are wired headsets.
        { 0x0A5C, RGBDeviceType.Headset, "G633", LedMappings.ZoneHeadset, (LogitechDeviceType.Headset, 2, 0) },
        { 0x0A89, RGBDeviceType.Headset, "G635", LedMappings.ZoneHeadset, (LogitechDeviceType.Headset, 2, 0) },

        //The other ones are wireless only. These PIDs correpond to the dongles.
        { 0x0A5B, RGBDeviceType.Headset, "G933", LedMappings.ZoneHeadset, (LogitechDeviceType.Headset, 2, 0) },
        { 0x0A87, RGBDeviceType.Headset, "G935", LedMappings.ZoneHeadset, (LogitechDeviceType.Headset, 2, 0) },

        { 0x0A78, RGBDeviceType.Speaker, "G560", LedMappings.ZoneSpeaker, (LogitechDeviceType.Speaker, 4, 0) },
    };

    /// <summary>
    /// Gets the HID-definitions for wireless per-zone-devices.
    /// </summary>
    public static LightspeedHIDLoader<int, (LogitechDeviceType deviceType, int zones, int zoneOffset)> PerZoneWirelessDeviceDefinitions { get; } = new()
    {
        { 0x4053, RGBDeviceType.Mouse, "G900", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x405D, RGBDeviceType.Mouse, "G403", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x4067, RGBDeviceType.Mouse, "G903", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x4070, RGBDeviceType.Mouse, "G703", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x4079, RGBDeviceType.Mouse, "G Pro Wireless", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x407F, RGBDeviceType.Mouse, "G502 Lightspeed", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x4086, RGBDeviceType.Mouse, "G703 Lightspeed", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
        { 0x4087, RGBDeviceType.Mouse, "G903 Lightspeed", LedMappings.ZoneMouse, (LogitechDeviceType.Mouse, 2, 0) },
    };

    /// <summary>
    /// Gets the HID-definitions for wired per-device-devices.
    /// </summary>
    public static HIDLoader<int, int> PerDeviceDeviceDefinitions { get; } = new(VENDOR_ID)
    {
        { 0xC228, RGBDeviceType.Keyboard, "G19", LedMappings.Device, 0 },
        { 0xC229, RGBDeviceType.Keyboard, "G19s", LedMappings.Device, 0 },
        { 0xC24A, RGBDeviceType.Mouse, "G600", LedMappings.Device, 0 },
        { 0xC246, RGBDeviceType.Mouse, "G300s", LedMappings.Device, 0 },
        { 0xC22D, RGBDeviceType.Keyboard, "G510", LedMappings.Device, 0 },
        { 0xC22E, RGBDeviceType.Keyboard, "G510s", LedMappings.Device, 0 },
        { 0xC21C, RGBDeviceType.Keypad, "G13", LedMappings.Device, 0 },
        { 0xC22B, RGBDeviceType.Keyboard, "G110", LedMappings.Device, 0 },
        { 0xC24D, RGBDeviceType.Keyboard, "G710+", LedMappings.Device, 0 },
        { 0xC248, RGBDeviceType.Keyboard, "G105", LedMappings.Device, 0 },
        { 0xC222, RGBDeviceType.Keyboard, "G15", LedMappings.Device, 0 },
        { 0xC225, RGBDeviceType.Keyboard, "G11", LedMappings.Device, 0 },
        { 0x0AB5, RGBDeviceType.Headset, "G733", LedMappings.Device, 0 },
        { 0xC53A, RGBDeviceType.Mousepad, "POWERPLAY", LedMappings.Device, 0 },
    };

    /// <summary>
    /// Gets the HID-definitions for wireless per-device-devices.
    /// </summary>
    public static LightspeedHIDLoader<int, int> PerDeviceWirelessDeviceDefinitions { get; } = [];

    #endregion

    #region Constructors

    /// <summary>
    /// Initializes a new instance of the <see cref="LogitechDeviceProvider"/> class.
    /// </summary>
    /// <exception cref="InvalidOperationException">Thrown if this constructor is called even if there is already an instance of this class.</exception>
    public LogitechDeviceProvider()
    {
        lock (_lock)
        {
            if (_instance != null)
                throw new InvalidOperationException($"There can be only one instance of type {nameof(LogitechDeviceProvider)}");
            _instance = this;
        }
    }

    #endregion

    #region Methods

    /// <inheritdoc />
    protected override void InitializeSDK()
    {
        _perDeviceUpdateQueue = new LogitechPerDeviceUpdateQueue(GetUpdateTrigger());
        _perKeyUpdateQueue = new LogitechPerKeyUpdateQueue(GetUpdateTrigger());

        _LogitechGSDK.Reload();
        if (!_LogitechGSDK.LogiLedInit()) Throw(new RGBDeviceException("Failed to initialize Logitech-SDK."), true);

        _LogitechGSDK.LogiLedSaveCurrentLighting();
    }

    /// <inheritdoc />
    protected override IEnumerable<IRGBDevice> GetLoadedDevices(RGBDeviceType loadFilter)
    {
        PerKeyDeviceDefinitions.LoadFilter = loadFilter;
        PerZoneDeviceDefinitions.LoadFilter = loadFilter;
        PerDeviceDeviceDefinitions.LoadFilter = loadFilter;

        return base.GetLoadedDevices(loadFilter);
    }

    /// <inheritdoc />
    protected override IEnumerable<IRGBDevice> LoadDevices()
    {
        #region PerKey

        if (_perKeyUpdateQueue != null)
        {
            (HIDDeviceDefinition<LogitechLedId, int> definition, HidDevice device) perKeyDevice = PerKeyDeviceDefinitions.GetConnectedDevices().FirstOrDefault();
            if (perKeyDevice != default)
            {
                (HIDDeviceDefinition<LogitechLedId, int> definition, _) = perKeyDevice;
                yield return new LogitechPerKeyRGBDevice(new LogitechRGBDeviceInfo(definition.DeviceType, definition.Name, LogitechDeviceCaps.PerKeyRGB, 0), _perKeyUpdateQueue, definition.LedMapping);
            }
            else
            {
                HIDDeviceDefinition<LogitechLedId, int>? perKeyWirelessDevice = PerKeyWirelessDeviceDefinitions.GetConnectedDevices().FirstOrDefault();
                if (perKeyWirelessDevice != null)
                    yield return new LogitechPerKeyRGBDevice(new LogitechRGBDeviceInfo(perKeyWirelessDevice.DeviceType, perKeyWirelessDevice.Name, LogitechDeviceCaps.PerKeyRGB, 0), _perKeyUpdateQueue, perKeyWirelessDevice.LedMapping);
            }
        }

        #endregion

        #region PerZone

        IEnumerable<HIDDeviceDefinition<int, (LogitechDeviceType deviceType, int zones, int zoneOffset)>> wiredPerZoneDevices = PerZoneDeviceDefinitions.GetConnectedDevices().Select(x => x.definition);
        IEnumerable<HIDDeviceDefinition<int, (LogitechDeviceType deviceType, int zones, int zoneOffset)>> wirelessPerZoneDevices = PerZoneWirelessDeviceDefinitions.GetConnectedDevices();
        foreach (HIDDeviceDefinition<int, (LogitechDeviceType deviceType, int zones, int zoneOffset)> definition in wiredPerZoneDevices.Concat(wirelessPerZoneDevices)
                                                                                                                                       .GroupBy(x => x.CustomData.deviceType)
                                                                                                                                       .Select(group => group.First()))
        {
            LogitechZoneUpdateQueue updateQueue = new(GetUpdateTrigger(), definition.CustomData.deviceType);
            yield return new LogitechZoneRGBDevice(new LogitechRGBDeviceInfo(definition.DeviceType, definition.Name, LogitechDeviceCaps.DeviceRGB, definition.CustomData.zones, definition.CustomData.zoneOffset),
                                                   updateQueue, definition.LedMapping);
        }

        #endregion

        #region PerDevice

        if (_perDeviceUpdateQueue != null)
        {
            (HIDDeviceDefinition<int, int> definition, HidDevice device) perDeviceDevice = PerDeviceDeviceDefinitions.GetConnectedDevices().FirstOrDefault();
            if (perDeviceDevice != default)
            {
                (HIDDeviceDefinition<int, int> definition, _) = perDeviceDevice;
                yield return new LogitechPerDeviceRGBDevice(new LogitechRGBDeviceInfo(definition.DeviceType, definition.Name, LogitechDeviceCaps.DeviceRGB, 0), _perDeviceUpdateQueue, definition.LedMapping);
            }
            else
            {
                HIDDeviceDefinition<int, int>? perDeviceWirelessDevice = PerDeviceWirelessDeviceDefinitions.GetConnectedDevices().FirstOrDefault();
                if (perDeviceWirelessDevice != null)
                    yield return new LogitechPerDeviceRGBDevice(new LogitechRGBDeviceInfo(perDeviceWirelessDevice.DeviceType, perDeviceWirelessDevice.Name, LogitechDeviceCaps.DeviceRGB, 0), _perDeviceUpdateQueue, perDeviceWirelessDevice.LedMapping);
            }
        }

        #endregion
    }

    /// <inheritdoc />
    protected override void Dispose(bool disposing)
    {
        lock (_lock)
        {
            base.Dispose(disposing);

            try { _LogitechGSDK.LogiLedRestoreLighting(); }
            catch { /* at least we tried */ }

            try { _LogitechGSDK.LogiLedShutdown(); }
            catch { /* at least we tried */ }

            try { _LogitechGSDK.UnloadLogitechGSDK(); }
            catch { /* at least we tried */ }

            _instance = null;
        }
    }

    #endregion
}