Skip to content

Commit

Permalink
完成生成代码
Browse files Browse the repository at this point in the history
  • Loading branch information
@lindexi
lindexi committed Aug 28, 2023
1 parent 423ce81 commit a35a4d0
Show file tree
Hide file tree
Showing 2 changed files with 270 additions and 22 deletions.
286 changes: 264 additions & 22 deletions ...enerator/Analyzers/ExportTypeToMethod_/TelescopeExportTypeToMethodIncrementalGenerator.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,9 @@
using System.Collections.Immutable;
using System.Data;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text;

using dotnetCampus.Telescope.SourceGeneratorAnalyzers.Core;

Expand Down Expand Up @@ -118,12 +120,15 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
var funcTypeSymbol = (INamedTypeSymbol) valueTupleInfo.ItemList[2].ItemType;
// 准备导出的类型的基类型
var expectedClassBaseType = funcTypeSymbol.TypeArguments[0];
// 表示的特性
var expectedClassAttributeType = valueTupleInfo.ItemList[1].ItemType;
return new ExportMethodReturnTypeCollectionResult(expectedClassBaseType, expectedClassAttributeType,
exportTypeCollectionResult, new ValueTupleExportMethodReturnTypeInfo(valueTupleInfo));
exportTypeCollectionResult, new ValueTupleExportMethodReturnTypeInfo(valueTupleInfo)
{
IsIEnumerable = true
});
}
}
}
Expand Down Expand Up @@ -205,22 +210,23 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
.Collect();

// 先收集整个项目里面所有的类型
var candidateClassCollectionResultIncrementalValuesProvider = context.SyntaxProvider.CreateSyntaxProvider((syntaxNode, _) =>
{
return syntaxNode.IsKind(SyntaxKind.ClassDeclaration);
}, (generatorSyntaxContext, token) =>
{
var classDeclarationSyntax = (ClassDeclarationSyntax) generatorSyntaxContext.Node;
// 从语法转换为语义，用于后续判断是否标记了特性
INamedTypeSymbol? assemblyClassTypeSymbol =
generatorSyntaxContext.SemanticModel.GetDeclaredSymbol(classDeclarationSyntax, token);
if (assemblyClassTypeSymbol is not null && !assemblyClassTypeSymbol.IsAbstract)
var candidateClassCollectionResultIncrementalValuesProvider = context.SyntaxProvider.CreateSyntaxProvider(
(syntaxNode, _) =>
{
return assemblyClassTypeSymbol;
}
return syntaxNode.IsKind(SyntaxKind.ClassDeclaration);
}, (generatorSyntaxContext, token) =>
{
var classDeclarationSyntax = (ClassDeclarationSyntax) generatorSyntaxContext.Node;
// 从语法转换为语义，用于后续判断是否标记了特性
INamedTypeSymbol? assemblyClassTypeSymbol =
generatorSyntaxContext.SemanticModel.GetDeclaredSymbol(classDeclarationSyntax, token);
if (assemblyClassTypeSymbol is not null && !assemblyClassTypeSymbol.IsAbstract)
{
return assemblyClassTypeSymbol;
}
return null;
})
return null;
})
.Where(t => t != null)
.Select((t, _) => t!)
.Combine(returnTypeCollectionIncrementalValuesProvider)
Expand All @@ -236,7 +242,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
if (exportMethodReturnTypeCollectionResult.IsMatch(assemblyClassTypeSymbol))
{
result.Add(new CandidateClassTypeResult(exportMethodReturnTypeCollectionResult,
new[] { assemblyClassTypeSymbol}));
new[] { assemblyClassTypeSymbol }));
}
}
Expand All @@ -249,25 +255,197 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
return null;
}
})
.Where(t => t is not null);
.Where(t => t is not null)
.Select((t, _) => t!);

var collectionResultIncrementalValueProvider = referenceAssemblyTypeIncrementalValueProvider.Combine(candidateClassCollectionResultIncrementalValuesProvider.Collect())
.SelectMany((tuple, _) => { return tuple.Right.Add(tuple.Left); })
.Collect()
.Select((array, _) =>
{
// 去重
return array.Distinct().ToList();
var dictionary = new Dictionary<ExportMethodReturnTypeCollectionResult, List<INamedTypeSymbol>>();
foreach (var candidateClassCollectionResult in array)
{
foreach (var candidateClassTypeResult in candidateClassCollectionResult.CandidateClassTypeResultList)
{
var result = candidateClassTypeResult.ExportMethodReturnTypeCollectionResult;
if (!dictionary.TryGetValue(result, out var list))
{
list = new List<INamedTypeSymbol>();
dictionary.Add(result, list);
}
list.AddRange(candidateClassTypeResult.AssemblyClassTypeSymbolList);
}
}
return dictionary;
});

// 可以被 IDE 选择不生成的代码，但是在完全生成输出时将会跑
// 这里可以用来存放具体实现的代码，将不影响用户代码的语义，而不是用来做定义的代码
context.RegisterImplementationSourceOutput(collectionResultIncrementalValueProvider,
(productionContext, result) =>
{
foreach (var candidateClassCollectionResult in result)
foreach (var item in result)
{
/*
private static partial IEnumerable<(Type, F1Attribute xx, Func<DemoLib1.F1> xxx)> ExportFooEnumerable()
{
yield return (typeof(CurrentFoo), new F1Attribute(), () => new CurrentFoo());
}
*/
var exportMethodReturnTypeCollectionResult = item.Key;
var list = item.Value;
var methodSource = new StringBuilder();
if (exportMethodReturnTypeCollectionResult.ExportMethodReturnTypeInfo is ValueTupleExportMethodReturnTypeInfo valueTupleExportMethodReturnTypeInfo)
{
var exportPartialMethodSymbol = exportMethodReturnTypeCollectionResult.ExportPartialMethodSymbol;
var accessibilityCode =
exportPartialMethodSymbol.DeclaredAccessibility switch
{
Accessibility.Public=>"public",
Accessibility.Private=>"private",
Accessibility.Internal=>"internal",
Accessibility.Protected=>"protected",
Accessibility.ProtectedAndInternal=>"private protected",
_ => string.Empty
};
methodSource.Append(accessibilityCode).Append(' ');
if (exportPartialMethodSymbol.IsStatic)
{
methodSource.Append("static ");
}
methodSource.Append("partial ");
if (!valueTupleExportMethodReturnTypeInfo.IsIEnumerable)
{
// 还没支持其他返回值的情况
throw new NotSupportedException();
}
methodSource.Append("global::System.Collections.Generic.IEnumerable<");
methodSource.Append('(');
var valueTupleInfo = valueTupleExportMethodReturnTypeInfo.ValueTupleInfo;
for (var i = 0; i < valueTupleInfo.ItemList.Count; i++)
{
var info = valueTupleInfo.ItemList[i];
if (i != valueTupleInfo.ItemList.Count - 1)
{
var type = TypeSymbolHelper.TypeSymbolToFullName(info.ItemType);
methodSource.Append(type).Append(' ');
methodSource.Append(info.ItemName);
methodSource.Append(',');
}
else
{
var type = TypeSymbolHelper.TypeSymbolToFullName(exportMethodReturnTypeCollectionResult
.ExpectedClassBaseType);
methodSource.Append($"global::System.Func<{type}> {info.ItemName}");
}
}
methodSource.Append(')');
methodSource.Append('>');
methodSource.Append(' ');
methodSource.Append(exportPartialMethodSymbol.Name);
methodSource.AppendLine("()");
methodSource.AppendLine("{");
foreach (var namedTypeSymbol in list)
{
// yield return (typeof(CurrentFoo), new F1Attribute(), () => new CurrentFoo());
var attribute = namedTypeSymbol.GetAttributes().First(t =>
SymbolEqualityComparer.Default.Equals(t.AttributeClass,
exportMethodReturnTypeCollectionResult
.ExpectedClassAttributeType));
var attributeCreatedCode = AttributeCodeReWriter.GetAttributeCreatedCode(attribute);
var typeName = TypeSymbolHelper.TypeSymbolToFullName(namedTypeSymbol);
methodSource.Append($"yield return (typeof({typeName}), {attributeCreatedCode}, () => new {typeName}());");
}
methodSource.AppendLine("}");
}
var source = methodSource.ToString();
var partialClassType = (INamedTypeSymbol) exportMethodReturnTypeCollectionResult.ExportPartialMethodSymbol.ReceiverType!;
var symbolDisplayFormat = new SymbolDisplayFormat
(
// 带上命名空间和类型名
SymbolDisplayGlobalNamespaceStyle.Omitted,
// 命名空间之前加上 global 防止冲突
SymbolDisplayTypeQualificationStyle
.NameAndContainingTypesAndNamespaces
);
var @namespace = partialClassType.ContainingNamespace?.ToDisplayString(symbolDisplayFormat);
if (TryGetClassDeclarationList(partialClassType, out var declarationList))
{
int declarationCount = declarationList!.Count;
/* 以下代码用来解决嵌套类型
for (int i = 0; i < declarationCount - 1; i++)
{
string declarationSource = $@"
{declarationList[declarationCount - 1 - i]}
{{";
sb.Append($@"
{IndentSource(declarationSource, numIndentations: i + 1)}
");
}
*/
var isIncludeNamespace = !string.IsNullOrEmpty(@namespace);
var stringBuilder = new StringBuilder();
stringBuilder.AppendLine("// <auto-generated/>");
if (isIncludeNamespace)
{
stringBuilder.Append(@$"
namespace {@namespace}
{{");
}
var generatedCodeAttributeSource =
"[global::System.CodeDom.Compiler.GeneratedCode(\"Telescope\", \"1.0.0\")]";
// Add the core implementation for the derived context class.
string partialContextImplementation = $@"
{generatedCodeAttributeSource}{declarationList[0]}
{{
{IndentSource(source, Math.Max(1, declarationCount - 1))}
}}";
stringBuilder.AppendLine(IndentSource(partialContextImplementation, numIndentations: declarationCount));
if (!isIncludeNamespace)
{
stringBuilder.AppendLine("}");
}
var fileName = $"{partialClassType.Name}-{exportMethodReturnTypeCollectionResult.ExportPartialMethodSymbol.Name}";
productionContext.AddSource(fileName,stringBuilder.ToString());
//#if DEBUG
// var debugFolder = @"F:\temp";
// if (Directory.Exists(debugFolder))
// {
// var debugFile = Path.Combine(debugFolder, fileName);
// File.WriteAllText(debugFile, stringBuilder.ToString());
// }
//#endif
}
}
});

Expand All @@ -284,6 +462,64 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
});
}



private static string IndentSource(string source, int numIndentations)
{
Debug.Assert(numIndentations >= 1);
return source.Replace("\r", "").Replace("\n", $"\r\n{new string(' ', 4 * numIndentations)}");
//return source.Replace(Environment.NewLine, $"{Environment.NewLine}{new string(' ', 4 * numIndentations)}"); // 4 spaces per indentation.
}

private static bool TryGetClassDeclarationList(INamedTypeSymbol typeSymbol, out List<string>? classDeclarationList)
{
INamedTypeSymbol currentSymbol = typeSymbol;
classDeclarationList = null;

while (currentSymbol != null)
{
ClassDeclarationSyntax? classDeclarationSyntax = currentSymbol.DeclaringSyntaxReferences.First().GetSyntax() as ClassDeclarationSyntax;

if (classDeclarationSyntax != null)
{
SyntaxTokenList tokenList = classDeclarationSyntax.Modifiers;
int tokenCount = tokenList.Count;

bool isPartial = false;

string[] declarationElements = new string[tokenCount + 2];

for (int i = 0; i < tokenCount; i++)
{
SyntaxToken token = tokenList[i];
declarationElements[i] = token.Text;

if (token.IsKind(SyntaxKind.PartialKeyword))
{
isPartial = true;
}
}

if (!isPartial)
{
classDeclarationList = null;
return false;
}

declarationElements[tokenCount] = "class";
declarationElements[tokenCount + 1] = currentSymbol.Name;

(classDeclarationList ??= new List<string>()).Add(string.Join(" ", declarationElements));
}

currentSymbol = currentSymbol.ContainingType;
}

Debug.Assert(classDeclarationList != null);
Debug.Assert(classDeclarationList!.Count > 0);
return true;
}

class ExportTypeCollectionResult : IEquatable<ExportTypeCollectionResult>
{
public ExportTypeCollectionResult(IMethodSymbol methodSymbol, GeneratorSyntaxContext generatorSyntaxContext)
Expand Down Expand Up @@ -377,6 +613,12 @@ public ExportMethodReturnTypeCollectionResult(ITypeSymbol expectedClassBaseType,
/// <returns></returns>
public bool IsMatch(INamedTypeSymbol assemblyClassTypeSymbol)
{
if (assemblyClassTypeSymbol.IsAbstract)
{
// 抽象类不能提供
return false;
}

// 先判断是否继承，再判断是否标记特性
if (!TypeSymbolHelper.IsInherit(assemblyClassTypeSymbol, ExpectedClassBaseType))
{
Expand Down
6 changes: 6 additions & 0 deletions ...copeSourceGenerator/Analyzers/ExportTypeToMethod_/ValueTupleExportMethodReturnTypeInfo.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,4 +13,10 @@ public ValueTupleExportMethodReturnTypeInfo(ValueTupleInfo valueTupleInfo)
}

public ValueTupleInfo ValueTupleInfo { get; }

/// <summary>
/// 是否采用 global::System.Collections.Generic.IEnumerable 返回值
/// <code>IEnumerable&lt;(Type, F1Attribute xx, Func&lt;DemoLib1.F1&gt; xxx)&gt;</code>
/// </summary>
public bool IsIEnumerable { set; get; }
}

0 comments on commit a35a4d0

Please sign in to comment.