From a0ce75c902c434d471480df8c1f9b9603c4e9d5d Mon Sep 17 00:00:00 2001
From: donteatfriedrice <chs@bupt.cn>
Date: Wed, 3 Jul 2024 07:34:07 +0000
Subject: [PATCH] fix: ai button should not display in readonly mode (#7415)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

[BS-737](https://linear.app/affine-design/issue/BS-737/shared-page-也可以用-ai)
---
 .../presets/ai/entries/code-toolbar/setup-code-toolbar.ts       | 2 +-
 .../presets/ai/entries/image-toolbar/setup-image-toolbar.ts     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts b/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts
index c37064620d..9ff5375cb8 100644
--- a/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/entries/code-toolbar/setup-code-toolbar.ts
@@ -33,7 +33,7 @@ export function setupCodeToolbarEntry(codeToolbar: AffineCodeToolbarWidget) {
         name: 'Ask AI',
         tooltip: 'Ask AI',
         icon: AIStarIcon,
-        showWhen: () => true,
+        showWhen: CodeBlockComponent => !CodeBlockComponent.doc.readonly,
         render(codeBlock: CodeBlockComponent, onClick?: () => void) {
           return html`<ask-ai-button
             class="code-toolbar-button ask-ai"
diff --git a/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts b/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts
index b4f9e4a9ba..33a951516b 100644
--- a/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/entries/image-toolbar/setup-image-toolbar.ts
@@ -44,7 +44,7 @@ export function setupImageToolbarEntry(imageToolbar: AffineImageToolbarWidget) {
             }}
           ></ask-ai-button>`;
         },
-        showWhen: () => true,
+        showWhen: imageBlockComponent => !imageBlockComponent.doc.readonly,
       },
     ],
     0