using Microsoft.Extensions.Logging;
using System;
using Xunit.Abstractions;

namespace k8s.Tests.Logging
{
    /// <summary>
    ///     Extension methods for logging to Xunit text output.
    /// </summary>
    public static class TestOutputLoggingExtensions
    {
        /// <summary>
        ///     Log to test output.
        /// </summary>
        /// <param name="logging">
        ///     The global logging configuration.
        /// </param>
        /// <param name="testOutput">
        ///     Output for the current test.
        /// </param>
        /// <param name="minLogLevel">
        ///     The minimum level to log at.
        /// </param>
        public static void AddTestOutput(this ILoggingBuilder logging, ITestOutputHelper testOutput, LogLevel minLogLevel = LogLevel.Information)
        {
            if (logging == null)
                throw new ArgumentNullException(nameof(logging));

            if (testOutput == null)
                throw new ArgumentNullException(nameof(testOutput));

            logging.AddProvider(
                new TestOutputLoggerProvider(testOutput, minLogLevel)
            );
        }

        /// <summary>
        ///     Log to test output.
        /// </summary>
        /// <param name="loggers">
        ///     The logger factory.
        /// </param>
        /// <param name="testOutput">
        ///     Output for the current test.
        /// </param>
        /// <param name="minLogLevel">
        ///     The minimum level to log at.
        /// </param>
        /// <returns>
        ///     The logger factory (enables inline use / method-chaining).
        /// </returns>
        public static ILoggerFactory AddTestOutput(this ILoggerFactory loggers, ITestOutputHelper testOutput, LogLevel minLogLevel = LogLevel.Information)
        {
            if (loggers == null)
                throw new ArgumentNullException(nameof(loggers));

            if (testOutput == null)
                throw new ArgumentNullException(nameof(testOutput));

            loggers.AddProvider(
                new TestOutputLoggerProvider(testOutput, minLogLevel)
            );

            return loggers;
        }
    }
}