sourcegit/src/Views/ChangeStatusIcon.cs

using System;
using System.Globalization;

using Avalonia;
using Avalonia.Controls;
using Avalonia.Media;
using Avalonia.Styling;

namespace SourceGit.Views
{
    public class ChangeStatusIcon : Control
    {
        private static readonly string[] INDICATOR = ["?", "±", "T", "+", "−", "➜", "❏", "★", "!"];
        private static readonly Color[] COLOR =
        [
            Colors.Transparent,
            Colors.Goldenrod,
            Colors.Goldenrod,
            Colors.LimeGreen,
            Colors.Tomato,
            Colors.Orchid,
            Colors.Goldenrod,
            Colors.LimeGreen,
            Colors.OrangeRed,
        ];

        public static readonly StyledProperty<bool> IsUnstagedChangeProperty =
            AvaloniaProperty.Register<ChangeStatusIcon, bool>(nameof(IsUnstagedChange));

        public bool IsUnstagedChange
        {
            get => GetValue(IsUnstagedChangeProperty);
            set => SetValue(IsUnstagedChangeProperty, value);
        }

        public static readonly StyledProperty<Models.Change> ChangeProperty =
            AvaloniaProperty.Register<ChangeStatusIcon, Models.Change>(nameof(Change));

        public Models.Change Change
        {
            get => GetValue(ChangeProperty);
            set => SetValue(ChangeProperty, value);
        }

        public override void Render(DrawingContext context)
        {
            if (Change == null || Bounds.Width <= 0)
                return;

            var typeface = new Typeface("fonts:SourceGit#JetBrains Mono");

            var idx = (int)(IsUnstagedChange ? Change.WorkTree : Change.Index);
            var indicator = INDICATOR[idx];
            var color = COLOR[idx];
            var hsl = color.ToHsl();
            var color2 = ActualThemeVariant == ThemeVariant.Dark
                ? new HslColor(hsl.A, hsl.H, hsl.S, hsl.L - 0.1).ToRgb()
                : new HslColor(hsl.A, hsl.H, hsl.S, hsl.L + 0.1).ToRgb();

            var background = new LinearGradientBrush
            {
                GradientStops = [new GradientStop(color, 0), new GradientStop(color2, 1)],
                StartPoint = new RelativePoint(0, 0, RelativeUnit.Relative),
                EndPoint = new RelativePoint(0, 1, RelativeUnit.Relative),
            };

            var txt = new FormattedText(
                indicator,
                CultureInfo.CurrentCulture,
                FlowDirection.LeftToRight,
                typeface,
                Bounds.Width * 0.8,
                Brushes.White);

            var corner = (float)Math.Max(2, Bounds.Width / 16);
            var textOrigin = new Point((Bounds.Width - txt.Width) * 0.5, (Bounds.Height - txt.Height) * 0.5);
            context.DrawRectangle(background, null, new Rect(0, 0, Bounds.Width, Bounds.Height), corner, corner);
            context.DrawText(txt, textOrigin);
        }

        protected override void OnPropertyChanged(AvaloniaPropertyChangedEventArgs change)
        {
            base.OnPropertyChanged(change);

            if (change.Property == IsUnstagedChangeProperty ||
                change.Property == ChangeProperty ||
                (change.Property.Name == "ActualThemeVariant" && change.NewValue != null))
                InvalidateVisual();
        }
    }
}