Replace updater custom stage to default dialog window

Author: introfog

src/main/java/com/github/introfog/gitwave/controller/ExecuteController.java

@@ -28,6 +28,7 @@
 import java.util.Objects;
 import javafx.fxml.FXML;
 import javafx.scene.control.Button;
+import javafx.scene.control.ButtonType;
 import javafx.scene.control.TextField;
 import javafx.stage.DirectoryChooser;
 import javafx.stage.Stage;
@@ -133,11 +134,14 @@ protected void saveCommand() {
             specifySourceCommand(commandDto);
             // TODO MINOR if DTO is already existed, nothing was happened, is it OK?
         } else {
-            FxmlStageHolder holder = StageFactory.createModalStage("view/updater.fxml", "Command updater");
-            UpdateController updateController = holder.getFxmlLoader().getController();
-            // TODO add required methods for all stages where necessary
-            updateController.setRequiredFields(this, sourceCommand, commandDto);
-            holder.getStage().showAndWait();
+            ButtonType result = DialogFactory.createSaveOrUpdateAlert();
+            if (ButtonType.YES == result) {
+                AppConfig.getInstance().addCommand(commandDto);
+                specifySourceCommand(commandDto);
+            } else if (ButtonType.NO == result) {
+                AppConfig.getInstance().updateExistedCommand(sourceCommand, commandDto);
+                specifySourceCommand(commandDto);
+            }
         }
     }
 

src/main/java/com/github/introfog/gitwave/controller/UpdateController.java

@@ -52,4 +52,17 @@ public static void createCloseConfirmationAlert(Stage primaryStage) {
             }
         });
     }
+
+    public static ButtonType createSaveOrUpdateAlert() {
+        Alert confirmationDialog = new Alert(AlertType.CONFIRMATION);
+        Stage stage = (Stage) confirmationDialog.getDialogPane().getScene().getWindow();
+        stage.getIcons().add(new Image(StageFactory.class.getResourceAsStream(AppConstants.PATH_TO_LOGO)));
+        confirmationDialog.setTitle("Confirmation");
+        confirmationDialog.setHeaderText("Do you want to save command as a new instance?\nSelect 'No' if update existed.");
+        confirmationDialog.getButtonTypes().setAll(ButtonType.YES, ButtonType.NO, ButtonType.CANCEL);
+
+        ButtonType[] pressedButton = {null};
+        confirmationDialog.showAndWait().ifPresent(response -> pressedButton[0] = response);
+        return pressedButton[0];
+    }
 }