FlowLogger.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System;
using System.Linq;
using Microsoft.Build.Framework;
using Microsoft.Build.Logging;

namespace RepoTasks
{
    public class FlowLogger : ConsoleLogger
    {
        // disabled until we can fix some of the whitespace and flow issues caused by invoking other shell commands from MSBuild
        // private static readonly bool IsTeamCity = !string.IsNullOrEmpty(Environment.GetEnvironmentVariable("TEAMCITY_PROJECT_NAME"));
        private static readonly bool IsTeamCity = false;

        private volatile bool _initialized;

        public FlowLogger()
        {
        }

        public override void Initialize(IEventSource eventSource, int nodeCount)
        {
            PreInit(eventSource);
            base.Initialize(eventSource, nodeCount);
        }

        public override void Initialize(IEventSource eventSource)
        {
            PreInit(eventSource);
            base.Initialize(eventSource);
        }

        private void PreInit(IEventSource eventSource)
        {
            if (_initialized) return;
            _initialized = true;

            var _flowId = GetFlowId();

            var writer = IsTeamCity
                ? (IWriter)new TeamCityMessageWriter(WriteHandler, _flowId)
                : new DefaultPrefixMessageWriter(WriteHandler, _flowId);

            WriteHandler = writer.WriteHandler;
            eventSource.BuildStarted += (o, e) =>
            {
                writer.OnBuildStarted(e);
            };
            eventSource.BuildFinished += (o, e) =>
            {
                writer.OnBuildFinished(e);
            };
        }

        private string GetFlowId()
        {
            var parameters = Parameters?.Split(new[] { ';' }, StringSplitOptions.RemoveEmptyEntries);
            if (parameters == null || parameters.Length == 0)
            {
                return null;
            }

            const string flowIdParamName = "FlowId=";
            return parameters
                .FirstOrDefault(p => p.StartsWith(flowIdParamName, StringComparison.Ordinal))
                ?.Substring(flowIdParamName.Length);
        }
    }
}