Added support for ASFEnhance features

Citrinate · Mar 3, 2024 · 82d48ed · 82d48ed
1 parent 86af6c9
commit 82d48ed
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 1 deletion.
diff --git a/FreePackages/AdapterBridge.cs b/FreePackages/AdapterBridge.cs
@@ -0,0 +1,57 @@
+using System;
+using System.Reflection;
+using ArchiSteamFarm.Core;
+
+// ASFEnhanced Adapter https://github.com/chr233/ASFEnhanceAdapterDemoPlugin
+
+namespace FreePackages;
+internal static class AdapterBridge
+{
+    /// <summary>
+    /// 注册子模块
+    /// </summary>
+    /// <param name="pluginName">插件名称</param>
+    /// <param name="pluginId">插件唯一标识符</param>
+    /// <param name="cmdPrefix">命令前缀</param>
+    /// <param name="repoName">自动更新仓库</param>
+    /// <param name="cmdHandler">命令处理函数</param>
+    /// <returns></returns>
+    public static bool InitAdapter(string pluginName, string pluginId, string? cmdPrefix, string? repoName, MethodInfo? cmdHandler)
+    {
+        try
+        {
+            var adapterEndpoint = Assembly.Load("ASFEnhance").GetType("ASFEnhance._Adapter_.Endpoint");
+            var registerModule = adapterEndpoint?.GetMethod("RegisterModule", BindingFlags.Static | BindingFlags.Public);
+            var pluinVersion = Assembly.GetExecutingAssembly().GetName().Version;
+
+            if (registerModule != null && adapterEndpoint != null)
+            {
+                var result = registerModule?.Invoke(null, new object?[] { pluginName, pluginId, cmdPrefix, repoName, pluinVersion, cmdHandler });
+
+                if (result is string str)
+                {
+                    if (str == pluginName)
+                    {
+                        return true;
+                    }
+                    else
+                    {
+                        ASF.ArchiLogger.LogGenericWarning(str);
+                    }
+                }
+            }
+        }
+#if DEBUG
+        catch (Exception ex)
+        {
+            ASF.ArchiLogger.LogGenericException(ex, "Community with ASFEnhance failed");
+        }
+#else
+        catch (Exception)
+        {
+            ASF.ArchiLogger.LogGenericDebug("Community with ASFEnhance failed");
+        }
+#endif
+        return false;
+    }
+}
diff --git a/FreePackages/Commands.cs b/FreePackages/Commands.cs
@@ -22,6 +22,9 @@ internal static class Commands {
 			switch (args.Length) {
 				case 1:
 					switch (args[0].ToUpperInvariant()) {
+						case "FREEPACKAGES" when access >= EAccess.FamilySharing:
+							return String.Format("{0} {1}", nameof(FreePackages), (typeof(FreePackages).Assembly.GetName().Version ?? new Version("0")).ToString());
+
 						case "CLEARFREEPACKAGESQUEUE":
 							return ResponseClearQueue(bot, access);
 

diff --git a/FreePackages/FreePackages.cs b/FreePackages/FreePackages.cs
@@ -8,20 +8,38 @@
 using SteamKit2;
 using System.Text.Json;
 using ArchiSteamFarm.Helpers.Json;
+using System.Reflection;
 
 namespace FreePackages {
 	[Export(typeof(IPlugin))]
 	public sealed class FreePackages : IASF, IBotModules, ISteamPICSChanges, IBotSteamClient, IBotConnection, IBotCommand2 {
 		public string Name => nameof(FreePackages);
 		public Version Version => typeof(FreePackages).Assembly.GetName().Version ?? new Version("0");
 		internal static GlobalCache? GlobalCache;
+		private bool ASFEnhanceEnabled = false;
 
 		public Task OnLoaded() {
 			ASF.ArchiLogger.LogGenericInfo("Free Packages ASF Plugin by Citrinate");
+
+			// ASFEnhanced Adapter https://github.com/chr233/ASFEnhanceAdapterDemoPlugin
+			var flag = BindingFlags.Static | BindingFlags.Public | BindingFlags.NonPublic;
+			var handler = typeof(Commands).GetMethod(nameof(Commands.Response), flag);
+			const string pluginId = nameof(FreePackages);
+			const string cmdPrefix = "FREEPACKAGES";
+			const string repoName = "Citrinate/FreePackages";
+			var registered = AdapterBridge.InitAdapter(Name, pluginId, cmdPrefix, repoName, handler);
+			ASFEnhanceEnabled = registered;
+
 			return Task.CompletedTask;
 		}
 
-		public Task<string?> OnBotCommand(Bot bot, EAccess access, string message, string[] args, ulong steamID = 0) => Task.FromResult(Commands.Response(bot, access, steamID, message, args));
+		public Task<string?> OnBotCommand(Bot bot, EAccess access, string message, string[] args, ulong steamID = 0) {
+			if (ASFEnhanceEnabled) {
+				return Task.FromResult<string?>(null);
+			}
+
+			return Task.FromResult(Commands.Response(bot, access, steamID, message, args));
+		}
 
 		public async Task OnASFInit(IReadOnlyDictionary<string, JsonElement>? additionalConfigProperties = null) {
 			if (GlobalCache == null) {