blob: 9604ecd9f1ffa560f34199ee267269ac8936abef (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
using System;
using System.Collections.Generic;
using System.Linq;
using Mono.Cecil;
namespace Mono.Linker
{
public class CustomAttributeSource
{
private readonly Dictionary<ICustomAttributeProvider, IEnumerable<CustomAttribute>> _xmlCustomAttributes;
private readonly Dictionary<ICustomAttributeProvider, IEnumerable<Attribute>> _internalAttributes;
public CustomAttributeSource ()
{
_xmlCustomAttributes = new Dictionary<ICustomAttributeProvider, IEnumerable<CustomAttribute>> ();
_internalAttributes = new Dictionary<ICustomAttributeProvider, IEnumerable<Attribute>> ();
}
public void AddCustomAttributes (ICustomAttributeProvider provider, IEnumerable<CustomAttribute> customAttributes)
{
if (!_xmlCustomAttributes.ContainsKey (provider))
_xmlCustomAttributes[provider] = customAttributes;
else
_xmlCustomAttributes[provider] = _xmlCustomAttributes[provider].Concat (customAttributes);
}
public IEnumerable<CustomAttribute> GetCustomAttributes (ICustomAttributeProvider provider)
{
if (provider.HasCustomAttributes) {
foreach (var customAttribute in provider.CustomAttributes)
yield return customAttribute;
}
if (_xmlCustomAttributes.TryGetValue (provider, out var annotations)) {
foreach (var customAttribute in annotations)
yield return customAttribute;
}
}
public bool HasCustomAttributes (ICustomAttributeProvider provider)
{
if (provider.HasCustomAttributes)
return true;
if (_xmlCustomAttributes.ContainsKey (provider))
return true;
return false;
}
public void AddInternalAttributes (ICustomAttributeProvider provider, IEnumerable<Attribute> attributes)
{
if (!_internalAttributes.ContainsKey (provider))
_internalAttributes[provider] = attributes;
else
_internalAttributes[provider] = _internalAttributes[provider].Concat (attributes);
}
public IEnumerable<Attribute> GetInternalAttributes (ICustomAttributeProvider provider)
{
if (_internalAttributes.TryGetValue (provider, out var annotations)) {
foreach (var attribute in annotations)
yield return attribute;
}
}
public bool HasInternalAttributes (ICustomAttributeProvider provider)
{
return _internalAttributes.ContainsKey (provider) ? true : false;
}
public bool HasAttributes (ICustomAttributeProvider provider) =>
HasCustomAttributes (provider) || HasInternalAttributes (provider);
}
}
|