// Copyright (c) .NET Foundation and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
 
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using Mono.Cecil;
 
namespace Mono.Linker
{
    public class AttributeInfo
    {
        public Dictionary<ICustomAttributeProvider, CustomAttribute[]> CustomAttributes { get; }
 
        public Dictionary<CustomAttribute, MessageOrigin> CustomAttributesOrigins { get; }
 
        public AttributeInfo()
        {
            CustomAttributes = new Dictionary<ICustomAttributeProvider, CustomAttribute[]>();
            CustomAttributesOrigins = new Dictionary<CustomAttribute, MessageOrigin>();
        }
 
        public void AddCustomAttributes(ICustomAttributeProvider provider, CustomAttribute[] customAttributes, MessageOrigin[] origins)
        {
            Debug.Assert(customAttributes.Length == origins.Length);
 
            AddCustomAttributes(provider, customAttributes);
 
            foreach (var (customAttribute, origin) in customAttributes.Zip(origins))
            {
                CustomAttributesOrigins.Add(customAttribute, origin);
            }
        }
 
        public void AddCustomAttributes(ICustomAttributeProvider provider, CustomAttribute[] customAttributes)
        {
            if (!CustomAttributes.TryGetValue(provider, out var existing))
            {
                CustomAttributes.Add(provider, customAttributes);
            }
            else
            {
                int existingLength = existing.Length;
                Array.Resize(ref existing, existingLength + customAttributes.Length);
                customAttributes.CopyTo(existing, existingLength);
 
                CustomAttributes[provider] = existing;
            }
        }
    }
}