Merge pull request #28 from Death111/feature/#10_showDueTasksInBar

Feature/#10 show due tasks in bar

Author: Death111

src/main/java/de/doubleslash/keeptask/App.java

@@ -24,6 +24,7 @@
 import de.doubleslash.keeptask.common.Resources.RESOURCE;
 import de.doubleslash.keeptask.controller.Controller;
 import de.doubleslash.keeptask.model.Model;
+import de.doubleslash.keeptask.view.IconController;
 import de.doubleslash.keeptask.view.MainWindowController;
 import java.io.IOException;
 import java.io.PrintWriter;
@@ -60,6 +61,7 @@ public class App extends Application {
   private Controller controller;
 
   private MainWindowController viewController;
+  private IconController iconController;
 
   @Override
   public void init() throws Exception {
@@ -90,7 +92,7 @@ public void start(final Stage primaryStage) {
       LOG.info("UI successfully initialised.");
     } catch (final Exception e) {
       LOG.error("There was an error while initialising the UI", e);
-      showExceptionAndExit(e);
+      showExceptionAndExit(e, primaryStage);
     }
   }
 
@@ -101,13 +103,17 @@ private void initialiseApplication(final Stage primaryStage) throws Exception {
     controller.init();
 
     initialiseAndShowUI(primaryStage);
+    iconController = new IconController(model, controller, primaryStage);
+    iconController.initialize();
   }
 
   private void initialiseAndShowUI(final Stage primaryStage) throws IOException {
     LOG.debug("Initialising main UI.");
     primaryStage.setTitle("KeepTask");
     primaryStage.initStyle(StageStyle.TRANSPARENT);
-    primaryStage.getIcons().add(new Image(Resources.getResource(RESOURCE.ICON_MAIN).toString()));
+    Image applicationIcon = new Image(Resources.getResource(RESOURCE.ICON_MAIN).toString());
+    primaryStage.getIcons().setAll(applicationIcon);
+
     primaryStage.setAlwaysOnTop(true);
     primaryStage.setResizable(false);
     primaryStage.setOnCloseRequest(windowEvent -> LOG.info("On close request"));
@@ -124,7 +130,7 @@ private void initialiseAndShowUI(final Stage primaryStage) throws IOException {
     primaryStage.show();
   }
 
-  private static void showExceptionAndExit(Exception e) {
+  private static void showExceptionAndExit(Exception e, Stage primaryStage) {
     final Alert alert = new Alert(AlertType.ERROR);
     alert.setTitle("Error");
     alert.setHeaderText("Could not start application");
@@ -150,6 +156,7 @@ private static void showExceptionAndExit(Exception e) {
 
     alert.getDialogPane().setExpandableContent(expContent);
 
+    alert.initOwner(primaryStage);
     alert.showAndWait();
     System.exit(1);
   }

src/main/java/de/doubleslash/keeptask/view/IconController.java

@@ -0,0 +1,115 @@
+/*
+ * Copyright 2023 - Death111
+ *
+ * This file is part of KeepTask.
+ * KeepTask is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package de.doubleslash.keeptask.view;
+
+import de.doubleslash.keeptask.common.Resources;
+import de.doubleslash.keeptask.common.Resources.RESOURCE;
+import de.doubleslash.keeptask.controller.Controller;
+import de.doubleslash.keeptask.model.Model;
+import de.doubleslash.keeptask.model.WorkItem;
+import java.awt.image.BufferedImage;
+import java.time.LocalDateTime;
+import javafx.collections.ListChangeListener;
+import javafx.embed.swing.SwingFXUtils;
+import javafx.scene.SnapshotParameters;
+import javafx.scene.canvas.Canvas;
+import javafx.scene.canvas.GraphicsContext;
+import javafx.scene.image.Image;
+import javafx.scene.image.WritableImage;
+import javafx.scene.paint.Color;
+import javafx.scene.text.Font;
+import javafx.scene.text.TextAlignment;
+import javafx.stage.Stage;
+
+public class IconController {
+
+  private final Model model;
+  private final Controller controller;
+  private final Stage primaryStage;
+
+  private final Image applicationIcon = new Image(
+      Resources.getResource(RESOURCE.ICON_MAIN).toString());
+
+  private final Canvas taskbarCanvas = new Canvas(64, 64);
+
+  public IconController(Model model, Controller controller, Stage primaryStage) {
+    this.model = model;
+    this.controller = controller;
+    this.primaryStage = primaryStage;
+  }
+
+  public void initialize() {
+    model.getWorkItems().addListener((ListChangeListener<? super WorkItem>) change -> {
+      updateProjectFilterButtons();
+    });
+    updateProjectFilterButtons();
+  }
+
+  private void updateProjectFilterButtons() {
+    LocalDateTime now = LocalDateTime.now();
+    int expiredTasks = (int) model.getWorkItems().stream().filter(item -> {
+      LocalDateTime dueDateTime = item.getDueDateTime();
+      if (dueDateTime == null) {
+        return false;
+      }
+      return !item.isFinished() && now.isAfter(dueDateTime);
+    }).count();
+
+    updateTaskbarIcon(expiredTasks, primaryStage);
+  }
+
+  private void updateTaskbarIcon(int expiredTasks, Stage primaryStage) {
+    final GraphicsContext gcIcon = taskbarCanvas.getGraphicsContext2D();
+
+    gcIcon.clearRect(0, 0, taskbarCanvas.getWidth(), taskbarCanvas.getHeight());
+    gcIcon.drawImage(applicationIcon, 0, 0, 64, 64);
+
+    if (expiredTasks > 0) {
+      drawExpiredTasksToGC(expiredTasks, gcIcon);
+    }
+
+    final Image icon = canvasToImage();
+
+    primaryStage.getIcons().setAll(icon);
+  }
+
+  private static void drawExpiredTasksToGC(int expiredTasks, GraphicsContext gcIcon) {
+    gcIcon.setFill(Color.RED);
+    gcIcon.fillOval(32, 32, 32, 32);
+
+    gcIcon.setStroke(Color.WHITE);
+    gcIcon.setTextAlign(TextAlignment.CENTER);
+    Font aDefault = Font.getDefault();
+    gcIcon.setFont(new Font(aDefault.getName(), 30));
+    gcIcon.setStroke(Color.WHITE);
+    gcIcon.setLineWidth(2);
+
+    gcIcon.strokeText(Integer.toString(expiredTasks), 47, 57);
+  }
+
+  private Image canvasToImage() {
+    final SnapshotParameters snapshotParameters = new SnapshotParameters();
+    snapshotParameters.setFill(Color.TRANSPARENT);
+    final WritableImage image = taskbarCanvas.snapshot(snapshotParameters, null);
+
+    final BufferedImage bi = SwingFXUtils.fromFXImage(image, null);
+    final Image icon = SwingFXUtils.toFXImage(bi, null);
+    return icon;
+  }
+}