src/QueryCat.Cli/Infrastructure/PagingOutput.cs - develop - krasninja/querycat

src/QueryCat.Cli/Infrastructure/PagingOutput.cs
84 строки2 KB
Ivan Kozhin
Move more messages to resources
23 апр 2024, 06:45
23 апр 2024, 06:4584f0d65
using QueryCat.Backend.Core.Data;
using QueryCat.Backend.Core.Types;

namespace QueryCat.Cli.Infrastructure;

/// <summary>
/// Render rows by pages.
/// </summary>
public class PagingOutput : IRowsOutput
{
    public const int NoLimit = -1;

    private static readonly string _clearText = new('\r', Resources.Messages.MoreRows.Length);

    private readonly IRowsOutput _rowsOutput;
    private readonly CancellationTokenSource? _cancellationTokenSource;
    private int _rowsCounter;

    /// <summary>
    /// Return specific amount of rows and stop for user input. -1 means no paging.
    /// </summary>
    public int PagingRowsCount { get; set; }

    /// <inheritdoc />
    public QueryContext QueryContext
    {
        get => _rowsOutput.QueryContext;
        set => _rowsOutput.QueryContext = value;
    }

    /// <inheritdoc />
    public RowsOutputOptions Options { get; } = new()
    {
        RequiresColumnsLengthAdjust = true,
    };

    public PagingOutput(IRowsOutput rowsOutput, int pagingRowsCount = 20, CancellationTokenSource? cancellationTokenSource = null)
    {
        _rowsOutput = rowsOutput;
        _cancellationTokenSource = cancellationTokenSource;
        PagingRowsCount = pagingRowsCount;
    }

    /// <inheritdoc />
    public void Open() => _rowsOutput.Open();

    /// <inheritdoc />
    public void Close() => _rowsOutput.Close();

    /// <inheritdoc />
    public void Reset()
    {
        _rowsCounter = 0;
        _rowsOutput.Reset();
    }

    /// <inheritdoc />
    public void WriteValues(in VariantValue[] values)
    {
        _rowsOutput.WriteValues(values);
        if (PagingRowsCount != NoLimit
            && Environment.UserInteractive
            && ++_rowsCounter >= PagingRowsCount
            && !Console.IsInputRedirected
            && !Console.IsOutputRedirected)
        {
            _rowsCounter = 0;
            Console.Write(Resources.Messages.MoreRows);
            var consoleKey = Console.ReadKey();
            Console.Write(_clearText);

            // Show all next content.
            if (consoleKey.Key == ConsoleKey.A)
            {
                PagingRowsCount = -1;
            }
            // Quit.
            else if (consoleKey.Key == ConsoleKey.Q)
            {
                _cancellationTokenSource?.Cancel();
            }
        }
    }
}