Fix update manager not displaying self-update errors

SKCraft · Feb 23, 2023 · 6003069 · 6003069
1 parent 8dfb049
commit 6003069
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/launcher/src/main/java/com/skcraft/launcher/Launcher.java b/launcher/src/main/java/com/skcraft/launcher/Launcher.java
@@ -105,7 +105,7 @@ public void run() {
             }
         });
 
-        updateManager.checkForUpdate();
+        updateManager.checkForUpdate(null);
     }
 
     /**
@@ -401,8 +401,11 @@ public URL propUrl(String key, String... args) {
     /**
      * Show the launcher.
      */
-    public void showLauncherWindow() {
-        mainWindowSupplier.get().setVisible(true);
+    public Window showLauncherWindow() {
+        Window window = mainWindowSupplier.get();
+        window.setVisible(true);
+
+        return window;
     }
 
     /**

diff --git a/launcher/src/main/java/com/skcraft/launcher/dialog/LauncherFrame.java b/launcher/src/main/java/com/skcraft/launcher/dialog/LauncherFrame.java
@@ -132,7 +132,7 @@ public void tableChanged(TableModelEvent e) {
             @Override
             public void actionPerformed(ActionEvent e) {
                 loadInstances();
-                launcher.getUpdateManager().checkForUpdate();
+                launcher.getUpdateManager().checkForUpdate(LauncherFrame.this);
                 webView.browse(launcher.getNewsURL(), false);
             }
         });
@@ -394,8 +394,8 @@ public void gameStarted() {
 
         @Override
         public void gameClosed() {
-            launcher.showLauncherWindow();
-            launcher.getUpdateManager().checkForUpdate();
+            Window newLauncherWindow = launcher.showLauncherWindow();
+            launcher.getUpdateManager().checkForUpdate(newLauncherWindow);
         }
     }
 

diff --git a/launcher/src/main/java/com/skcraft/launcher/update/UpdateManager.java b/launcher/src/main/java/com/skcraft/launcher/update/UpdateManager.java
@@ -50,7 +50,7 @@ public boolean getPendingUpdate() {
         return pendingUpdate != null;
     }
 
-    public void checkForUpdate() {
+    public void checkForUpdate(final Window window) {
         ListenableFuture<LatestVersionInfo> future = launcher.getExecutor().submit(new UpdateChecker(launcher));
 
         Futures.addCallback(future, new FutureCallback<LatestVersionInfo>() {
@@ -63,9 +63,11 @@ public void onSuccess(LatestVersionInfo result) {
 
             @Override
             public void onFailure(Throwable t) {
-
+                // Error handler attached below.
             }
         }, SwingExecutor.INSTANCE);
+
+        SwingHelper.addErrorDialogCallback(window, future);
     }
 
     public void performUpdate(final Window window) {