// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Diagnostics;
using Aspire.Dashboard.Otlp.Storage;
using Google.Protobuf.Collections;
using OpenTelemetry.Proto.Common.V1;
 
namespace Aspire.Dashboard.Otlp.Model;
 
[DebuggerDisplay("Application = {Application}, Properties = {Properties.Count}")]
public class OtlpApplicationView
{
    public ApplicationKey ApplicationKey => Application.ApplicationKey;
    public OtlpApplication Application { get; }
    public KeyValuePair<string, string>[] Properties { get; }
 
    public OtlpApplicationView(OtlpApplication application, RepeatedField<KeyValue> attributes)
    {
        Application = application;
 
        List<KeyValuePair<string, string>>? properties = null;
        foreach (var attribute in attributes)
        {
            switch (attribute.Key)
            {
                case OtlpApplication.SERVICE_NAME:
                case OtlpApplication.SERVICE_INSTANCE_ID:
                    // Values passed in via ctor and set to members. Don't add to properties collection.
                    break;
                default:
                    properties ??= [];
                    properties.Add(new KeyValuePair<string, string>(attribute.Key, attribute.Value.GetString()));
                    break;
 
            }
        }
 
        if (properties != null)
        {
            // Sort so keys are in a consistent order for equality check.
            properties.Sort((p1, p2) => string.Compare(p1.Key, p2.Key, StringComparisons.OtlpAttribute));
            Properties = properties.ToArray();
        }
        else
        {
            Properties = [];
        }
    }
 
    public Dictionary<string, string> AllProperties()
    {
        var props = new Dictionary<string, string>(StringComparers.OtlpAttribute)
        {
            { OtlpApplication.SERVICE_NAME, Application.ApplicationName },
            { OtlpApplication.SERVICE_INSTANCE_ID, Application.InstanceId }
        };
 
        foreach (var kv in Properties)
        {
            props.TryAdd(kv.Key, kv.Value);
        }
 
        return props;
    }
}