// Copyright (c) Microsoft. All rights reserved.

using Harness.ConsoleReactiveFramework;

namespace Harness.ConsoleReactiveComponents;

/// <summary>
/// Props for <see cref="TextPanel"/>.
/// </summary>
public record TextPanelProps : ConsoleReactiveProps
{
    /// <summary>Gets the items to render in the panel. Each item is a pre-rendered
    /// console string (may include ANSI escape sequences and newlines).</summary>
    public IReadOnlyList<string> Items { get; init; } = [];
}

/// <summary>
/// A component that renders a list of pre-rendered string items vertically.
/// Designed for rendering dynamic items in a non-scroll region that may be
/// re-rendered on each update. If the component's <see cref="ConsoleReactiveProps.Height"/>
/// exceeds the number of output lines, leftover lines are erased.
/// </summary>
public class TextPanel : ConsoleReactiveComponent<TextPanelProps, ConsoleReactiveState>
{
    /// <summary>
    /// Calculates the height (in lines) needed to render all items.
    /// </summary>
    /// <param name="items">The items to measure.</param>
    /// <returns>The total number of lines all items will occupy.</returns>
    public static int CalculateHeight(IReadOnlyList<string> items)
    {
        int total = 0;
        for (int i = 0; i < items.Count; i++)
        {
            total += CountLines(items[i]);
        }

        return total;
    }

    /// <inheritdoc />
    public override void RenderCore(TextPanelProps props, ConsoleReactiveState state)
    {
        int currentRow = 0;

        for (int i = 0; i < props.Items.Count; i++)
        {
            string text = props.Items[i];
            string[] lines = text.Split('\n');
            int lineCount = CountLines(text);

            for (int j = 0; j < lineCount; j++)
            {
                Console.Write(AnsiEscapes.MoveAndEraseLine(props.Y + currentRow));
                Console.Write(lines[j]);
                currentRow++;
            }
        }

        // If the component height exceeds the output, erase leftover lines
        if (props.Height > currentRow)
        {
            for (int i = currentRow; i < props.Height; i++)
            {
                Console.Write(AnsiEscapes.MoveAndEraseLine(props.Y + i));
            }
        }
    }

    private static int CountLines(string text)
    {
        if (string.IsNullOrEmpty(text))
        {
            return 0;
        }

        int count = 1;
        for (int i = 0; i < text.Length; i++)
        {
            if (text[i] == '\n')
            {
                count++;
            }
        }

        // If text ends with a newline, don't count the trailing empty line
        if (text[text.Length - 1] == '\n')
        {
            count--;
        }

        return count;
    }
}