From 8fd7c5d2843d146cbc4e17ce9b9a2ededb2dc9ec Mon Sep 17 00:00:00 2001
From: Sogomn <Sogomn@Sogomn-PC>
Date: Sun, 7 Feb 2016 15:49:21 +0100
Subject: [PATCH] Small changes Builder improvements

---
 .../src/de/sogomn/rat/builder/StubBuilder.java  | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/Ratty/src/de/sogomn/rat/builder/StubBuilder.java b/Ratty/src/de/sogomn/rat/builder/StubBuilder.java
index 0403cd4..5102fd2 100644
--- a/Ratty/src/de/sogomn/rat/builder/StubBuilder.java
+++ b/Ratty/src/de/sogomn/rat/builder/StubBuilder.java
@@ -27,6 +27,7 @@ public final class StubBuilder {
 	private static final String ADDRESS_MESSAGE = "Which address should the client connect to?";
 	private static final String PORT_MESSAGE = "Which port?";
 	
+	private static final String FILE_EXTENSION = ".jar";
 	private static final String FILE_NAME = "/connection_data.txt";
 	
 	private StubBuilder() {
@@ -44,7 +45,13 @@ public final class StubBuilder {
 		final int input = open ? fileChooser.showOpenDialog(null) : fileChooser.showSaveDialog(null);
 		
 		if (input == JFileChooser.APPROVE_OPTION) {
-			final File file = fileChooser.getSelectedFile();
+			File file = fileChooser.getSelectedFile();
+			
+			final String name = file.getName();
+			
+			if (!name.endsWith(FILE_EXTENSION)) {
+				file = new File(file + FILE_EXTENSION);
+			}
 			
 			return file;
 		}
@@ -76,13 +83,17 @@ public final class StubBuilder {
 	private static void replaceFile(final File jarFile) {
 		final String address = JOptionPane.showInputDialog(ADDRESS_MESSAGE);
 		
-		if (address == null) {
+		if (address == null || address.isEmpty()) {
+			jarFile.delete();
+			
 			return;
 		}
 		
 		final String port = JOptionPane.showInputDialog(PORT_MESSAGE);
 		
-		if (port == null) {
+		if (port == null || port.isEmpty()) {
+			jarFile.delete();
+			
 			return;
 		}