From df6101ddc5440a3d0d2089f33a28165a98b153ec Mon Sep 17 00:00:00 2001
From: Brendan Robert <brendan.robert@gmail.com>
Date: Sat, 8 Aug 2015 15:05:43 -0500
Subject: [PATCH] Dispose of temporary files when doing test compilation

---
 src/main/java/jace/applesoft/ApplesoftHandler.java | 5 ++++-
 src/main/java/jace/assembly/AssemblyHandler.java   | 5 +++++
 src/main/java/jace/ide/Program.java                | 3 +++
 src/main/java/jace/ide/TextHandler.java            | 6 +++++-
 4 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/src/main/java/jace/applesoft/ApplesoftHandler.java b/src/main/java/jace/applesoft/ApplesoftHandler.java
index 74cc154..f60cbc7 100644
--- a/src/main/java/jace/applesoft/ApplesoftHandler.java
+++ b/src/main/java/jace/applesoft/ApplesoftHandler.java
@@ -65,5 +65,8 @@ public class ApplesoftHandler implements LanguageHandler<ApplesoftProgram> {
     public void execute(CompileResult<ApplesoftProgram> lastResult) {
         lastResult.getCompiledAsset().run();
     }
-    
+
+    @Override
+    public void clean(CompileResult<ApplesoftProgram> lastResult) {
+    }    
 }
diff --git a/src/main/java/jace/assembly/AssemblyHandler.java b/src/main/java/jace/assembly/AssemblyHandler.java
index 0d0c531..52a4226 100644
--- a/src/main/java/jace/assembly/AssemblyHandler.java
+++ b/src/main/java/jace/assembly/AssemblyHandler.java
@@ -48,6 +48,11 @@ public class AssemblyHandler implements LanguageHandler<File> {
                 Logger.getLogger(AssemblyHandler.class.getName()).log(Level.SEVERE, null, ex);
             }
         }
+        clean(lastResult);
+    }
+
+    @Override
+    public void clean(CompileResult<File> lastResult) {
         lastResult.getCompiledAsset().delete();
     }
 }
diff --git a/src/main/java/jace/ide/Program.java b/src/main/java/jace/ide/Program.java
index f63fb8d..908f02b 100644
--- a/src/main/java/jace/ide/Program.java
+++ b/src/main/java/jace/ide/Program.java
@@ -219,12 +219,15 @@ public class Program {
         manageCompileResult(lastResult);
         if (lastResult.isSuccessful()) {
             getHandler().execute(lastResult);
+        } else {
+            getHandler().clean(lastResult);
         }
     }
 
     public void test() {
         lastResult = getHandler().compile(this);
         manageCompileResult(lastResult);
+        getHandler().clean(lastResult);
     }
 
     private void manageCompileResult(CompileResult lastResult) {
diff --git a/src/main/java/jace/ide/TextHandler.java b/src/main/java/jace/ide/TextHandler.java
index 535ab41..6dde305 100644
--- a/src/main/java/jace/ide/TextHandler.java
+++ b/src/main/java/jace/ide/TextHandler.java
@@ -9,7 +9,7 @@ import java.util.Map;
  *
  * @author blurry
  */
-class TextHandler implements LanguageHandler<String> {
+public class TextHandler implements LanguageHandler<String> {
     public TextHandler() {
     }
 
@@ -57,4 +57,8 @@ class TextHandler implements LanguageHandler<String> {
     public void execute(CompileResult<String> lastResult) {
         Keyboard.pasteFromString(lastResult.getCompiledAsset());
     }
+
+    @Override
+    public void clean(CompileResult<String> lastResult) {
+    }
 }