Shrink-IDC/Login/BotService.cs

using System.Text.Json;
using System.Text.Json.Serialization;
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.API;
using Shrink.Utility;
using Console = System.Console;

namespace Shrink.Login;

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() =>
        ReadOrCreateJsonFile(DeviceInfoFilePath, BotDeviceInfo.GenerateInfo);
    
    // 登录方法
    public async Task Login()
    {
        const int port = 50051;
        var server = new Server
        {
            Services = { APIService.BindService(new BotServiceImpl()) },
            Ports = { new ServerPort("localhost", port, ServerCredentials.Insecure) }
        };
        
        var deviceInfo = File.Exists(DeviceInfoFilePath)
            ? ReadJsonFromFile<BotDeviceInfo>(DeviceInfoFilePath)
            : GetDeviceInfo();

        var keyStore = File.Exists(KeystoreFilePath)
            ? 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($"gRPC server listening on port {port}");
        };
        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()));
    }

    private static T ReadJsonFromFile<T>(string filePath)
    {
        try
        {
            var json = File.ReadAllText(filePath);
            return JsonSerializer.Deserialize<T>(json,
                new JsonSerializerOptions { ReferenceHandler = ReferenceHandler.Preserve })!;
        }
        catch (Exception ex)
        {
            Console.WriteLine($"读取文件出错： {filePath}: {ex.Message}");
            throw;
        }
    }

    private static T ReadOrCreateJsonFile<T>(string filePath, Func<T> createFunc)
    {
        if (File.Exists(filePath))
        {
            return ReadJsonFromFile<T>(filePath);
        }

        var newData = createFunc();
        WriteJsonToFile(filePath, newData);
        return newData;
    }

    private static void WriteJsonToFile<T>(string filePath, T data)
    {
        try
        {
            var json = JsonSerializer.Serialize(data);
            File.WriteAllText(filePath, json);
        }
        catch (Exception ex)
        {
            Console.WriteLine($"写入文件出错： {filePath}: {ex.Message}");
            throw;
        }
    }
}