[OneBot] Custom SignServer url

Lagrange.OneBot/Core/Network/MsgRecvEventArgs.cs

@@ -1,5 +1,3 @@
-using Microsoft.Extensions.Hosting;
-
 namespace Lagrange.OneBot.Core.Network;
 
 public class MsgRecvEventArgs(string data, string? identifier = null) : EventArgs

Lagrange.OneBot/LagrangeApp.cs

@@ -1,6 +1,7 @@
 using System.Text.Json;
 using Lagrange.Core;
 using Lagrange.Core.Common.Interface.Api;
+using Lagrange.Core.Utility.Sign;
 using Lagrange.OneBot.Core.Message;
 using Lagrange.OneBot.Core.Network;
 using Lagrange.OneBot.Core.Operation;
@@ -44,7 +45,9 @@ internal LagrangeApp(IHost host)
     {
         await _hostApp.StartAsync(cancellationToken);
         Logger.LogInformation("Lagrange.OneBot Implementation has started");
-        Logger.LogInformation($"Protocol: {Configuration.GetValue<string>("Protocol")} | {Instance.ContextCollection.AppInfo.CurrentVersion}");
+        Logger.LogInformation($"Protocol: {Configuration["Protocol"]} | {Instance.ContextCollection.AppInfo.CurrentVersion}");
+
+        Instance.ContextCollection.Packet.SignProvider = Services.GetRequiredService<SignProvider>();
 
         Instance.Invoker.OnBotLogEvent += (_, args) => Logger.Log(args.Level switch
         {
@@ -71,11 +74,10 @@ internal LagrangeApp(IHost host)
             Instance.ContextCollection.Keystore.Session.TempPassword == null)
         {
             Logger.LogInformation("Session expired or Password not filled in, try to login by QrCode");
-
-            var qrCode = await Instance.FetchQrCode();
-            if (qrCode != null)
+
+            if (await Instance.FetchQrCode() is { } qrCode)
             {
-                QrCodeHelper.Output(qrCode.Value.Url ?? "");
+                QrCodeHelper.Output(qrCode.Url ?? "");
                 await Instance.LoginByQrCode();
             }
         }

Lagrange.OneBot/LagrangeAppBuilder.cs

@@ -1,10 +1,12 @@
 using System.Text.Json;
 using Lagrange.Core.Common;
 using Lagrange.Core.Common.Interface;
+using Lagrange.Core.Utility.Sign;
 using Lagrange.OneBot.Core.Message;
 using Lagrange.OneBot.Core.Network;
 using Lagrange.OneBot.Core.Operation;
 using Lagrange.OneBot.Database;
+using Lagrange.OneBot.Utility;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Hosting;
@@ -77,6 +79,7 @@ public LagrangeAppBuilder ConfigureOneBot()
     {
         Services.AddSingleton<ContextBase, LiteDbContext>();
         Services.AddSingleton<ILagrangeWebService, ReverseWSService>();
+        Services.AddSingleton<SignProvider, OneBotSigner>();
 
         Services.AddSingleton<MessageService>();
         Services.AddSingleton<OperationService>();

Lagrange.OneBot/Utility/OneBotSigner.cs

@@ -0,0 +1,67 @@
+using System.Diagnostics.CodeAnalysis;
+using System.Text;
+using System.Text.Json;
+using System.Text.Json.Nodes;
+using Lagrange.Core.Utility.Extension;
+using Lagrange.Core.Utility.Network;
+using Lagrange.Core.Utility.Sign;
+using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.Logging;
+
+namespace Lagrange.OneBot.Utility;
+
+public class OneBotSigner : SignProvider
+{
+    private const string Tag = nameof(OneBotSigner);
+    private readonly string _signServer;
+    private readonly HttpClient _client;
+    private readonly ILogger _logger;
+
+    public OneBotSigner(IConfiguration config, ILogger<LagrangeApp> logger)
+    {
+        _signServer = config["SignServerUrl"] ?? "";
+        _client = new HttpClient();
+        _logger = logger;
+
+        if (string.IsNullOrEmpty(_signServer))
+        {
+            Available = false;
+            logger.LogWarning($"[{Tag}]: Signature Service is not available, login may be failed");
+        }
+        else
+        {
+            logger.LogInformation($"[{Tag}]: Signature Service is successfully established");
+        }
+    }
+
+    public override byte[]? Sign(string cmd, uint seq, byte[] body, [UnscopedRef] out byte[]? ver, [UnscopedRef] out string? token)
+    {
+        ver = null;
+        token = null;
+        if (!WhiteListCommand.Contains(cmd)) return null;
+        if (!Available || string.IsNullOrEmpty(_signServer)) return new byte[35]; // Dummy signature
+
+        var payload = new Dictionary<string, string>
+        {
+            { "cmd", cmd },
+            { "seq", seq.ToString() },
+            { "src", body.Hex() },
+        };
+
+        try
+        {
+            string response = Http.GetAsync(_signServer, payload).GetAwaiter().GetResult();
+            var json = JsonSerializer.Deserialize<JsonObject>(response);
+
+            ver = json?["value"]?["extra"]?.ToString().UnHex() ?? Array.Empty<byte>();
+            token = Encoding.ASCII.GetString(json?["value"]?["token"]?.ToString().UnHex() ?? Array.Empty<byte>());
+            return json?["value"]?["sign"]?.ToString().UnHex() ?? new byte[35];
+        }
+        catch
+        {
+            Available = false;
+            _logger.LogWarning($"[{Tag}] Failed to get signature, using dummy signature");
+            return new byte[35]; // Dummy signature
+        }
+    }
+}