spectre.console/src/Spectre.Console/Enrichment/ProfileEnricher.cs

using System;
using System.Collections.Generic;
using System.Linq;

namespace Spectre.Console.Enrichment
{
    internal static class ProfileEnricher
    {
        private static readonly List<IProfileEnricher> _defaultEnrichers = new List<IProfileEnricher>
        {
            new WindowsTerminalEnricher(),
            new AppVeyorEnricher(),
            new BambooEnricher(),
            new BitbucketEnricher(),
            new BitriseEnricher(),
            new ContinuaEnricher(),
            new GitHubEnricher(),
            new GitLabEnricher(),
            new GoCDEnricher(),
            new JenkinsEnricher(),
            new MyGetEnricher(),
            new TeamCityEnricher(),
            new TfsEnricher(),
            new TravisEnricher(),
        };

        public static void Enrich(
            Profile profile,
            ProfileEnrichment settings,
            IDictionary<string, string>? environmentVariables)
        {
            if (profile is null)
            {
                throw new ArgumentNullException(nameof(profile));
            }

            settings ??= new ProfileEnrichment();

            var variables = GetEnvironmentVariables(environmentVariables);
            foreach (var enricher in GetEnrichers(settings))
            {
                if (string.IsNullOrWhiteSpace(enricher.Name))
                {
                    throw new InvalidOperationException($"Profile enricher of type '{enricher.GetType().FullName}' does not have a name.");
                }

                if (enricher.Enabled(variables))
                {
                    enricher.Enrich(profile);
                    profile.AddEnricher(enricher.Name);
                }
            }
        }

        private static List<IProfileEnricher> GetEnrichers(ProfileEnrichment settings)
        {
            var enrichers = new List<IProfileEnricher>();

            if (settings.UseDefaultEnrichers)
            {
                enrichers.AddRange(_defaultEnrichers);
            }

            if (settings.Enrichers?.Count > 0)
            {
                enrichers.AddRange(settings.Enrichers);
            }

            return enrichers;
        }

        private static IDictionary<string, string> GetEnvironmentVariables(IDictionary<string, string>? variables)
        {
            if (variables != null)
            {
                return new Dictionary<string, string>(variables, StringComparer.OrdinalIgnoreCase);
            }

            return Environment.GetEnvironmentVariables()
                .Cast<System.Collections.DictionaryEntry>()
                .Aggregate(
                    new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase),
                    (dictionary, entry) =>
                    {
                        var key = (string)entry.Key;
                        if (!dictionary.TryGetValue(key, out _))
                        {
                            dictionary.Add(key, entry.Value as string ?? string.Empty);
                        }

                        return dictionary;
                    },
                    dictionary => dictionary);
        }
    }
}