RGB.NET/RGB.NET.Devices.Corsair_Legacy/Custom/CorsairCustomRGBDeviceInfo.cs

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

using System;
using System.Runtime.InteropServices;
using System.Text.RegularExpressions;
using RGB.NET.Core;
using RGB.NET.Devices.CorsairLegacy.Native;

namespace RGB.NET.Devices.CorsairLegacy;

/// <inheritdoc />
/// <summary>
/// Represents a generic information for a <see cref="T:RGB.NET.Devices.CorsairLegacy.CorsairCustomRGBDevice" />.
/// </summary>
public class CorsairCustomRGBDeviceInfo : CorsairRGBDeviceInfo
{
    #region Properties & Fields

    /// <summary>
    /// Gets the amount of LEDs this device contains.
    /// </summary>
    public int LedCount { get; }

    /// <summary>
    /// Gets the offset used to access the LEDs of this device.
    /// </summary>
    internal int LedOffset { get; }

    #endregion

    #region Constructors

    /// <inheritdoc />
    /// <summary>
    /// Internal constructor of managed <see cref="T:RGB.NET.Devices.Corsair.CorsairCustomRGBDeviceInfo" />.
    /// </summary>
    /// <param name="deviceIndex">The index of the <see cref="T:RGB.NET.Devices.Corsair._CorsairChannelDeviceInfo" />.</param>
    /// <param name="nativeInfo">The native <see cref="T:RGB.NET.Devices.Corsair.Native._CorsairDeviceInfo" />-struct</param>
    /// <param name="channelDeviceInfo">The native <see cref="T:RGB.NET.Devices.Corsair.Native._CorsairChannelDeviceInfo"/> representing this device.</param>
    /// <param name="ledOffset">The offset used to find the LEDs of this device.</param>
    internal CorsairCustomRGBDeviceInfo(int deviceIndex, _CorsairDeviceInfo nativeInfo, _CorsairChannelDeviceInfo channelDeviceInfo, int ledOffset)
        : base(deviceIndex, GetDeviceType(channelDeviceInfo.type), nativeInfo,
               GetModelName(nativeInfo.model == 0 ? string.Empty : Regex.Replace(Marshal.PtrToStringAnsi(nativeInfo.model) ?? string.Empty, " ?DEMO", string.Empty, RegexOptions.IgnoreCase), channelDeviceInfo))
    {
        this.LedOffset = ledOffset;

        LedCount = channelDeviceInfo.deviceLedCount;
    }

    internal CorsairCustomRGBDeviceInfo(int deviceIndex, _CorsairDeviceInfo nativeInfo, int ledCount)
        : base(deviceIndex, GetDeviceType(nativeInfo.type), nativeInfo)
    {
        this.LedCount = ledCount;

        LedOffset = 0;
    }

    #endregion

    #region Methods

    private static RGBDeviceType GetDeviceType(CorsairChannelDeviceType deviceType)
        => deviceType switch
        {
            CorsairChannelDeviceType.Invalid => RGBDeviceType.Unknown,
            CorsairChannelDeviceType.FanHD => RGBDeviceType.Fan,
            CorsairChannelDeviceType.FanSP => RGBDeviceType.Fan,
            CorsairChannelDeviceType.FanLL => RGBDeviceType.Fan,
            CorsairChannelDeviceType.FanML => RGBDeviceType.Fan,
            CorsairChannelDeviceType.DAP => RGBDeviceType.Fan,
            CorsairChannelDeviceType.FanQL => RGBDeviceType.Fan,
            CorsairChannelDeviceType.EightLedSeriesFan => RGBDeviceType.Fan,
            CorsairChannelDeviceType.Strip => RGBDeviceType.LedStripe,
            CorsairChannelDeviceType.Pump => RGBDeviceType.Cooler,
            CorsairChannelDeviceType.WaterBlock => RGBDeviceType.Cooler,
            _ => throw new ArgumentOutOfRangeException(nameof(deviceType), deviceType, null)
        };

    private static RGBDeviceType GetDeviceType(CorsairDeviceType deviceType)
        => deviceType switch
        {
            CorsairDeviceType.Unknown => RGBDeviceType.Unknown,
            CorsairDeviceType.Mouse => RGBDeviceType.Mouse,
            CorsairDeviceType.Keyboard => RGBDeviceType.Keyboard,
            CorsairDeviceType.Headset => RGBDeviceType.Headset,
            CorsairDeviceType.Mousepad => RGBDeviceType.Mousepad,
            CorsairDeviceType.HeadsetStand => RGBDeviceType.HeadsetStand,
            CorsairDeviceType.CommanderPro => RGBDeviceType.LedController,
            CorsairDeviceType.LightningNodePro => RGBDeviceType.LedController,
            CorsairDeviceType.MemoryModule => RGBDeviceType.DRAM,
            CorsairDeviceType.Cooler => RGBDeviceType.Cooler,
            CorsairDeviceType.Mainboard => RGBDeviceType.Mainboard,
            CorsairDeviceType.GraphicsCard => RGBDeviceType.GraphicsCard,
            _ => throw new ArgumentOutOfRangeException(nameof(deviceType), deviceType, null)
        };

    private static string GetModelName(string model, _CorsairChannelDeviceInfo channelDeviceInfo)
    {
        switch (channelDeviceInfo.type)
        {
            case CorsairChannelDeviceType.Invalid:
                return model;

            case CorsairChannelDeviceType.FanHD:
                return "HD Fan";

            case CorsairChannelDeviceType.FanSP:
                return "SP Fan";

            case CorsairChannelDeviceType.FanLL:
                return "LL Fan";

            case CorsairChannelDeviceType.FanML:
                return "ML Fan";

            case CorsairChannelDeviceType.FanQL:
                return "QL Fan";

            case CorsairChannelDeviceType.EightLedSeriesFan:
                return "8-Led-Series Fan Fan";

            case CorsairChannelDeviceType.Strip:
                // LS100 Led Strips are reported as one big strip if configured in monitor mode in iCUE, 138 LEDs for dual monitor, 84 for single
                if ((model == "LS100 Starter Kit") && (channelDeviceInfo.deviceLedCount == 138))
                    return "LS100 LED Strip (dual monitor)";
                else if ((model == "LS100 Starter Kit") && (channelDeviceInfo.deviceLedCount == 84))
                    return "LS100 LED Strip (single monitor)";
                // Any other value means an "External LED Strip" in iCUE, these are reported per-strip, 15 for short strips, 27 for long
                else if ((model == "LS100 Starter Kit") && (channelDeviceInfo.deviceLedCount == 15))
                    return "LS100 LED Strip (short)";
                else if ((model == "LS100 Starter Kit") && (channelDeviceInfo.deviceLedCount == 27))
                    return "LS100 LED Strip (long)";
                // Device model is "Commander Pro" for regular LED strips
                else
                    return "LED Strip";

            case CorsairChannelDeviceType.DAP:
                return "DAP Fan";

            case CorsairChannelDeviceType.WaterBlock:
                return "Water Block";

            case CorsairChannelDeviceType.Pump:
                return "Pump";

            default:
#pragma warning disable CA2208 // Instantiate argument exceptions correctly
                throw new ArgumentOutOfRangeException($"{nameof(channelDeviceInfo)}.{nameof(channelDeviceInfo.type)}", channelDeviceInfo.type, null);
#pragma warning restore CA2208 // Instantiate argument exceptions correctly
        }
    }

    #endregion
}