Corsair device changed event

RGB.NET.Core/Devices/AbstractRGBDeviceProvider.cs

@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
 using System.Linq;
+using System.Runtime.CompilerServices;
 
 namespace RGB.NET.Core;
 
@@ -94,6 +95,15 @@ public bool Initialize(RGBDeviceType loadFilter = RGBDeviceType.All, bool throwE
             Reset();
             throw;
         }
+        catch (RGBDeviceException)
+        {
+            Reset();
+            if (throwExceptions)
+            {
+                throw;
+            }
+            return false;
+        }
         catch (Exception ex)
         {
             Reset();
@@ -158,6 +168,7 @@ protected virtual IEnumerable<IRGBDevice> GetLoadedDevices(RGBDeviceType loadFil
     /// <param name="id">The id of the update trigger.</param>
     /// <param name="updateRateHardLimit">The update rate hard limit to be set in the update trigger.</param>
     /// <returns>The update trigger mapped to the specified id.</returns>
+    [MethodImpl(MethodImplOptions.Synchronized)]
     protected virtual IDeviceUpdateTrigger GetUpdateTrigger(int id = -1, double? updateRateHardLimit = null)
     {
         if (_isDisposed) throw new ObjectDisposedException(GetType().FullName);

RGB.NET.Core/Ids/IdGenerator.cs

@@ -13,8 +13,8 @@ public static class IdGenerator
 
     // ReSharper disable InconsistentNaming
     private static readonly HashSet<string> _registeredIds = [];
-    private static readonly Dictionary<Assembly, Dictionary<string, string>> _idMappings = [];
-    private static readonly Dictionary<Assembly, Dictionary<string, int>> _counter = [];
+    private static readonly Dictionary<object, Dictionary<string, string>> _idMappings = [];
+    private static readonly Dictionary<object, Dictionary<string, int>> _counter = [];
     // ReSharper restore InconsistentNaming
 
     #endregion
@@ -29,7 +29,7 @@ public static class IdGenerator
     [MethodImpl(MethodImplOptions.NoInlining)]
     public static string MakeUnique(string id) => MakeUnique(Assembly.GetCallingAssembly(), id);
 
-    internal static string MakeUnique(Assembly callingAssembly, string id)
+    public static string MakeUnique(object callingAssembly, string id)
     {
         if (!_idMappings.TryGetValue(callingAssembly, out Dictionary<string, string>? idMapping))
         {
@@ -63,7 +63,7 @@ internal static string MakeUnique(Assembly callingAssembly, string id)
     [MethodImpl(MethodImplOptions.NoInlining)]
     public static void ResetCounter() => ResetCounter(Assembly.GetCallingAssembly());
 
-    internal static void ResetCounter(Assembly callingAssembly)
+    public static void ResetCounter(object callingAssembly)
     {
         if (_counter.TryGetValue(callingAssembly, out Dictionary<string, int>? counter))
             counter.Clear();

RGB.NET.Core/Update/Devices/DeviceUpdateTrigger.cs

@@ -141,11 +141,15 @@ public virtual async void Stop()
 
         UpdateTokenSource?.Cancel();
         if (UpdateTask != null)
-            try { await UpdateTask.ConfigureAwait(false); }
+            try
+            {
+                await UpdateTask.ConfigureAwait(false);
+                UpdateTask?.Dispose();
+            }
             catch (TaskCanceledException) { }
             catch (OperationCanceledException) { }
+            catch (InvalidOperationException) { }
 
-        UpdateTask?.Dispose();
         UpdateTask = null;
     }
 

RGB.NET.Devices.Corsair/Enum/CorsairEventId.cs

@@ -0,0 +1,8 @@
+namespace RGB.NET.Devices.Corsair;
+
+internal enum CorsairEventId
+{
+    Invalid = 0,
+    DeviceConnectionStatusChangedEvent = 1,
+    KeyEvent = 2,
+}

RGB.NET.Devices.Corsair/Generic/CorsairRGBDevice.cs

@@ -14,6 +14,11 @@ public abstract class CorsairRGBDevice<TDeviceInfo> : AbstractRGBDevice<TDeviceI
 {
     #region Properties & Fields
 
+    /// <summary>
+    /// <inheritdoc cref="ICorsairRGBDevice"/>
+    /// </summary>
+    public string DeviceId => DeviceInfo.DeviceId;
+
     /// <summary>
     /// Gets the mapping of <see cref="LedId"/> to <see cref="CorsairLedId"/> used to update the LEDs of this device.
     /// </summary>
@@ -29,14 +34,32 @@ public abstract class CorsairRGBDevice<TDeviceInfo> : AbstractRGBDevice<TDeviceI
     /// <param name="info">The generic information provided by CUE for the device.</param>
     /// <param name="mapping">The mapping <see cref="LedId"/> to <see cref="CorsairLedId"/> used to update the LEDs of this device.</param>
     /// <param name="updateQueue">The queue used to update this device.</param>
-    protected CorsairRGBDevice(TDeviceInfo info, CorsairDeviceUpdateQueue updateQueue)
+    protected CorsairRGBDevice(TDeviceInfo info, IUpdateQueue updateQueue)
         : base(info, updateQueue)
     { }
 
     #endregion
 
     #region Methods
 
+    protected bool Equals(CorsairRGBDevice<TDeviceInfo> other)
+    {
+        return DeviceId == other.DeviceId;
+    }
+
+    public override bool Equals(object? obj)
+    {
+        if (ReferenceEquals(null, obj)) return false;
+        if (ReferenceEquals(this, obj)) return true;
+        if (obj.GetType() != this.GetType()) return false;
+        return Equals((CorsairRGBDevice<TDeviceInfo>)obj);
+    }
+
+    public override int GetHashCode()
+    {
+        return DeviceId.GetHashCode();
+    }
+
     void ICorsairRGBDevice.Initialize() => InitializeLayout();
 
     /// <summary>

RGB.NET.Devices.Corsair/Generic/CorsairRGBDeviceInfo.cs

@@ -1,4 +1,7 @@
-using System.Text.RegularExpressions;
+using System;
+using System.Security.Cryptography;
+using System.Text;
+using System.Text.RegularExpressions;
 using RGB.NET.Core;
 using RGB.NET.Devices.Corsair.Native;
 
@@ -33,7 +36,7 @@ public class CorsairRGBDeviceInfo : IRGBDeviceInfo
     /// Returns the unique ID provided by the Corsair-SDK.
     /// Returns string.Empty for Custom devices.
     /// </summary>
-    public string DeviceId { get; }
+    public string DeviceId { get; init; }
 
     /// <inheritdoc />
     public object? LayoutMetadata { get; set; }
@@ -67,7 +70,14 @@ internal CorsairRGBDeviceInfo(RGBDeviceType deviceType, _CorsairDeviceInfo nativ
         this.LedCount = ledCount;
         this.LedOffset = ledOffset;
 
-        DeviceName = DeviceHelper.CreateDeviceName(Manufacturer, Model);
+        if (nativeInfo.id == null)  // this device is 99% unpluggable
+        {
+            DeviceName = IdGenerator.MakeUnique(typeof(CorsairDeviceProvider), Manufacturer + " " + Model);
+        }
+        else
+        {
+            DeviceName = Manufacturer + " " + Model + " #" + HashAndShorten(DeviceId);
+        }
     }
 
     /// <summary>
@@ -86,7 +96,34 @@ internal CorsairRGBDeviceInfo(RGBDeviceType deviceType, _CorsairDeviceInfo nativ
         this.LedCount = ledCount;
         this.LedOffset = ledOffset;
 
-        DeviceName = DeviceHelper.CreateDeviceName(Manufacturer, Model);
+        if (nativeInfo.id == null)
+        {
+            DeviceName = IdGenerator.MakeUnique(typeof(CorsairDeviceProvider),Manufacturer + " " + Model) + " " + ledOffset;;
+        }
+        else
+        {
+            DeviceName = Manufacturer + " " + Model + " #" + HashAndShorten(DeviceId) + " " + ledOffset;
+        }
+    }
+
+    #endregion
+
+    #region Methods
+
+    private static string HashAndShorten(string input)
+    {
+        using SHA256 sha256Hash = SHA256.Create();
+        byte[] bytes = sha256Hash.ComputeHash(Encoding.UTF8.GetBytes(input));
+        // Take the first 4 bytes of the hash
+        byte[] shortenedBytes = new byte[4];
+        Array.Copy(bytes, shortenedBytes, 4);
+        // Convert the bytes to a string
+        StringBuilder shortenedHash = new();
+        foreach (byte b in shortenedBytes)
+        {
+            shortenedHash.Append(b.ToString("X2"));
+        }
+        return shortenedHash.ToString();
     }
 
     #endregion

RGB.NET.Devices.Corsair/Generic/ICorsairRGBDevice.cs

@@ -7,5 +7,7 @@ namespace RGB.NET.Devices.Corsair;
 /// </summary>
 public interface ICorsairRGBDevice : IRGBDevice
 {
+    internal string DeviceId { get; }
+
     internal void Initialize();
 }

RGB.NET.Devices.Corsair/Native/_CUESDK.cs

@@ -12,6 +12,7 @@
 namespace RGB.NET.Devices.Corsair.Native;
 
 internal delegate void CorsairSessionStateChangedHandler(nint context, _CorsairSessionStateChanged eventData);
+internal delegate void CorsairEventHandler(nint context, _CorsairEvent corsairEvent);
 
 // ReSharper disable once InconsistentNaming
 internal static unsafe class _CUESDK
@@ -47,23 +48,59 @@ internal static unsafe class _CUESDK
 
     #region Properties & Fields
 
-    internal static bool IsConnected => SesionState == CorsairSessionState.Connected;
-    internal static CorsairSessionState SesionState { get; private set; }
+    internal static bool IsConnected => SessionState == CorsairSessionState.Connected;
+    internal static CorsairSessionState SessionState { get; private set; }
 
     #endregion
 
     #region Events
 
     internal static event EventHandler<CorsairSessionState>? SessionStateChanged;
+    internal static event EventHandler<_CorsairDeviceConnectionStatusChangedEvent>? DeviceConnectionEvent;
 
     #endregion
 
     #region Methods
 
-    private static void CorsairSessionStateChangedCallback(nint context, _CorsairSessionStateChanged eventdata)
+    private static void CorsairSessionStateChangedCallback(nint context, _CorsairSessionStateChanged eventData)
     {
-        SesionState = eventdata.state;
-        SessionStateChanged?.Invoke(null, eventdata.state);
+        SessionState = eventData.state;
+        try
+        {
+            SessionStateChanged?.Invoke(null, eventData.state);
+        }
+        catch { /* dont let exception go to sdk */ }
+
+        switch (eventData.state)
+        {
+            case CorsairSessionState.Connected:
+                _corsairSubscribeForEvents(CorsairEventCallback, 0);
+                break;
+            case CorsairSessionState.Closed:
+                _corsairUnsubscribeForEvents();
+                break;
+        }
+    }
+
+    private static void CorsairEventCallback(nint context, _CorsairEvent eventData)
+    {
+        if (eventData.id != CorsairEventId.DeviceConnectionStatusChangedEvent)
+        {
+            return;
+        }
+
+        try
+        {
+            if (eventData.eventPointer == 0)
+            {
+                return;
+            }
+
+            _CorsairDeviceConnectionStatusChangedEvent connectionStatusChangedEvent =
+                Marshal.PtrToStructure<_CorsairDeviceConnectionStatusChangedEvent>(eventData.eventPointer)!;
+
+            DeviceConnectionEvent?.Invoke(null, connectionStatusChangedEvent);
+        }catch { /* dont let exception go to sdk */ }
     }
 
     #endregion
@@ -97,7 +134,7 @@ private static void LoadCUESDK()
         _corsairGetSessionDetails = (delegate* unmanaged[Cdecl]<nint, CorsairError>)LoadFunction("CorsairGetSessionDetails");
         _corsairDisconnect = (delegate* unmanaged[Cdecl]<CorsairError>)LoadFunction("CorsairDisconnect");
         _corsairGetDevices = (delegate* unmanaged[Cdecl]<_CorsairDeviceFilter, int, nint, out int, CorsairError>)LoadFunction("CorsairGetDevices");
-        _corsairGetDeviceInfo = (delegate* unmanaged[Cdecl]<string, _CorsairDeviceInfo, CorsairError>)LoadFunction("CorsairGetDeviceInfo");
+        _corsairGetDeviceInfo = (delegate* unmanaged[Cdecl]<string, ref _CorsairDeviceInfo, CorsairError>)LoadFunction("CorsairGetDeviceInfo");
         _corsairGetLedPositions = (delegate* unmanaged[Cdecl]<string, int, nint, out int, CorsairError>)LoadFunction("CorsairGetLedPositions");
         _corsairSetLedColors = (delegate* unmanaged[Cdecl]<string, int, nint, CorsairError>)LoadFunction("CorsairSetLedColors");
         _corsairSetLayerPriority = (delegate* unmanaged[Cdecl]<uint, CorsairError>)LoadFunction("CorsairSetLayerPriority");
@@ -106,6 +143,8 @@ private static void LoadCUESDK()
         _corsairReleaseControl = (delegate* unmanaged[Cdecl]<string, CorsairError>)LoadFunction("CorsairReleaseControl");
         _getDevicePropertyInfo = (delegate* unmanaged[Cdecl]<string, CorsairDevicePropertyId, uint, out CorsairDataType, out CorsairPropertyFlag, CorsairError>)LoadFunction("CorsairGetDevicePropertyInfo");
         _readDeviceProperty = (delegate* unmanaged[Cdecl]<string, CorsairDevicePropertyId, uint, nint, CorsairError>)LoadFunction("CorsairReadDeviceProperty");
+        _corsairSubscribeForEvents = (delegate* unmanaged[Cdecl]<CorsairEventHandler, nint, CorsairError>)LoadFunction("CorsairSubscribeForEvents");
+        _corsairUnsubscribeForEvents = (delegate* unmanaged[Cdecl]<CorsairError>)LoadFunction("CorsairSubscribeForEvents");
     }
 
     private static nint LoadFunction(string function)
@@ -158,7 +197,7 @@ internal static void UnloadCUESDK()
     private static delegate* unmanaged[Cdecl]<nint, CorsairError> _corsairGetSessionDetails;
     private static delegate* unmanaged[Cdecl]<CorsairError> _corsairDisconnect;
     private static delegate* unmanaged[Cdecl]<_CorsairDeviceFilter, int, nint, out int, CorsairError> _corsairGetDevices;
-    private static delegate* unmanaged[Cdecl]<string, _CorsairDeviceInfo, CorsairError> _corsairGetDeviceInfo;
+    private static delegate* unmanaged[Cdecl]<string, ref _CorsairDeviceInfo, CorsairError> _corsairGetDeviceInfo;
     private static delegate* unmanaged[Cdecl]<string, int, nint, out int, CorsairError> _corsairGetLedPositions;
     private static delegate* unmanaged[Cdecl]<string, int, nint, CorsairError> _corsairSetLedColors;
     private static delegate* unmanaged[Cdecl]<uint, CorsairError> _corsairSetLayerPriority;
@@ -167,13 +206,18 @@ internal static void UnloadCUESDK()
     private static delegate* unmanaged[Cdecl]<string, CorsairError> _corsairReleaseControl;
     private static delegate* unmanaged[Cdecl]<string, CorsairDevicePropertyId, uint, out CorsairDataType, out CorsairPropertyFlag, CorsairError> _getDevicePropertyInfo;
     private static delegate* unmanaged[Cdecl]<string, CorsairDevicePropertyId, uint, nint, CorsairError> _readDeviceProperty;
+    private static delegate* unmanaged[Cdecl]<CorsairEventHandler, nint, CorsairError> _corsairSubscribeForEvents;
+    private static delegate* unmanaged[Cdecl]<CorsairError> _corsairUnsubscribeForEvents;
 
     #endregion
 
     internal static CorsairError CorsairConnect()
     {
         if (_corsairConnectPtr == null) throw new RGBDeviceException("The Corsair-SDK is not initialized.");
-        if (IsConnected) throw new RGBDeviceException("The Corsair-SDK is already connected.");
+        if (SessionState is CorsairSessionState.Connecting or CorsairSessionState.Timeout)
+        {
+            return CorsairError.Success;
+        }
         return _corsairConnectPtr(CorsairSessionStateChangedCallback, 0);
     }
 
@@ -197,7 +241,6 @@ internal static CorsairError CorsairGetSessionDetails(out _CorsairSessionDetails
 
     internal static CorsairError CorsairDisconnect()
     {
-        if (!IsConnected) throw new RGBDeviceException("The Corsair-SDK is not connected.");
         return _corsairDisconnect();
     }
 
@@ -220,10 +263,12 @@ internal static CorsairError CorsairGetDevices(_CorsairDeviceFilter filter, out
         }
     }
 
-    internal static CorsairError CorsairGetDeviceInfo(string deviceId, _CorsairDeviceInfo deviceInfo)
+    internal static CorsairError CorsairGetDeviceInfo(string deviceId, out _CorsairDeviceInfo deviceInfo)
     {
         if (!IsConnected) throw new RGBDeviceException("The Corsair-SDK is not connected.");
-        return _corsairGetDeviceInfo(deviceId, deviceInfo);
+
+        deviceInfo = new _CorsairDeviceInfo();
+        return _corsairGetDeviceInfo(deviceId, ref deviceInfo);
     }
 
     internal static CorsairError CorsairGetLedPositions(string deviceId, out _CorsairLedPosition[] ledPositions)

RGB.NET.Devices.Corsair/Native/_CorsairDeviceConnectionStatusChangedEvent.cs

@@ -0,0 +1,19 @@
+using System.Runtime.InteropServices;
+
+namespace RGB.NET.Devices.Corsair.Native;
+
+[StructLayout(LayoutKind.Sequential)]
+public class _CorsairDeviceConnectionStatusChangedEvent
+{
+    /// <summary>
+    /// iCUE-SDK: null terminated Unicode string that contains unique device identifier
+    /// </summary>
+    [MarshalAs(UnmanagedType.ByValTStr, SizeConst = _CUESDK.CORSAIR_STRING_SIZE_M)]
+    internal string? deviceId;
+
+    /// <summary>
+    /// iCUE-SDK: true if connected, false if disconnected
+    /// </summary>
+    [MarshalAs(UnmanagedType.U1)]
+    internal bool isConnected;
+}

RGB.NET.Devices.Corsair/Native/_CorsairDeviceInfo.cs

@@ -12,7 +12,7 @@ namespace RGB.NET.Devices.Corsair.Native;
 /// iCUE-SDK: contains information about device
 /// </summary>
 [StructLayout(LayoutKind.Sequential)]
-internal sealed class _CorsairDeviceInfo
+internal struct _CorsairDeviceInfo
 {
     /// <summary>
     /// iCUE-SDK: enum describing device type