В теории должен работать обновленные бэкэнд

2025-03-08 19:46:59 +03:00
parent 6aa11ba7c0
commit 4863010b47
9 changed files with 264 additions and 69 deletions
@@ -3,9 +3,7 @@
 using System;
 using System.Collections.Generic;
 using System.Collections.Immutable;
-using System.IO;
 using System.Linq;
-using System.Reflection;
 using System.Text;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
@@ -33,6 +31,7 @@ namespace mROA.Codegen
            var frontendContextRepo = new List<string>();

            List<IMethodSymbol> innerMethods = new List<IMethodSymbol>();
+            List<string> invokers = new List<string>();
            var declarations = classes.ToList().OrderBy(i => i.Identifier.Text).ToList();
            foreach (var classDeclarationSyntax in declarations)
            {
@@ -43,7 +42,7 @@ namespace mROA.Codegen

                var namespaceName = classSymbol.ContainingNamespace.ToDisplayString();
                var className = classDeclarationSyntax.Identifier.Text;
-                innerMethods = CollectMembers(classSymbol);
+                innerMethods.AddRange(CollectMembers(classSymbol));
                var associated = innerMethods.Select(i => i.AssociatedSymbol).Where(i => i != null)
                    .Distinct(SymbolEqualityComparer.Default).Cast<ISymbol>().ToList();

@@ -58,16 +57,16 @@ namespace mROA.Codegen

                var declaredMethods = new List<string>();
                var propertiesAccessMethods = new List<(string, IMethodSymbol)>();
-                var propertiesImplementations = new List<string>(associated.Count);
+
                foreach (var method in innerMethods)
                {
                    switch (method.MethodKind)
                    {
                        case MethodKind.PropertyGet or MethodKind.PropertySet:
-                            GeneratePropertyMethod(method, innerMethods, propertiesAccessMethods);
+                            GeneratePropertyMethod(method, innerMethods, propertiesAccessMethods, invokers);
                            continue;
                        default:
-                            GenerateDeclaretedMethod(method, declaredMethods, innerMethods);
+                            GenerateDeclaredMethod(method, declaredMethods, innerMethods, invokers);
                            break;
                    }
                }
@@ -135,7 +134,8 @@ namespace {namespaceName}

            if (innerMethods.Count != 0)
            {
-                var methodsStringed = Array.Empty<string>();
+                var methodsStringed = invokers;
+

                var coCodegenRepoCode = @$"// <auto-generated/>
 using System.Collections.Generic;
@@ -143,6 +143,7 @@ using System.Reflection;
 using mROA.Abstract;
 using mROA.Implementation;
 using System;
+using System.Threading;

 namespace mROA.Codegen
 {{
@@ -169,9 +170,11 @@ namespace mROA.Codegen
    }}
 }}
 ";
+#if !DONT_ADD
                context.AddSource("CoCodegenMethodRepository.g.cs", SourceText.From(coCodegenRepoCode, Encoding.UTF8));
+#endif
            }
-
+            
            if (frontendContextRepo.Count != 0)
            {
                var fronendRepoCode = @$"// <auto-generated/>
@@ -198,8 +201,17 @@ namespace mROA.Codegen
            }
        }

-        private void GenerateDeclaretedMethod(IMethodSymbol method, List<string> declaretedMethods,
-            List<IMethodSymbol> methods)
+        public static string Caster(ITypeSymbol type, string inner)
+        {
+            if (!type.IsValueType)
+                return inner +
+                       " as " +
+                       type.ToDisplayString();
+            return $"({type.ToDisplayString()})" + inner;
+        }
+
+        private void GenerateDeclaredMethod(IMethodSymbol method, List<string> declaredMethods,
+            List<IMethodSymbol> methods, List<string> invokers)
        {
            var index = methods.IndexOf(method);
            var sb = new StringBuilder();
@@ -259,40 +271,173 @@ namespace mROA.Codegen

            sb.AppendLine("\t\t}");

-            declaretedMethods.Add(sb.ToString());
+            declaredMethods.Add(sb.ToString());
+            var parameterTypes = string.Join(", ",
+                $"{string.Join(", ", parameters.Select(p => $"typeof({p.Type.ToDisplayString()})"))}");
+            var level = "\t\t\t";
+
+            var parametersInsertList = new List<string>();
+
+            for (var i = 0; i < method.Parameters.Length; i++)
+            {
+                var parameter = method.Parameters[i];
+                switch (parameter.Type.Name)
+                {
+                    case "CancellationToken":
+                        parametersInsertList.Add("(CancellationToken)special[1]");
+                        break;
+                    case "RequestContext":
+                        parametersInsertList.Add("special[1] as RequestContext");
+                        break;
+                    default:
+                        parametersInsertList.Add(Caster(parameter.Type,
+                            $"parameters[{parameters.IndexOf(parameter)}]"));
+                        break;
+                }
+            }
+
+            var parametersInsert = string.Join(", ", parametersInsertList);
+            var backend = string.Empty;
+
+            var funcInvoking = string.Empty;
+
+            if (isAsync && !isVoid)
+                funcInvoking =
+                    $"(i as {method.ContainingType.ToDisplayString()}).{method.Name}({parametersInsert}).ContinueWith(t => {{ post(t.Result); }})";
+            else if (isAsync && isVoid)
+                funcInvoking =
+                    $"(i as {method.ContainingType.ToDisplayString()}).{method.Name}({parametersInsert}).ContinueWith(t => {{ post(null); }})";
+            else if (!isAsync && isVoid)
+            {
+                funcInvoking = $@"{{
+{level}        (i as {method.ContainingType.ToDisplayString()}).{method.Name}({parametersInsert});
+{level}        return null;
+{level}    }}";
+            }
+            else
+            {
+                funcInvoking = $"(i as {method.ContainingType.ToDisplayString()}).{method.Name}({parametersInsert})";
+            }
+
+            if (isAsync)
+                backend = $@"new mROA.Implementation.AsyncMethodInvoker 
+{level}{{
+{level}    IsVoid = {isVoid.ToString().ToLower()},
+{(isVoid ? String.Empty : (level + "\t" + "ReturnType = typeof(" + ExtractTaskType(method.ReturnType)) + "),")}
+{level}    ParameterTypes = new Type[] {{ {parameterTypes} }},
+{level}    Invoking = (i, parameters, special, post) => {funcInvoking},
+{level}}}";
+            else
+                backend = $@"new mROA.Implementation.MethodInvoker 
+{level}{{
+{level}    IsVoid = {isVoid.ToString().ToLower()},
+{level}    ReturnType = typeof({method.ReturnType.ToDisplayString()}),
+{level}    ParameterTypes = new Type[] {{ {parameterTypes} }},
+{level}    Invoking = (i, parameters, special) => {funcInvoking}
+{level}}}";
+
+            invokers.Add(backend);
        }

        private static Predicate<IParameterSymbol> ParameterFilter =
            i => i.Type.Name is "CancellationToken" or "RequestContext";

        public void GeneratePropertyMethod(IMethodSymbol method, List<IMethodSymbol> methods,
-            List<(string, IMethodSymbol)> propsCollection)
+            List<(string, IMethodSymbol)> propsCollection, List<string> invokers)
        {
+            var level = "\t\t\t";
            var index = methods.IndexOf(method);
-            var sb = "";
+            string frontend;
+            string backend = string.Empty;
            if (method.MethodKind == MethodKind.PropertyGet)
            {
                var parametersArray = "";
                if (method.Parameters.Length != 0)
                {
                    parametersArray = $", new object[] {{{string.Join(", ", method.Parameters.Select(p => p.Name))}}}";
+
+                    var parameterTypes = string.Join(", ",
+                        $"{string.Join(", ", method.Parameters.Select(p => "typeof(" + p.Type.ToDisplayString() + ")"))}");
+                    var parameterInserts = string.Join(", ",
+                        method.Parameters.Select(p =>
+                            {
+                                return Caster(p.Type, "parameters[" + method.Parameters.IndexOf(p) + "]");
+                                // if (!p.Type.IsValueType)
+                                //     return "parameters[" + method.Parameters.IndexOf(p) + "] as " +
+                                //            p.Type.ToDisplayString();
+                                // return $"({p.Type.ToDisplayString()})parameters[{method.Parameters.IndexOf(p)}]";
+                            }
+                        ));
+                    backend = $@"new mROA.Implementation.MethodInvoker 
+{level}{{
+{level}    IsVoid = false,
+{level}    ReturnType = typeof({method.ReturnType.ToDisplayString()}),
+{level}    ParameterTypes = new Type[] {{ {parameterTypes} }},
+{level}    Invoking = (i, parameters, _) => (i as {method.ContainingType.ToDisplayString()})[{parameterInserts}],
+{level}}}";
+                }
+                else
+                {
+                    backend = $@"new mROA.Implementation.MethodInvoker 
+{level}{{
+{level}    IsVoid = false,
+{level}    ReturnType = typeof({method.ReturnType.ToDisplayString()}),
+{level}    Invoking = (i, _, _) => (i as {method.ContainingType.ToDisplayString()}).{(method.AssociatedSymbol as IPropertySymbol)!.Name},
+{level}}}";
                }

-                sb =
+                frontend =
                    $"get => GetResultAsync<{method.ReturnType.ToDisplayString()}>({index}{parametersArray}).GetAwaiter().GetResult();";
            }
            else
            {
                var parametersArray = "value";
-                if (method.Parameters.Length != 0)
+                if (method.Parameters.Length != 1)
                {
                    parametersArray = string.Join(", ", method.Parameters.Select(p => p.Name));
+
+                    var parameterTypes = string.Join(", ",
+                        $"{string.Join(", ", method.Parameters.Select(p => $"typeof({p.Type.ToDisplayString()})"))}");
+                    var parameterInserts = string.Join(", ",
+                        method.Parameters.Take(method.Parameters.Length - 1).Select(p =>
+                            {
+                                return Caster(p.Type, "parameters[" + method.Parameters.IndexOf(p) + "]");
+                                // if (!p.Type.IsValueType)
+                                //     return "parameters[" + method.Parameters.IndexOf(p) + "] as " +
+                                //            p.Type.ToDisplayString();
+                                // return $"({p.Type.ToDisplayString()})parameters[{method.Parameters.IndexOf(p)}]";
+                            }
+                        ));
+
+                    // var valueInsert = !method.Parameters.Last().Type.IsValueType
+                    //     ? "parameters[" + (method.Parameters.Length - 1) + "] as " +
+                    //       method.Parameters.Last().Type.ToDisplayString()
+                    //     : $"({method.Parameters.Last().Type.ToDisplayString()})parameters[{method.Parameters.Length - 1}]";
+                    var valueInsert = Caster(method.Parameters.Last().Type,
+                        "parameters[" + (method.Parameters.Length - 1) + "]");
+                    backend = $@"new mROA.Implementation.MethodInvoker 
+{level}{{
+{level}    IsVoid = false,
+{level}    ReturnType = typeof({method.ReturnType.ToDisplayString()}),
+{level}    ParameterTypes = new Type[] {{ {parameterTypes} }},
+{level}    Invoking = (i, parameters, _) => (i as {method.ContainingType.ToDisplayString()})[{parameterInserts}] = {valueInsert},
+{level}}}";
+                }
+                else
+                {
+                    backend = $@"new mROA.Implementation.MethodInvoker 
+{level}{{
+{level}    IsVoid = false,
+{level}    ReturnType = typeof({method.ReturnType.ToDisplayString()}),
+{level}    Invoking = (i, parameters, _) => (i as {method.ContainingType.ToDisplayString()}).{(method.AssociatedSymbol as IPropertySymbol)!.Name} = {Caster((method.AssociatedSymbol as IPropertySymbol)!.Type, "parameters[0]")},
+{level}}}";
                }

-                sb = $"set => CallAsync({index}, new object[] {{ {parametersArray} }}).Wait();";
+                frontend = $"set => CallAsync({index}, new object[] {{ {parametersArray} }}).Wait();";
            }

-            propsCollection.Add((sb.ToString(), method));
+            propsCollection.Add((frontend, method));
+            invokers.Add(backend);
        }

        private static string ToFullString(IParameterSymbol parameter)
@@ -307,7 +452,7 @@ namespace mROA.Codegen

        private string ExtractTaskType(ITypeSymbol taskType)
        {
-            return (taskType as INamedTypeSymbol).TypeParameters[0].ToDisplayString();
+            return (taskType as INamedTypeSymbol).TypeArguments[0].ToDisplayString();
        }

        public void Initialize(GeneratorInitializationContext context)