Lua向C#逻辑迁移一期 #13

将整个插件代码上传
2025-10-26 21:48:39 +08:00
parent 56994b3927
commit 648386cd73
785 changed files with 53683 additions and 2 deletions
--- a/Plugins/UnrealSharp/Managed/UnrealSharpPrograms/UnrealSharpWeaver/WeaverImporter.cs
+++ b/Plugins/UnrealSharp/Managed/UnrealSharpPrograms/UnrealSharpWeaver/WeaverImporter.cs
@ -0,0 +1,206 @@
+using Mono.Cecil;
+using Mono.Cecil.Rocks;
+using UnrealSharpWeaver.Utilities;
+
+namespace UnrealSharpWeaver;
+
+public class WeaverImporter
+{
+    private static WeaverImporter? _instance;
+    public static WeaverImporter Instance => _instance ??= new WeaverImporter();
+
+    private const string Attributes = ".Attributes";
+
+    public const string UnrealSharpNamespace = "UnrealSharp";
+    public const string UnrealSharpAttributesNamespace = UnrealSharpNamespace + Attributes;
+
+    public const string UnrealSharpCoreNamespace = UnrealSharpNamespace + ".Core";
+    public const string UnrealSharpCoreAttributesNamespace = UnrealSharpCoreNamespace + Attributes;
+    public const string UnrealSharpCoreMarshallers = UnrealSharpCoreNamespace + ".Marshallers";
+
+    public const string InteropNameSpace = UnrealSharpNamespace + ".Interop";
+    public const string AttributeNamespace = UnrealSharpNamespace + Attributes;
+    public const string CoreUObjectNamespace = UnrealSharpNamespace + ".CoreUObject";
+    public const string EngineNamespace = UnrealSharpNamespace + ".Engine";
+
+    public const string UnrealSharpObject = "UnrealSharpObject";
+    public const string FPropertyCallbacks = "FPropertyExporter";
+
+    public const string CoreUObjectCallbacks = "UCoreUObjectExporter";
+    public const string UObjectCallbacks = "UObjectExporter";
+    public const string UScriptStructCallbacks = "UScriptStructExporter";
+    public const string UFunctionCallbacks = "UFunctionExporter";
+    public const string MulticastDelegatePropertyCallbacks = "FMulticastDelegatePropertyExporter";
+    public const string UStructCallbacks = "UStructExporter";
+
+    public const string GeneratedTypeAttribute = "GeneratedTypeAttribute";
+    
+    public MethodReference? UFunctionAttributeConstructor => UnrealSharpAssembly.FindType("UFunctionAttribute", "UnrealSharp.Attributes")?.FindMethod(".ctor");
+    public MethodReference? BlueprintInternalUseAttributeConstructor => UnrealSharpAssembly.FindType("BlueprintInternalUseOnlyAttribute", "UnrealSharp.Attributes.MetaTags")?.FindMethod(".ctor");
+    
+    public AssemblyDefinition UnrealSharpAssembly => FindAssembly(UnrealSharpNamespace);
+    public AssemblyDefinition UnrealSharpCoreAssembly => FindAssembly(UnrealSharpNamespace + ".Core");
+    
+    public AssemblyDefinition CurrentWeavingAssembly = null!;
+    public List<AssemblyDefinition> AllProjectAssemblies = [];
+    
+    public MethodReference NativeObjectGetter = null!;
+    public TypeDefinition IntPtrType = null!;
+    public MethodReference IntPtrAdd = null!;
+    public FieldReference IntPtrZero = null!;
+    public MethodReference IntPtrEqualsOperator = null!;
+    public TypeReference UnrealSharpObjectType = null!;
+    public TypeDefinition IInterfaceType = null!;
+    public MethodReference GetNativeFunctionFromInstanceAndNameMethod = null!;
+    public TypeReference Int32TypeRef = null!;
+    public TypeReference UInt64TypeRef = null!;
+    public TypeReference VoidTypeRef = null!;
+    public TypeReference ByteTypeRef = null!;
+    
+    public TypeReference MarshalledStructReference = null!;
+    
+    public MethodReference GetNativeClassFromNameMethod = null!;
+    public MethodReference GetNativeInterfaceFromNameMethod = null!;
+    public MethodReference GetNativeStructFromNameMethod = null!;
+    public MethodReference GetPropertyOffsetFromNameMethod = null!;
+    public MethodReference GetPropertyOffset = null!;
+    public MethodReference GetNativePropertyFromNameMethod = null!;
+    public MethodReference GetNativeFunctionFromClassAndNameMethod = null!;
+    public MethodReference GetNativeFunctionParamsSizeMethod = null!;
+    public MethodReference GetNativeStructSizeMethod = null!;
+    public MethodReference GetSignatureFunction = null!;
+    public MethodReference InitializeStructMethod = null!;
+    
+    public MethodReference InvokeNativeFunctionMethod = null!;
+    public MethodReference InvokeNativeNetFunction = null!;
+    public MethodReference InvokeNativeFunctionOutParms = null!;
+
+    public MethodReference GeneratedTypeCtor = null!;
+    
+    public TypeDefinition UObjectDefinition = null!;
+    public TypeDefinition UActorComponentDefinition = null!;
+    
+    public TypeDefinition ScriptInterfaceWrapper = null!;
+    public TypeDefinition ScriptInterfaceMarshaller = null!;
+    public TypeReference ManagedObjectHandle = null!;
+    public TypeReference UnmanagedDataStore = null!;
+    
+    public MethodReference BlittableTypeConstructor = null!;
+
+    public DefaultAssemblyResolver AssemblyResolver = null!;
+    
+    public static void Shutdown()
+    {
+        if (_instance == null)
+        {
+            return;
+        }
+        
+        foreach (AssemblyDefinition assembly in _instance.AllProjectAssemblies)
+        {
+            assembly.Dispose();
+        }
+        
+        _instance.AllProjectAssemblies = [];
+        _instance.CurrentWeavingAssembly = null!;
+        _instance = null;
+    }
+    
+    static AssemblyDefinition FindAssembly(string assemblyName)
+    {
+        return Instance.AssemblyResolver.Resolve(new AssemblyNameReference(assemblyName, new Version(0, 0)));
+    }
+
+    public void ImportCommonTypes(AssemblyDefinition userAssembly)
+    {
+        CurrentWeavingAssembly = userAssembly;
+        
+        TypeSystem typeSystem = CurrentWeavingAssembly.MainModule.TypeSystem;
+        
+        Int32TypeRef = typeSystem.Int32;
+        UInt64TypeRef = typeSystem.UInt64;
+        VoidTypeRef = typeSystem.Void;
+        ByteTypeRef = typeSystem.Byte;
+        
+        IntPtrType = typeSystem.IntPtr.Resolve();
+        IntPtrAdd = IntPtrType.FindMethod("Add")!;
+        IntPtrZero = IntPtrType.FindField("Zero");
+        IntPtrEqualsOperator = IntPtrType.FindMethod("op_Equality")!;
+
+        UnrealSharpObjectType = UnrealSharpCoreAssembly.FindType(UnrealSharpObject, UnrealSharpCoreNamespace)!;
+        IInterfaceType = UnrealSharpAssembly.FindType("IInterface", CoreUObjectNamespace)!.Resolve();
+        
+        MarshalledStructReference = UnrealSharpCoreAssembly.FindType("MarshalledStruct`1", "UnrealSharp")!.Resolve();
+        
+        TypeDefinition unrealSharpObjectType = UnrealSharpObjectType.Resolve();
+        NativeObjectGetter = unrealSharpObjectType.FindMethod("get_NativeObject")!;
+
+        GetNativeFunctionFromInstanceAndNameMethod = FindExporterMethod(TypeDefinitionUtilities.UClassCallbacks, "CallGetNativeFunctionFromInstanceAndName");
+        
+        GetNativeStructFromNameMethod = FindExporterMethod(CoreUObjectCallbacks, "CallGetNativeStructFromName");
+        GetNativeClassFromNameMethod = FindExporterMethod(CoreUObjectCallbacks, "CallGetNativeClassFromName");
+        GetNativeInterfaceFromNameMethod = FindExporterMethod(CoreUObjectCallbacks, "CallGetNativeInterfaceFromName");
+        
+        GetPropertyOffsetFromNameMethod = FindExporterMethod(FPropertyCallbacks, "CallGetPropertyOffsetFromName");
+        GetPropertyOffset = FindExporterMethod(FPropertyCallbacks, "CallGetPropertyOffset");
+        
+        GetNativePropertyFromNameMethod = FindExporterMethod(FPropertyCallbacks, "CallGetNativePropertyFromName");
+        
+        GetNativeFunctionFromClassAndNameMethod = FindExporterMethod(TypeDefinitionUtilities.UClassCallbacks, "CallGetNativeFunctionFromClassAndName");
+        GetNativeFunctionParamsSizeMethod = FindExporterMethod(UFunctionCallbacks, "CallGetNativeFunctionParamsSize");
+        
+        GetNativeStructSizeMethod = FindExporterMethod(UScriptStructCallbacks, "CallGetNativeStructSize");
+        
+        InvokeNativeFunctionMethod = FindExporterMethod(UObjectCallbacks, "CallInvokeNativeFunction");
+        InvokeNativeNetFunction = FindExporterMethod(UObjectCallbacks, "CallInvokeNativeNetFunction");
+        InvokeNativeFunctionOutParms = FindExporterMethod(UObjectCallbacks, "CallInvokeNativeFunctionOutParms");
+        
+        GetSignatureFunction = FindExporterMethod(MulticastDelegatePropertyCallbacks, "CallGetSignatureFunction");
+        
+        InitializeStructMethod = FindExporterMethod(UStructCallbacks, "CallInitializeStruct");
+        
+        UObjectDefinition = UnrealSharpAssembly.FindType("UObject", CoreUObjectNamespace)!.Resolve();
+        UActorComponentDefinition = UnrealSharpAssembly.FindType("UActorComponent", EngineNamespace)!.Resolve();
+        
+        TypeReference blittableType = UnrealSharpCoreAssembly.FindType(TypeDefinitionUtilities.BlittableTypeAttribute, UnrealSharpCoreAttributesNamespace)!;
+        BlittableTypeConstructor = blittableType.FindMethod(".ctor")!;
+
+        TypeReference generatedType = UnrealSharpCoreAssembly.FindType(GeneratedTypeAttribute, UnrealSharpCoreAttributesNamespace)!;
+        GeneratedTypeCtor = generatedType.FindMethod(".ctor")!;
+        
+        ScriptInterfaceWrapper = UnrealSharpAssembly.FindType("IScriptInterface", CoreUObjectNamespace)!.Resolve();
+        ScriptInterfaceMarshaller = UnrealSharpAssembly.FindType("ScriptInterfaceMarshaller`1", CoreUObjectNamespace)!.Resolve();
+        
+        ManagedObjectHandle = UnrealSharpAssembly.FindType("FSharedGCHandle", "UnrealSharp.UnrealSharpCore")!.Resolve();
+        UnmanagedDataStore = UnrealSharpAssembly.FindType("FUnmanagedDataStore", "UnrealSharp.UnrealSharpCore")!.Resolve();
+    }
+
+    private static MethodReference FindBindingsStaticMethod(string findNamespace, string findClass, string findMethod)
+    {
+        foreach (var module in Instance.UnrealSharpAssembly.Modules)
+        {
+            foreach (var type in module.GetAllTypes())
+            {
+                if (type.Namespace != findNamespace || type.Name != findClass)
+                {
+                    continue;
+                }
+
+                foreach (var method in type.Methods)
+                {
+                    if (method.IsStatic && method.Name == findMethod)
+                    {
+                        return Instance.CurrentWeavingAssembly.MainModule.ImportReference(method);
+                    }
+                }
+            }
+        }
+        
+        throw new Exception("Could not find method " + findMethod + " in class " + findClass + " in namespace " + findNamespace);
+    }
+
+    private static MethodReference FindExporterMethod(string exporterName, string functionName)
+    {
+        return FindBindingsStaticMethod(InteropNameSpace, exporterName, functionName);
+    }
+}