// ./LogCollector/LogCollector.cs
using System.Collections.Generic;
using System.Xml;
using Microsoft.VisualStudio.TestPlatform.ObjectModel;
using Microsoft.VisualStudio.TestPlatform.ObjectModel.DataCollection;

namespace LogCollector
{
    [DataCollectorFriendlyName("LogCollector")]
    [DataCollectorTypeUri("datacollector://LogCollector")]
    public class LogCollector : DataCollector
    {
        private readonly List<TestCase> _failedTests = new List<TestCase>();
        private DataCollectionLogger _logger;

        public override void Initialize(XmlElement configurationElement, DataCollectionEvents events,
            DataCollectionSink dataSink,
            DataCollectionLogger logger, DataCollectionEnvironmentContext environmentContext)
        {
            events.TestCaseEnd += TestCaseEnd_Handler;
            events.SessionEnd += SessionEnd_Handler;
            _logger = logger;
        }

        private void SessionEnd_Handler(object? sender, SessionEndEventArgs e)
        {
            _failedTests.ForEach(testCase =>
                _logger.LogError(new DataCollectionContext(testCase), testCase.FullyQualifiedName)
            );
        }

        private void TestCaseEnd_Handler(object? sender, TestCaseEndEventArgs testCaseEndEventArgs)
        {
            if (testCaseEndEventArgs.TestOutcome == TestOutcome.Failed)
            {
                _failedTests.Add(testCaseEndEventArgs.TestElement);
            }
        }
    }
}