| | | 1 | | using System; |
| | | 2 | | using System.Collections.Immutable; |
| | | 3 | | using Microsoft.CodeAnalysis; |
| | | 4 | | |
| | | 5 | | namespace Hertzole.ScriptableValues.Generator; |
| | | 6 | | |
| | | 7 | | internal static class SymbolExtensions |
| | | 8 | | { |
| | | 9 | | public static bool StringEquals(this ISymbol symbol, string value) |
| | | 10 | | { |
| | 200 | 11 | | return string.Equals(symbol.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat), value, StringComparison.O |
| | | 12 | | } |
| | | 13 | | |
| | | 14 | | public static bool StringEquals(this ISymbol symbol, ISymbol other) |
| | | 15 | | { |
| | 0 | 16 | | return StringEquals(symbol, other.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat)); |
| | | 17 | | } |
| | | 18 | | |
| | | 19 | | public static string GetFullyQualifiedMetadataName(this ITypeSymbol symbol) |
| | | 20 | | { |
| | 25 | 21 | | using ArrayBuilder<char> builder = new ArrayBuilder<char>(); |
| | | 22 | | |
| | 25 | 23 | | symbol.AppendFullyQualifiedMetadataName(in builder); |
| | | 24 | | |
| | 25 | 25 | | return builder.ToString(); |
| | 25 | 26 | | } |
| | | 27 | | |
| | | 28 | | private static void AppendFullyQualifiedMetadataName(this ITypeSymbol symbol, in ArrayBuilder<char> builder) |
| | | 29 | | { |
| | 25 | 30 | | BuildFrom(symbol, in builder); |
| | 25 | 31 | | return; |
| | | 32 | | |
| | | 33 | | static void BuildFrom(ISymbol symbol, in ArrayBuilder<char> builder) |
| | | 34 | | { |
| | | 35 | | switch (symbol) |
| | | 36 | | { |
| | | 37 | | // Namespaces that are nested also append a leading '.' |
| | | 38 | | case INamespaceSymbol { ContainingNamespace.IsGlobalNamespace: false }: |
| | 12 | 39 | | BuildFrom(symbol.ContainingNamespace, in builder); |
| | 12 | 40 | | builder.Add('.'); |
| | 12 | 41 | | builder.AddRange(symbol.MetadataName.AsSpan()); |
| | 12 | 42 | | break; |
| | | 43 | | |
| | | 44 | | // Other namespaces (ie. the one right before global) skip the leading '.' |
| | | 45 | | case INamespaceSymbol { IsGlobalNamespace: false }: |
| | 12 | 46 | | builder.AddRange(symbol.MetadataName.AsSpan()); |
| | 12 | 47 | | break; |
| | | 48 | | |
| | | 49 | | // Types with no namespace just have their metadata name directly written |
| | | 50 | | case ITypeSymbol { ContainingSymbol: INamespaceSymbol { IsGlobalNamespace: true } }: |
| | 13 | 51 | | builder.AddRange(symbol.MetadataName.AsSpan()); |
| | 13 | 52 | | break; |
| | | 53 | | |
| | | 54 | | // Types with a containing non-global namespace also append a leading '.' |
| | | 55 | | case ITypeSymbol { ContainingSymbol: INamespaceSymbol namespaceSymbol }: |
| | 12 | 56 | | BuildFrom(namespaceSymbol, in builder); |
| | 12 | 57 | | builder.Add('.'); |
| | 12 | 58 | | builder.AddRange(symbol.MetadataName.AsSpan()); |
| | 12 | 59 | | break; |
| | | 60 | | |
| | | 61 | | // Nested types append a leading '+' |
| | | 62 | | case ITypeSymbol { ContainingSymbol: ITypeSymbol typeSymbol }: |
| | 0 | 63 | | BuildFrom(typeSymbol, in builder); |
| | 0 | 64 | | builder.Add('+'); |
| | 0 | 65 | | builder.AddRange(symbol.MetadataName.AsSpan()); |
| | | 66 | | break; |
| | | 67 | | } |
| | 0 | 68 | | } |
| | | 69 | | } |
| | | 70 | | |
| | | 71 | | public static bool HasAttribute(this ISymbol symbol, string fullyQualifiedAttributeName) |
| | | 72 | | { |
| | 62 | 73 | | ImmutableArray<AttributeData> attributes = symbol.GetAttributes(); |
| | 62 | 74 | | if (attributes.IsDefaultOrEmpty) |
| | | 75 | | { |
| | 0 | 76 | | return false; |
| | | 77 | | } |
| | | 78 | | |
| | 396 | 79 | | for (int i = 0; i < attributes.Length; i++) |
| | | 80 | | { |
| | 161 | 81 | | AttributeData attribute = attributes.ItemRef(i); |
| | | 82 | | |
| | 161 | 83 | | if (attribute.AttributeClass == null) |
| | | 84 | | { |
| | 0 | 85 | | return false; |
| | | 86 | | } |
| | | 87 | | |
| | 161 | 88 | | string name = attribute.AttributeClass.ToDisplayString(NullableFlowState.NotNull, SymbolDisplayFormat.FullyQ |
| | 161 | 89 | | if (string.Equals(name, fullyQualifiedAttributeName, StringComparison.Ordinal)) |
| | | 90 | | { |
| | 25 | 91 | | return true; |
| | | 92 | | } |
| | | 93 | | } |
| | | 94 | | |
| | 37 | 95 | | return false; |
| | | 96 | | } |
| | | 97 | | |
| | | 98 | | public static bool TryGetCallbackAttribute(this ISymbol symbol, out CallbackType callbackType) |
| | | 99 | | { |
| | | 100 | | string name; |
| | 0 | 101 | | if (symbol is ITypeSymbol typeSymbol) |
| | | 102 | | { |
| | 0 | 103 | | name = typeSymbol.ToDisplayString(NullableFlowState.NotNull, SymbolDisplayFormat.FullyQualifiedFormat); |
| | | 104 | | } |
| | | 105 | | else |
| | | 106 | | { |
| | 0 | 107 | | name = symbol.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat); |
| | | 108 | | } |
| | | 109 | | |
| | 0 | 110 | | if (name.Equals(Types.GLOBAL_GENERATE_VALUE_CALLBACK_ATTRIBUTE, StringComparison.Ordinal)) |
| | | 111 | | { |
| | 0 | 112 | | callbackType = CallbackType.Value; |
| | 0 | 113 | | return true; |
| | | 114 | | } |
| | | 115 | | |
| | 0 | 116 | | if (name.Equals(Types.GLOBAL_GENERATE_EVENT_CALLBACK_ATTRIBUTE, StringComparison.Ordinal)) |
| | | 117 | | { |
| | 0 | 118 | | callbackType = CallbackType.Event; |
| | 0 | 119 | | return true; |
| | | 120 | | } |
| | | 121 | | |
| | 0 | 122 | | if (name.Equals(Types.GLOBAL_GENERATE_POOL_CALLBACK_ATTRIBUTE, StringComparison.Ordinal)) |
| | | 123 | | { |
| | 0 | 124 | | callbackType = CallbackType.Pool; |
| | 0 | 125 | | return true; |
| | | 126 | | } |
| | | 127 | | |
| | 0 | 128 | | if (name.Equals(Types.GLOBAL_GENERATE_COLLECTION_CALLBACK_ATTRIBUTE, StringComparison.Ordinal)) |
| | | 129 | | { |
| | 0 | 130 | | callbackType = CallbackType.Collection; |
| | 0 | 131 | | return true; |
| | | 132 | | } |
| | | 133 | | |
| | 0 | 134 | | callbackType = (CallbackType) byte.MaxValue; |
| | 0 | 135 | | return false; |
| | | 136 | | } |
| | | 137 | | } |