micialware/mROA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

extract TemplateDocument loading logic

Browse Source
This commit is contained in:
Ivan Ansimov
2025-07-14 20:01:40 +03:00
parent 26cdf86bd6
commit abc3bda88e
2 changed files with 93 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mROA.Codegen/CodeTemplate.cs
+49
View File
@@ -0,0 +1,49 @@
using System;
using mROA.CodegenTools;
using mROA.CodegenTools.Reading;
namespace mROA.Codegen
{
public class CodeTemplate
{
private const string MethodRepoTemplateName = "MethodRepo.cstmpl";
private const string ProxyTemplateName = "Proxy.cstmpl";
private const string RemoteTypeBinderTemplateName = "RemoteTypeBinder.cstmpl";
private const string PartialInterfaceTemplateName = "PartialInterface.cstmpl";
private const string IndexProviderTemplateName = "IndexProvider.cstmpl";
private const string SyncInvokerTag = "syncInvoker";
private TemplateDocument? _proxy;
private TemplateDocument? _indexerProvider;
private TemplateDocument? _remoteTypeBinder;
private TemplateDocument? _partialInterface;
private TemplateDocument? _methodRepo;
private TemplateDocument? _methodInvoker;
public TemplateDocument Proxy => _proxy
?? throw new InvalidOperationException();
public TemplateDocument IndexerProvider => _indexerProvider
?? throw new InvalidOperationException();
public TemplateDocument RemoteTypeBinder => _remoteTypeBinder
?? throw new InvalidOperationException();
public TemplateDocument PartialInterface => _partialInterface
?? throw new InvalidOperationException();
public TemplateDocument MethodRepo => _methodRepo
?? throw new InvalidOperationException();
public TemplateDocument MethodInvoker => _methodInvoker
?? throw new InvalidOperationException();
public void LoadTemplates()
{
_proxy = TemplateReader.FromEmbeddedResource(ProxyTemplateName);
_indexerProvider = TemplateReader.FromEmbeddedResource(IndexProviderTemplateName);
_remoteTypeBinder = TemplateReader.FromEmbeddedResource(RemoteTypeBinderTemplateName);
_partialInterface = TemplateReader.FromEmbeddedResource(PartialInterfaceTemplateName);
_methodRepo = TemplateReader.FromEmbeddedResource(MethodRepoTemplateName);
var innerMethodSection = MethodRepo[SyncInvokerTag] as InnerTemplateSection
?? throw new InvalidOperationException();
_methodInvoker = innerMethodSection.InnerTemplate;
}
}
}
mROA.Codegen/mROASourceGenerator.cs
+44 -58
View File
@@ -9,7 +9,6 @@ using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CSharp.Syntax; using Microsoft.CodeAnalysis.CSharp.Syntax;
using Microsoft.CodeAnalysis.Text; using Microsoft.CodeAnalysis.Text;
using mROA.CodegenTools; using mROA.CodegenTools;
using mROA.CodegenTools.Reading;
#pragma warning disable CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider adding the 'required' modifier or declaring as nullable. #pragma warning disable CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider adding the 'required' modifier or declaring as nullable.
@@ -30,26 +29,13 @@ namespace mROA.Codegen
private static readonly Predicate<ITypeSymbol> ParameterFilterForType = private static readonly Predicate<ITypeSymbol> ParameterFilterForType =
i => i.Name is "CancellationToken" or "RequestContext"; i => i.Name is "CancellationToken" or "RequestContext";
private TemplateDocument _indexerTemplate; private readonly CodeTemplate _codeTemplate = new();
private TemplateDocument _binderTemplate;
private TemplateDocument _classTemplate;
private TemplateDocument _classTemplateOriginal;
private TemplateDocument _interfaceTemplate;
private TemplateDocument _interfaceTemplateOriginal;
private TemplateDocument _methodInvokerOriginal;
private TemplateDocument _methodRepoTemplate;
private int _currentInternalCallIndex; private int _currentInternalCallIndex;
public void Initialize(IncrementalGeneratorInitializationContext context) public void Initialize(IncrementalGeneratorInitializationContext context)
{ {
_methodRepoTemplate = TemplateReader.FromEmbeddedResource("MethodRepo.cstmpl"); _codeTemplate.LoadTemplates();
_methodInvokerOriginal =
((InnerTemplateSection)_methodRepoTemplate["syncInvoker"]!).InnerTemplate;
_classTemplateOriginal = TemplateReader.FromEmbeddedResource("Proxy.cstmpl");
_binderTemplate = TemplateReader.FromEmbeddedResource("RemoteTypeBinder.cstmpl");
_interfaceTemplateOriginal = TemplateReader.FromEmbeddedResource("PartialInterface.cstmpl");
_indexerTemplate = TemplateReader.FromEmbeddedResource("IndexProvider.cstmpl");
var syntaxes = context.SyntaxProvider.CreateSyntaxProvider( var syntaxes = context.SyntaxProvider.CreateSyntaxProvider(
(static (node, _) => node is InterfaceDeclarationSyntax), (static (node, _) => node is InterfaceDeclarationSyntax),
@@ -65,7 +51,7 @@ namespace mROA.Codegen
{ {
var totalMethods = new List<IMethodSymbol>(); var totalMethods = new List<IMethodSymbol>();
_indexerTemplate.AddDefine("namespace", compilation.AssemblyName!); _codeTemplate.IndexerProvider.AddDefine("namespace", compilation.AssemblyName!);
var invokers = new List<string>(); var invokers = new List<string>();
var declarations = classes.ToList(); var declarations = classes.ToList();
var apiLevel = 0; var apiLevel = 0;
@@ -93,7 +79,7 @@ namespace mROA.Codegen
className = className.TrimStart('I') + "Proxy"; className = className.TrimStart('I') + "Proxy";
_classTemplate = (TemplateDocument)_classTemplateOriginal.Clone(); var classTemplate = (TemplateDocument)_codeTemplate.Proxy.Clone();
var propertiesAccessMethods = new List<(string, IMethodSymbol)>(); var propertiesAccessMethods = new List<(string, IMethodSymbol)>();
int startInvokers = invokers.Count; int startInvokers = invokers.Count;
@@ -105,7 +91,7 @@ namespace mROA.Codegen
classSymbol); classSymbol);
continue; continue;
default: default:
GenerateDeclaredMethod(method, invokers, classSymbol); GenerateDeclaredMethod(classTemplate, method, invokers, classSymbol);
break; break;
} }
@@ -126,43 +112,43 @@ namespace mROA.Codegen
impl = impl =
$"public {propertySymbol.Type.ToUnityString()} {symbol.Name} {{ {getter.Item1} {setter.Item1} }}"; $"public {propertySymbol.Type.ToUnityString()} {symbol.Name} {{ {getter.Item1} {setter.Item1} }}";
_classTemplate.Insert("methods", impl); classTemplate.Insert("methods", impl);
// declaredMethods.Add(impl); // declaredMethods.Add(impl);
break; break;
case IEventSymbol eventSymbol: case IEventSymbol eventSymbol:
_classTemplate.Insert("methods", classTemplate.Insert("methods",
$"public event {eventSymbol.Type.ToUnityString()}? {eventSymbol.Name};"); $"public event {eventSymbol.Type.ToUnityString()}? {eventSymbol.Name};");
// declaredMethods.Add( // declaredMethods.Add(
// $"public event {eventSymbol.Type.ToDisplayString()}? {eventSymbol.Name};"); // $"public event {eventSymbol.Type.ToDisplayString()}? {eventSymbol.Name};");
break; break;
} }
GenerateEventImplementation(classSymbol, invokers, context); GenerateEventImplementation(classTemplate, classSymbol, invokers, context);
var endInvokers = invokers.Count; var endInvokers = invokers.Count;
_indexerTemplate.Insert("indexSpan", _codeTemplate.IndexerProvider.Insert("indexSpan",
$"{{ typeof({originalName}), new[] {{ {CodegenUtilities.JoinWithComa(Enumerable.Range(startInvokers, endInvokers - startInvokers).Select(i => i.ToString()))} }} }},"); $"{{ typeof({originalName}), new[] {{ {CodegenUtilities.JoinWithComa(Enumerable.Range(startInvokers, endInvokers - startInvokers).Select(i => i.ToString()))} }} }},");
_classTemplate.AddDefine("className", className); classTemplate.AddDefine("className", className);
_classTemplate.AddDefine("originalName", originalName); classTemplate.AddDefine("originalName", originalName);
_classTemplate.AddDefine("namespaceName", namespaceName); classTemplate.AddDefine("namespaceName", namespaceName);
var code = _classTemplate.Compile(); var code = classTemplate.Compile();
// Add the source code to the compilation. // Add the source code to the compilation.
#if !DONT_ADD #if !DONT_ADD
context.AddSource($"{className}.g.cs", SourceText.From(code, Encoding.UTF8)); context.AddSource($"{className}.g.cs", SourceText.From(code, Encoding.UTF8));
#endif #endif
_indexerTemplate.Insert("remoteTypePair", _codeTemplate.IndexerProvider.Insert("remoteTypePair",
$"{{ typeof({classSymbol.ToUnityString()}), (id, r, c, indices) => new {namespaceName}.{className}(id, r, c, indices) }}"); $"{{ typeof({classSymbol.ToUnityString()}), (id, r, c, indices) => new {namespaceName}.{className}(id, r, c, indices) }}");
} }
if (totalMethods.Count != 0) if (totalMethods.Count != 0)
{ {
var coCodegenRepoCode = _methodRepoTemplate.Compile(); var coCodegenRepoCode = _codeTemplate.MethodRepo.Compile();
_indexerTemplate.AddDefine("level", apiLevel.ToString()); _codeTemplate.IndexerProvider.AddDefine("level", apiLevel.ToString());
_indexerTemplate.AddDefine("len", invokers.Count.ToString()); _codeTemplate.IndexerProvider.AddDefine("len", invokers.Count.ToString());
var providerCode = _indexerTemplate.Compile(); var providerCode = _codeTemplate.IndexerProvider.Compile();
#if !DONT_ADD #if !DONT_ADD
context.AddSource("GeneratedInvokersCollection.g.cs", context.AddSource("GeneratedInvokersCollection.g.cs",
SourceText.From(coCodegenRepoCode, Encoding.UTF8)); SourceText.From(coCodegenRepoCode, Encoding.UTF8));
@@ -171,51 +157,51 @@ namespace mROA.Codegen
#endif #endif
} }
if (_indexerTemplate["remoteTypePair+"] != null) if (_codeTemplate.IndexerProvider["remoteTypePair+"] != null)
{ {
var frontendRepoCode = _binderTemplate.Compile(); var frontendRepoCode = _codeTemplate.RemoteTypeBinder.Compile();
#if !DONT_ADD #if !DONT_ADD
context.AddSource("RemoteTypeBinder.g.cs", SourceText.From(frontendRepoCode, Encoding.UTF8)); context.AddSource("RemoteTypeBinder.g.cs", SourceText.From(frontendRepoCode, Encoding.UTF8));
#endif #endif
} }
} }
private void GenerateEventImplementation(INamedTypeSymbol classSymbol, List<string> invokers, private void GenerateEventImplementation(TemplateDocument classTemplate,INamedTypeSymbol classSymbol, List<string> invokers,
SourceProductionContext context) SourceProductionContext context)
{ {
var events = classSymbol.AllInterfaces.Add(classSymbol).SelectMany(i => i.GetMembers()) var events = classSymbol.AllInterfaces.Add(classSymbol).SelectMany(i => i.GetMembers())
.OfType<IEventSymbol>().ToList(); .OfType<IEventSymbol>().ToList();
if (events.Count == 0) if (events.Count == 0)
return; return;
_interfaceTemplate = (TemplateDocument)_interfaceTemplateOriginal.Clone(); var interfaceTemplate = (TemplateDocument)_codeTemplate.PartialInterface.Clone();
_interfaceTemplate.AddDefine("name", classSymbol.Name); interfaceTemplate.AddDefine("name", classSymbol.Name);
_interfaceTemplate.AddDefine("namespace", classSymbol.ContainingNamespace.ToDisplayString()); interfaceTemplate.AddDefine("namespace", classSymbol.ContainingNamespace.ToDisplayString());
var objectBinderTemplate = var objectBinderTemplate =
(TemplateDocument)((InnerTemplateSection)_binderTemplate["objectBinderTemplate"]!).InnerTemplate (TemplateDocument)((InnerTemplateSection)_codeTemplate.RemoteTypeBinder["objectBinderTemplate"]!).InnerTemplate
.Clone(); .Clone();
foreach (var currentEvent in events) foreach (var currentEvent in events)
{ {
var additionalMethod = GenerateMethodExternalCaller(currentEvent, out var signature); var additionalMethod = GenerateMethodExternalCaller(currentEvent, out var signature);
_classTemplate.Insert("methods", additionalMethod); classTemplate.Insert("methods", additionalMethod);
// declaredMethods.Add(additionalMethod); // declaredMethods.Add(additionalMethod);
_interfaceTemplate.Insert("signature", signature); interfaceTemplate.Insert("signature", signature);
GenerateEventCode(currentEvent, invokers, classSymbol); GenerateEventCode(currentEvent, invokers, classSymbol);
GenerateBinderCode(currentEvent, classSymbol, objectBinderTemplate); GenerateBinderCode(currentEvent, classSymbol, objectBinderTemplate);
} }
objectBinderTemplate.AddDefine("type", classSymbol.ToUnityString()); objectBinderTemplate.AddDefine("type", classSymbol.ToUnityString());
var partialInterface = _interfaceTemplate.Compile(); var partialInterface = interfaceTemplate.Compile();
var binder = objectBinderTemplate.Compile(); var binder = objectBinderTemplate.Compile();
_binderTemplate.Insert("eventBinder", binder); _codeTemplate.RemoteTypeBinder.Insert("eventBinder", binder);
#if !DONT_ADD #if !DONT_ADD
context.AddSource($"{classSymbol.Name}.g.cs", SourceText.From(partialInterface, Encoding.UTF8)); context.AddSource($"{classSymbol.Name}.g.cs", SourceText.From(partialInterface, Encoding.UTF8));
#endif #endif
} }
private string GenerateMethodExternalCaller(IEventSymbol eventSymbol, out string interfaceSignature) private static string GenerateMethodExternalCaller(IEventSymbol eventSymbol, out string interfaceSignature)
{ {
const string level = "\t\t"; const string level = "\t\t";
var parameters = ((INamedTypeSymbol)eventSymbol.Type).TypeArguments; var parameters = ((INamedTypeSymbol)eventSymbol.Type).TypeArguments;
@@ -232,7 +218,7 @@ namespace mROA.Codegen
return caller; return caller;
} }
private void GenerateDeclaredMethod(IMethodSymbol method, List<string> invokers, INamedTypeSymbol baseInterface) private void GenerateDeclaredMethod(TemplateDocument classTemplate, IMethodSymbol method, List<string> invokers, INamedTypeSymbol baseInterface)
{ {
var sb = new StringBuilder(); var sb = new StringBuilder();
@@ -303,7 +289,7 @@ namespace mROA.Codegen
sb.AppendLine("\t\t\t" + prefix + caller + postfix + ";"); sb.AppendLine("\t\t\t" + prefix + caller + postfix + ";");
sb.AppendLine("\t\t}"); sb.AppendLine("\t\t}");
_classTemplate.Insert("methods", sb.ToString()); classTemplate.Insert("methods", sb.ToString());
// declaredMethods.Add(sb.ToString()); // declaredMethods.Add(sb.ToString());
var parameterTypes = string.Join(", ", var parameterTypes = string.Join(", ",
@@ -349,7 +335,7 @@ namespace mROA.Codegen
if (isAsync) if (isAsync)
{ {
var invokerTemplate = var invokerTemplate =
(TemplateDocument)((InnerTemplateSection)_methodRepoTemplate["asyncInvoker"]!).InnerTemplate (TemplateDocument)((InnerTemplateSection)_codeTemplate.MethodRepo["asyncInvoker"]!).InnerTemplate
.Clone(); .Clone();
invokerTemplate.AddDefine("isVoid", isVoid.ToString().ToLower()); invokerTemplate.AddDefine("isVoid", isVoid.ToString().ToLower());
invokerTemplate.AddDefine("returnType", CodegenUtilities.ExtractTaskType(method.ReturnType)); invokerTemplate.AddDefine("returnType", CodegenUtilities.ExtractTaskType(method.ReturnType));
@@ -361,7 +347,7 @@ namespace mROA.Codegen
} }
else else
{ {
var invokerTemplate = (TemplateDocument)_methodInvokerOriginal.Clone(); var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone();
invokerTemplate.AddDefine("isVoid", isVoid.ToString().ToLower()); invokerTemplate.AddDefine("isVoid", isVoid.ToString().ToLower());
invokerTemplate.AddDefine("returnType", isVoid ? "void" : method.ReturnType.ToUnityString()); invokerTemplate.AddDefine("returnType", isVoid ? "void" : method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.AddDefine("parametersType", parameterTypes);
@@ -371,7 +357,7 @@ namespace mROA.Codegen
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
_methodRepoTemplate.Insert("invoker", backend); _codeTemplate.MethodRepo.Insert("invoker", backend);
invokers.Add(backend); invokers.Add(backend);
} }
@@ -445,7 +431,7 @@ namespace mROA.Codegen
{level} return null; {level} return null;
{level} }}"; {level} }}";
var invokerTemplate = (TemplateDocument)_methodInvokerOriginal.Clone(); var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone();
invokerTemplate.AddDefine("isVoid", "true"); invokerTemplate.AddDefine("isVoid", "true");
invokerTemplate.AddDefine("returnType", "void"); invokerTemplate.AddDefine("returnType", "void");
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.AddDefine("parametersType", parameterTypes);
@@ -453,7 +439,7 @@ namespace mROA.Codegen
invokerTemplate.AddDefine("funcInvoking", funcInvoking); invokerTemplate.AddDefine("funcInvoking", funcInvoking);
invokerTemplate.AddDefine("isTrusted", "true"); invokerTemplate.AddDefine("isTrusted", "true");
var backend = invokerTemplate.Compile(); var backend = invokerTemplate.Compile();
_methodRepoTemplate.Insert("invoker", backend); _codeTemplate.MethodRepo.Insert("invoker", backend);
invokers.Add(backend); invokers.Add(backend);
} }
@@ -477,7 +463,7 @@ namespace mROA.Codegen
method.Parameters.Select(p => method.Parameters.Select(p =>
CodegenUtilities.Caster(p.Type, "parameters[" + method.Parameters.IndexOf(p) + "]"))); CodegenUtilities.Caster(p.Type, "parameters[" + method.Parameters.IndexOf(p) + "]")));
var invokerTemplate = (TemplateDocument)_methodInvokerOriginal.Clone(); var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone();
invokerTemplate.AddDefine("isVoid", "false"); invokerTemplate.AddDefine("isVoid", "false");
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.AddDefine("parametersType", parameterTypes);
@@ -490,7 +476,7 @@ namespace mROA.Codegen
} }
else else
{ {
var invokerTemplate = (TemplateDocument)_methodInvokerOriginal.Clone(); var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone();
invokerTemplate.AddDefine("isVoid", "false"); invokerTemplate.AddDefine("isVoid", "false");
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("suitableType", baseInterace.ToUnityString()); invokerTemplate.AddDefine("suitableType", baseInterace.ToUnityString());
@@ -519,7 +505,7 @@ namespace mROA.Codegen
var valueInsert = CodegenUtilities.Caster(method.Parameters.Last().Type, var valueInsert = CodegenUtilities.Caster(method.Parameters.Last().Type,
"parameters[" + (method.Parameters.Length - 1) + "]"); "parameters[" + (method.Parameters.Length - 1) + "]");
var invokerTemplate = (TemplateDocument)_methodInvokerOriginal.Clone(); var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone();
invokerTemplate.AddDefine("isVoid", "true"); invokerTemplate.AddDefine("isVoid", "true");
@@ -534,7 +520,7 @@ namespace mROA.Codegen
} }
else else
{ {
var invokerTemplate = (TemplateDocument)_methodInvokerOriginal.Clone(); var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone();
invokerTemplate.AddDefine("isVoid", "true"); invokerTemplate.AddDefine("isVoid", "true");
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString());
@@ -553,7 +539,7 @@ namespace mROA.Codegen
} }
propsCollection.Add((frontend, method)); propsCollection.Add((frontend, method));
_methodRepoTemplate.Insert("invoker", backend); _codeTemplate.MethodRepo.Insert("invoker", backend);
invokers.Add(backend); invokers.Add(backend);
} }
} }