From 38308ec5ba3fa4de21f5cfdaa474c670227de4d7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jindra=20Pet=C5=99=C3=ADk?= <jindra.petrik@gmail.com>
Date: Thu, 22 Dec 2022 19:03:04 +0100
Subject: [PATCH] autosave for AS3 editing

---
 .../decompiler/flash/gui/abc/ABCPanel.java    | 23 +++++++++++++++----
 .../decompiler/flash/gui/abc/DetailPanel.java | 22 ++++++++++++++----
 2 files changed, 36 insertions(+), 9 deletions(-)

diff --git a/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java b/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java
index 6b3b52009..4b1284659 100644
--- a/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java
+++ b/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java
@@ -1612,7 +1612,7 @@ public class ABCPanel extends JPanel implements ItemListener, SearchListener<Scr
         return (TreeItem) scriptsPath.getLastPathComponent();
     }
 
-    private void saveDecompiledButtonActionPerformed(ActionEvent evt) {
+    private void saveDecompiled(boolean refreshTree) {
         final ABC localAbc = abc;
         int oldIndex = pack.scriptIndex;
         SWF.uncache(pack);
@@ -1633,7 +1633,9 @@ public class ABCPanel extends JPanel implements ItemListener, SearchListener<Scr
             }
             reload();
             mainPanel.clearEditingStatus();
-            ViewMessages.showMessageDialog(this, AppStrings.translate("message.action.saved"), AppStrings.translate("dialog.message.title"), JOptionPane.INFORMATION_MESSAGE, Configuration.showCodeSavedMessage);
+            if (refreshTree) {
+                ViewMessages.showMessageDialog(this, AppStrings.translate("message.action.saved"), AppStrings.translate("dialog.message.title"), JOptionPane.INFORMATION_MESSAGE, Configuration.showCodeSavedMessage);       
+            }
         } catch (As3ScriptReplaceException asre) {
             StringBuilder sb = new StringBuilder();
             int firstErrorLine = As3ScriptReplaceExceptionItem.LINE_UNKNOWN;
@@ -1667,13 +1669,17 @@ public class ABCPanel extends JPanel implements ItemListener, SearchListener<Scr
                 ViewMessages.showMessageDialog(this, AppStrings.translate("error.action.save").replace("%error%", firstErrorText).replace("%line%", Long.toString(firstErrorLine)), AppStrings.translate("error"), JOptionPane.ERROR_MESSAGE);
             } else {
                 ViewMessages.showMessageDialog(this, sb.toString(), AppStrings.translate("error"), JOptionPane.ERROR_MESSAGE);
-            }
+            }            
             decompiledTextArea.requestFocus();
 
         } catch (Throwable ex) {
             Logger.getLogger(ABCPanel.class.getName()).log(Level.SEVERE, null, ex);
         }
     }
+    
+    private void saveDecompiledButtonActionPerformed(ActionEvent evt) {
+        saveDecompiled(true);
+    }
 
     private void deobfuscateButtonActionPerformed(ActionEvent evt) {
         JToggleButton toggleButton = (JToggleButton)evt.getSource();
@@ -1901,8 +1907,15 @@ public class ABCPanel extends JPanel implements ItemListener, SearchListener<Scr
     public boolean tryAutoSave() {
         View.checkAccess();
 
-        // todo: implement
-        return false;
+        boolean ok = true;
+        if (saveDecompiledButton.isVisible() && saveDecompiledButton.isEnabled() && Configuration.autoSaveTagModifications.get()) {
+            saveDecompiled(false);
+            ok = ok && !(saveDecompiledButton.isVisible() && saveDecompiledButton.isEnabled());
+        }
+        
+        ok = ok && detailPanel.tryAutoSave();
+        
+        return ok;
     }
 
     @Override
diff --git a/src/com/jpexs/decompiler/flash/gui/abc/DetailPanel.java b/src/com/jpexs/decompiler/flash/gui/abc/DetailPanel.java
index 46618cf74..4e1dcd445 100644
--- a/src/com/jpexs/decompiler/flash/gui/abc/DetailPanel.java
+++ b/src/com/jpexs/decompiler/flash/gui/abc/DetailPanel.java
@@ -16,6 +16,7 @@
  */
 package com.jpexs.decompiler.flash.gui.abc;
 
+import com.jpexs.decompiler.flash.SWF;
 import com.jpexs.decompiler.flash.abc.ABC;
 import com.jpexs.decompiler.flash.abc.types.Multiname;
 import com.jpexs.decompiler.flash.abc.types.traits.Trait;
@@ -331,10 +332,16 @@ public class DetailPanel extends JPanel implements TagEditorPanel {
         mainPanel.clearEditingStatus();
     }
 
-    private void saveButtonActionPerformed(ActionEvent evt) {
+    private void save(boolean refreshTree) {
         if (cardMap.get(selectedCard) instanceof TraitDetail) {
             if (((TraitDetail) cardMap.get(selectedCard)).save()) {
+                
                 DecompiledEditorPane decompiledTextArea = abcPanel.decompiledTextArea;
+                if (!refreshTree) {
+                    decompiledTextArea.reloadClass();
+                    setEditMode(false);
+                    return;
+                }
                 int lastTrait = decompiledTextArea.lastTraitIndex;
 
                 Runnable reloadComplete = new Runnable() {
@@ -347,8 +354,8 @@ public class DetailPanel extends JPanel implements TagEditorPanel {
                             decompiledTextArea.gotoTrait(lastTrait);
                         }
                         setEditMode(false);
-                        mainPanel.clearEditingStatus();
-                        ViewMessages.showMessageDialog(DetailPanel.this, AppStrings.translate("message.trait.saved"), AppStrings.translate("dialog.message.title"), JOptionPane.INFORMATION_MESSAGE, Configuration.showTraitSavedMessage);
+                        mainPanel.clearEditingStatus();                        
+                        ViewMessages.showMessageDialog(DetailPanel.this, AppStrings.translate("message.trait.saved"), AppStrings.translate("dialog.message.title"), JOptionPane.INFORMATION_MESSAGE, Configuration.showTraitSavedMessage);                        
                     }
                 };
 
@@ -357,10 +364,17 @@ public class DetailPanel extends JPanel implements TagEditorPanel {
             }
         }
     }
+    
+    private void saveButtonActionPerformed(ActionEvent evt) {
+        save(true);
+    }
 
     @Override
     public boolean tryAutoSave() {
-        // todo: implement
+        if (saveButton.isVisible() && saveButton.isEnabled() && Configuration.autoSaveTagModifications.get()) {
+            save(false);
+            return !(saveButton.isVisible() && saveButton.isEnabled());
+        }
         return false;
     }