From 4161d5fd2d4a9fcc19f3dbad183970e40e16d5e7 Mon Sep 17 00:00:00 2001
From: fantasticit <fantasticit@163.com>
Date: Tue, 5 Jul 2022 15:21:06 +0800
Subject: [PATCH] tiptap: fix doc url

---
 .../src/tiptap/core/extensions/document-reference.ts   |  4 ++++
 .../tiptap/core/menus/document-reference/bubble.tsx    |  4 ++--
 .../tiptap/core/wrappers/document-children/index.tsx   |  6 ++++--
 .../tiptap/core/wrappers/document-reference/index.tsx  | 10 ++++++----
 4 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/packages/client/src/tiptap/core/extensions/document-reference.ts b/packages/client/src/tiptap/core/extensions/document-reference.ts
index 2f84914e..aec83207 100644
--- a/packages/client/src/tiptap/core/extensions/document-reference.ts
+++ b/packages/client/src/tiptap/core/extensions/document-reference.ts
@@ -25,6 +25,10 @@ export const DocumentReference = Node.create({
 
   addAttributes() {
     return {
+      organizationId: {
+        default: '',
+        parseHTML: getDatasetAttribute('organizationId'),
+      },
       wikiId: {
         default: '',
         parseHTML: getDatasetAttribute('wikiId'),
diff --git a/packages/client/src/tiptap/core/menus/document-reference/bubble.tsx b/packages/client/src/tiptap/core/menus/document-reference/bubble.tsx
index 3bc721bd..8b8a0dc5 100644
--- a/packages/client/src/tiptap/core/menus/document-reference/bubble.tsx
+++ b/packages/client/src/tiptap/core/menus/document-reference/bubble.tsx
@@ -33,11 +33,11 @@ export const DocumentReferenceBubbleMenu = ({ editor }) => {
   const shouldShow = useCallback(() => editor.isActive(DocumentReference.name), [editor]);
   const selectDoc = useCallback(
     (item) => {
-      const { wikiId, title, id: documentId } = item;
+      const { organizationId, wikiId, title, id: documentId } = item;
 
       editor
         .chain()
-        .updateAttributes(DocumentReference.name, { wikiId, documentId, title })
+        .updateAttributes(DocumentReference.name, { organizationId, wikiId, documentId, title })
         .setNodeSelection(editor.state.selection.from)
         .focus()
         .run();
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 d8f7f1f0..bba3f1ab 100644
--- a/packages/client/src/tiptap/core/wrappers/document-children/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/document-children/index.tsx
@@ -59,8 +59,10 @@ export const DocumentChildrenWrapper = ({ editor, node, updateAttributes }) => {
                       <Link
                         key={doc.id}
                         href={{
-                          pathname: `${!isShare ? '' : '/share'}/wiki/[wikiId]/document/[documentId]`,
-                          query: { wikiId: doc.wikiId, documentId: doc.id },
+                          pathname: isShare
+                            ? `/share/wiki/[wikiId]/document/[documentId]`
+                            : `/app/org/[organizationId]/wiki/[wikiId]/doc/[documentId]`,
+                          query: { organizationId: doc.organizationId, wikiId: doc.wikiId, documentId: doc.id },
                         }}
                       >
                         <a className={styles.itemWrap} target="_blank">
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 cc41c19e..5a5ca181 100644
--- a/packages/client/src/tiptap/core/wrappers/document-reference/index.tsx
+++ b/packages/client/src/tiptap/core/wrappers/document-reference/index.tsx
@@ -11,7 +11,7 @@ export const DocumentReferenceWrapper = ({ editor, node, updateAttributes }) =>
   const { pathname } = useRouter();
   const isShare = pathname.includes('share');
   const isEditable = editor.isEditable;
-  const { wikiId, documentId, title } = node.attrs;
+  const { organizationId, wikiId, documentId, title } = node.attrs;
 
   const content = useMemo(() => {
     if (!wikiId && !documentId) {
@@ -35,8 +35,10 @@ export const DocumentReferenceWrapper = ({ editor, node, updateAttributes }) =>
       <Link
         key={documentId}
         href={{
-          pathname: `${!isShare ? '' : '/share'}/wiki/[wikiId]/document/[documentId]`,
-          query: { wikiId, documentId },
+          pathname: isShare
+            ? `/share/wiki/[wikiId]/document/[documentId]`
+            : `/app/org/[organizationId]/wiki/[wikiId]/doc/[documentId]`,
+          query: { organizationId, wikiId, documentId },
         }}
       >
         <a className={cls(styles.itemWrap, !isEditable && 'render-wrapper')} target="_blank">
@@ -45,7 +47,7 @@ export const DocumentReferenceWrapper = ({ editor, node, updateAttributes }) =>
         </a>
       </Link>
     );
-  }, [wikiId, documentId, isEditable, isShare, title]);
+  }, [organizationId, wikiId, documentId, isEditable, isShare, title]);
 
   return (
     <DragableWrapper editor={editor} as="div" className={cls(styles.wrap, isEditable && 'render-wrapper')}>