mirror of
https://github.com/DarthAffe/RGB.NET.git
synced 2026-11-11 04:57:15 +00:00
Added gRPC sdk client
This commit is contained in:
14 files changed
+426
-130
No files matched your search
@@ -0,0 +1,139 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading;
|
||||
using Grpc.Net.Client;
|
||||
using RGB.NET.Core;
|
||||
using RGB.NET.Devices.Wooting.Enum;
|
||||
using RGB.NET.Devices.Wooting.Keyboard;
|
||||
using RGB.NET.Devices.Wooting.Keypad;
|
||||
using WootingRgbSdk;
|
||||
|
||||
namespace RGB.NET.Devices.Wooting.Grpc;
|
||||
|
||||
/// <inheritdoc />
|
||||
/// <summary>
|
||||
/// Represents a device provider responsible for Wooting devices.
|
||||
/// </summary>
|
||||
public sealed class WootingGrpcDeviceProvider : AbstractRGBDeviceProvider
|
||||
{
|
||||
#region Properties & Fields
|
||||
|
||||
// ReSharper disable once InconsistentNaming
|
||||
private static readonly Lock _lock = new();
|
||||
|
||||
private GrpcChannel? _channel;
|
||||
private RgbSdkService.RgbSdkServiceClient? _client;
|
||||
|
||||
private static WootingGrpcDeviceProvider? _instance;
|
||||
/// <summary>
|
||||
/// Gets the singleton <see cref="WootingGrpcDeviceProvider"/> instance.
|
||||
/// </summary>
|
||||
public static WootingGrpcDeviceProvider Instance
|
||||
{
|
||||
get
|
||||
{
|
||||
lock (_lock)
|
||||
return _instance ?? new WootingGrpcDeviceProvider();
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Constructors
|
||||
|
||||
/// <summary>
|
||||
/// Initializes a new instance of the <see cref="WootingGrpcDeviceProvider"/> class.
|
||||
/// </summary>
|
||||
/// <exception cref="InvalidOperationException">Thrown if this constructor is called even if there is already an instance of this class.</exception>
|
||||
public WootingGrpcDeviceProvider()
|
||||
{
|
||||
lock (_lock)
|
||||
{
|
||||
if (_instance != null)
|
||||
throw new InvalidOperationException($"There can be only one instance of type {nameof(WootingGrpcDeviceProvider)}");
|
||||
_instance = this;
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Methods
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void InitializeSDK()
|
||||
{
|
||||
_channel = GrpcChannel.ForAddress("http://localhost:50051");
|
||||
_client = new RgbSdkService.RgbSdkServiceClient(_channel);
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override IEnumerable<IRGBDevice> LoadDevices()
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(_client, nameof(_client));
|
||||
|
||||
int i = 0;
|
||||
foreach (RgbGetConnectedDevicesResponse.Types.RgbDevice? device in _client.GetConnectedDevices(new()).Devices)
|
||||
{
|
||||
if (device.DeviceType == RgbDeviceType.None)
|
||||
continue; // Skip devices that are not supported
|
||||
|
||||
_client.Initialize(new RgbInitializeRequest { Id = device.Id });
|
||||
|
||||
WootingGrpcUpdateQueue updateQueue = new(GetUpdateTrigger(i++), device, _client);
|
||||
|
||||
WootingDeviceType deviceType = device.DeviceType switch
|
||||
{
|
||||
RgbDeviceType.Tkl => WootingDeviceType.KeyboardTKL,
|
||||
RgbDeviceType.FullSize => WootingDeviceType.Keyboard,
|
||||
RgbDeviceType.SixtyPercent => WootingDeviceType.KeyboardSixtyPercent,
|
||||
RgbDeviceType.ThreeKey => WootingDeviceType.Keypad3Keys,
|
||||
RgbDeviceType.EighyPercent => WootingDeviceType.KeyboardEightyPercent,
|
||||
_ or RgbDeviceType.None => throw new ArgumentOutOfRangeException()
|
||||
};
|
||||
KeyboardLayoutType layoutType = device.LayoutType switch
|
||||
{
|
||||
RgbDeviceLayout.Ansi => KeyboardLayoutType.ANSI,
|
||||
RgbDeviceLayout.Iso => KeyboardLayoutType.ISO,
|
||||
RgbDeviceLayout.Jis => KeyboardLayoutType.JIS,
|
||||
RgbDeviceLayout.AnsiSplitSpacebar => KeyboardLayoutType.ANSI,
|
||||
RgbDeviceLayout.IsoSplitSpacebar => KeyboardLayoutType.ISO,
|
||||
RgbDeviceLayout.Unknown => KeyboardLayoutType.Unknown,
|
||||
_ => throw new ArgumentOutOfRangeException()
|
||||
};
|
||||
|
||||
//NOTE: this model name ends up kind of ugly, since `ModelName` is like `Wooting 60HE`. We cannot remove the `Wooting` prefix,
|
||||
//since that makes loadouts fail to load. The deviceName part, however, is fine to strip.
|
||||
string model = device.ModelName;
|
||||
string name =
|
||||
DeviceHelper.CreateDeviceName("Wooting", $"{device.ModelName.Replace("Wooting", "").Trim()} ({device.SerialNumber})");
|
||||
|
||||
yield return deviceType switch
|
||||
{
|
||||
WootingDeviceType.Keypad3Keys => new WootingKeypadRGBDevice(deviceType, new(model, name), updateQueue),
|
||||
_ => new WootingKeyboardRGBDevice(deviceType, new(layoutType, model, name), updateQueue)
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void Dispose(bool disposing)
|
||||
{
|
||||
lock (_lock)
|
||||
{
|
||||
base.Dispose(disposing);
|
||||
|
||||
try
|
||||
{
|
||||
_client = null;
|
||||
_channel?.Dispose();
|
||||
}
|
||||
catch
|
||||
{ /* at least we tried */
|
||||
}
|
||||
|
||||
_instance = null;
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
@@ -0,0 +1,97 @@
|
||||
using System;
|
||||
using System.Runtime.InteropServices;
|
||||
using Google.Protobuf;
|
||||
using RGB.NET.Core;
|
||||
using RGB.NET.Devices.Wooting.Generic;
|
||||
using WootingRgbSdk;
|
||||
|
||||
namespace RGB.NET.Devices.Wooting.Grpc;
|
||||
|
||||
/// <inheritdoc />
|
||||
/// <summary>
|
||||
/// Represents the update-queue performing updates for cooler master devices.
|
||||
/// </summary>
|
||||
public sealed class WootingGrpcUpdateQueue : UpdateQueue
|
||||
{
|
||||
#region Properties & Fields
|
||||
|
||||
private readonly RgbSdkService.RgbSdkServiceClient _client;
|
||||
private readonly RgbGetConnectedDevicesResponse.Types.RgbDevice _wootDevice;
|
||||
private readonly WootingColor[] _colors;
|
||||
|
||||
#endregion
|
||||
|
||||
#region Constructors
|
||||
|
||||
/// <summary>
|
||||
/// Initializes a new instance of the <see cref="WootingGrpcUpdateQueue"/> class.
|
||||
/// </summary>
|
||||
/// <param name="updateTrigger">The update trigger used by this queue.</param>
|
||||
public WootingGrpcUpdateQueue(IDeviceUpdateTrigger updateTrigger, RgbGetConnectedDevicesResponse.Types.RgbDevice wootDevice,
|
||||
RgbSdkService.RgbSdkServiceClient client)
|
||||
: base(updateTrigger)
|
||||
{
|
||||
this._client = client;
|
||||
this._wootDevice = wootDevice;
|
||||
this._colors = new WootingColor[WootingLedMappings.COLUMNS * WootingLedMappings.ROWS];
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Methods
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override bool Update(ReadOnlySpan<(object key, Color color)> dataSet)
|
||||
{
|
||||
try
|
||||
{
|
||||
foreach ((object key, Color color) in dataSet)
|
||||
{
|
||||
(int row, int column) = ((int, int))key;
|
||||
long index = (WootingLedMappings.COLUMNS * row) + column;
|
||||
|
||||
_colors[index] = new WootingColor(color.GetR(), color.GetG(), color.GetB());
|
||||
}
|
||||
|
||||
_client.SetColors(new RgbSetColorsRequest
|
||||
{
|
||||
Id = _wootDevice.Id,
|
||||
Colors = ByteString.CopyFrom(MemoryMarshal.AsBytes(_colors.AsSpan()))
|
||||
});
|
||||
return true;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
WootingGrpcDeviceProvider.Instance.Throw(ex);
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override void Dispose()
|
||||
{
|
||||
_client.Close(new RgbCloseRequest { Id = _wootDevice.Id });
|
||||
base.Dispose();
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
|
||||
[StructLayout(LayoutKind.Sequential, Pack = 1)]
|
||||
internal struct WootingColor
|
||||
{
|
||||
public byte r;
|
||||
public byte g;
|
||||
public byte b;
|
||||
public byte a;
|
||||
|
||||
public WootingColor(byte r, byte g, byte b)
|
||||
{
|
||||
this.r = r;
|
||||
this.g = g;
|
||||
this.b = b;
|
||||
this.a = 0; // Alpha is not used in Wooting devices
|
||||
}
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user