实现API功能并编写日志过滤方法

Signed-off-by: Eicy <im@crash.work>
2024-12-21 01:05:49 +08:00
parent 2537d11f53
commit 3a41f098fe
7 changed files with 87 additions and 100 deletions
--- a/Service/BotService.cs
+++ b/Service/BotService.cs
@@ -0,0 +1,96 @@
+using System.Text.Json;
+using Debugger;
+using Grpc.Core;
+using Lagrange.Core;
+using Lagrange.Core.Common;
+using Lagrange.Core.Common.Interface;
+using Lagrange.Core.Common.Interface.Api;
+using Shrink.Logger;
+using Shrink.Utility;
+using Console = System.Console;
+
+namespace Shrink.Service;
+
+public class BotService
+{
+    private static readonly Lazy<BotService> _instance = new(() => new BotService());
+    public static BotService Instance => _instance.Value;
+
+    public BotContext? Client;
+    private bool _isOnline;
+    private static string KeystoreFilePath => "Keystore.json";
+    private static string DeviceInfoFilePath => "DeviceInfo.json";
+
+
+    private static BotDeviceInfo GetDeviceInfo() =>
+        JsonUtility.ReadOrCreateJsonFile(DeviceInfoFilePath, BotDeviceInfo.GenerateInfo);
+
+    // 登录方法
+    public async Task Login()
+    {
+        const int port = 50051;
+        var server = new Server
+        {
+            Services = { APIService.BindService(new ApiService()) },
+            Ports = { new ServerPort("localhost", port, ServerCredentials.Insecure) }
+        };
+
+        var deviceInfo = File.Exists(DeviceInfoFilePath)
+            ? JsonUtility.ReadJsonFromFile<BotDeviceInfo>(DeviceInfoFilePath)
+            : GetDeviceInfo();
+
+        var keyStore = File.Exists(KeystoreFilePath)
+            ? JsonUtility.ReadJsonFromFile<BotKeystore>(KeystoreFilePath)
+            : new BotKeystore();
+
+        Client = BotFactory.Create(new BotConfig
+        {
+            UseIPv6Network = false,
+            GetOptimumServer = true,
+            AutoReconnect = true,
+            Protocol = Protocols.Linux
+        }, deviceInfo, keyStore);
+
+        // Log模式
+        Client.Invoker.OnBotLogEvent += (_, @event) =>
+        {
+            @event.Level.ChangeColorByTitle();
+            Console.WriteLine(@event.ToString());
+        };
+
+        // bot信息保存
+        Client.Invoker.OnBotOnlineEvent += (_, @event) =>
+        {
+            Console.WriteLine(@event.ToString());
+            _isOnline = true;
+            server.Start();
+            Console.WriteLine($"[{DateTime.Now}] [gRPC Server] [Info]: Listening on port: {port}");
+            /*LogManager.Instance.LogChain.LogMessage($"{DateTime.Now}", "LoginEvent", LogLevel.Info, MsgType.Public,
+                620902312, "Login");*/
+        };
+        Client.Invoker.OnBotOfflineEvent += async (_, @event) =>
+        {
+            Console.WriteLine(@event.ToString());
+            await server.ShutdownAsync();
+            _isOnline = false;
+        };
+
+        if (File.Exists(KeystoreFilePath))
+        {
+            await Client.LoginByPassword();
+            if (!_isOnline)
+            {
+                Console.WriteLine("账密登录失败，请尝试二维码登录。");
+                var qrCode = await Client.FetchQrCode();
+                if (qrCode != null)
+                {
+                    await File.WriteAllBytesAsync("qr.png", qrCode.Value.QrCode);
+                    await Client.LoginByQrCode();
+                }
+            }
+        }
+
+        await File.WriteAllTextAsync(KeystoreFilePath, JsonSerializer.Serialize(Client.UpdateKeystore()));
+        await File.WriteAllTextAsync(DeviceInfoFilePath, JsonSerializer.Serialize(Client.UpdateDeviceInfo()));
+    }
+}