From 30354fd7620d35b67250c0590c07246346bdd063 Mon Sep 17 00:00:00 2001
From: viettranx <viettranx@gmail.com>
Date: Sun, 12 Apr 2026 18:09:42 +0700
Subject: [PATCH] refactor(vault): merge rescan and stop enrichment buttons
 into single toggle

---
 ui/web/src/pages/vault/vault-page.tsx | 29 +++++++++++++--------------
 1 file changed, 14 insertions(+), 15 deletions(-)

diff --git a/ui/web/src/pages/vault/vault-page.tsx b/ui/web/src/pages/vault/vault-page.tsx
index 262a575b..f810cd58 100644
--- a/ui/web/src/pages/vault/vault-page.tsx
+++ b/ui/web/src/pages/vault/vault-page.tsx
@@ -152,25 +152,24 @@ export function VaultPage() {
           <TooltipProvider>
             <Tooltip>
               <TooltipTrigger asChild>
-                <Button size="sm" variant="outline" onClick={async () => { await rescan(); loadRoot(); }} disabled={rescanPending || enriching}>
-                  {(rescanPending || enriching) ? <Loader2 className="h-3.5 w-3.5 animate-spin" /> : <FolderSync className="h-3.5 w-3.5" />}
+                <Button
+                  size="sm"
+                  variant="outline"
+                  onClick={enriching ? stopEnrich : async () => { await rescan(); loadRoot(); }}
+                  disabled={rescanPending || stopPending}
+                >
+                  {rescanPending || stopPending ? (
+                    <Loader2 className="h-3.5 w-3.5 animate-spin" />
+                  ) : enriching ? (
+                    <StopCircle className="h-3.5 w-3.5 text-destructive" />
+                  ) : (
+                    <FolderSync className="h-3.5 w-3.5" />
+                  )}
                 </Button>
               </TooltipTrigger>
-              <TooltipContent>{enriching ? t("enriching", "Enriching documents...") : t("rescanTooltip", "Rescan workspace")}</TooltipContent>
+              <TooltipContent>{enriching ? t("stopEnrich", "Stop enrichment") : t("rescanTooltip", "Rescan workspace")}</TooltipContent>
             </Tooltip>
           </TooltipProvider>
-          {enriching && (
-            <TooltipProvider>
-              <Tooltip>
-                <TooltipTrigger asChild>
-                  <Button size="sm" variant="outline" onClick={stopEnrich} disabled={stopPending}>
-                    {stopPending ? <Loader2 className="h-3.5 w-3.5 animate-spin" /> : <StopCircle className="h-3.5 w-3.5 text-destructive" />}
-                  </Button>
-                </TooltipTrigger>
-                <TooltipContent>{t("stopEnrich", "Stop enrichment")}</TooltipContent>
-              </Tooltip>
-            </TooltipProvider>
-          )}
           <TooltipProvider>
             <Tooltip>
               <TooltipTrigger asChild>