feat: 支持导出 docx 文件

LeafYeeXYZ · Dec 8, 2024 · b363b30 · b363b30
1 parent aad6d73
commit b363b30
Show file tree

Hide file tree

Showing 7 changed files with 82 additions and 2 deletions.
diff --git a/bun.lockb b/bun.lockb
diff --git a/package.json b/package.json
@@ -45,6 +45,7 @@
     "@electron-toolkit/preload": "^3.0.1",
     "@electron-toolkit/utils": "^3.0.0",
     "antd": "^5.21.1",
+    "html-docx-js-typescript": "^0.1.5",
     "katex": "^0.16.11",
     "marked": "^14.1.2",
     "marked-katex-extension": "^5.1.2",

diff --git a/src/main/index.ts b/src/main/index.ts
@@ -7,6 +7,7 @@ import path from 'node:path'
 import { mdToHtml } from '../../lib/render'
 import { getTheme } from '../../lib/utils'
 import { embedImageIntoHtml } from './utils'
+import { asBlob } from 'html-docx-js-typescript'
 
 function createWindow(): void {
   // Create the browser window.
@@ -158,6 +159,44 @@ app.whenReady().then(() => {
       return true
     }
   )
+  ipcMain.handle(
+    'createDocx',
+    async (
+      _,
+      markdown: string,
+      themeName: string = 'aps',
+      filepath: string,
+      filename: string
+    ): Promise<boolean> => {
+      const { response } = await dialog.showMessageBox({
+        type: 'warning',
+        title: '导出 Word',
+        message:
+          '导出的 DOCX 文件可能存在部分样式丢失，请手动调整或或选择导出为 PDF 文件. 是否继续？',
+        buttons: ['取消', '继续']
+      })
+      if (response === 0) {
+        return false
+      }
+      const { filePath, canceled } = await dialog.showSaveDialog({
+        title: '导出论文',
+        filters: [{ name: 'Word 文件', extensions: ['docx'] }],
+        defaultPath: filename.split('.')[0] + '.docx',
+        showsTagField: false,
+        properties: ['createDirectory']
+      })
+      if (canceled) {
+        return false
+      }
+      const theme = getTheme(themeName)
+      const dist = path.resolve(filePath)
+      const html = await embedImageIntoHtml(await mdToHtml(markdown, theme), filepath)
+      const docx = await asBlob(html)
+      const docxBuffer = Buffer.from(docx instanceof Blob ? await docx.arrayBuffer() : docx)
+      await fs.writeFile(dist, docxBuffer)
+      return true
+    }
+  )
 
   createWindow()
 

diff --git a/src/preload/index.d.ts b/src/preload/index.d.ts
@@ -16,6 +16,12 @@ declare global {
         filepath: string,
         filename: string
       ) => Promise<boolean>
+      createDocx: (
+        markdown: string,
+        themeName: string,
+        filepath: string,
+        filename: string
+      ) => Promise<boolean>
     }
   }
 }
diff --git a/src/preload/index.ts b/src/preload/index.ts
@@ -28,6 +28,14 @@ const api = {
     filename: string
   ): Promise<boolean> => {
     return electronAPI.ipcRenderer.invoke('createPdf', markdown, themeName, filepath, filename)
+  },
+  createDocx: (
+    markdown: string,
+    themeName: string,
+    filepath: string,
+    filename: string
+  ): Promise<boolean> => {
+    return electronAPI.ipcRenderer.invoke('createDocx', markdown, themeName, filepath, filename)
   }
 }
 

diff --git a/src/renderer/index.html b/src/renderer/index.html
@@ -2,7 +2,7 @@
 <html>
   <head>
     <meta charset="UTF-8" />
-    <title>EasyPaper v1.5.0</title>
+    <title>EasyPaper v1.6.0</title>
     <!-- https://developer.mozilla.org/en-US/docs/Web/HTTP/CSP -->
     <meta
       http-equiv="Content-Security-Policy"

diff --git a/src/renderer/src/components/Toolbar.tsx b/src/renderer/src/components/Toolbar.tsx
@@ -6,7 +6,8 @@ import {
   QuestionCircleOutlined,
   FileMarkdownOutlined,
   FilePdfOutlined,
-  FileTextOutlined
+  FileTextOutlined,
+  FileWordOutlined
 } from '@ant-design/icons'
 import { useStore } from '../lib/useStore'
 import { mdToHtml } from '../../../../lib/render'
@@ -136,6 +137,31 @@ export default function Toolbar(): JSX.Element {
       >
         <FileTextOutlined /> HTML
       </Button>
+      <Button
+        disabled={!filepath || !filename || !markdown.length || !messageApi || disabled}
+        onClick={async () => {
+          flushSync(() => setDisabled(true))
+          messageApi!.open({
+            content: '导出中...',
+            type: 'loading',
+            duration: 0,
+            key: 'exporting'
+          })
+          await window.api
+            .createDocx(markdown, theme.themeName, filepath, filename)
+            .then((res) => {
+              messageApi!.destroy()
+              res && messageApi!.success('导出成功')
+            })
+            .catch(() => {
+              messageApi!.destroy()
+              messageApi!.error('导出失败')
+            })
+            .finally(() => setDisabled(false))
+        }}
+      >
+        <FileWordOutlined /> DOCX
+      </Button>
       <p className="text-xs px-1 font-bold text-nowrap text-gray-600 dark:text-white">论文格式:</p>
       <Select
         defaultValue={getDefalutTheme().themeName}