< Summary - Jellyfin

Information
Class: Jellyfin.Server.Program
Assembly: jellyfin
File(s): /srv/git/jellyfin/Jellyfin.Server/Program.cs
Line coverage
86%
Covered lines: 19
Uncovered lines: 3
Coverable lines: 22
Total lines: 351
Line coverage: 86.3%
Branch coverage
50%
Covered branches: 2
Total branches: 4
Branch coverage: 50%
Method coverage

Feature is only available for sponsors

Upgrade to PRO version

Coverage history

Coverage history 0 25 50 75 100

Metrics

MethodBranch coverage Crap Score Cyclomatic complexity Line coverage
.cctor()100%11100%
Main(...)100%210%
CreateAppConfiguration(...)100%11100%
ConfigureAppConfiguration(...)50%4490%
PrepareDatabaseProvider(...)100%11100%

File(s)

/srv/git/jellyfin/Jellyfin.Server/Program.cs

#LineLine coverage
 1using System;
 2using System.Collections.Generic;
 3using System.Diagnostics;
 4using System.Globalization;
 5using System.IO;
 6using System.Linq;
 7using System.Reflection;
 8using System.Threading;
 9using System.Threading.Tasks;
 10using CommandLine;
 11using Emby.Server.Implementations;
 12using Emby.Server.Implementations.Configuration;
 13using Emby.Server.Implementations.Serialization;
 14using Jellyfin.Database.Implementations;
 15using Jellyfin.Server.Extensions;
 16using Jellyfin.Server.Helpers;
 17using Jellyfin.Server.Implementations.DatabaseConfiguration;
 18using Jellyfin.Server.Implementations.Extensions;
 19using Jellyfin.Server.Implementations.StorageHelpers;
 20using Jellyfin.Server.Implementations.SystemBackupService;
 21using Jellyfin.Server.Migrations;
 22using Jellyfin.Server.Migrations.Stages;
 23using Jellyfin.Server.ServerSetupApp;
 24using MediaBrowser.Common.Configuration;
 25using MediaBrowser.Common.Net;
 26using MediaBrowser.Controller;
 27using Microsoft.AspNetCore.Hosting;
 28using Microsoft.EntityFrameworkCore;
 29using Microsoft.Extensions.Configuration;
 30using Microsoft.Extensions.DependencyInjection;
 31using Microsoft.Extensions.Hosting;
 32using Microsoft.Extensions.Logging;
 33using Microsoft.Extensions.Logging.Abstractions;
 34using Serilog;
 35using Serilog.Extensions.Logging;
 36using static MediaBrowser.Controller.Extensions.ConfigurationExtensions;
 37using ILogger = Microsoft.Extensions.Logging.ILogger;
 38
 39namespace Jellyfin.Server
 40{
 41    /// <summary>
 42    /// Class containing the entry point of the application.
 43    /// </summary>
 44    public static class Program
 45    {
 46        /// <summary>
 47        /// The name of logging configuration file containing application defaults.
 48        /// </summary>
 49        public const string LoggingConfigFileDefault = "logging.default.json";
 50
 51        /// <summary>
 52        /// The name of the logging configuration file containing the system-specific override settings.
 53        /// </summary>
 54        public const string LoggingConfigFileSystem = "logging.json";
 55
 156        private static readonly SerilogLoggerFactory _loggerFactory = new SerilogLoggerFactory();
 57        private static SetupServer? _setupServer;
 58        private static CoreAppHost? _appHost;
 159        private static IHost? _jellyfinHost = null;
 60        private static long _startTimestamp;
 161        private static ILogger _logger = NullLogger.Instance;
 62        private static bool _restartOnShutdown;
 63        private static IStartupLogger<JellyfinMigrationService>? _migrationLogger;
 64        private static string? _restoreFromBackup;
 65
 66        /// <summary>
 67        /// The entry point of the application.
 68        /// </summary>
 69        /// <param name="args">The command line arguments passed.</param>
 70        /// <returns><see cref="Task" />.</returns>
 71        public static Task Main(string[] args)
 72        {
 73            static Task ErrorParsingArguments(IEnumerable<Error> errors)
 74            {
 75                Environment.ExitCode = 1;
 76                return Task.CompletedTask;
 77            }
 78
 79            // Parse the command line arguments and either start the app or exit indicating error
 080            return Parser.Default.ParseArguments<StartupOptions>(args)
 081                .MapResult(StartApp, ErrorParsingArguments);
 82        }
 83
 84        private static async Task StartApp(StartupOptions options)
 85        {
 86            _restoreFromBackup = options.RestoreArchive;
 87            _startTimestamp = Stopwatch.GetTimestamp();
 88            ServerApplicationPaths appPaths = StartupHelpers.CreateApplicationPaths(options);
 89            appPaths.MakeSanityCheckOrThrow();
 90
 91            // $JELLYFIN_LOG_DIR needs to be set for the logger configuration manager
 92            Environment.SetEnvironmentVariable("JELLYFIN_LOG_DIR", appPaths.LogDirectoryPath);
 93
 94            // Enable cl-va P010 interop for tonemapping on Intel VAAPI
 95            Environment.SetEnvironmentVariable("NEOReadDebugKeys", "1");
 96            Environment.SetEnvironmentVariable("EnableExtendedVaFormats", "1");
 97
 98            await StartupHelpers.InitLoggingConfigFile(appPaths).ConfigureAwait(false);
 99
 100            // Create an instance of the application configuration to use for application startup
 101            IConfiguration startupConfig = CreateAppConfiguration(options, appPaths);
 102            StartupHelpers.InitializeLoggingFramework(startupConfig, appPaths);
 103            _setupServer = new SetupServer(static () => _jellyfinHost?.Services?.GetService<INetworkManager>(), appPaths
 104            await _setupServer.RunAsync().ConfigureAwait(false);
 105            _logger = _loggerFactory.CreateLogger("Main");
 106            StartupLogger.Logger = new StartupLogger(_logger);
 107
 108            // Use the logging framework for uncaught exceptions instead of std error
 109            AppDomain.CurrentDomain.UnhandledException += (_, e)
 110                => _logger.LogCritical((Exception)e.ExceptionObject, "Unhandled Exception");
 111
 112            _logger.LogInformation(
 113                "Jellyfin version: {Version}",
 114                Assembly.GetEntryAssembly()!.GetName().Version!.ToString(3));
 115
 116            StartupHelpers.LogEnvironmentInfo(_logger, appPaths);
 117
 118            // If hosting the web client, validate the client content path
 119            if (startupConfig.HostWebClient())
 120            {
 121                var webContentPath = appPaths.WebPath;
 122                if (!Directory.Exists(webContentPath) || !Directory.EnumerateFiles(webContentPath).Any())
 123                {
 124                    _logger.LogError(
 125                        "The server is expected to host the web client, but the provided content directory is either " +
 126                        "invalid or empty: {WebContentPath}. If you do not want to host the web client with the " +
 127                        "server, you may set the '--nowebclient' command line flag, or set" +
 128                        "'{ConfigKey}=false' in your config settings",
 129                        webContentPath,
 130                        HostWebClientKey);
 131                    Environment.ExitCode = 1;
 132                    return;
 133                }
 134            }
 135
 136            StorageHelper.TestCommonPathsForStorageCapacity(appPaths, StartupLogger.Logger.With(_loggerFactory.CreateLog
 137
 138            StartupHelpers.PerformStaticInitialization();
 139
 140            await ApplyStartupMigrationAsync(appPaths, startupConfig).ConfigureAwait(false);
 141
 142            do
 143            {
 144                await StartServer(appPaths, options, startupConfig).ConfigureAwait(false);
 145
 146                if (_restartOnShutdown)
 147                {
 148                    _startTimestamp = Stopwatch.GetTimestamp();
 149                    await _setupServer.StopAsync().ConfigureAwait(false);
 150                    await _setupServer.RunAsync().ConfigureAwait(false);
 151                }
 152            } while (_restartOnShutdown);
 153
 154            _setupServer.Dispose();
 155        }
 156
 157        private static async Task StartServer(IServerApplicationPaths appPaths, StartupOptions options, IConfiguration s
 158        {
 159            using CoreAppHost appHost = new CoreAppHost(
 160                            appPaths,
 161                            _loggerFactory,
 162                            options,
 163                            startupConfig);
 164            _appHost = appHost;
 165            var configurationCompleted = false;
 166            try
 167            {
 168                _jellyfinHost = Host.CreateDefaultBuilder()
 169                    .UseConsoleLifetime()
 170                    .ConfigureServices(services => appHost.Init(services))
 171                    .ConfigureWebHostDefaults(webHostBuilder =>
 172                    {
 173                        webHostBuilder.ConfigureWebHostBuilder(appHost, startupConfig, appPaths, _logger);
 174                        if (bool.TryParse(Environment.GetEnvironmentVariable("JELLYFIN_ENABLE_IIS"), out var iisEnabled)
 175                        {
 176                            _logger.LogCritical("UNSUPPORTED HOSTING ENVIRONMENT Microsoft Internet Information Services
 177                            webHostBuilder.UseIIS();
 178                        }
 179                    })
 180                    .ConfigureAppConfiguration(config => config.ConfigureAppConfiguration(options, appPaths, startupConf
 181                    .UseSerilog()
 182                    .ConfigureServices(e => e
 183                        .RegisterStartupLogger()
 184                        .AddSingleton<IServiceCollection>(e))
 185                    .Build();
 186
 187                // Re-use the host service provider in the app host since ASP.NET doesn't allow a custom service collect
 188                appHost.ServiceProvider = _jellyfinHost.Services;
 189                PrepareDatabaseProvider(appHost.ServiceProvider);
 190
 191                if (!string.IsNullOrWhiteSpace(_restoreFromBackup))
 192                {
 193                    await appHost.ServiceProvider.GetService<IBackupService>()!.RestoreBackupAsync(_restoreFromBackup).C
 194                    _restoreFromBackup = null;
 195                    _restartOnShutdown = true;
 196                    return;
 197                }
 198
 199                var jellyfinMigrationService = ActivatorUtilities.CreateInstance<JellyfinMigrationService>(appHost.Servi
 200                await jellyfinMigrationService.PrepareSystemForMigration(_logger).ConfigureAwait(false);
 201                await jellyfinMigrationService.MigrateStepAsync(JellyfinMigrationStageTypes.CoreInitialisation, appHost.
 202
 203                await appHost.InitializeServices(startupConfig).ConfigureAwait(false);
 204
 205                await jellyfinMigrationService.MigrateStepAsync(JellyfinMigrationStageTypes.AppInitialisation, appHost.S
 206                await jellyfinMigrationService.CleanupSystemAfterMigration(_logger).ConfigureAwait(false);
 207                try
 208                {
 209                    configurationCompleted = true;
 210                    await _setupServer!.StopAsync().ConfigureAwait(false);
 211                    await _jellyfinHost.StartAsync().ConfigureAwait(false);
 212
 213                    if (!OperatingSystem.IsWindows() && startupConfig.UseUnixSocket())
 214                    {
 215                        var socketPath = StartupHelpers.GetUnixSocketPath(startupConfig, appPaths);
 216
 217                        StartupHelpers.SetUnixSocketPermissions(startupConfig, socketPath, _logger);
 218                    }
 219                }
 220                catch (Exception)
 221                {
 222                    _logger.LogError("Kestrel failed to start! This is most likely due to an invalid address or port bin
 223                    throw;
 224                }
 225
 226                await appHost.RunStartupTasksAsync().ConfigureAwait(false);
 227
 228                _logger.LogInformation("Startup complete {Time:g}", Stopwatch.GetElapsedTime(_startTimestamp));
 229
 230                await _jellyfinHost.WaitForShutdownAsync().ConfigureAwait(false);
 231                _restartOnShutdown = appHost.ShouldRestart;
 232                _restoreFromBackup = appHost.RestoreBackupPath;
 233            }
 234            catch (Exception ex)
 235            {
 236                _restartOnShutdown = false;
 237                _logger.LogCritical(ex, "Error while starting server");
 238                if (_setupServer!.IsAlive && !configurationCompleted)
 239                {
 240                    _setupServer!.SoftStop();
 241                    await Task.Delay(TimeSpan.FromMinutes(10)).ConfigureAwait(false);
 242                    await _setupServer!.StopAsync().ConfigureAwait(false);
 243                }
 244            }
 245            finally
 246            {
 247                // Don't throw additional exception if startup failed.
 248                if (appHost.ServiceProvider is not null)
 249                {
 250                    _logger.LogInformation("Running query planner optimizations in the database... This might take a whi
 251
 252                    var databaseProvider = appHost.ServiceProvider.GetRequiredService<IJellyfinDatabaseProvider>();
 253                    using var shutdownSource = new CancellationTokenSource();
 254                    shutdownSource.CancelAfter((int)TimeSpan.FromSeconds(60).TotalMicroseconds);
 255                    await databaseProvider.RunShutdownTask(shutdownSource.Token).ConfigureAwait(false);
 256                }
 257
 258                _appHost = null;
 259                _jellyfinHost?.Dispose();
 260            }
 261        }
 262
 263        /// <summary>
 264        /// [Internal]Runs the startup Migrations.
 265        /// </summary>
 266        /// <remarks>
 267        /// Not intended to be used other then by jellyfin and its tests.
 268        /// </remarks>
 269        /// <param name="appPaths">Application Paths.</param>
 270        /// <param name="startupConfig">Startup Config.</param>
 271        /// <returns>A task.</returns>
 272        public static async Task ApplyStartupMigrationAsync(ServerApplicationPaths appPaths, IConfiguration startupConfi
 273        {
 274            _migrationLogger = StartupLogger.Logger.BeginGroup<JellyfinMigrationService>($"Migration Service");
 275            var startupConfigurationManager = new ServerConfigurationManager(appPaths, _loggerFactory, new MyXmlSerializ
 276            startupConfigurationManager.AddParts([new DatabaseConfigurationFactory()]);
 277            var migrationStartupServiceProvider = new ServiceCollection()
 278                .AddLogging(d => d.AddSerilog())
 279                .AddJellyfinDbContext(startupConfigurationManager, startupConfig)
 280                .AddSingleton<IApplicationPaths>(appPaths)
 281                .AddSingleton<ServerApplicationPaths>(appPaths)
 282                .RegisterStartupLogger();
 283
 284            migrationStartupServiceProvider.AddSingleton(migrationStartupServiceProvider);
 285            var startupService = migrationStartupServiceProvider.BuildServiceProvider();
 286
 287            PrepareDatabaseProvider(startupService);
 288
 289            var jellyfinMigrationService = ActivatorUtilities.CreateInstance<JellyfinMigrationService>(startupService);
 290            await jellyfinMigrationService.CheckFirstTimeRunOrMigration(appPaths).ConfigureAwait(false);
 291            await jellyfinMigrationService.MigrateStepAsync(Migrations.Stages.JellyfinMigrationStageTypes.PreInitialisat
 292        }
 293
 294        /// <summary>
 295        /// [Internal]Runs the Jellyfin migrator service with the Core stage.
 296        /// </summary>
 297        /// <remarks>
 298        /// Not intended to be used other then by jellyfin and its tests.
 299        /// </remarks>
 300        /// <param name="serviceProvider">The service provider.</param>
 301        /// <param name="jellyfinMigrationStage">The stage to run.</param>
 302        /// <returns>A task.</returns>
 303        public static async Task ApplyCoreMigrationsAsync(IServiceProvider serviceProvider, Migrations.Stages.JellyfinMi
 304        {
 305            var jellyfinMigrationService = ActivatorUtilities.CreateInstance<JellyfinMigrationService>(serviceProvider, 
 306            await jellyfinMigrationService.MigrateStepAsync(jellyfinMigrationStage, serviceProvider).ConfigureAwait(fals
 307        }
 308
 309        /// <summary>
 310        /// Create the application configuration.
 311        /// </summary>
 312        /// <param name="commandLineOpts">The command line options passed to the program.</param>
 313        /// <param name="appPaths">The application paths.</param>
 314        /// <returns>The application configuration.</returns>
 315        public static IConfiguration CreateAppConfiguration(StartupOptions commandLineOpts, IApplicationPaths appPaths)
 316        {
 21317            return new ConfigurationBuilder()
 21318                .ConfigureAppConfiguration(commandLineOpts, appPaths)
 21319                .Build();
 320        }
 321
 322        private static IConfigurationBuilder ConfigureAppConfiguration(
 323            this IConfigurationBuilder config,
 324            StartupOptions commandLineOpts,
 325            IApplicationPaths appPaths,
 326            IConfiguration? startupConfig = null)
 327        {
 328            // Use the swagger API page as the default redirect path if not hosting the web client
 21329            var inMemoryDefaultConfig = ConfigurationOptions.DefaultConfiguration;
 21330            if (startupConfig is not null && !startupConfig.HostWebClient())
 331            {
 0332                inMemoryDefaultConfig[DefaultRedirectKey] = "api-docs/swagger";
 333            }
 334
 21335            return config
 21336                .SetBasePath(appPaths.ConfigurationDirectoryPath)
 21337                .AddInMemoryCollection(inMemoryDefaultConfig)
 21338                .AddJsonFile(LoggingConfigFileDefault, optional: false, reloadOnChange: true)
 21339                .AddJsonFile(LoggingConfigFileSystem, optional: true, reloadOnChange: true)
 21340                .AddEnvironmentVariables("JELLYFIN_")
 21341                .AddInMemoryCollection(commandLineOpts.ConvertToConfig());
 342        }
 343
 344        private static void PrepareDatabaseProvider(IServiceProvider services)
 345        {
 21346            var factory = services.GetRequiredService<IDbContextFactory<JellyfinDbContext>>();
 21347            var provider = services.GetRequiredService<IJellyfinDatabaseProvider>();
 21348            provider.DbContextFactory = factory;
 21349        }
 350    }
 351}