mirror of
https://github.com/AlexMacocian/Net.Sdk.Web.Extensions.git
synced 2026-07-15 13:39:29 +00:00
44 lines
2.1 KiB
C#
44 lines
2.1 KiB
C#
using Microsoft.CodeAnalysis;
|
|
using Sybil;
|
|
using System;
|
|
|
|
namespace Net.Sdk.Web.Extensions.SourceGenerators;
|
|
|
|
[Generator(LanguageNames.CSharp)]
|
|
#nullable enable
|
|
public class ControllerAttributeGenerator : IIncrementalGenerator
|
|
{
|
|
public void Initialize(IncrementalGeneratorInitializationContext context)
|
|
{
|
|
context.RegisterPostInitializationOutput(GenerateAttribute);
|
|
}
|
|
|
|
private static void GenerateAttribute(IncrementalGeneratorPostInitializationContext context)
|
|
{
|
|
var builder = SyntaxBuilder.CreateCompilationUnit()
|
|
.WithNamespace(SyntaxBuilder.CreateNamespace(Constants.Namespace)
|
|
.WithClass(SyntaxBuilder.CreateClass(Constants.RouteAttributeName)
|
|
.WithBaseClass(nameof(Attribute))
|
|
.WithModifiers($"{Constants.Public} {Constants.Sealed}")
|
|
.WithConstructor(SyntaxBuilder.CreateConstructor(Constants.RouteAttributeName)
|
|
.WithModifier(Constants.Public)
|
|
.WithBody($"this.{Constants.PatternPropertyName} = string.Empty;"))
|
|
.WithConstructor(SyntaxBuilder.CreateConstructor(Constants.RouteAttributeName)
|
|
.WithModifier(Constants.Public)
|
|
.WithParameter(Constants.StringType, Constants.PatternArgumentName)
|
|
.WithBody($"this.{Constants.PatternPropertyName} = {Constants.PatternArgumentName};"))
|
|
.WithAttribute(SyntaxBuilder.CreateAttribute("AttributeUsage")
|
|
.WithArgument(AttributeTargets.Class)
|
|
.WithArgument("Inherited", false)
|
|
.WithArgument("AllowMultiple", false))
|
|
.WithProperty(SyntaxBuilder.CreateProperty(Constants.StringType, Constants.PatternPropertyName)
|
|
.WithModifier(Constants.Public)
|
|
.WithAccessor(SyntaxBuilder.CreateGetter()))));
|
|
|
|
var syntax = builder.Build();
|
|
var source = syntax.ToFullString();
|
|
context.AddSource($"{Constants.RouteAttributeName}.g", source);
|
|
}
|
|
}
|
|
|
|
#nullable disable |