From 8949af5715d9516b5187a4bf8a589a2506640f19 Mon Sep 17 00:00:00 2001
From: fantasticit <fantasticit@163.com>
Date: Sun, 27 Mar 2022 14:18:19 +0800
Subject: [PATCH] feat: add image viewer support

---
 packages/client/package.json                  |  1 +
 .../src/components/document/reader/editor.tsx |  2 ++
 .../document/reader/public/index.tsx          |  2 ++
 .../client/src/components/image-viewer.tsx    | 30 +++++++++++++++++++
 .../src/components/template/reader/editor.tsx |  2 ++
 packages/client/src/pages/_app.tsx            |  1 +
 pnpm-lock.yaml                                |  6 ++++
 7 files changed, 44 insertions(+)
 create mode 100644 packages/client/src/components/image-viewer.tsx

diff --git a/packages/client/package.json b/packages/client/package.json
index fdf34b17..827712bd 100644
--- a/packages/client/package.json
+++ b/packages/client/package.json
@@ -80,6 +80,7 @@
     "scroll-into-view-if-needed": "^2.2.29",
     "swr": "^1.2.0",
     "tippy.js": "^6.3.7",
+    "viewerjs": "^1.10.4",
     "y-indexeddb": "^9.0.7",
     "y-prosemirror": "^1.0.14",
     "yjs": "^13.5.24"
diff --git a/packages/client/src/components/document/reader/editor.tsx b/packages/client/src/components/document/reader/editor.tsx
index d418be65..35c25f82 100644
--- a/packages/client/src/components/document/reader/editor.tsx
+++ b/packages/client/src/components/document/reader/editor.tsx
@@ -12,6 +12,7 @@ import {
   destoryProvider,
 } from 'components/tiptap';
 import { DataRender } from 'components/data-render';
+import { ImageViewer } from 'components/image-viewer';
 import { joinUser } from 'components/document/collaboration';
 import { CreateUser } from './user';
 import styles from './index.module.scss';
@@ -72,6 +73,7 @@ export const Editor: React.FC<IProps> = ({ user, documentId, document }) => {
           <>
             <Content className={styles.editorWrap}>
               <div id="js-reader-container">
+                <ImageViewer containerSelector="#js-reader-container" />
                 <EditorContent editor={editor} />
               </div>
               <CreateUser
diff --git a/packages/client/src/components/document/reader/public/index.tsx b/packages/client/src/components/document/reader/public/index.tsx
index 57753cde..e1019aa1 100644
--- a/packages/client/src/components/document/reader/public/index.tsx
+++ b/packages/client/src/components/document/reader/public/index.tsx
@@ -8,6 +8,7 @@ import { DataRender } from 'components/data-render';
 import { DocumentStyle } from 'components/document/style';
 import { User } from 'components/user';
 import { Theme } from 'components/theme';
+import { ImageViewer } from 'components/image-viewer';
 import { useDocumentStyle } from 'hooks/useDocumentStyle';
 import { useWindowSize } from 'hooks/useWindowSize';
 import { usePublicDocument } from 'data/document';
@@ -120,6 +121,7 @@ export const DocumentPublicReader: React.FC<IProps> = ({ documentId, hideLogo =
                     createUserContainerSelector="#js-share-document-editor-container .ProseMirror .title"
                   />
                 </div>
+                <ImageViewer containerSelector="#js-share-document-editor-container" />
                 <BackTop target={() => document.querySelector('#js-share-document-editor-container').parentNode} />
               </>
             );
diff --git a/packages/client/src/components/image-viewer.tsx b/packages/client/src/components/image-viewer.tsx
new file mode 100644
index 00000000..fe184239
--- /dev/null
+++ b/packages/client/src/components/image-viewer.tsx
@@ -0,0 +1,30 @@
+import { useEffect } from 'react';
+import Viewer from 'viewerjs';
+
+interface IProps {
+  containerSelector: string;
+}
+
+export const ImageViewer: React.FC<IProps> = ({ containerSelector }) => {
+  useEffect(() => {
+    const el = document.querySelector(containerSelector);
+    if (!el) {
+      return null;
+    }
+    const viewer = new Viewer(el as HTMLElement, { inline: false });
+    const io = new MutationObserver(() => {
+      viewer.update();
+    });
+    io.observe(el, {
+      childList: true,
+      subtree: true,
+    });
+
+    return () => {
+      io.disconnect();
+      viewer.destroy();
+    };
+  }, [containerSelector]);
+
+  return null;
+};
diff --git a/packages/client/src/components/template/reader/editor.tsx b/packages/client/src/components/template/reader/editor.tsx
index fb7b6475..f8e3fdec 100644
--- a/packages/client/src/components/template/reader/editor.tsx
+++ b/packages/client/src/components/template/reader/editor.tsx
@@ -5,6 +5,7 @@ import { Layout, Spin, Typography } from '@douyinfe/semi-ui';
 import { IUser, ITemplate } from '@think/domains';
 import { DEFAULT_EXTENSION, DocumentWithTitle } from 'components/tiptap';
 import { DataRender } from 'components/data-render';
+import { ImageViewer } from 'components/image-viewer';
 import { useDocumentStyle } from 'hooks/useDocumentStyle';
 import { safeJSONParse } from 'helpers/json';
 import styles from './index.module.scss';
@@ -61,6 +62,7 @@ export const Editor: React.FC<IProps> = ({ user, data, loading, error }) => {
                   <Title>{data.title}</Title>
                   <EditorContent editor={editor} />
                 </div>
+                <ImageViewer containerSelector={`.${styles.editorWrap}`} />
               </Content>
             );
           }}
diff --git a/packages/client/src/pages/_app.tsx b/packages/client/src/pages/_app.tsx
index 3f73b07a..bfc053a7 100644
--- a/packages/client/src/pages/_app.tsx
+++ b/packages/client/src/pages/_app.tsx
@@ -1,5 +1,6 @@
 import type { AppProps } from 'next/app';
 import { useSafari100vh } from 'hooks/use-safari-100vh';
+import 'viewerjs/dist/viewer.css';
 import 'styles/globals.scss';
 
 function MyApp({ Component, pageProps }: AppProps) {
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index e63fe000..187cb94c 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -118,6 +118,7 @@ importers:
       tippy.js: ^6.3.7
       tsconfig-paths-webpack-plugin: ^3.5.2
       typescript: 4.5.5
+      viewerjs: ^1.10.4
       y-indexeddb: ^9.0.7
       y-prosemirror: ^1.0.14
       yjs: ^13.5.24
@@ -193,6 +194,7 @@ importers:
       scroll-into-view-if-needed: 2.2.29
       swr: 1.2.0_react@17.0.2
       tippy.js: 6.3.7
+      viewerjs: 1.10.4
       y-indexeddb: 9.0.7_yjs@13.5.24
       y-prosemirror: 1.0.14_0fedec857d2fb730ad5b02a71124bf2a
       yjs: 13.5.24
@@ -8258,6 +8260,10 @@ packages:
     engines: {node: '>= 0.8'}
     dev: false
 
+  /viewerjs/1.10.4:
+    resolution: {integrity: sha512-CjMt64yC9D+XUx2t3F0TPbh/Yt5+/ke8/s3IizXa6NtksdJUFDoCcNxi/KRZ9eiZPR/D77pHnnQzAtCoLDaGIw==}
+    dev: false
+
   /vm2/3.9.5:
     resolution: {integrity: sha512-LuCAHZN75H9tdrAiLFf030oW7nJV5xwNMuk1ymOZwopmuK3d2H4L1Kv4+GFHgarKiLfXXLFU+7LDABHnwOkWng==}
     engines: {node: '>=6.0'}