Program.cs

using DD_Screen.Attributes;
using DD_Screen.DDBOT;
using DD_Screen.Middleware;
using DD_Screen.Timer;
using Microsoft.AspNetCore.HttpOverrides;
using Microsoft.Extensions.FileProviders;
using Microsoft.OpenApi;
using Serilog;
using Serilog.Events;
using System.Net;
using System.Reflection;
using System.Runtime.InteropServices;

namespace DD_Screen
{
    public class Program
    {
        public static void Main(string[] args)
        {
            var baseDir = AppContext.BaseDirectory;
            Directory.SetCurrentDirectory(baseDir);

            var logDir = Path.Combine(baseDir, "logs");
            var screenshotDir = Path.Combine(baseDir, "ScreenShotImg");
            var chromeRootDir = Path.Combine(baseDir, "Chrome");

            Directory.CreateDirectory(logDir);
            Directory.CreateDirectory(screenshotDir);

            Log.Logger = new LoggerConfiguration()
                .MinimumLevel.Information()
                .MinimumLevel.Override("Microsoft", LogEventLevel.Information)
                .MinimumLevel.Override("Microsoft.AspNetCore", LogEventLevel.Warning)
                .MinimumLevel.Override("System", LogEventLevel.Warning)
                .Enrich.FromLogContext()
                .WriteTo.Console(
                    outputTemplate: "[{Timestamp:HH:mm:ss} {Level:u3}] {SourceContext} - {Message:lj}{NewLine}{Exception}"
                )
                .WriteTo.File(
                    path: Path.Combine(logDir, "dd-.log"),
                    rollingInterval: RollingInterval.Day,
                    retainedFileCountLimit: 30,
                    outputTemplate: "[{Timestamp:yyyy-MM-dd HH:mm:ss.fff}] [{Level:u3}] {SourceContext} - {Message:lj}{NewLine}{Exception}"
                )
                .CreateLogger();

            try
            {
                Log.Information("DD_Screen 启动中");
                Log.Information("BaseDirectory: {BaseDirectory}", baseDir);
                Log.Information("CurrentDirectory: {CurrentDirectory}", Directory.GetCurrentDirectory());

                TimerManager.StartJob();

                var options = new WebApplicationOptions
                {
                    Args = args,
                    ContentRootPath = baseDir
                };

                var builder = WebApplication.CreateBuilder(options);

                builder.Host.UseSerilog();

                Config.ChromePath = builder.Configuration["ChromePath"];
                Config.Headless = Convert.ToBoolean(builder.Configuration["Headless"]);
                Config.Debug = builder.Configuration.GetValue("Debug", false);

                if (string.IsNullOrWhiteSpace(Config.ChromePath))
                {
                    if (Directory.Exists(chromeRootDir))
                    {
                        string chromeExecutable;
                        if (RuntimeInformation.IsOSPlatform(OSPlatform.Windows))
                        {
                            chromeExecutable = "chrome.exe";
                        }
                        else if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
                        {
                            chromeExecutable = "chrome";
                        }
                        else if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
                        {
                            chromeExecutable = Path.Combine("Google Chrome.app", "Contents", "MacOS", "Google Chrome");
                        }
                        else
                        {
                            throw new Exception("不支持的操作系统!");
                        }

                        Config.ChromePath = Path.Combine(chromeRootDir, chromeExecutable);
                        Log.Information("Chrome 路径: {ChromePath}", Config.ChromePath);
                    }
                    else
                    {
                        throw new Exception($"ChromePath 未配置, 且程序目录下未找到 Chrome 文件夹: {chromeRootDir}");
                    }
                }

                builder.Services.Configure<ForwardedHeadersOptions>(fho =>
                {
                    fho.ForwardedHeaders = ForwardedHeaders.XForwardedFor
                                         | ForwardedHeaders.XForwardedProto
                                         | ForwardedHeaders.XForwardedHost;
                    fho.ForwardLimit = null;
                    fho.KnownNetworks.Clear();
                    fho.KnownProxies.Clear();
                });

                builder.Services.AddControllers();
                builder.Services.AddEndpointsApiExplorer();
                builder.Services.AddHttpContextAccessor();
                builder.Services.AddSwaggerGen(c =>
                {
                    c.OperationFilter<DisableableOperationFilter>();
                    c.DocumentFilter<DisableableDocumentFilter>();
                    c.SwaggerDoc("v1", new OpenApiInfo
                    {
                        Title = "DDBOT Rendering Service",
                        Version = "v1",
                        Description =
                            "一枝独秀不是春,万紫千红春满园。\n\n" +
                            "**搭配 DDBOT 使用**\n\n" +
                            "DDBOT 交流群:**980848391**"
                    });
                    var xmlFile = $"{Assembly.GetExecutingAssembly().GetName().Name}.xml";
                    var xmlPath = Path.Combine(baseDir, xmlFile);
                    c.IncludeXmlComments(xmlPath, false);
                });

                var app = builder.Build();

                app.UseForwardedHeaders();

                app.UseSwagger();
                app.UseSwaggerUI(u =>
                {
                    u.SwaggerEndpoint("/swagger/v1/swagger.json", "DD_Screen");
                });

                app.Lifetime.ApplicationStopping.Register(() =>
                {
                    Log.Information("正在关闭浏览器...");
                    try
                    {
                        PupperManager.DisposeAsync().GetAwaiter().GetResult();
                        Log.Information("浏览器已关闭");
                    }
                    catch (Exception ex)
                    {
                        Log.Error(ex, "关闭浏览器失败");
                    }
                });

                app.UseRequestLogging();

                app.Use(async (context, next) =>
                {
                    await next();
                    if (context.Response.StatusCode == 404)
                    {
                        context.Response.Redirect("/swagger/index.html");
                    }
                });

                app.UseStaticFiles(new StaticFileOptions
                {
                    FileProvider = new PhysicalFileProvider(screenshotDir),
                    RequestPath = "/ScreenShotImg"
                });

                app.UseRouting();
                app.UseAuthorization();
                app.UseDisableable();
                app.MapControllers();

                Log.Information("ScreenShotImg 目录: {ScreenShotDir}", screenshotDir);
                Log.Information("DD_Screen 启动成功");

                app.Run();
            }
            catch (Exception ex)
            {
                Log.Fatal(ex, "DD_Screen 启动失败");
                throw;
            }
            finally
            {
                Log.Information("DD_Screen 关闭中");
                Log.CloseAndFlush();
            }
        }
    }
}