From f9d0a348c41d2b07ee7297af40fba913679aad14 Mon Sep 17 00:00:00 2001
From: fundon <fundon@pindash.io>
Date: Tue, 3 Sep 2024 03:52:10 +0000
Subject: [PATCH] feat(core): add custom configuration to database options
 (#7933)

Upstreams: https://github.com/toeverything/blocksuite/pull/8022

* add custom configuration to database options
* add `Copy link to block` button to database options

<img width="518" alt="Screenshot 2024-09-03 at 08 57 59" src="https://github.com/user-attachments/assets/a421cd82-abd4-456e-af17-c4db6c4ff3ae">
---
 .../specs/custom/database-block.ts            | 84 +++++++++++++++++++
 .../specs/custom/root-block.ts                |  3 +
 2 files changed, 87 insertions(+)
 create mode 100644 packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/database-block.ts

diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/database-block.ts b/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/database-block.ts
new file mode 100644
index 0000000000..7565a33138
--- /dev/null
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/database-block.ts
@@ -0,0 +1,84 @@
+import { notify } from '@affine/component';
+import {
+  generateUrl,
+  type UseSharingUrl,
+} from '@affine/core/hooks/affine/use-share-url';
+import { getAffineCloudBaseUrl } from '@affine/core/modules/cloud/services/fetch';
+import { EditorService } from '@affine/core/modules/editor';
+import { I18n } from '@affine/i18n';
+import type { DatabaseBlockModel, MenuOptions } from '@blocksuite/blocks';
+import { LinkIcon } from '@blocksuite/icons/lit';
+import { type FrameworkProvider } from '@toeverything/infra';
+import type { TemplateResult } from 'lit';
+
+export function createDatabaseOptionsConfig(framework: FrameworkProvider) {
+  return {
+    configure: (model: DatabaseBlockModel, options: MenuOptions) => {
+      const items = options.items;
+
+      const copyIndex = items.findIndex(
+        item => item.type === 'action' && item.name === 'Copy'
+      );
+
+      items.splice(
+        copyIndex + 1,
+        0,
+        createCopyLinkToBlockMenuItem(framework, model)
+      );
+
+      return options;
+    },
+  };
+}
+
+function createCopyLinkToBlockMenuItem(
+  framework: FrameworkProvider,
+  model: DatabaseBlockModel
+): {
+  type: 'action';
+  name: string;
+  icon?: TemplateResult<1>;
+  hide?: () => boolean;
+  select: () => void;
+} {
+  return {
+    type: 'action',
+    name: 'Copy link to block',
+    icon: LinkIcon({ width: '20', height: '20' }),
+    hide: () => {
+      const { editor } = framework.get(EditorService);
+      const mode = editor.mode$.value;
+      return mode === 'edgeless';
+    },
+    select: () => {
+      const baseUrl = getAffineCloudBaseUrl();
+      if (!baseUrl) return;
+
+      const { editor } = framework.get(EditorService);
+      const mode = editor.mode$.value;
+
+      if (mode === 'edgeless') return;
+
+      const pageId = editor.doc.id;
+      const workspaceId = editor.doc.workspace.id;
+      const options: UseSharingUrl = {
+        workspaceId,
+        pageId,
+        shareMode: mode,
+        blockIds: [model.id],
+      };
+
+      const str = generateUrl(options);
+      if (!str) return;
+
+      navigator.clipboard
+        .writeText(str)
+        .then(() => {
+          notify.success({
+            title: I18n['Copied link to clipboard'](),
+          });
+        })
+        .catch(console.error);
+    },
+  };
+}
diff --git a/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts b/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts
index 9c5cc59fb7..9ec77759d9 100644
--- a/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts
+++ b/packages/frontend/core/src/components/blocksuite/block-suite-editor/specs/custom/root-block.ts
@@ -19,6 +19,7 @@ import {
 } from '@blocksuite/blocks';
 import { type FrameworkProvider } from '@toeverything/infra';
 
+import { createDatabaseOptionsConfig } from './database-block';
 import { createLinkedWidgetConfig } from './widgets/linked';
 import { createToolbarMoreMenuConfig } from './widgets/toolbar';
 
@@ -72,6 +73,7 @@ export function createPageRootBlockSpec(
       linkedWidget: createLinkedWidgetConfig(framework),
       editorSetting: editorSettingService.editorSetting.settingSignal,
       toolbarMoreMenu: createToolbarMoreMenuConfig(framework),
+      databaseOptions: createDatabaseOptionsConfig(framework),
     }),
   ];
 }
@@ -95,6 +97,7 @@ export function createEdgelessRootBlockSpec(
       linkedWidget: createLinkedWidgetConfig(framework),
       editorSetting: editorSettingService.editorSetting.settingSignal,
       toolbarMoreMenu: createToolbarMoreMenuConfig(framework),
+      databaseOptions: createDatabaseOptionsConfig(framework),
     }),
   ];
 }