Restoring statuses to initial state after creating or loading notice-tree

Author: NoxChimaera

src/com/temporaryteam/noticeditor/controller/NoticeController.java

@@ -23,10 +23,7 @@
 import com.temporaryteam.noticeditor.io.ExportException;
 import com.temporaryteam.noticeditor.io.ExportStrategy;
 import com.temporaryteam.noticeditor.io.ExportStrategyHolder;
-import com.temporaryteam.noticeditor.model.NoticeItem;
-import com.temporaryteam.noticeditor.model.NoticeTree;
-import com.temporaryteam.noticeditor.model.NoticeTreeItem;
-import com.temporaryteam.noticeditor.model.PreviewStyles;
+import com.temporaryteam.noticeditor.model.*;
 import com.temporaryteam.noticeditor.view.Chooser;
 import com.temporaryteam.noticeditor.view.EditNoticeTreeCell;
 import java.util.ResourceBundle;
@@ -224,6 +221,7 @@ private void handleContextMenu(ActionEvent event) {
 	private void handleNew(ActionEvent event) {
 		rebuildTree(resources.getString("help"));
 		fileSaved = null;
+		NoticeStatus.load();
 	}
 
 	@FXML
@@ -242,6 +240,7 @@ private void handleOpen(ActionEvent event) {
 			createSearchBinding(noticeTree.getRoot());
 			currentTreeItem = null;
 			open();
+			noticeSettingsController.updateStatuses();
 		} catch (IOException | JSONException e) {
 			logger.log(Level.SEVERE, null, e);
 		}

src/com/temporaryteam/noticeditor/controller/NoticeSettingsController.java

@@ -44,6 +44,7 @@ public void setNoticeController(NoticeController noticeController) {
 	public void initialize(URL url, ResourceBundle rb) {
 		NoticeStatus.add(rb.getString("normal"));
 		NoticeStatus.add(rb.getString("important"));
+		NoticeStatus.save();
 
 		choiceBoxNoticeStatus.setItems(NoticeStatus.asObservable());
 		choiceBoxNoticeStatus.getSelectionModel().selectedItemProperty().addListener(new ChangeListener<String>() {
@@ -58,12 +59,15 @@ public void changed(ObservableValue<? extends String> observable, String oldValu
 		open(null);
 	}
 
+	public void updateStatuses() {
+		choiceBoxNoticeStatus.setItems(NoticeStatus.asObservable());
+	}
+	
 	public void open(NoticeTreeItem item) {
 		if (item == null || item.isBranch()) {
 			choiceBoxNoticeStatus.getSelectionModel().clearSelection();
 			settingsPane.setDisable(true);
 		} else {
-			choiceBoxNoticeStatus.setItems(NoticeStatus.asObservable());
 			choiceBoxNoticeStatus.getSelectionModel().select(NoticeStatus.getStatusName(item.getStatus()));
 			settingsPane.setDisable(false);
 		}

src/com/temporaryteam/noticeditor/io/JsonFormat.java

@@ -42,6 +42,8 @@ public NoticeTree importDocument() throws IOException, JSONException {
 				String key = it.next().toString();
 				NoticeStatus.add(key, status.getInt(key));
 			}
+		} else {
+			NoticeStatus.load();
 		}
 
 		return new NoticeTree(jsonToTree(json));

src/com/temporaryteam/noticeditor/model/NoticeStatus.java

@@ -13,14 +13,33 @@
  * @author Maximillian M.
  */
 public class NoticeStatus {
-	private static final HashMap<String, Integer> map;
+	private static HashMap<String, Integer> map;
 	private static int lastIndex;
 
+	private static HashMap<String, Integer> mementoMap;
+	private static int mementoIndex;
+	
 	static {
 		map = new HashMap<>();
 		lastIndex = -1;
 	}
 
+	/**
+	 * Saves statuses
+	 */
+	public static void save() {
+		mementoIndex = lastIndex;
+		mementoMap = (HashMap) map.clone();
+	}
+	
+	/**
+	 * Restores statuses to the saved state
+	 */
+	public static void load() {
+		lastIndex = mementoIndex;
+		map = mementoMap;
+	}
+	
 	/**
 	 * Associates the specified status with next status code
 	 * @param status Status name