-
Notifications
You must be signed in to change notification settings - Fork 0
/
ClassGenerator.cs
49 lines (42 loc) · 1.53 KB
/
ClassGenerator.cs
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
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Microsoft.CodeAnalysis;
using System;
using System.Collections.Generic;
using System.Linq;
namespace SourceGenerator
{
[Generator]
public class ClassGenerator : ISourceGenerator
{
public void Execute(GeneratorExecutionContext context)
{
if (context.SyntaxContextReceiver is not SyntaxContextReceiver scr)
{
return;
}
foreach (var (classSyntax, classSymbol) in scr.GenClasses)
{
var attributeName = "TestAttAttribute";
var attribute = classSymbol.GetAttributes().FirstOrDefault(a => a.AttributeClass.Name == attributeName);
var attributeArgument = attribute?.ConstructorArguments[0].Value;
}
}
public void Initialize(GeneratorInitializationContext context)
{
context.RegisterForSyntaxNotifications(() => new SyntaxContextReceiver());
}
}
internal class SyntaxContextReceiver : ISyntaxContextReceiver
{
public IList<(ClassDeclarationSyntax, ITypeSymbol)> GenClasses { get; } = new List<(ClassDeclarationSyntax, ITypeSymbol)>();
public void OnVisitSyntaxNode(GeneratorSyntaxContext context)
{
if (context.Node is not ClassDeclarationSyntax cds)
{
return;
}
var classSymbol = context.SemanticModel.GetDeclaredSymbol(cds) as ITypeSymbol;
GenClasses.Add((cds, classSymbol));
}
}
}