submodule对gitea不管用，所以直接拉了一份拉格兰

2025-02-04 16:29:43 +08:00
parent b0bfc803e3
commit d149a2ea0f
1023 changed files with 43308 additions and 18 deletions
--- a/Lagrange.Core/Lagrange.OneBot/Extensions/HostApplicationBuilderExtension.cs
+++ b/Lagrange.Core/Lagrange.OneBot/Extensions/HostApplicationBuilderExtension.cs
@@ -0,0 +1,168 @@
+using Lagrange.Core.Common;
+using Lagrange.Core.Common.Interface;
+using Lagrange.OneBot.Core.Login;
+using Lagrange.OneBot.Core.Network;
+using Lagrange.OneBot.Core.Network.Service;
+using Lagrange.OneBot.Core.Notify;
+using Lagrange.OneBot.Core.Operation;
+using Lagrange.OneBot.Database;
+using Lagrange.OneBot.Message;
+using Lagrange.OneBot.Utility;
+using LiteDB;
+using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Logging;
+
+using JsonSerializer = System.Text.Json.JsonSerializer;
+
+namespace Lagrange.OneBot.Extensions;
+
+public static class HostApplicationBuilderExtension
+{
+    public static HostApplicationBuilder ConfigureLagrangeCore(this HostApplicationBuilder builder)
+    {
+        builder.Services
+            .AddSingleton<OneBotSigner>() // Signer
+            .AddSingleton((services) => // BotConfig
+            {
+                var configuration = services.GetRequiredService<IConfiguration>();
+
+                return new BotConfig
+                {
+                    Protocol = configuration["Account:Protocol"] switch
+                    {
+                        "Windows" => Protocols.Windows,
+                        "MacOs" => Protocols.MacOs,
+                        _ => Protocols.Linux,
+                    },
+                    AutoReconnect = configuration.GetValue("Account:AutoReconnect", true),
+                    UseIPv6Network = configuration.GetValue("Account:UseIPv6Network", false),
+                    GetOptimumServer = configuration.GetValue("Account:GetOptimumServer", true),
+                    AutoReLogin = configuration.GetValue("Account:AutoReLogin", true),
+                    CustomSignProvider = services.GetRequiredService<OneBotSigner>()
+                };
+            })
+            .AddSingleton((services) => // Device
+            {
+                var configuration = services.GetRequiredService<IConfiguration>();
+                string path = configuration["ConfigPath:DeviceInfo"] ?? "device.json";
+
+                var device = File.Exists(path)
+                    ? JsonSerializer.Deserialize<BotDeviceInfo>(File.ReadAllText(path)) ?? BotDeviceInfo.GenerateInfo()
+                    : BotDeviceInfo.GenerateInfo();
+
+                string deviceJson = JsonSerializer.Serialize(device);
+                File.WriteAllText(path, deviceJson);
+
+                return device;
+            })
+            .AddSingleton((services) => // Keystore
+            {
+                var configuration = services.GetRequiredService<IConfiguration>();
+                string path = configuration["ConfigPath:Keystore"] ?? "keystore.json";
+
+                return File.Exists(path)
+                    ? JsonSerializer.Deserialize<BotKeystore>(File.ReadAllText(path)) ?? new()
+                    : new();
+            })
+            .AddSingleton((services) => services.GetRequiredService<OneBotSigner>().GetAppInfo()) // AppInfo
+            .AddSingleton((services) => BotFactory.Create( // BotContext
+                services.GetRequiredService<BotConfig>(),
+                services.GetRequiredService<BotDeviceInfo>(),
+                services.GetRequiredService<BotKeystore>(),
+                services.GetRequiredService<BotAppInfo>()
+            ))
+            .AddHostedService<LoginService>();
+
+        return builder;
+    }
+
+    public static HostApplicationBuilder ConfigureOneBot(this HostApplicationBuilder builder)
+    {
+        builder.Services.AddOptions()
+            .AddSingleton(services => // Database
+            {
+                var configuration = services.GetRequiredService<IConfiguration>();
+                var logger = services.GetRequiredService<ILogger<LiteDatabase>>();
+
+                BsonMapper.Global.TrimWhitespace = false;
+                BsonMapper.Global.EmptyStringToNull = false;
+
+                // Specify ctor for some classes
+                BsonMapper.Global.RegisterType(
+                    LiteDbUtility.IMessageEntitySerialize,
+                    LiteDbUtility.IMessageEntityDeserialize
+                );
+
+                string path = configuration["ConfigPath:Database"] ?? $"lagrange-{configuration["Account:Uin"]}.db";
+
+                bool isFirstCreate = false;
+                if (!File.Exists(path)) isFirstCreate = true;
+
+                var db = new LiteDatabase(path)
+                {
+                    CheckpointSize = 50
+                };
+
+                string[] expressions = ["$.Sequence", "$.MessageId", "$.FriendUin", "$.GroupUin"];
+
+                bool hasFirstIndex = false;
+                var indexes = db.GetCollection("$indexes");
+                foreach (var expression in expressions)
+                {
+                    if (indexes.Exists(Query.EQ("expression", expression))) continue;
+
+                    logger.LogWarning("In the database index");
+                    logger.LogWarning("Depending on the size of the database will consume some time and memory");
+                    logger.LogWarning("Not yet finished, please wait...");
+
+                    hasFirstIndex = true;
+                    break;
+                }
+
+                var records = db.GetCollection<MessageRecord>();
+                foreach (var expression in expressions)
+                {
+                    records.EnsureIndex(BsonExpression.Create(expression));
+                }
+
+                // Skipping the first database creation is a restart after indexing
+                if (!isFirstCreate && hasFirstIndex)
+                {
+                    db.Dispose(); // Ensure that the database is written correctly
+                    logger.LogInformation("Indexing Complete! Press any key to close and restart the program manually!");
+                    Console.ReadKey(true);
+                    Environment.Exit(0);
+                }
+                return db;
+            })
+
+            // // OneBot Netword Service
+            .AddSingleton<LagrangeWebSvcCollection>()
+
+            .AddScoped<ILagrangeWebServiceFactory<ForwardWSService>, ForwardWSServiceFactory>()
+            .AddScoped<ForwardWSService>()
+
+            .AddScoped<ILagrangeWebServiceFactory<ReverseWSService>, ReverseWSServiceFactory>()
+            .AddScoped<ReverseWSService>()
+
+            .AddScoped<ILagrangeWebServiceFactory<HttpService>, HttpServiceFactory>()
+            .AddScoped<HttpService>()
+
+            .AddScoped<ILagrangeWebServiceFactory<HttpPostService>, HttpPostServiceFactory>()
+            .AddScoped<HttpPostService>()
+
+            .AddScoped<ILagrangeWebServiceFactory, DefaultLagrangeWebServiceFactory>()
+            .AddScoped(services => services.GetRequiredService<ILagrangeWebServiceFactory>().Create()
+                ?? throw new Exception("Invalid conf detected"))
+
+            // // OneBot Misc Service
+            .AddSingleton<MusicSigner>()
+            .AddSingleton<NotifyService>()
+            .AddSingleton<MessageService>()
+            .AddSingleton<OperationService>();
+
+        return builder;
+    }
+}
--- a/Lagrange.Core/Lagrange.OneBot/Extensions/HostExtension.cs
+++ b/Lagrange.Core/Lagrange.OneBot/Extensions/HostExtension.cs
@@ -0,0 +1,15 @@
+using Lagrange.OneBot.Utility;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+
+namespace Lagrange.OneBot.Extensions;
+
+public static class HostExtension
+{
+    public static T InitializeMusicSigner<T>(this T host) where T : IHost
+    {
+        host.Services.GetRequiredService<MusicSigner>();
+
+        return host;
+    }
+}