BililiveRecorder/BililiveRecorder.Web/Startup.cs

using System;
using System.Diagnostics;
using System.IO;
using System.Threading.Tasks;
using AutoMapper;
using BililiveRecorder.Core;
using BililiveRecorder.Web.Graphql;
using BililiveRecorder.Web.Models.Rest;
using GraphQL;
using GraphQL.Execution;
using GraphQL.Server;
using GraphQL.SystemReactive;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Server.Kestrel.Core;
using Microsoft.AspNetCore.StaticFiles;
using Microsoft.AspNetCore.StaticFiles.Infrastructure;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.FileProviders;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using Microsoft.OpenApi.Models;

namespace BililiveRecorder.Web
{
    public class Startup
    {
        // TODO 减少引用的依赖数量

        public Startup(IConfiguration configuration, IWebHostEnvironment environment)
        {
            this.Configuration = configuration;
            this.Environment = environment;
        }

        public IConfiguration Configuration { get; }

        public IWebHostEnvironment Environment { get; }

        // This method gets called by the runtime. Use this method to add services to the container.
        // For more information on how to configure your application, visit https://go.microsoft.com/fwlink/?LinkID=398940
        public void ConfigureServices(IServiceCollection services)
        {
            // GraphQL API 使用 Newtonsoft.Json 会同步调用 System.IO.StreamReader.Peek
            // 来源是 GraphQL.Server.Transports.AspNetCore.NewtonsoftJson.GraphQLRequestDeserializer.DeserializeFromJsonBodyAsync
            // System.Text.Json 不使用构造函数传属性，所以无法创建 Optional<T> （需要加 attribute，回头改）
            // TODO 改 Optional<T>
            services.Configure<KestrelServerOptions>(o => o.AllowSynchronousIO = true);

            // 如果 IRecorder 没有被注册过才会添加，模拟调试用
            // 实际运行时在 BililiveRecorder.Web.Program 里会加上真的 IRecorder
            services.TryAddSingleton<IRecorder>(new FakeRecorderForWeb());

#if DEBUG
            // TODO 移动到一个单独的测试项目里
            if (Debugger.IsAttached)
            {
                var configuration = new MapperConfiguration(cfg => cfg.AddProfile<DataMappingProfile>());
                configuration.AssertConfigurationIsValid();
            }
#endif

            services
                .AddAutoMapper(c => c.AddProfile<DataMappingProfile>())
                .AddCors(o => o.AddDefaultPolicy(p => p.AllowAnyOrigin().AllowAnyMethod().AllowAnyHeader()))
                ;

            services.AddSingleton(new ManifestEmbeddedFileProvider(typeof(Startup).Assembly));

            // Graphql API
            GraphQL.MicrosoftDI.GraphQLBuilderExtensions.AddGraphQL(services)
                .AddServer(true)
                .AddWebSockets()
                .AddNewtonsoftJson()
                .AddSchema<RecorderSchema>()
                .AddSubscriptionDocumentExecuter()
                .AddDefaultEndpointSelectorPolicy()
                .AddGraphTypes(typeof(RecorderSchema).Assembly)
                .Configure<ErrorInfoProviderOptions>(opt => opt.ExposeExceptionStackTrace = this.Environment.IsDevelopment() || Debugger.IsAttached)
                .ConfigureExecution(options =>
                {
                    options.EnableMetrics = this.Environment.IsDevelopment() || Debugger.IsAttached;
                    var logger = options.RequestServices?.GetRequiredService<ILogger<Startup>>();
                    options.UnhandledExceptionDelegate = ctx => logger?.LogError(ctx.OriginalException, "Graphql Error");
                })
                ;

            // REST API
            services
                .AddSwaggerGen(c =>
                {
                    c.SwaggerDoc("brec", new OpenApiInfo
                    {
                        Title = "录播姬 REST API",
                        Description = "录播姬网站 [rec.danmuji.org](https://rec.danmuji.org/)  \n录播姬 GitHub [BililiveRecorder/BililiveRecorder](https://github.com/BililiveRecorder/BililiveRecorder)  \n\n" +
                        "除了 REST API 以外，录播姬还有 Graphql API 可以使用。\n\n" +
                        "API 中的 objectId 在重启后会重新生成，不保存到配置文件。",
                        Version = "v1"
                    });

                    var filePath = Path.Combine(AppContext.BaseDirectory, "BililiveRecorder.Web.xml");
                    c.IncludeXmlComments(filePath);
                })
                .AddRouting(c =>
                {
                    c.LowercaseUrls = true;
                    c.LowercaseQueryStrings = true;
                })
                .AddMvcCore(option =>
                {

                })
                .AddApiExplorer()
                .AddNewtonsoftJson();
        }

        // This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
        public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
        {
            const string PAGE404 = "/404.html";

            app.UseMiddleware<BasicAuthMiddleware>();

            app.UseCors().UseWebSockets();

            app.Use(static next => async context =>
            {
                // 404 页面处理
                await next(context);

                if (!context.Response.HasStarted
                && !context.Response.ContentLength.HasValue
                && string.IsNullOrEmpty(context.Response.ContentType)
                && context.Response.StatusCode == StatusCodes.Status404NotFound)
                {
                    var originalPath = context.Request.Path;
                    var originalQueryString = context.Request.QueryString;
                    try
                    {
                        if (originalPath.StartsWithSegments("/ui"))
                        {
                            context.Request.Path = "/ui/";
                        }
                        else
                        {
                            context.Request.Path = PAGE404;
                        }

                        context.Request.QueryString = new QueryString();
                        await next(context);
                    }
                    finally
                    {
                        context.Request.Path = originalPath;
                        context.Request.QueryString = originalQueryString;
                    }
                }
            });

            app.UseRouting();

            static void OnPrepareResponse(StaticFileResponseContext context)
            {
                if (context.Context.Request.Path == PAGE404)
                {
                    context.Context.Response.StatusCode = StatusCodes.Status404NotFound;
                    context.Context.Response.Headers.CacheControl = "no-cache";
                }
                else
                {
                    context.Context.Response.Headers.CacheControl = "max-age=86400, must-revalidate";
                }
            }

            var ctp = new FileExtensionContentTypeProvider();
            ctp.Mappings[".htm"] = "text/html; charset=utf-8";
            ctp.Mappings[".html"] = "text/html; charset=utf-8";
            ctp.Mappings[".css"] = "text/css; charset=utf-8";
            ctp.Mappings[".js"] = "text/javascript; charset=utf-8";
            ctp.Mappings[".mjs"] = "text/javascript; charset=utf-8";
            ctp.Mappings[".json"] = "application/json; charset=utf-8";

            var manifestEmbeddedFileProvider = app.ApplicationServices.GetRequiredService<ManifestEmbeddedFileProvider>();
            var sharedStaticFiles = new SharedOptions()
            {
                // 在运行的 exe 旁边新建一个 wwwroot 文件夹，会优先使用里面的内容，然后 fallback 到打包的资源文件
                FileProvider = new CompositeFileProvider(env.WebRootFileProvider, manifestEmbeddedFileProvider),
                RedirectToAppendTrailingSlash = true,
            };

            app
                .UseDefaultFiles(new DefaultFilesOptions(sharedStaticFiles))
                .UseStaticFiles(new StaticFileOptions(sharedStaticFiles)
                {
                    ContentTypeProvider = ctp,
                    OnPrepareResponse = OnPrepareResponse
                })
                //.UseDirectoryBrowser(new DirectoryBrowserOptions(shared))
                ;

            try
            {
                var sharedRecordingFiles = new SharedOptions
                {
                    FileProvider = new PhysicalFileProvider(app.ApplicationServices.GetRequiredService<IRecorder>().Config.Global.WorkDirectory),
                    RequestPath = "/file",
                    RedirectToAppendTrailingSlash = true,
                };
                app.UseStaticFiles(new StaticFileOptions(sharedRecordingFiles)).UseDirectoryBrowser(new DirectoryBrowserOptions(sharedRecordingFiles));
            }
            catch (Exception) { }

            app
                .UseEndpoints(endpoints =>
                {
                    endpoints.MapControllers();

                    endpoints.MapGraphQL<RecorderSchema>();
                    endpoints.MapGraphQLWebSockets<RecorderSchema>();

                    endpoints.MapSwagger();
                    endpoints.MapGraphQLPlayground("graphql/playground");
                    endpoints.MapGraphQLGraphiQL("graphql/graphiql");
                    endpoints.MapGraphQLAltair("graphql/altair");
                    endpoints.MapGraphQLVoyager("graphql/voyager");
                })
                .UseSwaggerUI(c =>
                {
                    c.SwaggerEndpoint("brec/swagger.json", "录播姬 REST API");
                })
                .Use(static next => static context =>
                {
                    context.Response.StatusCode = StatusCodes.Status404NotFound;
                    return Task.CompletedTask;
                });
        }
    }
}