Sync csproj when files are changed from the Godot FileSystem dock

6 files changed, 179 insertions, 15 deletions

diff --git a/modules/mono/editor/GodotTools/GodotTools.Core/StringExtensions.cs b/modules/mono/editor/GodotTools/GodotTools.Core/StringExtensions.cs
index b531b6aeee..326c49f096 100644
--- a/modules/mono/editor/GodotTools/GodotTools.Core/StringExtensions.cs
+++ b/modules/mono/editor/GodotTools/GodotTools.Core/StringExtensions.cs
@@ -30,7 +30,7 @@ namespace GodotTools.Core
 
             path = string.Join(Path.DirectorySeparatorChar.ToString(), parts).Trim();
 
-            return rooted ? Path.DirectorySeparatorChar.ToString() + path : path;
+            return rooted ? Path.DirectorySeparatorChar + path : path;
         }
 
         private static readonly string driveRoot = Path.GetPathRoot(Environment.CurrentDirectory);
diff --git a/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectExtensions.cs b/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectExtensions.cs
index 36961eb45e..f0e0d1b33d 100644
--- a/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectExtensions.cs
+++ b/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectExtensions.cs
@@ -1,5 +1,7 @@
 using GodotTools.Core;
 using System;
+using System.Collections.Generic;
+using System.IO;
 using DotNet.Globbing;
 using Microsoft.Build.Construction;
 
@@ -7,16 +9,15 @@ namespace GodotTools.ProjectEditor
 {
     public static class ProjectExtensions
     {
-        public static bool HasItem(this ProjectRootElement root, string itemType, string include)
+        public static ProjectItemElement FindItemOrNull(this ProjectRootElement root, string itemType, string include, bool noCondition = false)
         {
-            GlobOptions globOptions = new GlobOptions();
-            globOptions.Evaluation.CaseInsensitive = false;
+            GlobOptions globOptions = new GlobOptions {Evaluation = {CaseInsensitive = false}};
 
             string normalizedInclude = include.NormalizePath();
 
             foreach (var itemGroup in root.ItemGroups)
             {
-                if (itemGroup.Condition.Length != 0)
+                if (noCondition && itemGroup.Condition.Length != 0)
                     continue;
 
                 foreach (var item in itemGroup.Items)
@@ -27,18 +28,65 @@ namespace GodotTools.ProjectEditor
                     var glob = Glob.Parse(item.Include.NormalizePath(), globOptions);
 
                     if (glob.IsMatch(normalizedInclude))
-                    {
-                        return true;
-                    }
+                        return item;
                 }
             }
 
-            return false;
+            return null;
+        }
+        public static ProjectItemElement FindItemOrNullAbs(this ProjectRootElement root, string itemType, string include, bool noCondition = false)
+        {
+            GlobOptions globOptions = new GlobOptions {Evaluation = {CaseInsensitive = false}};
+
+            string normalizedInclude = Path.GetFullPath(include).NormalizePath();
+
+            foreach (var itemGroup in root.ItemGroups)
+            {
+                if (noCondition && itemGroup.Condition.Length != 0)
+                    continue;
+
+                foreach (var item in itemGroup.Items)
+                {
+                    if (item.ItemType != itemType)
+                        continue;
+
+                    var glob = Glob.Parse(Path.GetFullPath(item.Include).NormalizePath(), globOptions);
+
+                    if (glob.IsMatch(normalizedInclude))
+                        return item;
+                }
+            }
+
+            return null;
+        }
+
+        public static IEnumerable<ProjectItemElement> FindAllItemsInFolder(this ProjectRootElement root, string itemType, string folder)
+        {
+            string absFolderNormalizedWithSep = Path.GetFullPath(folder).NormalizePath() + Path.DirectorySeparatorChar;
+
+            foreach (var itemGroup in root.ItemGroups)
+            {
+                foreach (var item in itemGroup.Items)
+                {
+                    if (item.ItemType != itemType)
+                        continue;
+
+                    string absPathNormalized = Path.GetFullPath(item.Include).NormalizePath();
+
+                    if (absPathNormalized.StartsWith(absFolderNormalizedWithSep))
+                        yield return item;
+                }
+            }
+        }
+
+        public static bool HasItem(this ProjectRootElement root, string itemType, string include, bool noCondition = false)
+        {
+            return root.FindItemOrNull(itemType, include, noCondition) != null;
         }
 
         public static bool AddItemChecked(this ProjectRootElement root, string itemType, string include)
         {
-            if (!root.HasItem(itemType, include))
+            if (!root.HasItem(itemType, include, noCondition: true))
             {
                 root.AddItem(itemType, include);
                 return true;
@@ -47,6 +95,18 @@ namespace GodotTools.ProjectEditor
             return false;
         }
 
+        public static bool RemoveItemChecked(this ProjectRootElement root, string itemType, string include)
+        {
+            var item = root.FindItemOrNullAbs(itemType, include);
+            if (item != null)
+            {
+                item.Parent.RemoveChild(item);
+                return true;
+            }
+
+            return false;
+        }
+
         public static Guid GetGuid(this ProjectRootElement root)
         {
             foreach (var property in root.Properties)
diff --git a/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectUtils.cs b/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectUtils.cs
index af36f125f5..1776b46e6a 100644
--- a/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectUtils.cs
+++ b/modules/mono/editor/GodotTools/GodotTools.ProjectEditor/ProjectUtils.cs
@@ -23,6 +23,79 @@ namespace GodotTools.ProjectEditor
                 root.Save();
         }
 
+        public static void RenameItemInProjectChecked(string projectPath, string itemType, string oldInclude, string newInclude)
+        {
+            var dir = Directory.GetParent(projectPath).FullName;
+            var root = ProjectRootElement.Open(projectPath);
+            Debug.Assert(root != null);
+
+            var normalizedOldInclude = oldInclude.NormalizePath();
+            var normalizedNewInclude = newInclude.NormalizePath();
+
+            var item = root.FindItemOrNullAbs(itemType, normalizedOldInclude);
+
+            if (item == null)
+                return;
+
+            item.Include = normalizedNewInclude.RelativeToPath(dir).Replace("/", "\\");
+            root.Save();
+        }
+
+        public static void RemoveItemFromProjectChecked(string projectPath, string itemType, string include)
+        {
+            var dir = Directory.GetParent(projectPath).FullName;
+            var root = ProjectRootElement.Open(projectPath);
+            Debug.Assert(root != null);
+
+            var normalizedInclude = include.NormalizePath();
+
+            if (root.RemoveItemChecked(itemType, normalizedInclude))
+                root.Save();
+        }
+
+        public static void RenameItemsToNewFolderInProjectChecked(string projectPath, string itemType, string oldFolder, string newFolder)
+        {
+            var dir = Directory.GetParent(projectPath).FullName;
+            var root = ProjectRootElement.Open(projectPath);
+            Debug.Assert(root != null);
+
+            bool dirty = false;
+
+            var oldFolderNormalized = oldFolder.NormalizePath();
+            var newFolderNormalized = newFolder.NormalizePath();
+            string absOldFolderNormalized = Path.GetFullPath(oldFolderNormalized).NormalizePath();
+            string absNewFolderNormalized = Path.GetFullPath(newFolderNormalized).NormalizePath();
+
+            foreach (var item in root.FindAllItemsInFolder(itemType, oldFolderNormalized))
+            {
+                string absPathNormalized = Path.GetFullPath(item.Include).NormalizePath();
+                string absNewIncludeNormalized = absNewFolderNormalized + absPathNormalized.Substring(absOldFolderNormalized.Length);
+                item.Include = absNewIncludeNormalized.RelativeToPath(dir).Replace("/", "\\");
+                dirty = true;
+            }
+
+            if (dirty)
+                root.Save();
+        }
+
+        public static void RemoveItemsInFolderFromProjectChecked(string projectPath, string itemType, string folder)
+        {
+            var root = ProjectRootElement.Open(projectPath);
+            Debug.Assert(root != null);
+
+            var folderNormalized = folder.NormalizePath();
+
+            var itemsToRemove = root.FindAllItemsInFolder(itemType, folderNormalized).ToList();
+
+            if (itemsToRemove.Count > 0)
+            {
+                foreach (var item in itemsToRemove)
+                    item.Parent.RemoveChild(item);
+
+                root.Save();
+            }
+        }
+
         private static string[] GetAllFilesRecursive(string rootDirectory, string mask)
         {
             string[] files = Directory.GetFiles(rootDirectory, mask, SearchOption.AllDirectories);
diff --git a/modules/mono/editor/GodotTools/GodotTools/Export/ExportPlugin.cs b/modules/mono/editor/GodotTools/GodotTools/Export/ExportPlugin.cs
index 05f84f547b..022005ad0b 100644
--- a/modules/mono/editor/GodotTools/GodotTools/Export/ExportPlugin.cs
+++ b/modules/mono/editor/GodotTools/GodotTools/Export/ExportPlugin.cs
@@ -96,7 +96,7 @@ namespace GodotTools.Export
             if (type != Internal.CSharpLanguageType)
                 return;
 
-            if (Path.GetExtension(path) != $".{Internal.CSharpLanguageExtension}")
+            if (Path.GetExtension(path) != Internal.CSharpLanguageExtension)
                 throw new ArgumentException($"Resource of type {Internal.CSharpLanguageType} has an invalid file extension: {path}", nameof(path));
 
             // TODO What if the source file is not part of the game's C# project
diff --git a/modules/mono/editor/GodotTools/GodotTools/GodotSharpEditor.cs b/modules/mono/editor/GodotTools/GodotTools/GodotSharpEditor.cs
index e4e391765e..e6d5dd9895 100644
--- a/modules/mono/editor/GodotTools/GodotTools/GodotSharpEditor.cs
+++ b/modules/mono/editor/GodotTools/GodotTools/GodotSharpEditor.cs
@@ -13,6 +13,7 @@ using JetBrains.Annotations;
 using static GodotTools.Internals.Globals;
 using File = GodotTools.Utils.File;
 using OS = GodotTools.Utils.OS;
+using Path = System.IO.Path;
 
 namespace GodotTools
 {
@@ -61,7 +62,7 @@ namespace GodotTools
                     {
                         Guid = guid,
                         PathRelativeToSolution = name + ".csproj",
-                        Configs = new List<string> { "Debug", "ExportDebug", "ExportRelease" }
+                        Configs = new List<string> {"Debug", "ExportDebug", "ExportRelease"}
                     };
 
                     solution.AddNewProject(name, projectInfo);
@@ -161,6 +162,36 @@ namespace GodotTools
                     // Once shown a first time, it can be seen again via the Mono menu - it doesn't have to be exclusive from that time on.
                     aboutDialog.PopupExclusive = false;
                 }
+
+                var fileSystemDock = GetEditorInterface().GetFileSystemDock();
+
+                fileSystemDock.FilesMoved += (file, newFile) =>
+                {
+                    if (Path.GetExtension(file) == Internal.CSharpLanguageExtension)
+                    {
+                        ProjectUtils.RenameItemInProjectChecked(GodotSharpDirs.ProjectCsProjPath, "Compile",
+                            ProjectSettings.GlobalizePath(file), ProjectSettings.GlobalizePath(newFile));
+                    }
+                };
+
+                fileSystemDock.FileRemoved += file =>
+                {
+                    if (Path.GetExtension(file) == Internal.CSharpLanguageExtension)
+                        ProjectUtils.RemoveItemFromProjectChecked(GodotSharpDirs.ProjectCsProjPath, "Compile",
+                            ProjectSettings.GlobalizePath(file));
+                };
+
+                fileSystemDock.FolderMoved += (oldFolder, newFolder) =>
+                {
+                    ProjectUtils.RenameItemsToNewFolderInProjectChecked(GodotSharpDirs.ProjectCsProjPath, "Compile",
+                        ProjectSettings.GlobalizePath(oldFolder), ProjectSettings.GlobalizePath(newFolder));
+                };
+
+                fileSystemDock.FolderRemoved += oldFolder =>
+                {
+                    ProjectUtils.RemoveItemsInFolderFromProjectChecked(GodotSharpDirs.ProjectCsProjPath, "Compile",
+                        ProjectSettings.GlobalizePath(oldFolder));
+                };
             }
         }
 
@@ -339,7 +370,7 @@ namespace GodotTools
 
             bottomPanelBtn = AddControlToBottomPanel(BottomPanel, "Mono".TTR());
 
-            AddChild(new HotReloadAssemblyWatcher { Name = "HotReloadAssemblyWatcher" });
+            AddChild(new HotReloadAssemblyWatcher {Name = "HotReloadAssemblyWatcher"});
 
             menuPopup = new PopupMenu();
             menuPopup.Hide();
@@ -387,7 +418,7 @@ namespace GodotTools
                 EditorDef("mono/editor/show_info_on_start", true);
 
                 // CheckBox in main container
-                aboutDialogCheckBox = new CheckBox { Text = "Show this warning when starting the editor" };
+                aboutDialogCheckBox = new CheckBox {Text = "Show this warning when starting the editor"};
                 aboutDialogCheckBox.Toggled += enabled =>
                 {
                     bool showOnStart = (bool)editorSettings.GetSetting("mono/editor/show_info_on_start");
diff --git a/modules/mono/editor/GodotTools/GodotTools/Internals/Internal.cs b/modules/mono/editor/GodotTools/GodotTools/Internals/Internal.cs
index 2e121ba879..026a7db89c 100644
--- a/modules/mono/editor/GodotTools/GodotTools/Internals/Internal.cs
+++ b/modules/mono/editor/GodotTools/GodotTools/Internals/Internal.cs
@@ -8,7 +8,7 @@ namespace GodotTools.Internals
     public static class Internal
     {
         public const string CSharpLanguageType = "CSharpScript";
-        public const string CSharpLanguageExtension = "cs";
+        public const string CSharpLanguageExtension = ".cs";
 
         public static string UpdateApiAssembliesFromPrebuilt(string config) =>
             internal_UpdateApiAssembliesFromPrebuilt(config);