79 lines
3.8 KiB
C#
79 lines
3.8 KiB
C#
// 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.Collections.Generic;
|
|
using System.Text;
|
|
using Microsoft.CodeAnalysis;
|
|
|
|
namespace CommunityToolkit.Mvvm.SourceGenerators.Extensions;
|
|
|
|
/// <summary>
|
|
/// Extension methods for the <see cref="INamedTypeSymbol"/> type.
|
|
/// </summary>
|
|
internal static class INamedTypeSymbolExtensions
|
|
{
|
|
/// <summary>
|
|
/// Gets a valid filename for a given <see cref="INamedTypeSymbol"/> instance.
|
|
/// </summary>
|
|
/// <param name="symbol">The input <see cref="INamedTypeSymbol"/> instance.</param>
|
|
/// <returns>The full metadata name for <paramref name="symbol"/> that is also a valid filename.</returns>
|
|
public static string GetFullMetadataNameForFileName(this INamedTypeSymbol symbol)
|
|
{
|
|
static StringBuilder BuildFrom(ISymbol? symbol, StringBuilder builder)
|
|
{
|
|
return symbol switch
|
|
{
|
|
INamespaceSymbol ns when ns.IsGlobalNamespace => builder,
|
|
INamespaceSymbol ns when ns.ContainingNamespace is { IsGlobalNamespace: false }
|
|
=> BuildFrom(ns.ContainingNamespace, builder.Insert(0, $".{ns.MetadataName}")),
|
|
ITypeSymbol ts when ts.ContainingType is ISymbol pt
|
|
=> BuildFrom(pt, builder.Insert(0, $"+{ts.MetadataName}")),
|
|
ITypeSymbol ts when ts.ContainingNamespace is ISymbol pn and not INamespaceSymbol { IsGlobalNamespace: true }
|
|
=> BuildFrom(pn, builder.Insert(0, $".{ts.MetadataName}")),
|
|
ISymbol => BuildFrom(symbol.ContainingSymbol, builder.Insert(0, symbol.MetadataName)),
|
|
_ => builder
|
|
};
|
|
}
|
|
|
|
// Build the full metadata name by concatenating the metadata names of all symbols from the input
|
|
// one to the outermost namespace, if any. Additionally, the ` and + symbols need to be replaced
|
|
// to avoid errors when generating code. This is a known issue with source generators not accepting
|
|
// those characters at the moment, see: https://github.com/dotnet/roslyn/issues/58476.
|
|
return BuildFrom(symbol, new StringBuilder(256)).ToString().Replace('`', '-').Replace('+', '.');
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets all member symbols from a given <see cref="INamedTypeSymbol"/> instance, including inherited ones.
|
|
/// </summary>
|
|
/// <param name="symbol">The input <see cref="INamedTypeSymbol"/> instance.</param>
|
|
/// <returns>A sequence of all member symbols for <paramref name="symbol"/>.</returns>
|
|
public static IEnumerable<ISymbol> GetAllMembers(this INamedTypeSymbol symbol)
|
|
{
|
|
for (INamedTypeSymbol? currentSymbol = symbol; currentSymbol is { SpecialType: not SpecialType.System_Object }; currentSymbol = currentSymbol.BaseType)
|
|
{
|
|
foreach (ISymbol memberSymbol in currentSymbol.GetMembers())
|
|
{
|
|
yield return memberSymbol;
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets all member symbols from a given <see cref="INamedTypeSymbol"/> instance, including inherited ones.
|
|
/// </summary>
|
|
/// <param name="symbol">The input <see cref="INamedTypeSymbol"/> instance.</param>
|
|
/// <param name="name">The name of the members to look for.</param>
|
|
/// <returns>A sequence of all member symbols for <paramref name="symbol"/>.</returns>
|
|
public static IEnumerable<ISymbol> GetAllMembers(this INamedTypeSymbol symbol, string name)
|
|
{
|
|
for (INamedTypeSymbol? currentSymbol = symbol; currentSymbol is { SpecialType: not SpecialType.System_Object }; currentSymbol = currentSymbol.BaseType)
|
|
{
|
|
foreach (ISymbol memberSymbol in currentSymbol.GetMembers(name))
|
|
{
|
|
yield return memberSymbol;
|
|
}
|
|
}
|
|
}
|
|
}
|