From 5641b1173cafe59f97d3622ec58cace52640d3e3 Mon Sep 17 00:00:00 2001
From: snowleo <schneeleo@gmail.com>
Date: Wed, 26 Oct 2011 19:46:40 +0200
Subject: Abort the installation wizard if a runtime exception happens

---
 .../essentials/update/states/StateMachine.java     | 34 +++++++++++++++-------
 1 file changed, 23 insertions(+), 11 deletions(-)

(limited to 'EssentialsUpdate/src/com/earth2me/essentials/update/states/StateMachine.java')

diff --git a/EssentialsUpdate/src/com/earth2me/essentials/update/states/StateMachine.java b/EssentialsUpdate/src/com/earth2me/essentials/update/states/StateMachine.java
index 9a9c64134..17f6502fb 100644
--- a/EssentialsUpdate/src/com/earth2me/essentials/update/states/StateMachine.java
+++ b/EssentialsUpdate/src/com/earth2me/essentials/update/states/StateMachine.java
@@ -33,22 +33,33 @@ public class StateMachine extends WorkListener
 
 	public MachineResult askQuestion()
 	{
-		while (current.guessAnswer())
+		try
 		{
-			current = current.getNextState();
-			if (current == null)
+			while (current.guessAnswer())
 			{
-				result = MachineResult.DONE;
-				break;
+
+				current = current.getNextState();
+				if (current == null)
+				{
+					result = MachineResult.DONE;
+					break;
+				}
+
 			}
-		}
-		if (current != null)
-		{
-			if (player.isOnline())
+			if (current != null)
 			{
-				current.askQuestion(player);
+				if (player.isOnline())
+				{
+					current.askQuestion(player);
+				}
+				result = MachineResult.WAIT;
 			}
-			result = MachineResult.WAIT;
+		}
+		catch (RuntimeException ex)
+		{
+			player.sendMessage(ex.getMessage());
+			finish();
+			result = MachineResult.ABORT;
 		}
 		return result;
 	}
@@ -135,6 +146,7 @@ public class StateMachine extends WorkListener
 
 	private void finish()
 	{
+		current = null;
 		iterator = null;
 		states.clear();
 		getPlugin().getServer().getPluginManager().callEvent(new InstallationFinishedEvent());
-- 
cgit v1.2.3