[Core] Custom Signer

LagrangeDev · Oct 24, 2023 · 953b58d · 953b58d
1 parent d8d0456
commit 953b58d
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/Lagrange.Core/Common/BotConfig.cs b/Lagrange.Core/Common/BotConfig.cs
@@ -1,3 +1,5 @@
+using Lagrange.Core.Utility.Sign;
+
 namespace Lagrange.Core.Common;
 
 /// <summary>
@@ -25,6 +27,11 @@ public class BotConfig
     /// Get optimum server from Tencent MSF server, set to false to use hardcode server
     /// </summary>
     public bool GetOptimumServer { get; set; } = true;
+
+    /// <summary>
+    /// Custom Sign Provider
+    /// </summary>
+    public SignProvider? CustomSignProvider { get; set; } = null;
 }
 
 /// <summary>

diff --git a/Lagrange.Core/Internal/Context/ContextCollection.cs b/Lagrange.Core/Internal/Context/ContextCollection.cs
@@ -16,6 +16,7 @@ internal class ContextCollection
     public BotKeystore Keystore { get; }
     public BotAppInfo AppInfo { get; }
     public BotDeviceInfo Device { get; }
+    public BotConfig Config { get; }
 
     public TaskScheduler Scheduler { get; }
     public EventInvoker Invoker { get; }

diff --git a/Lagrange.Core/Internal/Context/PacketContext.cs b/Lagrange.Core/Internal/Context/PacketContext.cs
@@ -22,7 +22,7 @@ internal class PacketContext : ContextBase
     public PacketContext(ContextCollection collection, BotKeystore keystore, BotAppInfo appInfo, BotDeviceInfo device) 
         : base(collection, keystore, appInfo, device)
     {
-        _signProvider = appInfo.Os switch
+        _signProvider = Collection.Config.CustomSignProvider ?? appInfo.Os switch
         {
             "Windows" => new WindowsSigner(),
             "Mac" => new MacSigner(),

diff --git a/Lagrange.Core/Utility/Sign/SignProvider.cs b/Lagrange.Core/Utility/Sign/SignProvider.cs
@@ -1,6 +1,6 @@
 namespace Lagrange.Core.Utility.Sign;
 
-internal abstract class SignProvider
+public abstract class SignProvider
 {
     protected bool Available = true;