// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Collections.Immutable;
using Aspire.Hosting.ApplicationModel;
 
namespace Aspire.Hosting;
 
internal static class CustomResourceSnapshotExtensions
{
    internal static ImmutableArray<ResourcePropertySnapshot> SetResourceProperty(this ImmutableArray<ResourcePropertySnapshot> properties, string name, object value, bool isSensitive = false, string? displayName = null, bool isHighlighted = false, int? sortOrder = null)
    {
        for (var i = 0; i < properties.Length; i++)
        {
            var property = properties[i];
 
            if (string.Equals(property.Name, name, StringComparisons.ResourcePropertyName))
            {
                if (Equals(property.Value, value) &&
                    property.IsSensitive == isSensitive &&
                    string.Equals(property.DisplayName, displayName, StringComparison.Ordinal) &&
                    property.IsHighlighted == isHighlighted &&
                    property.SortOrder == sortOrder)
                {
                    // Unchanged.
                    return properties;
                }
 
                // Set value.
                return properties.SetItem(i, property with { Value = value, IsSensitive = isSensitive, DisplayName = displayName, IsHighlighted = isHighlighted, SortOrder = sortOrder });
            }
        }
 
        // Add property.
        return [.. properties, new ResourcePropertySnapshot(name, value) { IsSensitive = isSensitive, DisplayName = displayName, IsHighlighted = isHighlighted, SortOrder = sortOrder }];
    }
 
    internal static ImmutableArray<ResourcePropertySnapshot> SetResourcePropertyRange(this ImmutableArray<ResourcePropertySnapshot> properties, IEnumerable<ResourcePropertySnapshot> newValues)
    {
        var existingProperties = new List<ResourcePropertySnapshot>(properties);
        var propertiesToAdd = new List<ResourcePropertySnapshot>();
 
        foreach (var newValue in newValues)
        {
            var found = false;
            for (var i = 0; i < existingProperties.Count; i++)
            {
                var existingProperty = existingProperties[i];
 
                if (string.Equals(existingProperty.Name, newValue.Name, StringComparisons.ResourcePropertyName))
                {
                    if (existingProperty != newValue)
                    {
                        existingProperties[i] = newValue;
                    }
 
                    found = true;
                    break;
                }
            }
 
            if (!found)
            {
                propertiesToAdd.Add(newValue);
            }
        }
 
        return [.. existingProperties, .. propertiesToAdd];
    }
 
    internal static ImmutableArray<ResourcePropertySnapshot> RemoveResourceProperty(this ImmutableArray<ResourcePropertySnapshot> properties, string name)
    {
        for (var i = 0; i < properties.Length; i++)
        {
            if (string.Equals(properties[i].Name, name, StringComparisons.ResourcePropertyName))
            {
                return properties.RemoveAt(i);
            }
        }
 
        return properties;
    }
}