From 225e90f42c199a476c0e51092f7c94e1c7499457 Mon Sep 17 00:00:00 2001
From: chris <kokosias@yahoo.gr>
Date: Mon, 11 Jan 2021 18:53:12 +0200
Subject: [PATCH] track block nodes counter

---
 editors/editoria/src/layout/EditoriaLayout.js | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/editors/editoria/src/layout/EditoriaLayout.js b/editors/editoria/src/layout/EditoriaLayout.js
index c845fe9ca..7b905b7cb 100644
--- a/editors/editoria/src/layout/EditoriaLayout.js
+++ b/editors/editoria/src/layout/EditoriaLayout.js
@@ -207,6 +207,17 @@ const getCommentsTracks = main => {
   return commentsTracks;
 };
 
+const getTrackBlockNodes = main => {
+  const allBlockNodes = DocumentHelpers.findBlockNodes(main.state.doc);
+  const trackBlockNodes = [];
+  allBlockNodes.map(node => {
+    if (node.node.attrs.track && node.node.attrs.track.length > 0) {
+      trackBlockNodes.push(node);
+    }
+  });
+  return trackBlockNodes;
+};
+
 const LeftSideBar = ComponentPlugin('leftSideBar');
 const MainMenuToolBar = ComponentPlugin('mainMenuToolBar');
 const NotesArea = ComponentPlugin('notesArea');
@@ -238,7 +249,8 @@ const EditoriaLayout = ({ editor }) => {
   }
   const notes = main && getNotes(main);
   const commentsTracks = main && getCommentsTracks(main).length;
-  console.log('comments', commentsTracks);
+  const trackBlockNodes = main && getTrackBlockNodes(main).length;
+
   const areNotes = notes && !!notes.length && notes.length > 0;
 
   const [hasNotes, setHasNotes] = useState(areNotes);
@@ -280,7 +292,8 @@ const EditoriaLayout = ({ editor }) => {
                 <CommentsContainer>
                   <CommentTrackTools>
                     <span>
-                      {commentsTracks} COMMENTS AND SUGGESTIONS
+                      {commentsTracks + trackBlockNodes} COMMENTS AND
+                      SUGGESTIONS
                       <CommentTrackToolBar />
                     </span>
                   </CommentTrackTools>
-- 
GitLab