From 5b90fce1f5a77800e792d5eefceaa40c826c5b36 Mon Sep 17 00:00:00 2001
From: CocoTheOwner <sjoerdvdgoor02@gmail.com>
Date: Wed, 3 Feb 2021 16:09:14 +0100
Subject: [PATCH] Catch hosted server with gui

---
 .../java/com/volmit/iris/pregen/Pregenerator.java    | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/volmit/iris/pregen/Pregenerator.java b/src/main/java/com/volmit/iris/pregen/Pregenerator.java
index 5b1c894da..20c7233f7 100644
--- a/src/main/java/com/volmit/iris/pregen/Pregenerator.java
+++ b/src/main/java/com/volmit/iris/pregen/Pregenerator.java
@@ -21,6 +21,7 @@ import java.awt.event.KeyEvent;
 import java.awt.event.KeyListener;
 import java.awt.image.BufferedImage;
 import java.awt.image.ImageObserver;
+import java.awt.HeadlessException;
 import java.io.File;
 import java.io.IOException;
 import java.nio.file.Files;
@@ -74,7 +75,7 @@ public class Pregenerator implements Listener
 		this.onComplete.add(onComplete);
 	}
 
-	public Pregenerator(World world, int blockSize)
+	public Pregenerator(World world, int blockSize) throws HeadlessException
 	{
 		instance();
 		elapsed = M.ms();
@@ -582,9 +583,14 @@ public class Pregenerator implements Listener
 		}
 
 		@SuppressWarnings("deprecation")
-		private static MCAPregenGui createAndShowGUI(Pregenerator j)
+		private static MCAPregenGui createAndShowGUI(Pregenerator j) throws HeadlessException
 		{
-			JFrame frame = new JFrame("Pregen View");
+			JFrame frame;
+			try {
+				frame = new JFrame("Pregen View");
+			} catch (HeadlessException e) {
+				throw e;
+			}
 			MCAPregenGui nv = new MCAPregenGui();
 			frame.addKeyListener(nv);
 			nv.l = new ReentrantLock();