From 62c6f9a8856c92bb2abf9067415603a914955e63 Mon Sep 17 00:00:00 2001
From: 0xboobface <0xboobface@gmail.com>
Date: Thu, 11 Apr 2019 14:00:27 +0200
Subject: [PATCH] Remove notes from config, if new note is an empty string

---
 .../java/ctbrec/ui/action/EditNotesAction.java | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/client/src/main/java/ctbrec/ui/action/EditNotesAction.java b/client/src/main/java/ctbrec/ui/action/EditNotesAction.java
index 5358ea3c..372708f2 100644
--- a/client/src/main/java/ctbrec/ui/action/EditNotesAction.java
+++ b/client/src/main/java/ctbrec/ui/action/EditNotesAction.java
@@ -1,7 +1,11 @@
 package ctbrec.ui.action;
 
+import java.io.IOException;
 import java.util.Optional;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import ctbrec.Config;
 import ctbrec.Model;
 import ctbrec.ui.JavaFxModel;
@@ -12,6 +16,7 @@ import javafx.scene.Node;
 import javafx.scene.control.TableView;
 
 public class EditNotesAction {
+    private static final transient Logger LOG = LoggerFactory.getLogger(EditNotesAction.class);
 
     private Node source;
     private Model model;
@@ -29,7 +34,18 @@ public class EditNotesAction {
             Platform.runLater(() -> {
                 String notes = Config.getInstance().getSettings().modelNotes.getOrDefault(model.getUrl(), "");
                 Optional<String> newNotes = Dialogs.showTextInput("Model Notes", "Notes for " + model.getName(), notes);
-                newNotes.ifPresent(n -> Config.getInstance().getSettings().modelNotes.put(model.getUrl(), n));
+                newNotes.ifPresent(n -> {
+                    if(!n.trim().isEmpty()) {
+                        Config.getInstance().getSettings().modelNotes.put(model.getUrl(), n);
+                    } else {
+                        Config.getInstance().getSettings().modelNotes.remove(model.getUrl());
+                    }
+                    try {
+                        Config.getInstance().save();
+                    } catch (IOException e) {
+                        LOG.warn("Couldn't save config", e);
+                    }
+                });
                 table.refresh();
                 source.setCursor(Cursor.DEFAULT);
             });