ServiceCollectionExtensions.cs

Go to the documentation of this file.

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Serilog;
using Serilog.Configuration;
using System;
using System.Diagnostics;
using System.Globalization;
using Tgstation.Server.Host.Configuration;

namespace Tgstation.Server.Host.Extensions
{
    static class ServiceCollectionExtensions
    {
        public const string SerilogContextTemplate = "(Instance:{Instance}|Job:{Job}|Request:{Request}|User:{User}|Monitor:{Monitor}|Bridge:{Bridge}|Chat:{ChatMessage})";

        public static IServiceCollection UseStandardConfig<TConfig>(this IServiceCollection serviceCollection, IConfiguration configuration) where TConfig : class
        {
            if (serviceCollection == null)
                throw new ArgumentNullException(nameof(serviceCollection));
            if (configuration == null)
                throw new ArgumentNullException(nameof(configuration));

            const string SectionFieldName = nameof(GeneralConfiguration.Section);

            var configType = typeof(TConfig);
            var sectionField = configType.GetField(SectionFieldName);
            if (sectionField == null)
                throw new InvalidOperationException(String.Format(CultureInfo.InvariantCulture, "{0} has no {1} field!", configType, SectionFieldName));

            var stringType = typeof(string);
            if (sectionField.FieldType != stringType)
                throw new InvalidOperationException(String.Format(CultureInfo.InvariantCulture, "{0} has invalid {1} field type, must be {2}!", configType, SectionFieldName, stringType));

            var sectionName = (string)sectionField.GetValue(null);

            return serviceCollection.Configure<TConfig>(configuration.GetSection(sectionName));
        }

        public static IServiceCollection SetupLogging(
            this IServiceCollection serviceCollection,
            Action<LoggerConfiguration> configurationAction,
            Action<LoggerSinkConfiguration> sinkConfigurationAction = null)
            => serviceCollection.AddLogging(builder =>
            {
                builder.ClearProviders();

                var configuration = new LoggerConfiguration()
                    .MinimumLevel
                        .Verbose();

                configurationAction?.Invoke(configuration);

                configuration
                    .Enrich.FromLogContext()
                    .WriteTo
                    .Async(sinkConfiguration =>
                    {
                        sinkConfiguration.Console(
                            outputTemplate: "[{Timestamp:HH:mm:ss}] {Level:w3}: {SourceContext:l} "
                                + SerilogContextTemplate
                                + "{NewLine}    {Message:lj}{NewLine}{Exception}");
                        sinkConfigurationAction?.Invoke(sinkConfiguration);
                    });

                builder.AddSerilog(configuration.CreateLogger(), true);

                if (Debugger.IsAttached)
                    builder.AddDebug();
            });
    }
}