diff --git a/editors/editoria/src/layout/EditoriaLayout.js b/editors/editoria/src/layout/EditoriaLayout.js index c845fe9ca899964db0a328ad0e096451c8ae5d5a..7b905b7cb9bc08294fc7f1c80df014b5357b2621 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>