Merge branch 'module-remove' of https://github.com/Tryibion/FlaxEngine into Tryibion-module-remove

This commit is contained in:
Wojtek Figat
2024-09-11 13:41:30 +02:00
2 changed files with 91 additions and 0 deletions

View File

@@ -703,6 +703,10 @@ namespace FlaxEditor.Modules
}
else
{
// Try to remove module if build.cs file is being deleted
if (item.Path.Contains(".Build.cs", StringComparison.Ordinal) && item.ItemType == ContentItemType.Script)
Editor.Instance.CodeEditing.RemoveModule(item.Path);
// Check if it's an asset
if (item.IsAsset)
{

View File

@@ -404,6 +404,93 @@ namespace FlaxEditor.Modules.SourceCodeEditing
}
}
internal void RemoveModule(string path)
{
if (!File.Exists(path))
return;
// Read text, figure out if it is an editor module or game module
var editorModule = false;
var moduleTextIndex = -1;
var fileText = File.ReadAllText(path);
if (fileText.Contains("GameModule", StringComparison.Ordinal))
{
moduleTextIndex = fileText.IndexOf("GameModule", StringComparison.Ordinal);
}
else if (fileText.Contains("ThirdPartyModule", StringComparison.Ordinal))
{
moduleTextIndex = fileText.IndexOf("ThirdPartyModule", StringComparison.Ordinal);
}
else if (fileText.Contains("DepsModule", StringComparison.Ordinal))
{
moduleTextIndex = fileText.IndexOf("DepsModule", StringComparison.Ordinal);
}
else if (fileText.Contains("GameEditorModule", StringComparison.Ordinal))
{
moduleTextIndex = fileText.IndexOf("GameEditorModule", StringComparison.Ordinal);
editorModule = true;
}
else
{
// If it does not contain a module, then this could be target file and not a module file
return;
}
// Get module name
var classTextIndex = fileText.IndexOf("class ", StringComparison.Ordinal);
var className = fileText.Substring(classTextIndex, moduleTextIndex - classTextIndex).Replace("class ", "").Replace(":", "").Trim();
Editor.Log($"Removing Module: {className}");
// Find target files
// Assume Target files are in the source directory that is up 2 levels
var sourceDirectoryInfo = Directory.GetParent(path)?.Parent;
if (sourceDirectoryInfo != null)
{
var sourceFiles = Directory.GetFiles(sourceDirectoryInfo.FullName);
// Search target files for module name and remove it
foreach (var file in sourceFiles)
{
string fileName = Path.GetFileName(file);
if (file.Contains(".Build.cs", StringComparison.OrdinalIgnoreCase))
{
var targetText = File.ReadAllText(file);
// Skip game project if it is suppose to be an editor module
if (editorModule && targetText.Contains("GameProjectTarget", StringComparison.Ordinal))
continue;
var newText = targetText;
bool removedModuleText = false;
if (targetText.Contains($"Modules.Add(\"{className}\")", StringComparison.Ordinal))
{
newText = newText.Replace($"Modules.Add(\"{className}\");\n", "", StringComparison.Ordinal).Replace($"Modules.Add(\"{className}\");", "", StringComparison.Ordinal);
removedModuleText = true;
}
if (targetText.Contains($"Modules.Add(nameof({className}))", StringComparison.Ordinal))
{
newText = newText.Replace($"Modules.Add(nameof({className}));\n", "", StringComparison.Ordinal).Replace($"Modules.Add(nameof({className}));", "", StringComparison.Ordinal);
removedModuleText = true;
}
if (removedModuleText)
{
File.WriteAllText(file, newText);
Editor.Log($"Removed Module: {className} from {file}");
}
}
// Remove Generated module files
else if (fileName.Equals($"{className}.csproj", StringComparison.Ordinal) ||
fileName.Equals($"{className}.Gen.cs", StringComparison.Ordinal) ||
fileName.Equals($"{className}.Gen.cpp", StringComparison.Ordinal) ||
fileName.Equals($"{className}.Gen.h", StringComparison.Ordinal))
{
File.Delete(file);
Editor.Log($"Deleted generated modules file for module: {className}. File path {file}");
}
}
}
}
/// <inheritdoc />
public override void OnUpdate()
{