fix focus in all views

wax-prosemirror-components/src/components/notes/NoteEditorContainer.js

@@ -20,6 +20,10 @@ const NoteEditorContainerStyled = styled.div`
     padding-bottom: 5px;
     margin-right: 0;
   }
+  *::selection {
+    background-color: #c5d7fe;
+    color: #000;
+  }
 `;
 
 const NoteStyled = styled.div`

wax-prosemirror-components/src/components/various/FullScreen.js

@@ -16,10 +16,12 @@ const Button = ({ view = {}, item }) => {
   const handleMouseDown = (e, editorState, editorDispatch) => {
     e.preventDefault();
     options.fullScreen = !options.fullScreen;
-    const { selection } = editorState;
     activeView.dispatch(
       activeView.state.tr.setSelection(
-        new TextSelection(activeView.state.tr.doc.resolve(selection.from)),
+        TextSelection.between(
+          activeView.state.selection.$anchor,
+          activeView.state.selection.$head,
+        ),
       ),
     );
     activeView.focus();

wax-prosemirror-services/src/EssayService/components/EssayAnswerComponent.js

@@ -149,6 +149,11 @@ const EssayAnswerComponent = ({ node, view, getPos }) => {
             // the parent editor is focused.
             if (essayAnswerView.hasFocus()) essayAnswerView.focus();
           },
+          blur: (editorView, event) => {
+            if (essayAnswerView && event.relatedTarget === null) {
+              essayAnswerView.focus();
+            }
+          },
         },
 
         attributes: {

wax-prosemirror-services/src/EssayService/components/EssayQuestionComponent.js

@@ -149,6 +149,11 @@ const EssayQuestionComponent = ({ node, view, getPos }) => {
             // the parent editor is focused.
             if (essayQuestionView.hasFocus()) essayQuestionView.focus();
           },
+          blur: (editorView, event) => {
+            if (essayQuestionView && event.relatedTarget === null) {
+              essayQuestionView.focus();
+            }
+          },
         },
         handleClickOn: () => {
           context.updateView({}, questionId);

wax-prosemirror-services/src/NoteService/Editor.js

@@ -47,7 +47,7 @@ export default ({ node, view }) => {
         dispatchTransaction,
         disallowedTools: ['Tables', 'Images'],
         handleDOMEvents: {
-          blur: () => {
+          blur: (editorView, event) => {
             if (context.view[noteId]) {
               context.view[noteId].dispatch(
                 context.view[noteId].state.tr.setSelection(