From 332a5c66851b77fa1da5f1698a7d833798ce7ba0 Mon Sep 17 00:00:00 2001
From: JimmFly <447268514@qq.com>
Date: Thu, 21 Dec 2023 07:13:05 +0000
Subject: [PATCH] chore(core): temporarily hide export png and pdf in edgeless
 (#5364)

---
 .../page-list/operation-menu-items/export.tsx | 19 ++++++++++++++-----
 .../share-menu/share-export.tsx               |  4 ++++
 .../operation-menu.tsx                        |  2 +-
 ...se-register-blocksuite-editor-commands.tsx |  4 ++--
 4 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/packages/frontend/component/src/components/page-list/operation-menu-items/export.tsx b/packages/frontend/component/src/components/page-list/operation-menu-items/export.tsx
index d37bb64a18..939fd9f1ea 100644
--- a/packages/frontend/component/src/components/page-list/operation-menu-items/export.tsx
+++ b/packages/frontend/component/src/components/page-list/operation-menu-items/export.tsx
@@ -21,6 +21,7 @@ interface ExportMenuItemProps<T> {
 
 interface ExportProps {
   exportHandler: (type: 'pdf' | 'html' | 'png' | 'markdown') => Promise<void>;
+  pageMode?: 'page' | 'edgeless';
   className?: string;
 }
 
@@ -47,6 +48,7 @@ export function ExportMenuItem<T>({
 export const ExportMenuItems = ({
   exportHandler,
   className = transitionStyle,
+  pageMode = 'page',
 }: ExportProps) => {
   const t = useAFFiNEI18N();
   const itemMap = useMemo(
@@ -94,16 +96,23 @@ export const ExportMenuItems = ({
     ],
     [className, exportHandler, t]
   );
-  const items = itemMap.map(({ component: Component, props }) => (
-    <Component key={props.label} {...props} />
-  ));
+  const items = itemMap.map(({ component: Component, props }) =>
+    pageMode === 'edgeless' &&
+    (props.type === 'pdf' || props.type === 'png') ? null : (
+      <Component key={props.label} {...props} />
+    )
+  );
   return items;
 };
 
-export const Export = ({ exportHandler, className }: ExportProps) => {
+export const Export = ({ exportHandler, className, pageMode }: ExportProps) => {
   const t = useAFFiNEI18N();
   const items = (
-    <ExportMenuItems exportHandler={exportHandler} className={className} />
+    <ExportMenuItems
+      exportHandler={exportHandler}
+      className={className}
+      pageMode={pageMode}
+    />
   );
   return (
     <MenuSub
diff --git a/packages/frontend/core/src/components/affine/share-page-modal/share-menu/share-export.tsx b/packages/frontend/core/src/components/affine/share-page-modal/share-menu/share-export.tsx
index 2819f1bf53..6ce92be6cd 100644
--- a/packages/frontend/core/src/components/affine/share-page-modal/share-menu/share-export.tsx
+++ b/packages/frontend/core/src/components/affine/share-page-modal/share-menu/share-export.tsx
@@ -4,7 +4,9 @@ import { Divider } from '@affine/component/ui/divider';
 import { WorkspaceFlavour } from '@affine/env/workspace';
 import { useAFFiNEI18N } from '@affine/i18n/hooks';
 import { LinkIcon } from '@blocksuite/icons';
+import { useAtomValue } from 'jotai';
 
+import { currentModeAtom } from '../../../../atoms/mode';
 import { useExportPage } from '../../../../hooks/affine/use-export-page';
 import * as styles from './index.css';
 import type { ShareMenuProps } from './share-menu';
@@ -23,6 +25,7 @@ export const ShareExport = ({
     urlType: 'workspace',
   });
   const exportHandler = useExportPage(currentPage);
+  const currentMode = useAtomValue(currentModeAtom);
 
   return (
     <>
@@ -36,6 +39,7 @@ export const ShareExport = ({
         <ExportMenuItems
           exportHandler={exportHandler}
           className={styles.menuItemStyle}
+          pageMode={currentMode}
         />
       </div>
       {workspace.flavour !== WorkspaceFlavour.LOCAL ? (
diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-header-title/operation-menu.tsx b/packages/frontend/core/src/components/blocksuite/block-suite-header-title/operation-menu.tsx
index 4aa9080bb4..7bb2979a48 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-header-title/operation-menu.tsx
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-header-title/operation-menu.tsx
@@ -182,7 +182,7 @@ export const PageHeaderMenuButton = ({ rename, pageId }: PageMenuProps) => {
       >
         {t['Import']()}
       </MenuItem>
-      <Export exportHandler={exportHandler} />
+      <Export exportHandler={exportHandler} pageMode={currentMode} />
 
       {workspace.flavour === WorkspaceFlavour.AFFINE_CLOUD &&
       runtimeConfig.enablePageHistory ? (
diff --git a/packages/frontend/core/src/hooks/affine/use-register-blocksuite-editor-commands.tsx b/packages/frontend/core/src/hooks/affine/use-register-blocksuite-editor-commands.tsx
index 4667436032..302f34ad00 100644
--- a/packages/frontend/core/src/hooks/affine/use-register-blocksuite-editor-commands.tsx
+++ b/packages/frontend/core/src/hooks/affine/use-register-blocksuite-editor-commands.tsx
@@ -141,7 +141,7 @@ export function useRegisterBlocksuiteEditorCommands(
     unsubs.push(
       registerAffineCommand({
         id: `editor:${mode}-export-to-pdf`,
-        preconditionStrategy,
+        preconditionStrategy: () => mode === 'page' && !trash,
         category: `editor:${mode}`,
         icon: mode === 'page' ? <PageIcon /> : <EdgelessIcon />,
         label: t['Export to PDF'](),
@@ -167,7 +167,7 @@ export function useRegisterBlocksuiteEditorCommands(
     unsubs.push(
       registerAffineCommand({
         id: `editor:${mode}-export-to-png`,
-        preconditionStrategy,
+        preconditionStrategy: () => mode === 'page' && !trash,
         category: `editor:${mode}`,
         icon: mode === 'page' ? <PageIcon /> : <EdgelessIcon />,
         label: t['Export to PNG'](),