From 5f773ec98885e20ee344d67ee71df662cb33fdf5 Mon Sep 17 00:00:00 2001
From: fantasticit <fantasticit@163.com>
Date: Thu, 14 Jul 2022 21:49:39 +0800
Subject: [PATCH] tiptap: improve drag

---
 .../src/tiptap/core/extensions/dragable.ts    | 22 ++++++++++++++-----
 .../tiptap/core/wrappers/attachment/index.tsx |  7 +++++-
 .../tiptap/core/wrappers/callout/index.tsx    |  8 ++++++-
 .../tiptap/core/wrappers/code-block/index.tsx |  3 ++-
 .../tiptap/core/wrappers/countdown/index.tsx  |  7 +++---
 .../core/wrappers/document-children/index.tsx |  2 ++
 .../wrappers/document-reference/index.tsx     |  8 ++++++-
 .../core/wrappers/dragable/index.module.scss  |  2 +-
 .../tiptap/core/wrappers/dragable/index.tsx   |  7 ++++--
 .../src/tiptap/core/wrappers/flow/index.tsx   |  6 ++++-
 .../src/tiptap/core/wrappers/iframe/index.tsx |  3 ++-
 .../src/tiptap/core/wrappers/image/index.tsx  |  3 ++-
 .../src/tiptap/core/wrappers/katex/index.tsx  |  2 ++
 .../src/tiptap/core/wrappers/mind/index.tsx   |  6 ++++-
 14 files changed, 67 insertions(+), 19 deletions(-)

diff --git a/packages/client/src/tiptap/core/extensions/dragable.ts b/packages/client/src/tiptap/core/extensions/dragable.ts
index d5e94092..460d1026 100644
--- a/packages/client/src/tiptap/core/extensions/dragable.ts
+++ b/packages/client/src/tiptap/core/extensions/dragable.ts
@@ -28,6 +28,7 @@ export const Dragable = Extension.create({
   name: 'dragable',
 
   addProseMirrorPlugins() {
+    let scrollContainer;
     let dropElement;
     let currentNode;
     let editorView;
@@ -54,6 +55,11 @@ export const Dragable = Extension.create({
       editorView.dragging = { slice, move: true };
     }
 
+    function onScroll() {
+      if (!dropElement) return;
+      dropElement.style.opacity = 0;
+    }
+
     return [
       new Plugin({
         view(view) {
@@ -64,6 +70,12 @@ export const Dragable = Extension.create({
             dropElement.className = 'drag-handler';
             dropElement.addEventListener('dragstart', drag);
             view.dom.parentElement.appendChild(dropElement);
+
+            scrollContainer = view.dom.parentElement.parentElement?.parentElement?.parentElement;
+
+            if (scrollContainer) {
+              scrollContainer.addEventListener('scroll', onScroll);
+            }
           }
 
           return {
@@ -75,6 +87,10 @@ export const Dragable = Extension.create({
                 dropElement.removeEventListener('dragstart', drag);
                 dropElement.parentNode.removeChild(dropElement);
               }
+
+              if (scrollContainer) {
+                scrollContainer.removeEventListener('scroll', onScroll);
+              }
             },
           };
         },
@@ -91,7 +107,7 @@ export const Dragable = Extension.create({
                 }
               }, 50);
             },
-            mousemove(view, event) {
+            mousedown(view, event) {
               if (!dropElement) return;
 
               const coords = { left: event.clientX, top: event.clientY };
@@ -134,10 +150,6 @@ export const Dragable = Extension.create({
               dropElement.style.top = rect.top + 6 + 'px';
               dropElement.style.opacity = 1;
             },
-            mouseleave() {
-              if (!dropElement || currentNode) return;
-              dropElement.style.opacity = 0;
-            },
           },
         },
       }),
diff --git a/packages/client/src/tiptap/core/wrappers/attachment/index.tsx b/packages/client/src/tiptap/core/wrappers/attachment/index.tsx
index 50687a57..e16293e1 100644
--- a/packages/client/src/tiptap/core/wrappers/attachment/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/attachment/index.tsx
@@ -6,6 +6,7 @@ import { Tooltip } from 'components/tooltip';
 import { useToggle } from 'hooks/use-toggle';
 import { useCallback, useEffect, useRef, useState } from 'react';
 import { uploadFile } from 'services/file';
+import { Attachment } from 'tiptap/core/extensions/attachment';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 import { download, extractFileExtension, extractFilename, normalizeFileSize } from 'tiptap/prose-utils';
 
@@ -154,5 +155,9 @@ export const AttachmentWrapper = ({ editor, node, updateAttributes }) => {
     }
   })();
 
-  return <DragableWrapper editor={editor}>{content}</DragableWrapper>;
+  return (
+    <DragableWrapper editor={editor} extensionName={Attachment.name}>
+      {content}
+    </DragableWrapper>
+  );
 };
diff --git a/packages/client/src/tiptap/core/wrappers/callout/index.tsx b/packages/client/src/tiptap/core/wrappers/callout/index.tsx
index f9d03715..bea85295 100644
--- a/packages/client/src/tiptap/core/wrappers/callout/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/callout/index.tsx
@@ -4,6 +4,7 @@ import { EmojiPicker } from 'components/emoji-picker';
 import { convertColorToRGBA } from 'helpers/color';
 import { Theme, ThemeEnum } from 'hooks/use-theme';
 import { useCallback, useMemo } from 'react';
+import { Callout } from 'tiptap/core/extensions/callout';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 
 import styles from './index.module.scss';
@@ -26,7 +27,12 @@ export const CalloutWrapper = ({ editor, node, updateAttributes }) => {
   );
 
   return (
-    <DragableWrapper editor={editor} id="js-callout-container" className={cls(styles.wrap)}>
+    <DragableWrapper
+      editor={editor}
+      extensionName={Callout.name}
+      id="js-callout-container"
+      className={cls(styles.wrap)}
+    >
       <div
         className={cls(styles.innerWrap, 'render-wrapper')}
         style={{
diff --git a/packages/client/src/tiptap/core/wrappers/code-block/index.tsx b/packages/client/src/tiptap/core/wrappers/code-block/index.tsx
index a1794fea..8b2142e3 100644
--- a/packages/client/src/tiptap/core/wrappers/code-block/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/code-block/index.tsx
@@ -4,6 +4,7 @@ import { NodeViewContent } from '@tiptap/react';
 import cls from 'classnames';
 import { copy } from 'helpers/copy';
 import React, { useRef } from 'react';
+import { CodeBlock } from 'tiptap/core/extensions/code-block';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 
 import styles from './index.module.scss';
@@ -14,7 +15,7 @@ export const CodeBlockWrapper = ({ editor, node: { attrs }, updateAttributes, ex
   const $container = useRef<HTMLPreElement>();
 
   return (
-    <DragableWrapper editor={editor} className={cls(styles.wrap, 'render-wrapper')}>
+    <DragableWrapper editor={editor} extensionName={CodeBlock.name} className={cls(styles.wrap, 'render-wrapper')}>
       <div className={styles.handleWrap}>
         <Select
           size="small"
diff --git a/packages/client/src/tiptap/core/wrappers/countdown/index.tsx b/packages/client/src/tiptap/core/wrappers/countdown/index.tsx
index d129b31b..1cf3c0fc 100644
--- a/packages/client/src/tiptap/core/wrappers/countdown/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/countdown/index.tsx
@@ -1,6 +1,7 @@
 import { Space, Typography } from '@douyinfe/semi-ui';
 import cls from 'classnames';
-import Countdown from 'react-countdown';
+import ReactCountdown from 'react-countdown';
+import { Countdown } from 'tiptap/core/extensions/countdown';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 
 import styles from './index.module.scss';
@@ -32,10 +33,10 @@ export const CountdownWrapper = ({ editor, node }) => {
   const { title, date } = node.attrs;
 
   return (
-    <DragableWrapper editor={editor}>
+    <DragableWrapper editor={editor} extensionName={Countdown.name}>
       <div className={cls(styles.wrap, 'render-wrapper')}>
         <Text>{title}</Text>
-        <Countdown date={date} renderer={renderer}></Countdown>
+        <ReactCountdown date={date} renderer={renderer}></ReactCountdown>
       </div>
     </DragableWrapper>
   );
diff --git a/packages/client/src/tiptap/core/wrappers/document-children/index.tsx b/packages/client/src/tiptap/core/wrappers/document-children/index.tsx
index bba3f1ab..a477faa1 100644
--- a/packages/client/src/tiptap/core/wrappers/document-children/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/document-children/index.tsx
@@ -7,6 +7,7 @@ import { useChildrenDocument } from 'data/document';
 import Link from 'next/link';
 import { useRouter } from 'next/router';
 import { useEffect } from 'react';
+import { DocumentChildren } from 'tiptap/core/extensions/document-children';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 
 import styles from './index.module.scss';
@@ -37,6 +38,7 @@ export const DocumentChildrenWrapper = ({ editor, node, updateAttributes }) => {
   return (
     <DragableWrapper
       editor={editor}
+      extensionName={DocumentChildren.name}
       as="div"
       className={cls('render-wrapper', styles.wrap, isEditable && styles.isEditable, 'documentChildren')}
     >
diff --git a/packages/client/src/tiptap/core/wrappers/document-reference/index.tsx b/packages/client/src/tiptap/core/wrappers/document-reference/index.tsx
index 5a5ca181..484c82ab 100644
--- a/packages/client/src/tiptap/core/wrappers/document-reference/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/document-reference/index.tsx
@@ -3,6 +3,7 @@ import { IconDocument } from 'components/icons';
 import Link from 'next/link';
 import { useRouter } from 'next/router';
 import { useMemo } from 'react';
+import { DocumentReference } from 'tiptap/core/extensions/document-reference';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 
 import styles from './index.module.scss';
@@ -50,7 +51,12 @@ export const DocumentReferenceWrapper = ({ editor, node, updateAttributes }) =>
   }, [organizationId, wikiId, documentId, isEditable, isShare, title]);
 
   return (
-    <DragableWrapper editor={editor} as="div" className={cls(styles.wrap, isEditable && 'render-wrapper')}>
+    <DragableWrapper
+      editor={editor}
+      extensionName={DocumentReference.name}
+      as="div"
+      className={cls(styles.wrap, isEditable && 'render-wrapper')}
+    >
       {content}
     </DragableWrapper>
   );
diff --git a/packages/client/src/tiptap/core/wrappers/dragable/index.module.scss b/packages/client/src/tiptap/core/wrappers/dragable/index.module.scss
index 8487d957..979dbfb4 100644
--- a/packages/client/src/tiptap/core/wrappers/dragable/index.module.scss
+++ b/packages/client/src/tiptap/core/wrappers/dragable/index.module.scss
@@ -21,7 +21,7 @@
   }
 
   &.isEditable {
-    &:hover {
+    &.isActive {
       .dragHandle {
         opacity: 1;
       }
diff --git a/packages/client/src/tiptap/core/wrappers/dragable/index.tsx b/packages/client/src/tiptap/core/wrappers/dragable/index.tsx
index 75bd9d5a..eb1c0375 100644
--- a/packages/client/src/tiptap/core/wrappers/dragable/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/dragable/index.tsx
@@ -2,23 +2,26 @@ import { Editor } from '@tiptap/core';
 import { NodeViewWrapper } from '@tiptap/react';
 import cls from 'classnames';
 import React, { ElementType } from 'react';
+import { useActive } from 'tiptap/core/hooks/use-active';
 
 import styles from './index.module.scss';
 
 export const DragableWrapper: React.FC<{
   editor: Editor;
+  extensionName: string;
   as?: ElementType;
   id?: string;
   className?: string;
   style?: React.CSSProperties;
-}> = ({ editor, as = 'div', id, className, style = {}, children }) => {
+}> = ({ editor, extensionName, as = 'div', id, className, style = {}, children }) => {
   const isEditable = editor.isEditable;
+  const isActive = useActive(editor, extensionName);
 
   return (
     <NodeViewWrapper
       as={as}
       id={id}
-      className={cls(styles.draggableItem, isEditable && styles.isEditable, className)}
+      className={cls(styles.draggableItem, isEditable && styles.isEditable, isActive && styles.isActive, className)}
       style={style}
     >
       <div className={styles.dragHandle} contentEditable="false" draggable="true" data-drag-handle />
diff --git a/packages/client/src/tiptap/core/wrappers/flow/index.tsx b/packages/client/src/tiptap/core/wrappers/flow/index.tsx
index 9a25b233..9dbeaa9f 100644
--- a/packages/client/src/tiptap/core/wrappers/flow/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/flow/index.tsx
@@ -95,7 +95,11 @@ export const FlowWrapper = ({ editor, node, updateAttributes }) => {
   }, [toggleLoading, data]);
 
   return (
-    <DragableWrapper editor={editor} className={cls(styles.wrap, isActive && styles.isActive)}>
+    <DragableWrapper
+      editor={editor}
+      className={cls(styles.wrap, isActive && styles.isActive)}
+      extensionName={Flow.name}
+    >
       <VisibilitySensor onChange={onViewportChange}>
         <Resizeable isEditable={isEditable} width={width} height={height} maxWidth={maxWidth} onChangeEnd={onResize}>
           <div
diff --git a/packages/client/src/tiptap/core/wrappers/iframe/index.tsx b/packages/client/src/tiptap/core/wrappers/iframe/index.tsx
index 5d2f5cd5..b84e3953 100644
--- a/packages/client/src/tiptap/core/wrappers/iframe/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/iframe/index.tsx
@@ -2,6 +2,7 @@ import { Typography } from '@douyinfe/semi-ui';
 import cls from 'classnames';
 import { Resizeable } from 'components/resizeable';
 import { useCallback } from 'react';
+import { Iframe } from 'tiptap/core/extensions/iframe';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 import { getEditorContainerDOMSize } from 'tiptap/prose-utils';
 
@@ -22,7 +23,7 @@ export const IframeWrapper = ({ editor, node, updateAttributes }) => {
   );
 
   return (
-    <DragableWrapper editor={editor}>
+    <DragableWrapper editor={editor} extensionName={Iframe.name}>
       <Resizeable width={width} maxWidth={maxWidth} height={height} isEditable={isEditable} onChangeEnd={onResize}>
         <div className={cls(styles.wrap, 'render-wrapper')}>
           {url ? (
diff --git a/packages/client/src/tiptap/core/wrappers/image/index.tsx b/packages/client/src/tiptap/core/wrappers/image/index.tsx
index 34d2cbc6..96e3333a 100644
--- a/packages/client/src/tiptap/core/wrappers/image/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/image/index.tsx
@@ -4,6 +4,7 @@ import { useToggle } from 'hooks/use-toggle';
 import { useCallback, useEffect, useRef } from 'react';
 import { LazyLoadImage } from 'react-lazy-load-image-component';
 import { uploadFile } from 'services/file';
+import { Image } from 'tiptap/core/extensions/image';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 import {
   extractFileExtension,
@@ -69,7 +70,7 @@ export const ImageWrapper = ({ editor, node, updateAttributes }) => {
   }, [src, hasTrigger, selectFile, updateAttributes]);
 
   return (
-    <DragableWrapper editor={editor} style={{ textAlign, fontSize: 0, maxWidth: '100%' }}>
+    <DragableWrapper editor={editor} extensionName={Image.name} style={{ textAlign, fontSize: 0, maxWidth: '100%' }}>
       <Resizeable
         className={'render-wrapper'}
         width={width || maxWidth}
diff --git a/packages/client/src/tiptap/core/wrappers/katex/index.tsx b/packages/client/src/tiptap/core/wrappers/katex/index.tsx
index 5f42e632..c9f0dee1 100644
--- a/packages/client/src/tiptap/core/wrappers/katex/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/katex/index.tsx
@@ -2,6 +2,7 @@ import { convertColorToRGBA } from 'helpers/color';
 import { Theme, ThemeEnum } from 'hooks/use-theme';
 import katex from 'katex';
 import { useMemo } from 'react';
+import { Katex } from 'tiptap/core/extensions/katex';
 import { DragableWrapper } from 'tiptap/core/wrappers/dragable';
 
 import styles from './index.module.scss';
@@ -36,6 +37,7 @@ export const KatexWrapper = ({ node, editor }) => {
   return (
     <DragableWrapper
       editor={editor}
+      extensionName={Katex.name}
       className={'render-wrapper'}
       style={{
         backgroundColor,
diff --git a/packages/client/src/tiptap/core/wrappers/mind/index.tsx b/packages/client/src/tiptap/core/wrappers/mind/index.tsx
index 6be3df0e..9da84173 100644
--- a/packages/client/src/tiptap/core/wrappers/mind/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/mind/index.tsx
@@ -108,7 +108,11 @@ export const MindWrapper = ({ editor, node, updateAttributes }) => {
   }, [width, height, setCenter]);
 
   return (
-    <DragableWrapper editor={editor} className={cls(styles.wrap, isActive && styles.isActive)}>
+    <DragableWrapper
+      editor={editor}
+      extensionName={Mind.name}
+      className={cls(styles.wrap, isActive && styles.isActive)}
+    >
       <VisibilitySensor onChange={onViewportChange}>
         <Resizeable isEditable={isEditable} width={width} height={height} maxWidth={maxWidth} onChangeEnd={onResize}>
           <div