From 2c0749ef1d0dd67f903b22c1b9ad89745afe5d9e Mon Sep 17 00:00:00 2001
From: Bohdan Triapitsyn <artmore@protonmail.com>
Date: Wed, 10 Jun 2026 15:28:24 +0300
Subject: [PATCH] fix: align archive all action in VS Code header

Moves the archive all sessions action to the right side of the header
Keeps the sessions title focused on text only
---
 packages/ui/src/components/layout/VSCodeLayout.tsx     |  4 +++-
 .../ui/src/components/session/SessionsTabTitle.tsx     | 10 ++--------
 2 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/packages/ui/src/components/layout/VSCodeLayout.tsx b/packages/ui/src/components/layout/VSCodeLayout.tsx
index 2f5993be..2c450b28 100644
--- a/packages/ui/src/components/layout/VSCodeLayout.tsx
+++ b/packages/ui/src/components/layout/VSCodeLayout.tsx
@@ -10,6 +10,7 @@ import { useConfigStore } from '@/stores/useConfigStore';
 import { resolveGlobalSessionDirectory, useGlobalSessionsStore } from '@/stores/useGlobalSessionsStore';
 import { ContextUsageDisplay } from '@/components/ui/ContextUsageDisplay';
 import { McpDropdown } from '@/components/mcp/McpDropdown';
+import { ArchiveAllDropdown } from '@/components/session/ArchiveAllDropdown';
 import { SessionSwitcherDropdown } from '@/components/session/SessionSwitcherDropdown';
 import { SessionsTabTitle } from '@/components/session/SessionsTabTitle';
 import { useProjectsStore } from '@/stores/useProjectsStore';
@@ -813,9 +814,10 @@ const VSCodeHeader: React.FC<VSCodeHeaderProps> = ({ title, showBack, onBack, on
           </button>
         </SessionSwitcherDropdown>
       ) : (
-        <SessionsTabTitle title={title} onArchiveAll={onArchiveAll} />
+        <SessionsTabTitle title={title} />
       )}
       <div className="min-w-0 flex-1" />
+      {onArchiveAll && <ArchiveAllDropdown onArchiveAll={onArchiveAll} />}
       {onNewSession && (
         <button
           onClick={onNewSession}
diff --git a/packages/ui/src/components/session/SessionsTabTitle.tsx b/packages/ui/src/components/session/SessionsTabTitle.tsx
index d85012f3..efacdf35 100644
--- a/packages/ui/src/components/session/SessionsTabTitle.tsx
+++ b/packages/ui/src/components/session/SessionsTabTitle.tsx
@@ -1,16 +1,10 @@
 import React from 'react';
-import { ArchiveAllDropdown } from '@/components/session/ArchiveAllDropdown';
-
 interface SessionsTabTitleProps {
   title: string;
-  onArchiveAll?: () => void;
 }
 
-const SessionsTabTitle: React.FC<SessionsTabTitleProps> = ({ title, onArchiveAll }) => (
-  <>
-    <h1 className="text-sm font-medium truncate flex-1" title={title}>{title}</h1>
-    {onArchiveAll && <ArchiveAllDropdown onArchiveAll={onArchiveAll} />}
-  </>
+const SessionsTabTitle: React.FC<SessionsTabTitleProps> = ({ title }) => (
+  <h1 className="text-sm font-medium truncate" title={title}>{title}</h1>
 );
 
 export { SessionsTabTitle };