fix(editor): file modification status is not updated while closing files (fixes #1485)

Author: itsaky

.idea/misc.xml

@@ -465,6 +465,8 @@ abstract class BaseEditorActivity :
           viewContainer.displayedChild = 0
         }
       }
+
+      invalidateOptionsMenu()
     }
 
     setupNoEditorView()

app/src/main/java/com/itsaky/androidide/activities/editor/BaseEditorActivity.kt

@@ -28,7 +28,6 @@ import androidx.appcompat.view.menu.MenuBuilder
 import androidx.collection.MutableIntObjectMap
 import androidx.core.view.GravityCompat
 import com.blankj.utilcode.util.ImageUtils
-import com.blankj.utilcode.util.ThreadUtils
 import com.itsaky.androidide.R.string
 import com.itsaky.androidide.actions.ActionData
 import com.itsaky.androidide.actions.ActionItem.Location.EDITOR_TOOLBAR
@@ -51,7 +50,6 @@ import com.itsaky.androidide.models.Range
 import com.itsaky.androidide.models.SaveResult
 import com.itsaky.androidide.projects.ProjectManagerImpl
 import com.itsaky.androidide.tasks.executeAsync
-import com.itsaky.androidide.tasks.executeAsyncProvideError
 import com.itsaky.androidide.ui.CodeEditorView
 import com.itsaky.androidide.utils.DialogUtils.newYesNoDialog
 import com.itsaky.androidide.utils.IntentUtils.openImage
@@ -109,6 +107,9 @@ open class EditorHandlerActivity : ProjectHandlerActivity(), IEditorHandler {
       }
     }
 
+    editorViewModel._filesModified.observe(this) { invalidateOptionsMenu() }
+    editorViewModel._filesSaving.observe(this) { invalidateOptionsMenu() }
+
     editorViewModel.observeFiles(this) {
       // rewrite the cached files index if there are any opened files
       val currentFile =
@@ -413,41 +414,40 @@ open class EditorHandlerActivity : ProjectHandlerActivity(), IEditorHandler {
   }
 
   override suspend fun saveAllResult(progressConsumer: ((Int, Int) -> Unit)?): SaveResult {
-    setFilesSaving(true)
-    val result = SaveResult()
-    for (i in 0 until editorViewModel.getOpenedFileCount()) {
-      saveResultInternal(i, result, false)
-      progressConsumer?.invoke(i + 1, editorViewModel.getOpenedFileCount())
+    return performFileSave {
+      val result = SaveResult()
+      for (i in 0 until editorViewModel.getOpenedFileCount()) {
+        saveResultInternal(i, result)
+        progressConsumer?.invoke(i + 1, editorViewModel.getOpenedFileCount())
+      }
+
+      return@performFileSave result
     }
-    setFilesSaving(false)
-    return result
   }
 
   override suspend fun saveResult(index: Int, result: SaveResult) {
-    saveResultInternal(index, result, true)
+    performFileSave {
+      saveResultInternal(index, result)
+    }
   }
 
-  private suspend fun saveResultInternal(index: Int, result: SaveResult, updateSaving: Boolean) {
+  private suspend fun saveResultInternal(
+    index: Int,
+    result: SaveResult
+  ) : Boolean {
     if (index < 0 || index >= editorViewModel.getOpenedFileCount()) {
-      return
+      return false
     }
 
-    val frag = getEditorAtIndex(index) ?: return
-    val fileName = frag.file?.name ?: return
-
-    if (updateSaving) {
-      setFilesSaving(true)
-    }
+    val frag = getEditorAtIndex(index) ?: return false
+    val fileName = frag.file?.name ?: return false
 
     run {
       // Must be called before frag.save()
       // Otherwise, it'll always return false
       val modified = frag.isModified
       if (!frag.save()) {
-        if (updateSaving) {
-          setFilesSaving(false)
-        }
-        return
+        return false
       }
 
       val isGradle = fileName.endsWith(".gradle") || fileName.endsWith(".gradle.kts")
@@ -461,108 +461,117 @@ open class EditorHandlerActivity : ProjectHandlerActivity(), IEditorHandler {
       }
     }
 
-    var modified = false
-    for (file in editorViewModel.getOpenedFiles()) {
-      val editor = getEditorForFile(file) ?: continue
-      modified = modified || editor.isModified
-    }
-
-    val finalModified = modified
+    val hasUnsaved = hasUnsavedFiles()
 
     withContext(Dispatchers.Main) {
-      editorViewModel.apply {
-        setFilesModified(finalModified)
-      }
 
-      if (updateSaving) {
-        setFilesSaving(false)
-      }
+      editorViewModel.areFilesModified = hasUnsaved
 
       // set tab as unmodified
       val tab = binding.tabs.getTabAt(index) ?: return@withContext
       if (tab.text!!.startsWith('*')) {
         tab.text = tab.text!!.substring(startIndex = 1)
       }
     }
+
+    return true
   }
 
-  private suspend fun setFilesSaving(saving: Boolean) {
-    withContext(Dispatchers.Main.immediate) {
-      invalidateOptionsMenu()
-      editorViewModel.setFilesSaving(saving)
+  private fun hasUnsavedFiles() = editorViewModel.getOpenedFiles().any { file ->
+    getEditorForFile(file)?.isModified == true
+  }
+
+  private suspend inline fun <T : Any?> performFileSave(crossinline action: suspend () -> T) : T {
+    setFilesSaving(true)
+    try {
+      return action()
+    } finally {
+      setFilesSaving(false)
     }
   }
 
-  private fun onEditorContentChanged() {
-    editorViewModel.setFilesModified(true)
-    invalidateOptionsMenu()
+  private suspend fun setFilesSaving(saving: Boolean) {
+    withContext(Dispatchers.Main.immediate) {
+      editorViewModel.areFilesSaving = saving
+    }
   }
 
   override fun areFilesModified(): Boolean {
-    return editorViewModel.areFilesModified()
+    return editorViewModel.areFilesModified
   }
 
   override fun areFilesSaving(): Boolean {
-    return editorViewModel.areFilesSaving()
+    return editorViewModel.areFilesSaving
   }
 
   override fun closeFile(index: Int, runAfter: () -> Unit) {
-    if (index >= 0 && index < editorViewModel.getOpenedFileCount()) {
-      val opened: File = editorViewModel.getOpenedFile(index)
-      log.info("Closing file:", opened)
-      val editor = getEditorAtIndex(index)
-      if (editor != null && editor.isModified) {
-        notifyFilesUnsaved(listOf(editor)) { closeFile(index, runAfter) }
-        return
-      }
+    if (index < 0 || index >= editorViewModel.getOpenedFileCount()) {
+      log.error("Invalid file index. Cannot close.")
+      return
+    }
 
-      editor?.close() ?: run {
-        log.error("Cannot save file before close. Editor instance is null")
-      }
+    val opened = editorViewModel.getOpenedFile(index)
+    log.info("Closing file:", opened)
 
-      editorViewModel.removeFile(index)
-      binding.apply {
-        tabs.removeTabAt(index)
-        editorContainer.removeViewAt(index)
+    val editor = getEditorAtIndex(index)
+    if (editor?.isModified == true) {
+      log.info("File has been modified:", opened)
+      notifyFilesUnsaved(listOf(editor)) {
+        closeFile(index, runAfter)
       }
-
-      updateTabs()
-    } else {
-      log.error("Invalid file index. Cannot close.")
       return
     }
 
+    editor?.close() ?: run {
+      log.error("Cannot save file before close. Editor instance is null")
+    }
+
+    editorViewModel.removeFile(index)
+    binding.apply {
+      tabs.removeTabAt(index)
+      editorContainer.removeViewAt(index)
+    }
+
+    editorViewModel.areFilesModified = hasUnsavedFiles()
+
+    updateTabs()
     runAfter()
   }
 
   override fun closeOthers() {
+    if (editorViewModel.getOpenedFileCount() == 0) {
+      return
+    }
+
     val unsavedFiles =
       editorViewModel.getOpenedFiles().map(::getEditorForFile)
         .filter { it != null && it.isModified }
 
-    if (unsavedFiles.isEmpty()) {
-      val file = editorViewModel.getCurrentFile()
-      var index = 0
+    if (unsavedFiles.isNotEmpty()) {
+      notifyFilesUnsaved(unsavedFiles) { closeOthers() }
+      return
+    }
+
+    val file = editorViewModel.getCurrentFile()
+    var index = 0
 
-      // keep closing the file at index 0
-      // if openedFiles[0] == file, then keep closing files at index 1
-      while (editorViewModel.getOpenedFileCount() != 1) {
-        val editor = getEditorAtIndex(index)
+    // keep closing the file at index 0
+    // if openedFiles[0] == file, then keep closing files at index 1
+    while (editorViewModel.getOpenedFileCount() != 1) {
+      val editor = getEditorAtIndex(index)
 
-        // Index of files changes as we keep close files
-        // So we compare the files instead of index
-        if (editor != null) {
-          if (file != editor.file) {
-            closeFile(index)
-          } else {
-            index = 1
-          }
-        } else {
-          log.error("Unable to save file at index:", index)
-        }
+      if (editor == null) {
+        log.error("Unable to save file at index:", index)
+        continue
+      }
+
+      // Index of files changes as we keep close files
+      // So we compare the files instead of index
+      if (file != editor.file) {
+        closeFile(index)
+      } else {
+        index = 1
       }
-    } else {
-      notifyFilesUnsaved(unsavedFiles) { closeOthers() }
     }
   }
 
@@ -650,7 +659,7 @@ open class EditorHandlerActivity : ProjectHandlerActivity(), IEditorHandler {
   @Subscribe(threadMode = ThreadMode.MAIN)
   fun onDocumentChange(event: DocumentChangeEvent) {
     // update content modification status
-    onEditorContentChanged()
+    editorViewModel.areFilesModified = true
 
     val index = findIndexOfEditorByFile(event.file.toFile())
     if (index == -1) {
@@ -667,7 +676,7 @@ open class EditorHandlerActivity : ProjectHandlerActivity(), IEditorHandler {
   }
 
   private fun updateTabs() {
-    executeAsyncProvideError({
+    editorActivityScope.launch {
       val files = editorViewModel.getOpenedFiles()
       val dupliCount = mutableMapOf<String, Int>()
       val names = MutableIntObjectMap<String>()
@@ -690,15 +699,8 @@ open class EditorHandlerActivity : ProjectHandlerActivity(), IEditorHandler {
         names[i] = name
       }
 
-      names
-    }) { result, error ->
-      if (result == null || error != null) {
-        log.error("Failed to compute names for file tabs", error)
-        return@executeAsyncProvideError
-      }
-
-      ThreadUtils.runOnUiThread {
-        result.forEach { index, name ->
+      withContext(Dispatchers.Main) {
+        names.forEach { index, name ->
           binding.tabs.getTabAt(index)?.text = name
         }
       }

app/src/main/java/com/itsaky/androidide/activities/editor/EditorHandlerActivity.kt

@@ -208,7 +208,7 @@ class CodeEditorView(
    * Mark this files as saved. Even if it not saved.
    */
   fun markAsSaved() {
-    notifySaved()
+    editor?.markUnmodified()
   }
 
   /**
@@ -245,6 +245,7 @@ class CodeEditorView(
       _binding?.rwProgress?.isVisible = false
     }
 
+    markUnmodified()
     notifySaved()
 
     return true