From 247275d391a21e739852b434b43d07e73a50dafe Mon Sep 17 00:00:00 2001
From: azivner <adam.zivner@gmail.com>
Date: Wed, 14 Nov 2018 08:42:00 +0100
Subject: [PATCH] fix icon of newly created note

---
 src/public/javascripts/services/tree.js         | 3 ++-
 src/public/javascripts/services/tree_builder.js | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/public/javascripts/services/tree.js b/src/public/javascripts/services/tree.js
index 3acb635b5..2a506fec1 100644
--- a/src/public/javascripts/services/tree.js
+++ b/src/public/javascripts/services/tree.js
@@ -516,7 +516,8 @@ async function createNote(node, parentNoteId, target, isProtected, saveSelection
         refKey: branchEntity.noteId,
         branchId: branchEntity.branchId,
         isProtected: isProtected,
-        extraClasses: await treeBuilder.getExtraClasses(noteEntity)
+        extraClasses: await treeBuilder.getExtraClasses(noteEntity),
+        icon: treeBuilder.getIcon(noteEntity)
     };
 
     if (target === 'after') {
diff --git a/src/public/javascripts/services/tree_builder.js b/src/public/javascripts/services/tree_builder.js
index 60b5257a8..09c564470 100644
--- a/src/public/javascripts/services/tree_builder.js
+++ b/src/public/javascripts/services/tree_builder.js
@@ -159,5 +159,6 @@ async function getExtraClasses(note) {
 export default {
     prepareTree,
     prepareBranch,
-    getExtraClasses
+    getExtraClasses,
+    getIcon
 }
\ No newline at end of file