LANCommander/LANCommander.Server/Program.cs

using LANCommander.Server.Data;
using LANCommander.Server.Data.Models;
using LANCommander.Server.Hubs;
using LANCommander.Server.Services;
using Microsoft.AspNetCore.Identity;
using Microsoft.EntityFrameworkCore;
using Microsoft.IdentityModel.Tokens;
using System.Text;
using Hangfire;
using LANCommander.Server.Services.MediaGrabbers;
using LANCommander.Server.Extensions;
using Microsoft.AspNetCore.Http.Features;
using LANCommander.SDK.Enums;
using Serilog;
using Serilog.Sinks.AspNetCore.App.SignalR.Extensions;
using LANCommander.Server.Logging;
using LANCommander.Server.Data.Enums;
using System.Diagnostics;
using LANCommander.Server.Services.Factories;
using Microsoft.AspNetCore.Components.Authorization;
using Microsoft.AspNetCore.Authentication;
using LANCommander.Server.Jobs.Background;
using LANCommander.Server.Services.Models;
using Microsoft.AspNetCore.HttpOverrides;
using Microsoft.Data.Sqlite;
using Microsoft.OpenApi.Models;
using Microsoft.CodeAnalysis.Options;
using Scalar.AspNetCore;
using Microsoft.IdentityModel.Protocols.OpenIdConnect;

namespace LANCommander.Server
{
    internal class Program
    {
        static async Task Main(string[] args)
        {
            Log.Logger = new LoggerConfiguration()
                .WriteTo.Console()
                .CreateBootstrapLogger();

            Log.Information("Starting application...");

            var builder = WebApplication.CreateBuilder(args);

            ConfigurationManager configuration = builder.Configuration;

            #region Debug
            if (args.Contains("--debugger"))
            {
                var currentProcess = Process.GetCurrentProcess();

                Console.WriteLine($"Waiting for debugger to attach... Process ID: {currentProcess.Id}");

                while (!Debugger.IsAttached)
                {
                    Thread.Sleep(100);
                }

                Console.WriteLine("Debugger attached.");
            }
            #endregion

            // Add services to the container.
            Log.Debug("Loading settings");
            var settings = SettingService.GetSettings(true);

            var databaseProviderParameter = args.FirstOrDefault(arg => arg.StartsWith("--database-provider="))?.Split('=', 2).Last();
            var connectionStringParameter = args.FirstOrDefault(arg => arg.StartsWith("--connection-string="))?.Split('=', 2).Last();

            if (!String.IsNullOrWhiteSpace(databaseProviderParameter))
                DatabaseContext.Provider = Enum.Parse<DatabaseProvider>(databaseProviderParameter);
            else
                DatabaseContext.Provider = settings.DatabaseProvider;

            if (!String.IsNullOrWhiteSpace(connectionStringParameter))
                DatabaseContext.ConnectionString = connectionStringParameter;
            else
                DatabaseContext.ConnectionString = settings.DatabaseConnectionString;

            Log.Debug("Loaded!");


            Log.Debug("Configuring logging");

            builder.Services.AddSignalR().AddJsonProtocol(options =>
            {
                options.PayloadSerializerOptions.PropertyNamingPolicy = null;
            });

            builder.Services.AddSerilogHub<LoggingHub>();
            builder.Services.AddSerilog((serviceProvider, config) => config
                .WriteTo.Console()
                .WriteTo.File(Path.Combine(settings.Logs.StoragePath, "log-.txt"), rollingInterval: (RollingInterval)(int)settings.Logs.ArchiveEvery)
#if DEBUG
                .WriteTo.Seq("http://localhost:5341")
                .MinimumLevel.Debug()
#endif
                .WriteTo.SignalR<LoggingHub>(
                    serviceProvider,
                    (context, message, logEvent) => LoggingHub.Log(context, message, logEvent)
                ));

            #region Validate Settings
            Log.Debug("Validating settings");
            if (settings?.Authentication?.TokenSecret?.Length < 16)
            {
                Log.Debug("JWT token secret is too short. Regenerating...");
                settings.Authentication.TokenSecret = Guid.NewGuid().ToString();
                SettingService.SaveSettings(settings);
            }
            Log.Debug("Done validating settings");
            #endregion

            Log.Debug("Configuring MVC and Blazor");
            builder.Services
                .AddMvc(options => options.EnableEndpointRouting = false)
                .AddRazorOptions(options =>
                {
                    options.ViewLocationFormats.Clear();
                    options.ViewLocationFormats.Add("/UI/Views/{1}/{0}.cshtml");
                    options.ViewLocationFormats.Add("/UI/Views/Shared/{0}.cshtml");
                    options.ViewLocationFormats.Add("/UI/Pages/Shared/{0}.cshtml");

                    options.AreaViewLocationFormats.Clear();
                    options.AreaViewLocationFormats.Add("/Areas/{2}/Views/{1}/{0}.cshtml");
                    options.AreaViewLocationFormats.Add("/Areas/{2}/Views/Shared/{0}.cshtml");
                    options.AreaViewLocationFormats.Add("/UI/Views/Shared/{0}.cshtml");
                    options.AreaViewLocationFormats.Add("/UI/Pages/Shared/{0}.cshtml");

                    options.PageViewLocationFormats.Clear();
                    options.PageViewLocationFormats.Add("/UI/Pages/{1}/{0}.cshtml");
                    options.PageViewLocationFormats.Add("/UI/Pages/Shared/{0}.cshtml");
                    options.PageViewLocationFormats.Add("/UI/Views/Shared/{0}.cshtml");

                    options.AreaPageViewLocationFormats.Clear();
                    options.AreaPageViewLocationFormats.Add("/Areas/{2}/Pages/{1}/{0}.cshtml");
                    options.AreaPageViewLocationFormats.Add("/Areas/{2}/Pages/Shared/{0}.cshtml");
                    options.AreaPageViewLocationFormats.Add("/Areas/{2}/Views/Shared/{0}.cshtml");
                    options.AreaPageViewLocationFormats.Add("/UI/Pages/Shared/{0}.cshtml");
                    options.AreaPageViewLocationFormats.Add("/UI/Views/Shared/{0}.cshtml");
                });

            builder.Services.AddRazorPages(options =>
            {
                options.RootDirectory = "/UI/Pages";
            });

            builder.Services.AddRazorComponents()
                .AddInteractiveServerComponents();

            builder.Services.AddCascadingAuthenticationState();

            builder.Services.AddAutoMapper(typeof(AutoMapper));

            Log.Debug("Starting web server on port {Port}", settings.Port);
            builder.WebHost.ConfigureKestrel(options =>
            {
                // Configure as HTTP only
                options.ListenAnyIP(settings.Port);
            });

            builder.Services.AddCors(options => options.AddPolicy("CorsPolicy", builder =>
            {
                builder.AllowAnyHeader()
                       .AllowAnyMethod()
                       .SetIsOriginAllowed((host) => true)
                       .AllowCredentials();
            }));

            Log.Debug("Initializing DatabaseContext with connection string {ConnectionString}", settings.DatabaseConnectionString);

            builder.Services.AddDbContextFactory<DatabaseContext>();
            builder.Services.AddDbContext<DatabaseContext>();

            builder.Services.AddDatabaseDeveloperPageExceptionFilter();

            Log.Debug("Initializing Identity");
            builder.Services.AddIdentityCore<User>((IdentityOptions options) =>
            {
                options.SignIn.RequireConfirmedAccount = false;
                options.SignIn.RequireConfirmedEmail = false;

                options.Password.RequireNonAlphanumeric = settings.Authentication.PasswordRequireNonAlphanumeric;
                options.Password.RequireLowercase = settings.Authentication.PasswordRequireLowercase;
                options.Password.RequireUppercase = settings.Authentication.PasswordRequireUppercase;
                options.Password.RequireDigit = settings.Authentication.PasswordRequireDigit;
                options.Password.RequiredLength = settings.Authentication.PasswordRequiredLength;
            })
                .AddRoles<Role>()
                .AddEntityFrameworkStores<DatabaseContext>()
                .AddSignInManager()
                .AddDefaultTokenProviders();

            var authBuilder = builder.Services.AddAuthentication(options =>
            {
                options.DefaultScheme = IdentityConstants.ApplicationScheme;
                options.DefaultSignInScheme = IdentityConstants.ExternalScheme;
                /*options.DefaultAuthenticateScheme = JwtBearerDefaults.AuthenticationScheme;
                options.DefaultChallengeScheme = JwtBearerDefaults.AuthenticationScheme;
                options.DefaultScheme = JwtBearerDefaults.AuthenticationScheme;*/
            });

            foreach (var authenticationProvider in settings.Authentication.AuthenticationProviders)
            {
                try
                {
                    switch (authenticationProvider.Type)
                    {
                        case AuthenticationProviderType.OAuth2:
                            authBuilder.AddOAuth(authenticationProvider);
                            break;

                        case AuthenticationProviderType.OpenIdConnect:
                            authBuilder.AddOpenIdConnect(authenticationProvider);
                            break;

                        case AuthenticationProviderType.Saml:
                            throw new NotImplementedException("SAML providers are not supported at this time.");
                            break;
                    }
                }
                catch (Exception ex)
                {
                    Log.Error(ex, "Authentication Provider {Name} could not be registered",
                        authenticationProvider.Name);
                }
            }

            authBuilder.AddIdentityCookies();

            builder.Services.ConfigureApplicationCookie(options =>
            {
                options.LoginPath = "/Login";
                options.LogoutPath = "/Logout";
                options.AccessDeniedPath = "/AccessDenied";
            });

            authBuilder.AddJwtBearer(options =>
            {
                options.SaveToken = true;
                options.RequireHttpsMetadata = false;
                options.TokenValidationParameters = new TokenValidationParameters()
                {
                    ValidateIssuer = false,
                    ValidateAudience = false,
                    // ValidAudience = configuration["JWT:ValidAudience"],
                    // ValidIssuer = configuration["JWT:ValidIssuer"],
                    IssuerSigningKey = new SymmetricSecurityKey(Encoding.UTF8.GetBytes(settings.Authentication.TokenSecret))
                };
            });

            Log.Debug("Initializing Controllers");
            builder.Services.AddControllers().AddJsonOptions(x =>
            {
                x.JsonSerializerOptions.ReferenceHandler = System.Text.Json.Serialization.ReferenceHandler.IgnoreCycles;
            });

            Log.Debug("Initializing Hangfire");
            builder.Services.AddHangfire(configuration =>
                configuration
                    .SetDataCompatibilityLevel(CompatibilityLevel.Version_170)
                    .UseSimpleAssemblyNameTypeSerializer()
                    .UseRecommendedSerializerSettings()
                    .UseInMemoryStorage());
            builder.Services.AddHangfireServer();

            builder.Services.AddFusionCache();

            Log.Debug("Registering Swashbuckle");
            builder.Services.AddEndpointsApiExplorer();
            builder.Services.AddSwaggerGen(options =>
            {
                options.CustomSchemaIds(type => type.ToString());
                options.AddSecurityDefinition("Bearer", new OpenApiSecurityScheme
                {
                    In = ParameterLocation.Header,
                    Description = "Enter a valid access token",
                    Name = "Authorization",
                    Type = SecuritySchemeType.Http,
                    BearerFormat = "JWT",
                    Scheme = "Bearer"
                });

                options.AddSecurityRequirement(new OpenApiSecurityRequirement
                {
                    {
                        new OpenApiSecurityScheme
                        {
                            Reference = new OpenApiReference
                            {
                                Type = ReferenceType.SecurityScheme,
                                Id = "Bearer"
                            }
                        },
                        new string[]{}
                    }
                });
            });

            Log.Debug("Registering AntDesign Blazor");
            builder.Services.AddAntDesign();
            
            builder.Services.AddHttpClient();

            Log.Debug("Registering Services");
            builder.Services.AddSingleton<SDK.Client>(new SDK.Client("", ""));
            builder.Services.AddSingleton<RepositoryFactory>();
            builder.Services.AddScoped(typeof(Repository<>));
            builder.Services.AddScoped<DatabaseServiceFactory>();
            builder.Services.AddScoped<IdentityContextFactory>();
            builder.Services.AddScoped<SettingService>();
            builder.Services.AddScoped<ArchiveService>();
            builder.Services.AddScoped<StorageLocationService>();
            builder.Services.AddScoped<CategoryService>();
            builder.Services.AddScoped<CollectionService>();
            builder.Services.AddScoped<GameService>();
            builder.Services.AddScoped<LibraryService>();
            builder.Services.AddScoped<ScriptService>();
            builder.Services.AddScoped<GenreService>();
            builder.Services.AddScoped<PlatformService>();
            builder.Services.AddScoped<KeyService>();
            builder.Services.AddScoped<TagService>();
            builder.Services.AddScoped<EngineService>();
            builder.Services.AddScoped<CompanyService>();
            builder.Services.AddScoped<IGDBService>();
            builder.Services.AddScoped<ServerService>();
            builder.Services.AddScoped<ServerConsoleService>();
            builder.Services.AddScoped<GameSaveService>();
            builder.Services.AddScoped<PlaySessionService>();
            builder.Services.AddScoped<MediaService>();
            builder.Services.AddScoped<RedistributableService>();
            builder.Services.AddScoped<IMediaGrabberService, SteamGridDBMediaGrabber>();
            builder.Services.AddScoped<UpdateService>();
            builder.Services.AddScoped<IssueService>();
            builder.Services.AddScoped<PageService>();
            builder.Services.AddScoped<UserService>();
            builder.Services.AddScoped<UserCustomFieldService>();
            builder.Services.AddScoped<RoleService>();
            builder.Services.AddScoped<Services.AuthenticationService>();
            builder.Services.AddScoped<SetupService>();

            builder.Services.AddSingleton<ServerProcessService>();
            builder.Services.AddSingleton<IPXRelayService>();

            if (settings.Beacon?.Enabled ?? false)
            {
                Log.Debug("The beacons have been lit! LANCommander calls for players!");
                builder.Services.AddHostedService<BeaconService>();
            }

            builder.WebHost.UseStaticWebAssets();

            builder.WebHost.UseKestrel(options =>
            {
                options.Limits.MaxRequestBodySize = long.MaxValue;
                options.Limits.RequestHeadersTimeout = TimeSpan.FromMinutes(5);
            });

            builder.Services.Configure<FormOptions>(options =>
            {
                options.MultipartBodyLengthLimit = long.MaxValue;
            });

            Log.Debug("Building Application");
            var app = builder.Build();

            app.UseCors("CorsPolicy");
            app.UseHttpsRedirection();

            app.MapHub<GameServerHub>("/hubs/gameserver");
            
            app.UseForwardedHeaders(new ForwardedHeadersOptions
            {
                ForwardedHeaders = ForwardedHeaders.XForwardedFor | ForwardedHeaders.XForwardedProto
            });

            app.Use(async (context, next) =>
            {
                if (context.Request.Path.StartsWithSegments("/robots.txt"))
                {
                    context.Response.ContentType = "text/plain";

                    await context.Response.WriteAsync("User-agent: *\nDisallow: /Identity/");
                }
                else await next();
            });

            app.Use((context, next) =>
            {
                var headers = context.Response.Headers;

                headers.Append("X-API-Version", UpdateService.GetCurrentVersion().ToString());

                return next();
            });

            // Configure the HTTP request pipeline.
            if (app.Environment.IsDevelopment())
            {
                Log.Debug("App has been run in a development environment");
                app.UseMigrationsEndPoint();
            }
            else
            {
                app.UseExceptionHandler("/Home/Error");
                // The default HSTS value is 30 days. You may want to change this for production scenarios, see https://aka.ms/aspnetcore-hsts.
                app.UseHsts();
            }

            app.UseSwagger(options =>
            {
                options.RouteTemplate = "/openapi/{documentName}.json";
            });
            app.MapScalarApiReference();
            app.UseHangfireDashboard();

            // app.UseHttpsRedirection();

            app.UseRouting();

            app.UseAuthentication();
            app.UseAuthorization();

            app.UseMvcWithDefaultRoute();

            Log.Debug("Registering Endpoints");

            app.MapHub<LoggingHub>("/logging");

            app.UseAntiforgery();
            app.UseStaticFiles();
            
            app.UseForwardedHeaders(new ForwardedHeadersOptions
            {
                ForwardedHeaders = ForwardedHeaders.XForwardedProto
            });

            app.MapRazorComponents<UI.App>()
                .AddInteractiveServerRenderMode();

            app.UseEndpoints(endpoints =>
            {
                endpoints.MapFallbackToPage("/_Host");
                endpoints.MapControllers();
            });

            Log.Debug("Ensuring required directories exist");

            if (!Directory.Exists(settings.UserSaves.StoragePath))
                Directory.CreateDirectory(settings.UserSaves.StoragePath);

            if (!Directory.Exists(settings.Update.StoragePath))
                Directory.CreateDirectory(settings.Update.StoragePath);

            if (!Directory.Exists("Snippets"))
                Directory.CreateDirectory("Snippets");

            if (!Directory.Exists("Backups"))
                Directory.CreateDirectory("Backups");

            // Migrate
            Log.Debug("Migrating database if required");

            if (DatabaseContext.Provider != DatabaseProvider.Unknown)
            {
                await using var scope = app.Services.CreateAsyncScope();
                using var db = scope.ServiceProvider.GetService<DatabaseContext>();

                if ((await db.Database.GetPendingMigrationsAsync()).Any())
                {
                    if (DatabaseContext.Provider == DatabaseProvider.SQLite)
                    {
                        var dataSource = new SqliteConnectionStringBuilder(settings.DatabaseConnectionString).DataSource;

                        var backupName = Path.Combine("Backups", $"LANCommander.db.{DateTime.Now.ToString("dd-MM-yyyy-HH.mm.ss.bak")}");

                        if (File.Exists(dataSource))
                        {
                            Log.Information("Migrations pending, database will be backed up to {BackupName}", backupName);
                            File.Copy(dataSource, backupName);
                        }
                    }

                    await db.Database.MigrateAsync();
                }
                else
                    Log.Debug("No pending migrations are available. Skipping database migration.");

                // Autostart any server processes
                Log.Debug("Autostarting Servers");
                var serverService = scope.ServiceProvider.GetService<ServerService>();
                var serverProcessService = scope.ServiceProvider.GetService<ServerProcessService>();

                foreach (var server in await serverService.GetAsync(s => s.Autostart && s.AutostartMethod == ServerAutostartMethod.OnApplicationStart))
                {
                    try
                    {
                        Log.Debug("Autostarting server {ServerName} with a delay of {AutostartDelay} seconds", server.Name, server.AutostartDelay);

                        if (server.AutostartDelay > 0)
                            await Task.Delay(server.AutostartDelay);

                        serverProcessService.StartServerAsync(server.Id);
                    }
                    catch (Exception ex)
                    {
                        Log.Debug(ex, "An unexpected error occurred while trying to autostart the server {ServerName}", server.Name);
                    }
                }

                await db.DisposeAsync();
                await scope.DisposeAsync();

                BackgroundJob.Enqueue<GenerateThumbnailsJob>(x => x.ExecuteAsync());
            }
            else
                Log.Debug("No database provider has been setup, application is fresh and needs first time setup");

            app.Run();
        }
    }
}