From df13a54f3702c911b9af7e546034b66a080028a4 Mon Sep 17 00:00:00 2001
From: Bohdan Triapitsyn <artmore@protonmail.com>
Date: Sat, 18 Apr 2026 13:50:09 +0300
Subject: [PATCH] fix(vscode): hide Save-as-plan and Add-to-notes actions in VS
 Code runtime

Both flows write into ~/.config/openchamber which isn't reachable from the
VS Code webview, so the buttons were dead weight there.
---
 .../components/chat/message/MessageBody.tsx   | 40 ++++++-----
 .../chat/message/TextSelectionMenu.tsx        | 71 ++++++++++---------
 2 files changed, 60 insertions(+), 51 deletions(-)

diff --git a/packages/ui/src/components/chat/message/MessageBody.tsx b/packages/ui/src/components/chat/message/MessageBody.tsx
index 47081b2e..5f02c8bf 100644
--- a/packages/ui/src/components/chat/message/MessageBody.tsx
+++ b/packages/ui/src/components/chat/message/MessageBody.tsx
@@ -1498,25 +1498,27 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
                      </TooltipTrigger>
                      <TooltipContent sideOffset={6}>{isSharing ? 'Saving image...' : 'Save as image'}</TooltipContent>
                  </Tooltip>
-                <Tooltip delayDuration={1000}>
-                    <TooltipTrigger asChild>
-                        <Button
-                            type="button"
-                            size="icon"
-                            variant="ghost"
-                            disabled={!hasCopyableText || !currentProjectRef}
-                            className={cn(
-                                'h-8 w-8 text-muted-foreground bg-transparent hover:text-foreground hover:!bg-transparent active:!bg-transparent focus-visible:!bg-transparent focus-visible:ring-2 focus-visible:ring-primary/50',
-                                (!hasCopyableText || !currentProjectRef) && 'opacity-50'
-                            )}
-                            onPointerDown={(event) => event.stopPropagation()}
-                            onClick={handleSaveAsPlanClick}
-                        >
-                            <RiBookletLine className="h-4 w-4" />
-                        </Button>
-                    </TooltipTrigger>
-                    <TooltipContent sideOffset={6}>Save as plan</TooltipContent>
-                </Tooltip>
+                {!isVSCodeRuntime() ? (
+                    <Tooltip delayDuration={1000}>
+                        <TooltipTrigger asChild>
+                            <Button
+                                type="button"
+                                size="icon"
+                                variant="ghost"
+                                disabled={!hasCopyableText || !currentProjectRef}
+                                className={cn(
+                                    'h-8 w-8 text-muted-foreground bg-transparent hover:text-foreground hover:!bg-transparent active:!bg-transparent focus-visible:!bg-transparent focus-visible:ring-2 focus-visible:ring-primary/50',
+                                    (!hasCopyableText || !currentProjectRef) && 'opacity-50'
+                                )}
+                                onPointerDown={(event) => event.stopPropagation()}
+                                onClick={handleSaveAsPlanClick}
+                            >
+                                <RiBookletLine className="h-4 w-4" />
+                            </Button>
+                        </TooltipTrigger>
+                        <TooltipContent sideOffset={6}>Save as plan</TooltipContent>
+                    </Tooltip>
+                ) : null}
                 <Tooltip delayDuration={1000}>
                     <TooltipTrigger asChild>
                         <Button
diff --git a/packages/ui/src/components/chat/message/TextSelectionMenu.tsx b/packages/ui/src/components/chat/message/TextSelectionMenu.tsx
index 3e3c9a4d..ce2df569 100644
--- a/packages/ui/src/components/chat/message/TextSelectionMenu.tsx
+++ b/packages/ui/src/components/chat/message/TextSelectionMenu.tsx
@@ -11,6 +11,7 @@ import { copyTextToClipboard } from '@/lib/clipboard';
 import { toast } from '@/components/ui';
 import { getProjectNotesAndTodos, saveProjectNotesAndTodos } from '@/lib/openchamberConfig';
 import { summarizeText } from '@/lib/voice/summarize';
+import { isVSCodeRuntime } from '@/lib/desktop';
 
 interface TextSelectionMenuProps {
   containerRef: React.RefObject<HTMLElement | null>;
@@ -628,21 +629,23 @@ export const TextSelectionMenu: React.FC<TextSelectionMenuProps> = ({ containerR
           <span>Copy</span>
         </button>
 
-        <button
-          onClick={handleAddToNotes}
-          disabled={isAddingToNotes}
-          className={cn(
-            'flex items-center gap-2 px-3 py-2 rounded-lg',
-            'text-sm font-medium',
-            'bg-[var(--surface-muted)] text-[var(--surface-foreground)]',
-            'active:opacity-80 disabled:opacity-60 disabled:cursor-not-allowed',
-            'transition-opacity duration-150'
-          )}
-          type="button"
-        >
-          {isAddingToNotes ? <RiLoader4Line className="h-5 w-5 animate-spin" /> : <RiBookletLine className="h-5 w-5" />}
-          <span>Add to notes</span>
-        </button>
+        {!isVSCodeRuntime() ? (
+          <button
+            onClick={handleAddToNotes}
+            disabled={isAddingToNotes}
+            className={cn(
+              'flex items-center gap-2 px-3 py-2 rounded-lg',
+              'text-sm font-medium',
+              'bg-[var(--surface-muted)] text-[var(--surface-foreground)]',
+              'active:opacity-80 disabled:opacity-60 disabled:cursor-not-allowed',
+              'transition-opacity duration-150'
+            )}
+            type="button"
+          >
+            {isAddingToNotes ? <RiLoader4Line className="h-5 w-5 animate-spin" /> : <RiBookletLine className="h-5 w-5" />}
+            <span>Add to notes</span>
+          </button>
+        ) : null}
       </div>,
       document.body
     );
@@ -703,24 +706,28 @@ export const TextSelectionMenu: React.FC<TextSelectionMenuProps> = ({ containerR
           <span className="whitespace-nowrap">New session</span>
         </button>
 
-        <div className="w-px h-4 bg-[var(--interactive-border)]" />
+        {!isVSCodeRuntime() ? (
+          <>
+            <div className="w-px h-4 bg-[var(--interactive-border)]" />
 
-        <button
-          onClick={handleAddToNotes}
-          disabled={isAddingToNotes}
-          className={cn(
-            'flex items-center gap-1.5 px-2 py-1 rounded-md',
-            'text-sm font-medium',
-            'text-[var(--surface-foreground)]',
-            'hover:bg-[var(--interactive-hover)] disabled:opacity-60 disabled:cursor-not-allowed',
-            'transition-colors duration-150'
-          )}
-          title="Save distilled insight to notes"
-          type="button"
-        >
-          {isAddingToNotes ? <RiLoader4Line className="h-4 w-4 animate-spin" /> : <RiBookletLine className="h-4 w-4" />}
-          <span className="whitespace-nowrap">Add to notes</span>
-        </button>
+            <button
+              onClick={handleAddToNotes}
+              disabled={isAddingToNotes}
+              className={cn(
+                'flex items-center gap-1.5 px-2 py-1 rounded-md',
+                'text-sm font-medium',
+                'text-[var(--surface-foreground)]',
+                'hover:bg-[var(--interactive-hover)] disabled:opacity-60 disabled:cursor-not-allowed',
+                'transition-colors duration-150'
+              )}
+              title="Save distilled insight to notes"
+              type="button"
+            >
+              {isAddingToNotes ? <RiLoader4Line className="h-4 w-4 animate-spin" /> : <RiBookletLine className="h-4 w-4" />}
+              <span className="whitespace-nowrap">Add to notes</span>
+            </button>
+          </>
+        ) : null}
       </div>
     </div>,
     document.body