From f02a0b0520bdb3d41a7995b23e49724b78485126 Mon Sep 17 00:00:00 2001
From: R-Hart80 <R-Hart80@users.noreply.github.com>
Date: Mon, 11 May 2026 10:39:19 -0300
Subject: [PATCH] fix: mirror Expand layout in Collapse split-button group

The labeled main button now triggers collapse-one-level (conservative)
and the icon-only add-on triggers collapse-all, matching the Expand
group's convention. Fixes #208.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 components/editor/shared/EntityTreeToolbar.tsx | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/components/editor/shared/EntityTreeToolbar.tsx b/components/editor/shared/EntityTreeToolbar.tsx
index 234dec7c..165b2dc8 100644
--- a/components/editor/shared/EntityTreeToolbar.tsx
+++ b/components/editor/shared/EntityTreeToolbar.tsx
@@ -130,21 +130,21 @@ export function EntityTreeToolbar({
         {showCollapseGroup && (
           <div className="flex rounded-md border border-slate-200 dark:border-slate-600">
             <button
-              onClick={onCollapseAll}
+              onClick={onCollapseOneLevel}
               disabled={collapseDisabled}
               className={cn(btnBase, "rounded-l-md border-r border-slate-200 dark:border-slate-600", collapseDisabled ? btnDisabled : btnEnabled)}
-              aria-label="Collapse all"
+              aria-label="Collapse one level"
             >
-              <ChevronsRight className="h-3.5 w-3.5" />
+              <ChevronRight className="h-3.5 w-3.5" />
               <span className="hidden sm:inline">Collapse</span>
             </button>
             <button
-              onClick={onCollapseOneLevel}
+              onClick={onCollapseAll}
               disabled={collapseDisabled}
               className={cn(btnBase, "rounded-r-md px-1", collapseDisabled ? btnDisabled : btnEnabled)}
-              aria-label="Collapse one level"
+              aria-label="Collapse all"
             >
-              <ChevronRight className="h-3.5 w-3.5" />
+              <ChevronsRight className="h-3.5 w-3.5" />
             </button>
           </div>
         )}