Projucer: Mark modified files with an asterisk in the file tree panel

extras/Projucer/Source/Application/jucer_Application.cpp

@@ -1055,7 +1055,7 @@ bool ProjucerApplication::perform (const InvocationInfo& info)
         case CommandIDs::newPIP:                    createNewPIP(); break;
         case CommandIDs::open:                      askUserToOpenFile(); break;
         case CommandIDs::launchDemoRunner:          launchDemoRunner(); break;
-        case CommandIDs::saveAll:                   openDocumentManager.saveAll(); break;
+        case CommandIDs::saveAll:                   saveAllDocuments(); break;
         case CommandIDs::closeAllWindows:           closeAllMainWindowsAndQuitIfNeeded(); break;
         case CommandIDs::closeAllDocuments:         closeAllDocuments (true); break;
         case CommandIDs::clearRecentFiles:          clearRecentFiles(); break;
@@ -1120,6 +1120,15 @@ bool ProjucerApplication::openFile (const File& file)
     return mainWindowList.openFile (file);
 }
 
+void ProjucerApplication::saveAllDocuments()
+{
+    openDocumentManager.saveAll();
+
+    for (int i = 0; i < mainWindowList.windows.size(); ++i)
+        if (auto* pcc = mainWindowList.windows.getUnchecked(i)->getProjectContentComponent())
+            pcc->refreshProjectTreeFileStatuses();
+}
+
 bool ProjucerApplication::closeAllDocuments (bool askUserToSave)
 {
     return openDocumentManager.closeAll (askUserToSave);

extras/Projucer/Source/Application/jucer_Application.h

@@ -96,6 +96,7 @@ public:
     void createNewPIP();
     void askUserToOpenFile();
     bool openFile (const File&);
+    void saveAllDocuments();
     bool closeAllDocuments (bool askUserToSave);
     bool closeAllMainWindows();
     void closeAllMainWindowsAndQuitIfNeeded();

extras/Projucer/Source/CodeEditor/jucer_DocumentEditorComponent.cpp

@@ -47,13 +47,20 @@ bool DocumentEditorComponent::documentAboutToClose (OpenDocumentManager::Documen
     {
         jassert (document != nullptr);
 
-        if (ProjectContentComponent* pcc = findParentComponentOfClass<ProjectContentComponent>())
+        if (auto* pcc = findParentComponentOfClass<ProjectContentComponent>())
             pcc->hideDocument (document);
     }
 
     return true;
 }
 
-void DocumentEditorComponent::setEditedState (bool /*hasBeenEdited*/)
+void DocumentEditorComponent::setEditedState (bool hasBeenEdited)
 {
+    if (hasBeenEdited != lastEditedState)
+    {
+        if (auto* pcc = findParentComponentOfClass<ProjectContentComponent>())
+            pcc->refreshProjectTreeFileStatuses();
+
+        lastEditedState = hasBeenEdited;
+    }
 }

extras/Projucer/Source/CodeEditor/jucer_DocumentEditorComponent.h

@@ -43,6 +43,7 @@ public:
 
 protected:
     OpenDocumentManager::Document* document;
+    bool lastEditedState = false;
 
     void setEditedState (bool hasBeenEdited);
 

extras/Projucer/Source/Project/UI/Sidebar/jucer_FileTreeItems.h

@@ -465,6 +465,21 @@ public:
         return getFile().getFileName();
     }
 
+    void paintItem (Graphics& g, int width, int height) override
+    {
+        JucerTreeViewBase::paintItem (g, width, height);
+
+        if (item.needsSaving())
+        {
+            auto bounds = g.getClipBounds().withY (0).withHeight (height);
+
+            g.setFont (getFont());
+            g.setColour (getContentColour (false));
+
+            g.drawFittedText ("*", bounds.removeFromLeft (height), Justification::centred, 1);
+        }
+    }
+
     static File findCorrespondingHeaderOrCpp (const File& f)
     {
         if (f.hasFileExtension (sourceFileExtensions))  return f.withFileExtension (".h");

extras/Projucer/Source/Project/UI/jucer_ProjectContentComponent.cpp

@@ -334,6 +334,13 @@ void ProjectContentComponent::changeListenerCallback (ChangeBroadcaster*)
     updateMissingFileStatuses();
 }
 
+void ProjectContentComponent::refreshProjectTreeFileStatuses()
+{
+    if (auto* projectTab = getProjectTab())
+        if (auto* fileTree = projectTab->getFileTreePanel())
+            fileTree->repaint();
+}
+
 void ProjectContentComponent::updateMissingFileStatuses()
 {
     if (auto* pTab = getProjectTab())
@@ -482,6 +489,8 @@ void ProjectContentComponent::saveDocument()
     {
         if (! currentDocument->save())
             showSaveWarning (currentDocument);
+
+        refreshProjectTreeFileStatuses();
     }
     else
     {
@@ -491,8 +500,13 @@ void ProjectContentComponent::saveDocument()
 
 void ProjectContentComponent::saveAs()
 {
-    if (currentDocument != nullptr && ! currentDocument->saveAs())
-        showSaveWarning (currentDocument);
+    if (currentDocument != nullptr)
+    {
+        if (! currentDocument->saveAs())
+            showSaveWarning (currentDocument);
+
+        refreshProjectTreeFileStatuses();
+    }
 }
 
 bool ProjectContentComponent::goToPreviousFile()

extras/Projucer/Source/Project/UI/jucer_ProjectContentComponent.h

@@ -97,6 +97,7 @@ public:
 
     void deleteSelectedTreeItems();
 
+    void refreshProjectTreeFileStatuses();
     void updateMissingFileStatuses();
     void createProjectTabs();
     void deleteProjectTabs();

extras/Projucer/Source/Project/jucer_Project.cpp

@@ -1574,6 +1574,23 @@ bool Project::Item::isIconCrossedOut() const
                    || getFile().hasFileExtension (headerFileExtensions));
 }
 
+bool Project::Item::needsSaving() const noexcept
+{
+    auto& odm = ProjucerApplication::getApp().openDocumentManager;
+
+    if (odm.anyFilesNeedSaving())
+    {
+        for (int i = 0; i < odm.getNumOpenDocuments(); ++i)
+        {
+            auto* doc = odm.getOpenDocument (i);
+            if (doc->needsSaving() && doc->getFile() == getFile())
+                return true;
+        }
+    }
+
+    return false;
+}
+
 //==============================================================================
 ValueTree Project::getConfigNode()
 {

extras/Projucer/Source/Project/jucer_Project.h

@@ -300,6 +300,8 @@ public:
         Icon getIcon (bool isOpen = false) const;
         bool isIconCrossedOut() const;
 
+        bool needsSaving() const noexcept;
+
         Project& project;
         ValueTree state;