Source Code Generator

SourceCodeGenerator.cs

@@ -0,0 +1,839 @@
+// Copyright Epic Games, Inc. All Rights Reserved.
+
+using System;
+using System.Collections.Concurrent;
+using EpicGames.Core;
+using EpicGames.UHT.Types;
+using EpicGames.UHT.Utils;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using EpicGames.UHT.Tables;
+using UnrealBuildTool;
+
+namespace SourceCodeGeneratorUbtPlugin
+{
+    [UnrealHeaderTool]
+    internal class SourceCodeGenerator
+    {
+        [UhtExporter(Name = "UnrealCSharp", Description = "UnrealCSharp Source Code Generator",
+            Options = UhtExporterOptions.Default, ModuleName = "UnrealCSharp")]
+        private static void SourceCodeGeneratorExporter(IUhtExportFactory factory)
+        {
+            var SettingFilePath = Path.Combine(new string(factory.Session.ProjectDirectory),
+                "Config",
+                "DefaultUnrealCSharpEditorSetting.ini");
+
+            if (File.Exists(SettingFilePath))
+            {
+                var SettingConfigFile = new ConfigFile(new FileReference(SettingFilePath));
+
+                if (SettingConfigFile.TryGetSection("/Script/UnrealCSharpCore.UnrealCSharpEditorSetting",
+                        out var SettingConfigSection))
+                {
+                    var SettingConfigHierarchySection = new ConfigHierarchySection(
+                        new List<ConfigFileSection> { SettingConfigSection });
+
+                    if (SettingConfigHierarchySection.TryGetValue("bEnableExport",
+                            out var Value))
+                    {
+                        if (bool.Parse(Value.ToLower()))
+                        {
+                            new SourceCodeGenerator(factory).Generate();
+                        }
+                    }
+                }
+            }
+        }
+
+        private readonly IUhtExportFactory Factory;
+
+        private UhtSession Session => Factory.Session;
+
+        private ConcurrentQueue<UhtClass> ExportClasses = new();
+
+        private HashSet<string> Project = new();
+
+        private Dictionary<string, string> HeaderPath = new();
+
+        private const string BindingSuffix = ".binding.inl";
+
+        private const string HeaderSuffix = ".header.inl";
+
+        public SourceCodeGenerator(IUhtExportFactory factory)
+        {
+            Factory = factory;
+        }
+
+        /// <summary>
+        /// Export all the classes in all the packages
+        /// </summary>
+        private void Generate()
+        {
+            if (Session is { EngineDirectory: not null, ProjectDirectory: not null })
+            {
+                GetModules(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Source/")), Project);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Source/")), HeaderPath);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Plugins/")), HeaderPath);
+
+                GetPlugins(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Plugins/")), HeaderPath);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Developer/")), HeaderPath);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Editor/")), HeaderPath);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Programs/")), HeaderPath);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Runtime/")), HeaderPath);
+
+                GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Plugins/")), HeaderPath);
+
+                GetPlugins(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Plugins/")), HeaderPath);
+            }
+
+            var SettingFilePath = Path.Combine(new string(Session.ProjectDirectory),
+                "Config",
+                "DefaultUnrealCSharpEditorSetting.ini");
+
+            var ExportModule = new List<string>();
+
+            if (File.Exists(SettingFilePath))
+            {
+                var SettingConfigFile = new ConfigFile(new FileReference(SettingFilePath));
+
+                if (SettingConfigFile.TryGetSection("/Script/UnrealCSharpCore.UnrealCSharpEditorSetting",
+                        out var SettingConfigSection))
+                {
+                    var SettingConfigHierarchySection = new ConfigHierarchySection(
+                        new List<ConfigFileSection> { SettingConfigSection });
+
+                    if (SettingConfigHierarchySection.TryGetValues("ExportModule",
+                            out var Values))
+                    {
+                        foreach (var value in Values)
+                        {
+                            ExportModule.Add(value);
+                        }
+                    }
+                }
+            }
+
+            List<Task?> tasks = new();
+
+            foreach (var package in Session.Packages)
+            {
+                if (package.Module.ModuleType != UHTModuleType.EngineRuntime &&
+                    package.Module.ModuleType != UHTModuleType.GameRuntime)
+                {
+                    continue;
+                }
+
+                if (ExportModule.All(module =>
+                        string.Compare(module, package.Module.Name, StringComparison.OrdinalIgnoreCase) != 0))
+                {
+                    continue;
+                }
+
+                QueueClassExports(package, tasks);
+            }
+
+            var waitTasks = tasks.Where(Task => Task != null).Cast<Task>().ToArray();
+
+            if (waitTasks.Length > 0)
+            {
+                Task.WaitAll(waitTasks);
+            }
+
+            Finish();
+        }
+
+        /// <summary>
+        /// Collect the classes to be exported for the given package and type
+        /// </summary>
+        /// <param name="type">Type to test for exporting</param>
+        /// <param name="tasks">Collection of queued tasks</param>
+        private void QueueClassExports(UhtType type, List<Task?> tasks)
+        {
+            if (type is UhtClass classObj)
+            {
+                if (CanExportClass(classObj))
+                {
+                    tasks.Add(Factory.CreateTask(_ => { ExportClass(classObj); }));
+                }
+            }
+
+            foreach (var child in type.Children)
+            {
+                QueueClassExports(child, tasks);
+            }
+        }
+
+        /// <summary>
+        /// Test to see if the given class should be exported
+        /// </summary>
+        /// <param name="classObj">Class to test</param>
+        /// <returns>True if the class should be exported, false if not</returns>
+        protected virtual bool CanExportClass(UhtClass classObj)
+        {
+            return !classObj.ClassFlags.HasAnyFlags(EClassFlags.Interface) &&
+                   IsClassTypeSupported(classObj);
+        }
+
+        /// <summary>
+        /// Test to see if the given function should be exported
+        /// </summary>
+        /// <param name="function">Function to test</param>
+        /// <returns>True if the function should be exported</returns>
+        protected virtual bool CanExportFunction(UhtFunction function)
+        {
+            if (function.Outer is UhtClass classObj)
+            {
+                if (!classObj.ClassFlags.HasAnyFlags(EClassFlags.RequiredAPI) &&
+                    !function.FunctionExportFlags.HasAnyFlags(UhtFunctionExportFlags.RequiredAPI))
+                {
+                    return false;
+                }
+            }
+            else
+            {
+                return false;
+            }
+
+            if (function.FunctionType != UhtFunctionType.Function)
+            {
+                return false;
+            }
+
+            if (!function.FunctionFlags.HasAnyFlags(EFunctionFlags.Public))
+            {
+                return false;
+            }
+
+            if (function.Deprecated)
+            {
+                return false;
+            }
+
+            if (function.FunctionFlags.HasAnyFlags(EFunctionFlags.EditorOnly))
+            {
+                return false;
+            }
+
+            if (function.FunctionExportFlags.HasAnyFlags(UhtFunctionExportFlags.CustomThunk))
+            {
+                return false;
+            }
+
+            if (function is { HasReturnProperty: true, ReturnProperty: UhtMapProperty })
+            {
+                return false;
+            }
+
+            foreach (var child in function.Children)
+            {
+                if (child is UhtProperty property && !IsPropertyTypeSupported(property))
+                {
+                    return false;
+                }
+            }
+
+            return true;
+        }
+
+        /// <summary>
+        /// Test to see if the given property should be exported
+        /// </summary>
+        /// <param name="property">Property to test</param>
+        /// <returns>True if the property should be exported</returns>
+        protected virtual bool CanExportProperty(UhtProperty property)
+        {
+            return !property.Deprecated &&
+                   !property.PropertyFlags.HasAnyFlags(EPropertyFlags.EditorOnly) &&
+                   property.PropertyFlags.HasAnyFlags(EPropertyFlags.NativeAccessSpecifierPublic) &&
+                   IsPropertyTypeSupported(property);
+        }
+
+        private static bool IsClassTypeSupported(UhtClass classObj)
+        {
+            return !classObj.Deprecated &&
+                   (classObj == classObj.Session.UObject ||
+                    classObj == classObj.Session.UClass ||
+                    classObj.ClassFlags.HasAnyFlags(EClassFlags.RequiredAPI | EClassFlags.MinimalAPI));
+        }
+
+        private static bool IsPropertyTypeSupported(UhtProperty property)
+        {
+            if (property is UhtArrayProperty arrayProperty)
+            {
+                return IsPropertyTypeSupported(arrayProperty.ValueProperty);
+            }
+
+            if (property is UhtObjectProperty objectProperty)
+            {
+                if (!IsClassTypeSupported(objectProperty.Class))
+                {
+                    return false;
+                }
+            }
+
+            if (property is UhtObjectPtrProperty objectPtrProperty)
+            {
+                if (!IsClassTypeSupported(objectPtrProperty.Class))
+                {
+                    return false;
+                }
+            }
+
+            if (property is UhtInterfaceProperty interfaceProperty)
+            {
+                if (interfaceProperty.InterfaceClass == interfaceProperty.Session.IInterface)
+                {
+                    return false;
+                }
+            }
+
+            if (property is UhtDelegateProperty or UhtMulticastDelegateProperty or UhtFieldPathProperty)
+            {
+                return false;
+            }
+
+            if (property is UhtStructProperty { ScriptStruct.HasNoOpConstructor: false })
+            {
+                return false;
+            }
+
+            return property is { IsBitfield: false, IsStaticArray: false };
+        }
+
+        /// <summary>
+        /// Export the given class
+        /// </summary>
+        /// <param name="classObj">Class to export</param>
+        private void ExportClass(UhtClass classObj)
+        {
+            using BorrowStringBuilder borrower = new(StringBuilderCache.Big);
+
+            if (ExportClass(borrower.StringBuilder, classObj))
+            {
+                var fileName = Factory.MakePath(classObj.EngineName, BindingSuffix);
+
+                SaveIfChanged(fileName, borrower.StringBuilder);
+
+                ExportClasses.Enqueue(classObj);
+            }
+        }
+
+        private void Finish()
+        {
+            var packages = new Dictionary<UhtPackage, List<string>>();
+
+            foreach (var ExportClass in ExportClasses)
+            {
+                if (!packages.ContainsKey(ExportClass.Package))
+                {
+                    packages.Add(ExportClass.Package, new List<string>());
+                }
+
+                packages[ExportClass.Package].AddUnique(ExportClass.EngineName);
+            }
+
+            foreach (var package in packages)
+            {
+                package.Value.Sort();
+
+                var borrow = new BorrowStringBuilder(StringBuilderCache.Big);
+
+                var builder = borrow.StringBuilder;
+
+                builder.Append("#pragma once\r\n\r\n");
+
+                foreach (var value in package.Value)
+                {
+                    builder.Append(GenerateInclude($"{value}{BindingSuffix}"));
+                }
+
+                var fileName = Factory.MakePath(package.Key, HeaderSuffix);
+
+                SaveIfChanged(fileName, builder);
+            }
+        }
+
+        private bool ExportClass(StringBuilder builder, UhtClass classObj)
+        {
+            builder.Append("#pragma once\r\n\r\n" +
+                           "PRAGMA_DISABLE_DEPRECATION_WARNINGS\r\n\r\n");
+
+            var DependencyClasses = new HashSet<UhtClass> { classObj };
+
+            using BorrowStringBuilder bodyBorrower = new(StringBuilderCache.Big);
+
+            var bodyBuilder = bodyBorrower.StringBuilder;
+
+            bodyBuilder.AppendFormat("namespace\r\n" +
+                                     "{{\r\n" +
+                                     "\tstruct FRegister{0}\r\n" +
+                                     "\t{{\r\n" +
+                                     "\t\tFRegister{1}()\r\n" +
+                                     "\t\t{{\r\n" +
+                                     "\t\t\tTBindingClassBuilder<{2}>(NAMESPACE_BINDING)\r\n",
+                classObj.EngineName,
+                classObj.EngineName,
+                classObj.SourceName);
+
+            var bExportProperty = false;
+
+            foreach (var child in classObj.Children)
+            {
+                if (child is UhtProperty property && CanExportProperty(property))
+                {
+                    GetDependencyClasses(property, DependencyClasses);
+
+                    ExportProperty(bodyBuilder, classObj, property);
+
+                    bExportProperty = true;
+                }
+            }
+
+            var bExportFunction = false;
+
+            foreach (var function in classObj.Functions.Reverse())
+            {
+                if (CanExportFunction(function))
+                {
+                    foreach (var child in function.Children)
+                    {
+                        if (child is UhtProperty property)
+                        {
+                            GetDependencyClasses(property, DependencyClasses);
+                        }
+                    }
+
+                    ExportFunction(bodyBuilder, classObj, function);
+
+                    bExportFunction = true;
+                }
+            }
+
+            foreach (var DependencyClass in DependencyClasses)
+            {
+                builder.Append(GetInclude(DependencyClass));
+            }
+
+            builder.Append("\r\n");
+
+            bodyBuilder.Remove(bodyBuilder.Length - 2, 2);
+
+            bodyBuilder.AppendFormat(";\r\n" +
+                                     "\t\t}}\r\n" +
+                                     "\t}};\r\n" +
+                                     "\r\n" +
+                                     "\t[[maybe_unused]] FRegister{0} Register{1};\r\n" +
+                                     "}}\r\n\r\n",
+                classObj.EngineName,
+                classObj.EngineName);
+
+            builder.Append(bodyBuilder);
+
+            builder.Append("PRAGMA_ENABLE_DEPRECATION_WARNINGS\r\n");
+
+            return bExportProperty || bExportFunction;
+        }
+
+        private static void GetDependencyClasses(UhtProperty property, HashSet<UhtClass> DependencyClasses)
+        {
+            if (property is UhtClassProperty classProperty)
+            {
+                if (classProperty.MetaClass != null)
+                {
+                    DependencyClasses.Add(classProperty.MetaClass);
+                }
+            }
+            else if (property is UhtObjectPtrProperty objectPtrProperty)
+            {
+                DependencyClasses.Add(objectPtrProperty.Class);
+            }
+            else if (property is UhtObjectProperty objectProperty)
+            {
+                DependencyClasses.Add(objectProperty.Class);
+            }
+            else if (property is UhtSoftClassProperty softClassProperty)
+            {
+                if (softClassProperty.MetaClass != null)
+                {
+                    DependencyClasses.Add(softClassProperty.MetaClass);
+                }
+            }
+            else if (property is UhtSoftObjectProperty softObjectProperty)
+            {
+                DependencyClasses.Add(softObjectProperty.Class);
+            }
+            else if (property is UhtArrayProperty { ValueProperty: UhtObjectProperty valueProperty })
+            {
+                DependencyClasses.Add(valueProperty.Class);
+            }
+        }
+
+        private static void ExportFunction(StringBuilder builder, UhtClass classObj, UhtFunction function)
+        {
+            builder.Append(
+                $"\t\t\t\t.Function(\"{function.SourceName}\", BINDING_OVERLOAD({GetFunctionSignature(classObj, function)}, &{classObj.SourceName}::{function.SourceName}");
+
+            if (function.HasParameters)
+            {
+                builder.Append($", {GetFunctionParamName(function)}");
+            }
+
+            if (!function.FunctionFlags.HasAnyFlags(EFunctionFlags.Static))
+            {
+                builder.Append(", EFunctionInteract::Virtual");
+            }
+
+            if (function.HasParameters)
+            {
+                builder.Append(GetFunctionDefaultValue(function));
+            }
+
+            builder.Append("))\r\n");
+        }
+
+        private static void ExportProperty(StringBuilder builder, UhtClass classObj, UhtProperty property)
+        {
+            builder.Append(
+                $"\t\t\t\t.Property(\"{property.SourceName}\", BINDING_PROPERTY(&{classObj.SourceName}::{property.SourceName}))\r\n");
+        }
+
+        private static string GetParamPropertySignature(UhtProperty? property)
+        {
+            if (property == null)
+            {
+                return "void";
+            }
+
+            var builder = new StringBuilder();
+
+            var caps = property.PropertyCaps;
+
+            var isInterfaceProp = property is UhtInterfaceProperty;
+
+            var passCppArgsByRef = caps.HasAnyFlags(UhtPropertyCaps.PassCppArgsByRef);
+
+            var isConstParam = property.PropertyFlags.HasAnyFlags(EPropertyFlags.ConstParm) ||
+                               (isInterfaceProp && !property.PropertyFlags.HasAnyFlags(EPropertyFlags.OutParm));
+
+            var isConstArgsByRef = property.ArrayDimensions == null && passCppArgsByRef &&
+                                   !property.PropertyFlags.HasAnyFlags(
+                                       EPropertyFlags.ConstParm | EPropertyFlags.OutParm);
+
+            var isOnConstClass = false;
+
+            if (property is UhtObjectProperty objectProperty)
+            {
+                isOnConstClass = objectProperty.Class.ClassFlags.HasAnyFlags(EClassFlags.Const) ||
+                                 objectProperty.MetaData.TryGetValue(UhtNames.NativeConst, out _);
+            }
+
+            var shouldHaveRef =
+                property.PropertyFlags.HasAnyFlags(EPropertyFlags.OutParm | EPropertyFlags.ReferenceParm);
+
+            var constAtTheBeginning = isOnConstClass || isConstArgsByRef || (isConstParam && !shouldHaveRef);
+
+            if (constAtTheBeginning)
+            {
+                builder.Append("const ");
+            }
+
+            if (property is UhtEnumProperty enumProperty)
+            {
+                builder.Append(enumProperty.Enum.CppType);
+            }
+            else
+            {
+                property.AppendText(builder, UhtPropertyTextType.ExportMember);
+            }
+
+            var fromConstClass = false;
+
+            if (property.Outer is UhtClass outerClass)
+            {
+                fromConstClass = outerClass.ClassFlags.HasAnyFlags(EClassFlags.Const);
+            }
+
+            var constAtTheEnd = (fromConstClass || (isConstParam && shouldHaveRef)) && !constAtTheBeginning;
+
+            if (constAtTheEnd)
+            {
+                builder.Append(" const");
+            }
+
+            shouldHaveRef = passCppArgsByRef;
+
+            if (shouldHaveRef ||
+                property.ArrayDimensions == null &&
+                (passCppArgsByRef ||
+                 property.PropertyFlags.HasAnyFlags(EPropertyFlags.OutParm | EPropertyFlags.ReferenceParm)))
+            {
+                if (property is UhtArrayProperty or UhtStrProperty or UhtTextProperty or UhtInterfaceProperty
+                    or UhtSoftObjectProperty)
+                {
+                    if (property.RefQualifier != UhtPropertyRefQualifier.None)
+                    {
+                        builder.Append('&');
+                    }
+                }
+                else
+                {
+                    builder.Append('&');
+                }
+            }
+
+            return builder.ToString();
+        }
+
+        private static string GetReturnPropertySignature(UhtProperty? property)
+        {
+            if (property == null)
+            {
+                return "void";
+            }
+
+            var builder = new StringBuilder();
+
+            if (property.PropertyFlags.HasAnyFlags(EPropertyFlags.ConstParm) ||
+                property.RefQualifier == UhtPropertyRefQualifier.ConstRef)
+            {
+                builder.Append("const ");
+            }
+
+            if (property is UhtEnumProperty enumProperty)
+            {
+                builder.Append(enumProperty.Enum.CppType);
+            }
+            else
+            {
+                property.AppendText(builder, UhtPropertyTextType.ExportMember);
+            }
+
+            if (property.RefQualifier == UhtPropertyRefQualifier.ConstRef)
+            {
+                builder.Append('&');
+            }
+
+            return builder.ToString();
+        }
+
+        private static string GetFunctionSignature(UhtClass classObj, UhtFunction function)
+        {
+            var builder = new StringBuilder();
+
+            builder.Append(GetReturnPropertySignature(function.ReturnProperty));
+
+            if (function.FunctionFlags.HasAnyFlags(EFunctionFlags.Static))
+            {
+                builder.Append("(*)");
+            }
+            else
+            {
+                builder.Append($"({classObj.SourceName}::*)");
+            }
+
+            builder.Append('(');
+
+            foreach (var child in function.Children)
+            {
+                if (child is UhtProperty property)
+                {
+                    if (!property.PropertyFlags.HasAnyFlags(EPropertyFlags.ReturnParm))
+                    {
+                        builder.Append(GetParamPropertySignature(property));
+
+                        builder.Append(", ");
+                    }
+                }
+            }
+
+            if (function.HasParameters)
+            {
+                builder.Remove(builder.Length - 2, 2);
+            }
+
+            builder.Append(')');
+
+            if (function.FunctionFlags.HasAnyFlags(EFunctionFlags.Const))
+            {
+                builder.Append("const");
+            }
+
+            return builder.ToString();
+        }
+
+        private static string GetFunctionParamName(UhtFunction function)
+        {
+            var builder = new StringBuilder();
+
+            builder.Append("TArray<FString>{");
+
+            foreach (var child in function.Children)
+            {
+                if (child is UhtProperty property)
+                {
+                    if (!property.PropertyFlags.HasAnyFlags(EPropertyFlags.ReturnParm))
+                    {
+                        builder.Append($"\"{property.SourceName}\", ");
+                    }
+                }
+            }
+
+            builder.Remove(builder.Length - 2, 2);
+
+            builder.Append('}');
+
+            return builder.ToString();
+        }
+
+        private static string GetFunctionDefaultValue(UhtFunction function)
+        {
+            if (!function.FunctionFlags.HasAnyFlags(EFunctionFlags.BlueprintCallable))
+            {
+                return "";
+            }
+
+            var builder = new StringBuilder();
+
+            foreach (var child in function.Children)
+            {
+                if (child is UhtProperty property)
+                {
+                    if (!property.PropertyFlags.HasAnyFlags(EPropertyFlags.ReturnParm))
+                    {
+                        if (property.DefaultValueTokens != null)
+                        {
+                            builder.Append(", ");
+
+                            var DefaultValue = new StringBuilder();
+
+                            foreach (var ValueToken in property.DefaultValueTokens)
+                            {
+                                DefaultValue.Append(ValueToken.Value);
+                            }
+
+                            if (DefaultValue.ToString() == "NULL")
+                            {
+                                builder.Append("nullptr");
+                            }
+                            else
+                            {
+                                builder.Append(DefaultValue);
+                            }
+                        }
+                    }
+                }
+            }
+
+            return builder.ToString();
+        }
+
+        private string GetHeaderFile(UhtClass classObj)
+        {
+            return Path.Combine(HeaderPath[classObj.Package.ShortName], classObj.HeaderFile.FilePath);
+        }
+
+        private static string GenerateInclude(string file)
+        {
+            return $"#include \"{file}\"\r\n";
+        }
+
+        private string GetInclude(UhtClass classObj)
+        {
+            return GenerateInclude(GetHeaderFile(classObj));
+        }
+
+        private void SaveIfChanged(string file, StringBuilder builder)
+        {
+            if (File.Exists(file) && File.ReadAllText(file) == builder.ToString())
+            {
+                return;
+            }
+
+            Factory.CommitOutput(file, builder);
+        }
+
+        private void GetPlugins(string InPathName, Dictionary<string, string> Plugins)
+        {
+            var Suffix = "*.uplugin";
+
+            var DirectoryInfo = new DirectoryInfo(InPathName);
+
+            foreach (var Item in DirectoryInfo.GetFiles(Suffix))
+            {
+                if (Item.DirectoryName != null)
+                {
+                    Plugins[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName;
+                }
+            }
+
+            foreach (var Directories in DirectoryInfo.GetDirectories())
+            {
+                foreach (var Item in Directories.GetFiles(Suffix, SearchOption.AllDirectories))
+                {
+                    if (Item.DirectoryName != null)
+                    {
+                        Plugins[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName;
+                    }
+                }
+            }
+        }
+
+        private void GetModules(string InPathName, Dictionary<string, string> Modules)
+        {
+            var Suffix = "*.Build.cs";
+
+            var DirectoryInfo = new DirectoryInfo(InPathName);
+
+            foreach (var Item in DirectoryInfo.GetFiles(Suffix))
+            {
+                if (Item.DirectoryName != null)
+                {
+                    Modules[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName;
+                }
+            }
+
+            foreach (var Directories in DirectoryInfo.GetDirectories())
+            {
+                foreach (var Item in Directories.GetFiles(Suffix, SearchOption.AllDirectories))
+                {
+                    if (Item.DirectoryName != null)
+                    {
+                        Modules[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName;
+                    }
+                }
+            }
+        }
+
+        private void GetModules(string InPathName, HashSet<string> Modules)
+        {
+            var Suffix = "*.Build.cs";
+
+            var DirectoryInfo = new DirectoryInfo(InPathName);
+
+            foreach (var Item in DirectoryInfo.GetFiles(Suffix))
+            {
+                Modules.Add(Item.Name.Remove(Item.Name.Length - Suffix.Length + 1));
+            }
+
+            foreach (var Directories in DirectoryInfo.GetDirectories())
+            {
+                foreach (var Item in Directories.GetFiles(Suffix, SearchOption.AllDirectories))
+                {
+                    if (Item.DirectoryName != null)
+                    {
+                        Modules.Add(Item.Name.Remove(Item.Name.Length - Suffix.Length + 1));
+                    }
+                }
+            }
+        }
+    }
+}

SourceCodeGenerator.ubtplugin.csproj

@@ -0,0 +1,49 @@
+<Project Sdk="Microsoft.NET.Sdk">
+  <Import Project="SourceCodeGenerator.ubtplugin.csproj.props" Condition="Exists('SourceCodeGenerator.ubtplugin.csproj.props')" />
+  <Import Project="$(EngineDir)\Source\Programs\Shared\UnrealEngine.csproj.props" />
+  <PropertyGroup>
+    <TargetFramework>net6.0</TargetFramework>
+    <Configuration Condition=" '$(Configuration)' == '' ">Development</Configuration>
+    <OutputType>Library</OutputType>
+    <AppendTargetFrameworkToOutputPath>false</AppendTargetFrameworkToOutputPath>
+    <GenerateAssemblyInfo>false</GenerateAssemblyInfo>
+    <GenerateTargetFrameworkAttribute>false</GenerateTargetFrameworkAttribute>
+    <Configurations>Debug;Release;Development</Configurations>
+    <RootNamespace>SourceCodeGenerator</RootNamespace>
+    <AssemblyName>SourceCodeGenerator</AssemblyName>
+    <OutputPath>..\..\..\..\Binaries\DotNET\UnrealBuildTool\Plugins\SourceCodeGenerator\</OutputPath>
+    <TreatWarningsAsErrors>true</TreatWarningsAsErrors>
+    <DebugType>pdbonly</DebugType>
+    <Nullable>enable</Nullable>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Development|AnyCPU' ">
+    <DefineConstants>$(DefineConstants);TRACE</DefineConstants>
+    <Optimize>true</Optimize>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
+    <DefineConstants>$(DefineConstants);TRACE</DefineConstants>
+    <Optimize>true</Optimize>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
+    <DefineConstants>$(DefineConstants);DEBUG;TRACE</DefineConstants>
+    <DebugSymbols>true</DebugSymbols>
+    <DebugType>full</DebugType>
+  </PropertyGroup>
+  <ItemGroup>
+    <PackageReference Include="Microsoft.CSharp" Version="4.7.0" />
+  </ItemGroup>
+  <ItemGroup>
+    <Reference Include="EpicGames.Build">
+      <HintPath>$(EngineDir)\Binaries\DotNET\UnrealBuildTool\EpicGames.Build.dll</HintPath>
+    </Reference>
+    <Reference Include="EpicGames.Core">
+      <HintPath>$(EngineDir)\Binaries\DotNET\UnrealBuildTool\EpicGames.Core.dll</HintPath>
+    </Reference>
+    <Reference Include="EpicGames.UHT">
+      <HintPath>$(EngineDir)\Binaries\DotNET\UnrealBuildTool\EpicGames.UHT.dll</HintPath>
+    </Reference>
+    <Reference Include="UnrealBuildTool">
+      <HintPath>$(EngineDir)\Binaries\DotNET\UnrealBuildTool\UnrealBuildTool.dll</HintPath>
+    </Reference>
+  </ItemGroup>
+</Project>