feat: Optimize message box utility and add Icon support

ZZHow1024 · Sep 6, 2024 · f3c65b3 · f3c65b3
1 parent 7967ed4
commit f3c65b3
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/src/main/java/com/zzhow/magicencoding/utils/MessageBox.java b/src/main/java/com/zzhow/magicencoding/utils/MessageBox.java
@@ -1,6 +1,11 @@
 package com.zzhow.magicencoding.utils;
 
+import com.zzhow.magicencoding.MainClass;
 import javafx.scene.control.Alert;
+import javafx.scene.image.Image;
+import javafx.stage.Stage;
+
+import java.util.Objects;
 
 /**
  * @author ZZHow
@@ -14,25 +19,25 @@ public static void alert(Alert.AlertType type, String title, String headerText,
         alert.setHeaderText(headerText);
         alert.setContentText(contentText);
 
+        Stage stage = (Stage) alert.getDialogPane().getScene().getWindow();
+        Image icon = new Image(Objects.requireNonNull(MainClass.class.getResourceAsStream("/image/icon.png")));
+        stage.getIcons().add(icon);
+
         alert.showAndWait();
     }
 
     public static void error(String headerText, String contentText) {
-        Alert alert = new Alert(Alert.AlertType.ERROR);
-        alert.setTitle("错误");
-        alert.setHeaderText(headerText);
-        alert.setContentText(contentText);
-
-        alert.showAndWait();
+        alert(Alert.AlertType.ERROR,
+                "错误",
+                headerText,
+                contentText);
     }
 
     public static void success(String headerText, String contentText) {
-        Alert alert = new Alert(Alert.AlertType.INFORMATION);
-        alert.setTitle("成功");
-        alert.setHeaderText(headerText);
-        alert.setContentText(contentText);
-
-        alert.showAndWait();
+        alert(Alert.AlertType.INFORMATION,
+                "成功",
+                headerText,
+                contentText);
     }
 
 }