didn't rerender tools while typing

wax-prosemirror-components/src/components/Button.js

 /* eslint react/prop-types: 0 */
 
-import React, { useContext, useMemo } from 'react';
+import React, { useContext, useMemo, useState } from 'react';
 import { WaxContext } from 'wax-prosemirror-core';
 import MenuButton from '../ui/buttons/MenuButton';
 
@@ -16,9 +16,9 @@ const Button = ({ view = {}, item }) => {
 
   const { dispatch, state } = view;
 
-  const handleMouseDown = e => {
+  const handleMouseDown = (e, editorState, editorDispatch) => {
     e.preventDefault();
-    run(state, dispatch);
+    run(editorState, dispatch);
   };
 
   const isActive = active && active(state, activeViewId);
@@ -33,11 +33,13 @@ const Button = ({ view = {}, item }) => {
         disabled={isDisabled}
         iconName={icon}
         label={label}
-        onMouseDown={handleMouseDown}
+        onMouseDown={(e, editorState, editorDispatch) =>
+          handleMouseDown(e, view.state, view.dispatch)
+        }
         title={title}
       />
     ),
-    [],
+    [isActive, isDisabled],
   );
 
   return MenuButtonComponent;

wax-prosemirror-components/src/ui/buttons/MenuButton.js

@@ -78,7 +78,6 @@ const MenuButton = props => {
     title,
     onMouseDown,
   } = props;
-  console.log('here?');
   return (
     <Wrapper
       active={active}

wax-prosemirror-services/src/lib/ToolGroup.js

-import React, { useState } from 'react';
+import React, { useMemo } from 'react';
 import { injectable } from 'inversify';
 import { ToolGroupComponent, ToolGroups } from 'wax-prosemirror-components';
 import { v4 as uuidv4 } from 'uuid';
@@ -61,14 +61,20 @@ class ToolGroup {
     if (this._toolGroups > 0) {
       return <ToolGroups toolGroups={this._toolGroups} view={view} />;
     }
-    return (
-      <ToolGroupComponent
-        key={uuidv4()}
-        view={view}
-        tools={this._tools}
-        title={this.title}
-        name={name}
-      />
+
+    const MemorizedToolGroupComponent = useMemo(
+      () => (
+        <ToolGroupComponent
+          key={uuidv4()}
+          view={view}
+          tools={this._tools}
+          title={this.title}
+          name={name}
+        />
+      ),
+      [view],
     );
+
+    return MemorizedToolGroupComponent;
   }
 }