gui-cs.Terminal.Gui/Terminal.Gui/View/Adornment/Margin.cs

#nullable enable

using System.Drawing;

namespace Terminal.Gui;

/// <summary>The Margin for a <see cref="View"/>.</summary>
/// <remarks>
///     <para>See the <see cref="Adornment"/> class.</para>
/// </remarks>
public class Margin : Adornment
{
    /// <inheritdoc/>
    public Margin ()
    { /* Do nothing; A parameter-less constructor is required to support all views unit tests. */
    }

    /// <inheritdoc/>
    public Margin (View parent) : base (parent)
    {
        /* Do nothing; View.CreateAdornment requires a constructor that takes a parent */

        HighlightStyle |= HighlightStyle.Pressed;
        Highlight += Margin_Highlight;
        LayoutStarted += Margin_LayoutStarted;

        // Margin should not be focusable
        CanFocus = false;
    }

    private void Margin_LayoutStarted (object? sender, LayoutEventArgs e)
    {
        // Adjust the shadow such that it is drawn aligned with the Border
        if (ShadowStyle != Gui.ShadowStyle.None && _rightShadow is { } && _bottomShadow is { })
        {
            _rightShadow.Y = Parent.Border.Thickness.Top > 0 ? Parent.Border.Thickness.Top - (Parent.Border.Thickness.Top > 2 && Parent.Border.ShowTitle ? 1 : 0) : 1;
            _bottomShadow.X = Parent.Border.Thickness.Left > 0 ? Parent.Border.Thickness.Left : 1;
        }
    }

    private bool _pressed;
    private void Margin_Highlight (object? sender, CancelEventArgs<HighlightStyle> e)
    {
        if (ShadowStyle != Gui.ShadowStyle.None)
        {
            if (_pressed && e.CurrentValue == HighlightStyle.None)
            {
                Thickness = new (Thickness.Left - 1, Thickness.Top, Thickness.Right + 1, Thickness.Bottom);

                if (_rightShadow is { })
                {
                    _rightShadow.Visible = true;
                }

                if (_bottomShadow is { })
                {
                    _bottomShadow.Visible = true;
                }

                _pressed = false;
                return;
            }

            if (!_pressed && (e.CurrentValue.HasFlag (HighlightStyle.Pressed) /*|| e.HighlightStyle.HasFlag (HighlightStyle.PressedOutside)*/))
            {
                Thickness = new (Thickness.Left + 1, Thickness.Top, Thickness.Right - 1, Thickness.Bottom);
                _pressed = true;
                if (_rightShadow is { })
                {
                    _rightShadow.Visible = false;
                }

                if (_bottomShadow is { })
                {
                    _bottomShadow.Visible = false;
                }
            }
        }

    }

    /// <inheritdoc />
    public override void OnDrawContent (Rectangle viewport)
    {
        Rectangle screen = ViewportToScreen (viewport);
        Attribute normalAttr = GetNormalColor ();

        Driver?.SetAttribute (normalAttr);


        // This just draws/clears the thickness, not the insides.
        if (ShadowStyle != ShadowStyle.None)
        {
            screen = Rectangle.Inflate (screen, -1, -1);
        }
        Thickness.Draw (screen, ToString ());

        if (Subviews.Count > 0)
        {
            // Draw subviews
            // TODO: Implement OnDrawSubviews (cancelable);
            if (Subviews is { } && SubViewNeedsDisplay)
            {
                IEnumerable<View> subviewsNeedingDraw = Subviews.Where (
                                                                        view => view.Visible
                                                                                && (view.NeedsDisplay || view.SubViewNeedsDisplay || view.LayoutNeeded)
                                                                       );
                foreach (View view in subviewsNeedingDraw)
                {
                    if (view.LayoutNeeded)
                    {
                        view.LayoutSubviews ();
                    }
                    view.Draw ();
                }
            }
        }
    }

    /// <summary>
    ///     The color scheme for the Margin. If set to <see langword="null"/>, gets the <see cref="Adornment.Parent"/>'s
    ///     <see cref="View.SuperView"/> scheme. color scheme.
    /// </summary>
    public override ColorScheme ColorScheme
    {
        get
        {
            if (base.ColorScheme is { })
            {
                return base.ColorScheme;
            }

            return (Parent?.SuperView?.ColorScheme ?? Colors.ColorSchemes ["TopLevel"])!;
        }
        set
        {
            base.ColorScheme = value;
            Parent?.SetNeedsDisplay ();
        }
    }

    /// <inheritdoc />
    public override ShadowStyle ShadowStyle
    {
        get => base.ShadowStyle;
        set
        {
            base.ShadowStyle = SetShadow (value);
        }
    }

    /// <summary>
    ///    Sets whether the Margin includes a shadow effect. The shadow is drawn on the right and bottom sides of the
    ///     Margin.
    /// </summary>
    public ShadowStyle SetShadow (ShadowStyle style)
    {
        if (ShadowStyle == style)
        {
            // return style;
        }

        if (ShadowStyle != ShadowStyle.None)
        {
            // Turn off shadow
            Thickness = new (Thickness.Left, Thickness.Top, Thickness.Right - 1, Thickness.Bottom - 1);
        }

        if (style != ShadowStyle.None)
        {
            // Turn on shadow
            Thickness = new (Thickness.Left, Thickness.Top, Thickness.Right + 1, Thickness.Bottom + 1);
        }

        if (_rightShadow is { })
        {
            _rightShadow.ShadowStyle = style;
        }

        if (_bottomShadow is { })
        {
            _bottomShadow.ShadowStyle = style;
        }
        return style;
    }

    private ShadowView? _bottomShadow;
    private ShadowView? _rightShadow;

    /// <inheritdoc/>
    public override void BeginInit ()
    {
        base.BeginInit ();

        if (Parent is null)
        {
            return;
        }

        ShadowStyle = base.ShadowStyle;
        Add (
             _rightShadow = new ShadowView
             {
                 X = Pos.AnchorEnd (1),
                 Y = 0,
                 Width = 1,
                 Height = Dim.Fill (),
                 ShadowStyle = ShadowStyle,
                 Orientation = Orientation.Vertical
             },
             _bottomShadow = new ShadowView
             {
                 X = 0,
                 Y = Pos.AnchorEnd (1),
                 Width = Dim.Fill (),
                 Height = 1,
                 ShadowStyle = ShadowStyle,
                 Orientation = Orientation.Horizontal
             }
            );
    }
}