improve highlight toggling

packages/core/src/commands/toggleMark.ts

@@ -2,6 +2,7 @@ import { toggleMark as originalToggleMark } from 'prosemirror-commands'
 import { MarkType } from 'prosemirror-model'
 import { Command } from '../Editor'
 import getMarkType from '../utils/getMarkType'
+import markIsActive from '../utils/markIsActive'
 
 type ToggleMarkCommand = (typeOrName: string | MarkType, attrs?: {}) => Command
 
@@ -11,8 +12,13 @@ declare module '../Editor' {
   }
 }
 
-export const toggleMark: ToggleMarkCommand = (typeOrName, attrs) => ({ state, dispatch }) => {
+export const toggleMark: ToggleMarkCommand = (typeOrName, attrs) => ({ state, dispatch, commands }) => {
   const type = getMarkType(typeOrName, state.schema)
 
+  if (markIsActive(state, type) && !markIsActive(state, type, attrs)) {
+    // @ts-ignore
+    return commands.updateMark(type, attrs)
+  }
+
   return originalToggleMark(type, attrs)(state, dispatch)
 }