encapsulate template logic

This commit is contained in:
Ivan Ansimov
2025-07-15 16:37:06 +03:00
parent ec4c008d4c
commit 1cde4b0ed0
12 changed files with 453 additions and 171 deletions
-49
View File
@@ -1,49 +0,0 @@
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;
}
}
}
@@ -0,0 +1,16 @@
using System;
using mROA.CodegenTools;
namespace mROA.Codegen
{
public static class TemplateDocumentExtensions
{
public static TemplateDocument CloneInnerTemplate(this TemplateDocument template, string tag)
{
var innerTemplate = template[tag] as InnerTemplateSection
?? throw new InvalidOperationException();
var clone = innerTemplate.InnerTemplate.Clone();
return (TemplateDocument)clone;
}
}
}
@@ -0,0 +1,46 @@
using mROA.CodegenTools;
namespace mROA.Codegen.Templates
{
public class EventBinderTemplate : TemplateView
{
private const string CallFilterTag = "callFilter";
private const string TypeTag = "type";
private const string EventNameTag = "eventName";
private const string ParametersDeclarationTag = "parametersDeclaration";
private const string CommandIdTag = "commandId";
private const string TransferParametersTag = "transferParameters";
public EventBinderTemplate(TemplateDocument template) : base(template) { }
public void DefineCallFilter(string value)
{
Define(CallFilterTag, value);
}
public void DefineType(string value)
{
Define(TypeTag, value);
}
public void DefineEventName(string value)
{
Define(EventNameTag, value);
}
public void DefineParametersDeclaration(string value)
{
Define(ParametersDeclarationTag, value);
}
public void DefineCommandIdTag(string value)
{
Define(CommandIdTag, value);
}
public void DefineTransferParameters(string value)
{
Define(TransferParametersTag, value);
}
}
}
@@ -0,0 +1,45 @@
namespace mROA.Codegen.Templates
{
public class IndexProviderTemplate : TemplateView
{
private const string TemplateFile = "IndexProvider.cstmpl";
private const string NamespaceTag = "namespace";
private const string LevelTag = "level";
private const string LenTag = "len";
private const string IndexSpanTag = "indexSpan";
private const string RemoteTypePairTag = "remoteTypePair";
public IndexProviderTemplate() : base(TemplateFile) { }
public void DefineNamespace(string value)
{
Define(NamespaceTag, value);
}
public void DefineLevel(string value)
{
Define(LevelTag, value);
}
public void DefineLen(string value)
{
Define(LenTag, value);
}
public void InsertIndexSpan(string value)
{
Insert(IndexSpanTag, value);
}
public void InsertRemoteTypePair(string value)
{
Insert(RemoteTypePairTag, value);
}
public bool IsRemoteTypePairInserted()
{
return Inserted(RemoteTypePairTag);
}
}
}
+46
View File
@@ -0,0 +1,46 @@
using mROA.CodegenTools;
namespace mROA.Codegen.Templates
{
public class InvokerTemplate : TemplateView
{
private const string IsVoidTag = "isVoid";
private const string ReturnTypeTag = "returnType";
private const string ParametersTypeTag = "parametersType";
private const string SuitableTypeTag = "suitableType";
private const string FuncInvokingTag = "funcInvoking";
private const string IsTrustedTag = "isTrusted";
public InvokerTemplate(TemplateDocument template) : base(template) { }
public void DefineIsVoid(string value)
{
Define(IsVoidTag, value);
}
public void DefineReturnType(string value)
{
Define(ReturnTypeTag, value);
}
public void DefineParametersType(string value)
{
Define(ParametersTypeTag, value);
}
public void DefineSuitableType(string value)
{
Define(SuitableTypeTag, value);
}
public void DefineFuncInvoking(string value)
{
Define(FuncInvokingTag, value);
}
public void DefineIsTrusted(string value)
{
Define(IsTrustedTag, value);
}
}
}
@@ -0,0 +1,31 @@
namespace mROA.Codegen.Templates
{
public class MethodRepoTemplate : TemplateView
{
private const string TemplateFile = "MethodRepo.cstmpl";
private const string SyncInvokerTemplate = "syncInvoker";
private const string AsyncInvokerTemplate = "asyncInvoker";
private const string InvokerTag = "invoke";
public MethodRepoTemplate() : base(TemplateFile) { }
public void InsertInvoke(string value)
{
Insert(InvokerTag, value);
}
public InvokerTemplate CloneInnerSyncInvoker()
{
var template = CloneInner(SyncInvokerTemplate);
return new InvokerTemplate(template);
}
public InvokerTemplate CloneInnerAsyncInvoker()
{
var template = CloneInner(AsyncInvokerTemplate);
return new InvokerTemplate(template);
}
}
}
@@ -0,0 +1,30 @@
using mROA.CodegenTools;
namespace mROA.Codegen.Templates
{
public partial class ObjectBinderTemplate : TemplateView
{
private const string EventBinderTemplateTag = "eventBinderTemplate";
private const string TypeTag = "type";
private const string EventBinderTag = "eventBinder";
public ObjectBinderTemplate(TemplateDocument template) : base(template) { }
public void DefineType(string value)
{
Define(TypeTag, value);
}
public void InsertEventBinder(string value)
{
Insert(EventBinderTag, value);
}
public EventBinderTemplate CloneInnerEventBinder()
{
var template = CloneInner(EventBinderTemplateTag);
return new EventBinderTemplate(template);
}
}
}
@@ -0,0 +1,28 @@
namespace mROA.Codegen.Templates
{
public class PartialInterfaceTemplate : TemplateView
{
private const string TemplateFile = "PartialInterface.cstmpl";
private const string NameTag = "name";
private const string NamespaceTag = "namespace";
private const string SignatureTag = "signature";
public PartialInterfaceTemplate() : base(TemplateFile) { }
public void DefineName(string value)
{
Define(NameTag, value);
}
public void DefineNamespace(string value)
{
Define(NamespaceTag, value);
}
public void InsertSignature(string value)
{
Insert(SignatureTag, value);
}
}
}
+34
View File
@@ -0,0 +1,34 @@
namespace mROA.Codegen.Templates
{
public class ProxyTemplate : TemplateView
{
private const string TemplateFile = "Proxy.cstmpl";
private const string ClassNameTag = "className";
private const string OriginalNameTag = "originalName";
private const string NamespaceNameTag= "namespaceName";
private const string MethodsTag = "methods";
public ProxyTemplate() : base(TemplateFile) { }
public void DefineClassName(string value)
{
Define(ClassNameTag, value);
}
public void DefineOriginalName(string value)
{
Define(OriginalNameTag, value);
}
public void DefineNamespaceName(string value)
{
Define(NamespaceNameTag, value);
}
public void InsertMethods(string value)
{
Insert(MethodsTag, value);
}
}
}
@@ -0,0 +1,23 @@
namespace mROA.Codegen.Templates
{
public class RemoteTypeBinderTemplate : TemplateView
{
private const string TemplateFile = "RemoteTypeBinder.cstmpl";
private const string ObjectBinderTemplateTag = "objectBinderTemplate";
private const string EventBinderTag = "eventBinder";
public RemoteTypeBinderTemplate() : base(TemplateFile) { }
public ObjectBinderTemplate CloneInnerObjectBinder()
{
var template = CloneInner(ObjectBinderTemplateTag);
return new ObjectBinderTemplate(template);
}
public void InsertEventBinder(string value)
{
Insert(EventBinderTag, value);
}
}
}
+45
View File
@@ -0,0 +1,45 @@
using mROA.CodegenTools;
using mROA.CodegenTools.Reading;
namespace mROA.Codegen.Templates
{
public abstract class TemplateView
{
private readonly TemplateDocument _template;
protected TemplateView(string templateFile)
{
_template = TemplateReader.FromEmbeddedResource(templateFile);
}
protected TemplateView(TemplateDocument template)
{
_template = template;
}
public string Compile()
{
return _template.Compile();
}
protected void Define(string tag, string value)
{
_template.AddDefine(tag, value);
}
protected void Insert(string tag, string value)
{
_template.Insert(tag, value);
}
protected bool Inserted(string tag)
{
return _template[$"{tag}+"] != null;
}
protected TemplateDocument CloneInner(string tag)
{
return _template.CloneInnerTemplate(tag);
}
}
}
+109 -122
View File
@@ -9,7 +9,7 @@ using System.Threading;
using Microsoft.CodeAnalysis; 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.Codegen.Templates;
#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.
@@ -31,15 +31,11 @@ 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 readonly CodeTemplate _codeTemplate = new();
private int _currentInternalCallIndex; private int _currentInternalCallIndex;
public void Initialize(IncrementalGeneratorInitializationContext context) public void Initialize(IncrementalGeneratorInitializationContext context)
{ {
_codeTemplate.LoadTemplates();
var syntaxes = context.SyntaxProvider.CreateSyntaxProvider( var syntaxes = context.SyntaxProvider.CreateSyntaxProvider(
NodeIsInterfaceWithSharedObjectInterfaceAttribute, NodeIsInterfaceWithSharedObjectInterfaceAttribute,
TransformToInterfaceDeclarationSyntax); TransformToInterfaceDeclarationSyntax);
@@ -69,9 +65,15 @@ namespace mROA.Codegen
private void GenerateCode(SourceProductionContext context, Compilation compilation, private void GenerateCode(SourceProductionContext context, Compilation compilation,
ImmutableArray<InterfaceDeclarationSyntax> classes) ImmutableArray<InterfaceDeclarationSyntax> classes)
{ {
var methodRepoTemplate = new MethodRepoTemplate();
var typeBinder = new RemoteTypeBinderTemplate();
var indexProviderTemplate = new IndexProviderTemplate();
var totalMethods = new List<IMethodSymbol>(); var totalMethods = new List<IMethodSymbol>();
_codeTemplate.IndexerProvider.AddDefine("namespace", compilation.AssemblyName!); if (compilation.AssemblyName == null)
throw new InvalidOperationException();
indexProviderTemplate.DefineNamespace(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;
@@ -99,7 +101,7 @@ namespace mROA.Codegen
className = className.TrimStart('I') + "Proxy"; className = className.TrimStart('I') + "Proxy";
var classTemplate = (TemplateDocument)_codeTemplate.Proxy.Clone(); var proxyTemplate = new ProxyTemplate();
var propertiesAccessMethods = new List<(string, IMethodSymbol)>(); var propertiesAccessMethods = new List<(string, IMethodSymbol)>();
int startInvokers = invokers.Count; int startInvokers = invokers.Count;
@@ -107,11 +109,11 @@ namespace mROA.Codegen
switch (method.MethodKind) switch (method.MethodKind)
{ {
case MethodKind.PropertyGet or MethodKind.PropertySet: case MethodKind.PropertyGet or MethodKind.PropertySet:
GeneratePropertyMethod(method, propertiesAccessMethods, invokers, GeneratePropertyMethod(methodRepoTemplate, method, propertiesAccessMethods, invokers,
classSymbol); classSymbol);
continue; continue;
default: default:
GenerateDeclaredMethod(classTemplate, method, invokers, classSymbol); GenerateDeclaredMethod(proxyTemplate, methodRepoTemplate, method, invokers, classSymbol);
break; break;
} }
@@ -132,43 +134,40 @@ 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); proxyTemplate.InsertMethods(impl);
// declaredMethods.Add(impl); // declaredMethods.Add(impl);
break; break;
case IEventSymbol eventSymbol: case IEventSymbol eventSymbol:
classTemplate.Insert("methods", proxyTemplate.InsertMethods($"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(classTemplate, classSymbol, invokers, context); GenerateEventImplementation(proxyTemplate, methodRepoTemplate, typeBinder, classSymbol, invokers, context);
var endInvokers = invokers.Count; var endInvokers = invokers.Count;
_codeTemplate.IndexerProvider.Insert("indexSpan", indexProviderTemplate.InsertIndexSpan($"{{ 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()))} }} }},"); proxyTemplate.DefineClassName(className);
classTemplate.AddDefine("className", className); proxyTemplate.DefineOriginalName(originalName);
classTemplate.AddDefine("originalName", originalName); proxyTemplate.DefineNamespaceName(namespaceName);
classTemplate.AddDefine("namespaceName", namespaceName);
var code = classTemplate.Compile(); var code = proxyTemplate.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
_codeTemplate.IndexerProvider.Insert("remoteTypePair", indexProviderTemplate.InsertRemoteTypePair($"{{ 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 = _codeTemplate.MethodRepo.Compile(); var coCodegenRepoCode = methodRepoTemplate.Compile();
_codeTemplate.IndexerProvider.AddDefine("level", apiLevel.ToString()); indexProviderTemplate.DefineLevel(apiLevel.ToString());
_codeTemplate.IndexerProvider.AddDefine("len", invokers.Count.ToString()); indexProviderTemplate.DefineLen(invokers.Count.ToString());
var providerCode = _codeTemplate.IndexerProvider.Compile(); var providerCode = indexProviderTemplate.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));
@@ -177,16 +176,16 @@ namespace mROA.Codegen
#endif #endif
} }
if (_codeTemplate.IndexerProvider["remoteTypePair+"] != null) if (indexProviderTemplate.IsRemoteTypePairInserted())
{ {
var frontendRepoCode = _codeTemplate.RemoteTypeBinder.Compile(); var frontendRepoCode = typeBinder.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(TemplateDocument classTemplate, INamedTypeSymbol classSymbol, private void GenerateEventImplementation(ProxyTemplate proxyTemplate, MethodRepoTemplate methodRepoTemplate, RemoteTypeBinderTemplate remoteTypeBinder, INamedTypeSymbol classSymbol,
List<string> invokers, SourceProductionContext context) List<string> invokers, SourceProductionContext context)
{ {
var events = classSymbol.AllInterfaces.Add(classSymbol).SelectMany(i => i.GetMembers()) var events = classSymbol.AllInterfaces.Add(classSymbol).SelectMany(i => i.GetMembers())
@@ -194,28 +193,26 @@ namespace mROA.Codegen
if (events.Count == 0) if (events.Count == 0)
return; return;
var interfaceTemplate = (TemplateDocument)_codeTemplate.PartialInterface.Clone(); var partialInterfaceTemplate = new PartialInterfaceTemplate();
interfaceTemplate.AddDefine("name", classSymbol.Name); partialInterfaceTemplate.DefineName(classSymbol.Name);
interfaceTemplate.AddDefine("namespace", classSymbol.ContainingNamespace.ToDisplayString()); partialInterfaceTemplate.DefineNamespace(classSymbol.ContainingNamespace.ToDisplayString());
var objectBinderTemplate =
(TemplateDocument)((InnerTemplateSection)_codeTemplate.RemoteTypeBinder["objectBinderTemplate"]!) var objectBinderTemplate = remoteTypeBinder.CloneInnerObjectBinder();
.InnerTemplate
.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); proxyTemplate.InsertMethods(additionalMethod);
// declaredMethods.Add(additionalMethod); // declaredMethods.Add(additionalMethod);
interfaceTemplate.Insert("signature", signature); partialInterfaceTemplate.InsertSignature(signature);
GenerateEventCode(currentEvent, invokers, classSymbol); GenerateEventCode(methodRepoTemplate, currentEvent, invokers, classSymbol);
GenerateBinderCode(currentEvent, classSymbol, objectBinderTemplate); GenerateBinderCode(objectBinderTemplate, currentEvent, classSymbol);
} }
objectBinderTemplate.AddDefine("type", classSymbol.ToUnityString()); objectBinderTemplate.DefineType(classSymbol.ToUnityString());
var partialInterface = interfaceTemplate.Compile(); var partialInterface = partialInterfaceTemplate.Compile();
var binder = objectBinderTemplate.Compile(); var binder = objectBinderTemplate.Compile();
_codeTemplate.RemoteTypeBinder.Insert("eventBinder", binder); remoteTypeBinder.InsertEventBinder(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));
@@ -239,7 +236,7 @@ namespace mROA.Codegen
return caller; return caller;
} }
private void GenerateDeclaredMethod(TemplateDocument classTemplate, IMethodSymbol method, List<string> invokers, private void GenerateDeclaredMethod(ProxyTemplate proxyTemplate, MethodRepoTemplate methodRepoTemplate, IMethodSymbol method, List<string> invokers,
INamedTypeSymbol baseInterface) INamedTypeSymbol baseInterface)
{ {
var sb = new StringBuilder(); var sb = new StringBuilder();
@@ -311,7 +308,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()); proxyTemplate.InsertMethods(sb.ToString());
// declaredMethods.Add(sb.ToString()); // declaredMethods.Add(sb.ToString());
var parameterTypes = string.Join(", ", var parameterTypes = string.Join(", ",
@@ -356,38 +353,35 @@ namespace mROA.Codegen
if (isAsync) if (isAsync)
{ {
var invokerTemplate = var invokerTemplate = methodRepoTemplate.CloneInnerAsyncInvoker();
(TemplateDocument)((InnerTemplateSection)_codeTemplate.MethodRepo["asyncInvoker"]!).InnerTemplate invokerTemplate.DefineIsVoid(isVoid.ToString().ToLower());
.Clone(); invokerTemplate.DefineReturnType(CodegenUtilities.ExtractTaskType(method.ReturnType));
invokerTemplate.AddDefine("isVoid", isVoid.ToString().ToLower()); invokerTemplate.DefineParametersType(parameterTypes);
invokerTemplate.AddDefine("returnType", CodegenUtilities.ExtractTaskType(method.ReturnType)); invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.DefineFuncInvoking(funcInvoking);
invokerTemplate.AddDefine("suitableType", baseInterface.ToUnityString()); invokerTemplate.DefineIsTrusted((!isUntrusted).ToString().ToLower());
invokerTemplate.AddDefine("funcInvoking", funcInvoking);
invokerTemplate.AddDefine("isTrusted", (!isUntrusted).ToString().ToLower());
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
else else
{ {
var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone(); var invokerTemplate = methodRepoTemplate.CloneInnerSyncInvoker();
invokerTemplate.AddDefine("isVoid", isVoid.ToString().ToLower()); invokerTemplate.DefineIsVoid(isVoid.ToString().ToLower());
invokerTemplate.AddDefine("returnType", isVoid ? "void" : method.ReturnType.ToUnityString()); invokerTemplate.DefineReturnType(isVoid ? "void" : method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.DefineParametersType(parameterTypes);
invokerTemplate.AddDefine("suitableType", baseInterface.ToUnityString()); invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
invokerTemplate.AddDefine("funcInvoking", funcInvoking); invokerTemplate.DefineFuncInvoking(funcInvoking);
invokerTemplate.AddDefine("isTrusted", (!isUntrusted).ToString().ToLower()); invokerTemplate.DefineIsTrusted((!isUntrusted).ToString().ToLower());
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
_codeTemplate.MethodRepo.Insert("invoker", backend); methodRepoTemplate.InsertInvoke(backend);
invokers.Add(backend); invokers.Add(backend);
} }
private void GenerateBinderCode(IEventSymbol eventSymbol, INamedTypeSymbol baseType, TemplateDocument document) private void GenerateBinderCode(ObjectBinderTemplate objectBinderTemplate, IEventSymbol eventSymbol, INamedTypeSymbol baseType)
{ {
var eventBinderTemplate = var eventBinderTemplate = objectBinderTemplate.CloneInnerEventBinder();
(TemplateDocument)((InnerTemplateSection)document["eventBinderTemplate"]!).InnerTemplate.Clone();
var index = _currentInternalCallIndex; var index = _currentInternalCallIndex;
var parameters = (eventSymbol.Type as INamedTypeSymbol)!.TypeArguments.ToList(); var parameters = (eventSymbol.Type as INamedTypeSymbol)!.TypeArguments.ToList();
@@ -399,25 +393,24 @@ namespace mROA.Codegen
var transferParameters = CodegenUtilities.JoinWithComa(parameters.Select(i => (i, pi++)) var transferParameters = CodegenUtilities.JoinWithComa(parameters.Select(i => (i, pi++))
.Where(i => !ParameterFilterForType(i.i)) .Where(i => !ParameterFilterForType(i.i))
.Select(i => "p" + i.Item2)); .Select(i => "p" + i.Item2));
var requestIndex = parameters.FindIndex(i => i.Name == "RequestContext"); var requestIndex = parameters.FindIndex(i => i.Name == "RequestContext");
if (requestIndex != -1) if (requestIndex != -1)
{ {
var callFilter = $"\n\r\t\t\tif(ownerId == p{requestIndex}.OwnerId) return;"; var callFilter = $"\n\r\t\t\tif(ownerId == p{requestIndex}.OwnerId) return;";
eventBinderTemplate.AddDefine("callFilter", callFilter); eventBinderTemplate.DefineCallFilter(callFilter);
} }
eventBinderTemplate.AddDefine("type", baseType.ToUnityString()); eventBinderTemplate.DefineType(baseType.ToUnityString());
eventBinderTemplate.AddDefine("eventName", eventSymbol.Name); eventBinderTemplate.DefineEventName(eventSymbol.Name);
eventBinderTemplate.AddDefine("parametersDeclaration", parametersDeclaration); eventBinderTemplate.DefineParametersDeclaration(parametersDeclaration);
eventBinderTemplate.AddDefine("commandId", eventBinderTemplate.DefineCommandIdTag($"context.CallIndexProvider.GetIndices(typeof({baseType.ToUnityString()}))[{index}]");
$"context.CallIndexProvider.GetIndices(typeof({baseType.ToUnityString()}))[{index}]"); eventBinderTemplate.DefineTransferParameters(transferParameters);
eventBinderTemplate.AddDefine("transferParameters", transferParameters);
var eventBinderCode = eventBinderTemplate.Compile(); var eventBinderCode = eventBinderTemplate.Compile();
document.Insert("eventBinder", eventBinderCode); objectBinderTemplate.InsertEventBinder(eventBinderCode);
} }
private void GenerateEventCode(IEventSymbol eventSymbol, List<string> invokers, ITypeSymbol baseInterface) private void GenerateEventCode(MethodRepoTemplate methodRepoTemplate, IEventSymbol eventSymbol, List<string> invokers, ITypeSymbol baseInterface)
{ {
var level = "\t\t\t"; var level = "\t\t\t";
@@ -453,21 +446,21 @@ namespace mROA.Codegen
{level} return null; {level} return null;
{level} }}"; {level} }}";
var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone(); var invokerTemplate = methodRepoTemplate.CloneInnerSyncInvoker();
invokerTemplate.AddDefine("isVoid", "true"); invokerTemplate.DefineIsVoid("true");
invokerTemplate.AddDefine("returnType", "void"); invokerTemplate.DefineReturnType("void");
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.DefineParametersType(parameterTypes);
invokerTemplate.AddDefine("suitableType", baseInterface.ToUnityString()); invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
invokerTemplate.AddDefine("funcInvoking", funcInvoking); invokerTemplate.DefineFuncInvoking(funcInvoking);
invokerTemplate.AddDefine("isTrusted", "true"); invokerTemplate.DefineIsTrusted("true");
var backend = invokerTemplate.Compile(); var backend = invokerTemplate.Compile();
_codeTemplate.MethodRepo.Insert("invoker", backend); methodRepoTemplate.InsertInvoke(backend);
invokers.Add(backend); invokers.Add(backend);
} }
private void GeneratePropertyMethod(IMethodSymbol method, private void GeneratePropertyMethod(MethodRepoTemplate methodRepoTemplate, IMethodSymbol method,
List<(string, IMethodSymbol)> propsCollection, List<string> invokers, INamedTypeSymbol baseInterace) List<(string, IMethodSymbol)> propsCollection, List<string> invokers, INamedTypeSymbol baseInterface)
{ {
string frontend; string frontend;
string backend; string backend;
@@ -485,26 +478,25 @@ 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)_codeTemplate.MethodInvoker.Clone(); var invokerTemplate = methodRepoTemplate.CloneInnerSyncInvoker();
invokerTemplate.AddDefine("isVoid", "false"); invokerTemplate.DefineIsVoid("false");
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.DefineReturnType(method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.DefineParametersType(parameterTypes);
invokerTemplate.AddDefine("suitableType", baseInterace.ToUnityString()); invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
invokerTemplate.AddDefine("funcInvoking", invokerTemplate.DefineFuncInvoking($"(i as {method.ContainingType.ToUnityString()})[{parameterInserts}]");
$"(i as {method.ContainingType.ToUnityString()})[{parameterInserts}]"); invokerTemplate.DefineIsTrusted("true");
invokerTemplate.AddDefine("isTrusted", "true");
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
else else
{ {
var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone(); var invokerTemplate = methodRepoTemplate.CloneInnerSyncInvoker();
invokerTemplate.AddDefine("isVoid", "false"); invokerTemplate.DefineIsVoid("false");
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.DefineReturnType(method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("suitableType", baseInterace.ToUnityString()); invokerTemplate.DefineParametersType("");
invokerTemplate.AddDefine("funcInvoking", invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
$"(i as {method.ContainingType.ToUnityString()}).{(method.AssociatedSymbol as IPropertySymbol)!.Name}"); invokerTemplate.DefineFuncInvoking($"(i as {method.ContainingType.ToUnityString()}).{(method.AssociatedSymbol as IPropertySymbol)!.Name}");
invokerTemplate.AddDefine("isTrusted", "true"); invokerTemplate.DefineIsTrusted("true");
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
@@ -527,31 +519,26 @@ 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)_codeTemplate.MethodInvoker.Clone();
var invokerTemplate = methodRepoTemplate.CloneInnerSyncInvoker();
invokerTemplate.DefineIsVoid("true");
invokerTemplate.AddDefine("isVoid", "true"); invokerTemplate.DefineReturnType(method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.DefineParametersType(parameterTypes);
invokerTemplate.AddDefine("parametersType", parameterTypes); invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
invokerTemplate.AddDefine("suitableType", baseInterace.ToUnityString()); invokerTemplate.DefineFuncInvoking($"(i as {method.ContainingType.ToUnityString()})[{parameterInserts}] = {valueInsert}");
invokerTemplate.AddDefine("funcInvoking", invokerTemplate.DefineIsTrusted("true");
$"(i as {method.ContainingType.ToUnityString()})[{parameterInserts}] = {valueInsert}");
invokerTemplate.AddDefine("isTrusted", "true");
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
else else
{ {
var invokerTemplate = (TemplateDocument)_codeTemplate.MethodInvoker.Clone(); var invokerTemplate = methodRepoTemplate.CloneInnerSyncInvoker();
invokerTemplate.DefineIsVoid("true");
invokerTemplate.AddDefine("isVoid", "true"); invokerTemplate.DefineReturnType(method.ReturnType.ToUnityString());
invokerTemplate.AddDefine("returnType", method.ReturnType.ToUnityString()); invokerTemplate.DefineParametersType($"typeof({method.Parameters.First().Type.ToUnityString()})");
invokerTemplate.AddDefine("parametersType", invokerTemplate.DefineSuitableType(baseInterface.ToUnityString());
$"typeof({method.Parameters.First().Type.ToUnityString()})"); invokerTemplate.DefineFuncInvoking($"(i as {method.ContainingType.ToUnityString()}).{(method.AssociatedSymbol as IPropertySymbol)!.Name} = {CodegenUtilities.Caster((method.AssociatedSymbol as IPropertySymbol)!.Type, "parameters[0]")}");
invokerTemplate.AddDefine("suitableType", baseInterace.ToUnityString()); invokerTemplate.DefineIsTrusted("true");
invokerTemplate.AddDefine("funcInvoking",
$"(i as {method.ContainingType.ToUnityString()}).{(method.AssociatedSymbol as IPropertySymbol)!.Name} = {CodegenUtilities.Caster((method.AssociatedSymbol as IPropertySymbol)!.Type, "parameters[0]")}");
invokerTemplate.AddDefine("isTrusted", "true");
backend = invokerTemplate.Compile(); backend = invokerTemplate.Compile();
} }
@@ -561,7 +548,7 @@ namespace mROA.Codegen
} }
propsCollection.Add((frontend, method)); propsCollection.Add((frontend, method));
_codeTemplate.MethodRepo.Insert("invoker", backend); methodRepoTemplate.InsertInvoke(backend);
invokers.Add(backend); invokers.Add(backend);
} }
} }