using System;
using System.IO;

namespace Spectre.Console.Tests
{
    public sealed class AnsiConsoleFixture : IDisposable
    {
        private readonly StringWriter _writer;

        public IAnsiConsole Console { get; }

        public string Output => _writer.ToString();

        public AnsiConsoleFixture(ColorSystem system, AnsiSupport ansi = AnsiSupport.Yes, int width = 80)
        {
            _writer = new StringWriter();

            Console = new ConsoleWithWidth(
                AnsiConsole.Create(new AnsiConsoleSettings
                {
                    Ansi = ansi,
                    ColorSystem = (ColorSystemSupport)system,
                    Out = _writer,
                }), width);
        }

        public void Dispose()
        {
            _writer?.Dispose();
        }
    }
}