From 44ec2921d4583436488e16739612fdf50b7a9b4b Mon Sep 17 00:00:00 2001
From: mlogclub <mlog1@qq.com>
Date: Tue, 2 Jun 2026 16:19:11 +0800
Subject: [PATCH] feat: add context menu support to DocumentList, FAQList, and
 KnowledgeBaseList for improved user interaction

---
 .../knowledge/_components/document-list.tsx   | 21 +++++++++++++++----
 .../knowledge/_components/faq-list.tsx        | 15 ++++++++++---
 .../_components/knowledge-base-list.tsx       | 13 ++++++++++--
 3 files changed, 40 insertions(+), 9 deletions(-)

diff --git a/web/app/dashboard/knowledge/_components/document-list.tsx b/web/app/dashboard/knowledge/_components/document-list.tsx
index 1e4e7c8..20d3374 100644
--- a/web/app/dashboard/knowledge/_components/document-list.tsx
+++ b/web/app/dashboard/knowledge/_components/document-list.tsx
@@ -152,6 +152,7 @@ export function DocumentList({ knowledgeBaseId, onActionStateChange }: DocumentL
   const [moving, setMoving] = useState(false);
   const [bulkMoveOpen, setBulkMoveOpen] = useState(false);
   const [selectedIds, setSelectedIds] = useState<number[]>([]);
+  const [contextMenuDocumentId, setContextMenuDocumentId] = useState<number | null>(null);
   const [selectedDirectoryId, setSelectedDirectoryId] = useState<number | null>(null);
   const [actionLoadingMap, setActionLoadingMap] = useState<Record<number, { rebuildIndex: boolean; delete: boolean }>>({});
   const [dialogOpen, setDialogOpen] = useState(false);
@@ -484,10 +485,16 @@ export function DocumentList({ knowledgeBaseId, onActionStateChange }: DocumentL
             ) : null}
             {documents.results.map((item) => (
               viewMode === "grid" ? (
-                <ContextMenu key={item.id}>
+                <ContextMenu
+                  key={item.id}
+                  onOpenChange={(open) => setContextMenuDocumentId(open ? item.id : null)}
+                >
                   <ContextMenuTrigger className="w-full">
                     <div
-                      className="bg-background p-3 transition-colors hover:bg-accent w-full"
+                      className={cn(
+                        "bg-background p-3 transition-colors hover:bg-accent w-full",
+                        contextMenuDocumentId === item.id && "bg-accent text-accent-foreground",
+                      )}
                     >
                       <div className="flex items-start justify-between gap-2">
                         <div className="flex min-w-0 flex-1 items-start gap-2">
@@ -569,10 +576,16 @@ export function DocumentList({ knowledgeBaseId, onActionStateChange }: DocumentL
                   </ContextMenuContent>
                 </ContextMenu>
               ) : (
-                <ContextMenu key={item.id}>
+                <ContextMenu
+                  key={item.id}
+                  onOpenChange={(open) => setContextMenuDocumentId(open ? item.id : null)}
+                >
                   <ContextMenuTrigger className="w-full">
                     <div
-                      className="flex items-center gap-3 bg-background p-2 transition-colors hover:bg-accent w-full"
+                      className={cn(
+                        "flex items-center gap-3 bg-background p-2 transition-colors hover:bg-accent w-full",
+                        contextMenuDocumentId === item.id && "bg-accent text-accent-foreground",
+                      )}
                     >
                       <Checkbox
                         checked={selectedIds.includes(item.id)}
diff --git a/web/app/dashboard/knowledge/_components/faq-list.tsx b/web/app/dashboard/knowledge/_components/faq-list.tsx
index 73e232c..190892c 100644
--- a/web/app/dashboard/knowledge/_components/faq-list.tsx
+++ b/web/app/dashboard/knowledge/_components/faq-list.tsx
@@ -55,7 +55,7 @@ import {
 } from "@/lib/api/admin";
 import { KnowledgeDocumentIndexStatus } from "@/lib/generated/enums";
 import { useI18n } from "@/i18n/provider";
-import { formatDateTime } from "@/lib/utils";
+import { cn, formatDateTime } from "@/lib/utils";
 import { FAQEditDialog } from "./faq-edit";
 import { FAQImportDialog } from "./faq-import-dialog";
 import { KnowledgeBulkMoveDialog } from "./knowledge-bulk-move-dialog";
@@ -141,6 +141,7 @@ export function FAQList({
   const [moving, setMoving] = useState(false);
   const [bulkMoveOpen, setBulkMoveOpen] = useState(false);
   const [selectedIds, setSelectedIds] = useState<number[]>([]);
+  const [contextMenuFAQId, setContextMenuFAQId] = useState<number | null>(null);
   const [importing, setImporting] = useState(false);
   const [importDialogOpen, setImportDialogOpen] = useState(false);
   const [selectedDirectoryId, setSelectedDirectoryId] = useState<number | null>(null);
@@ -429,9 +430,17 @@ export function FAQList({
                   </label>
                 ) : null}
                 {faqs.results.map((item) => (
-                  <ContextMenu key={item.id}>
+                  <ContextMenu
+                    key={item.id}
+                    onOpenChange={(open) => setContextMenuFAQId(open ? item.id : null)}
+                  >
                     <ContextMenuTrigger className="w-full">
-                      <div className="flex items-center gap-3 bg-background p-2 transition-colors hover:bg-accent w-full">
+                      <div
+                        className={cn(
+                          "flex items-center gap-3 bg-background p-2 transition-colors hover:bg-accent w-full",
+                          contextMenuFAQId === item.id && "bg-accent text-accent-foreground",
+                        )}
+                      >
                         <Checkbox
                           checked={selectedIds.includes(item.id)}
                           onClick={(event) => event.stopPropagation()}
diff --git a/web/app/dashboard/knowledge/_components/knowledge-base-list.tsx b/web/app/dashboard/knowledge/_components/knowledge-base-list.tsx
index 666f1b1..075db61 100644
--- a/web/app/dashboard/knowledge/_components/knowledge-base-list.tsx
+++ b/web/app/dashboard/knowledge/_components/knowledge-base-list.tsx
@@ -82,7 +82,9 @@ function getStatusOptions(t: TFunction) {
 type SortableKnowledgeBaseCardProps = {
   item: KnowledgeBase;
   isSelected: boolean;
+  isContextMenuOpen: boolean;
   disabled: boolean;
+  onContextMenuOpenChange: (open: boolean) => void;
   onSelect: () => void;
   onEdit: () => void;
   onDelete: () => void;
@@ -95,7 +97,9 @@ type SortableKnowledgeBaseCardProps = {
 function SortableKnowledgeBaseCard({
   item,
   isSelected,
+  isContextMenuOpen,
   disabled,
+  onContextMenuOpenChange,
   onSelect,
   onEdit,
   onDelete,
@@ -122,13 +126,13 @@ function SortableKnowledgeBaseCard({
   };
 
   return (
-    <ContextMenu>
+    <ContextMenu onOpenChange={onContextMenuOpenChange}>
       <ContextMenuTrigger
         ref={setNodeRef}
         style={style}
         className={cn(
           "group flex items-center gap-1 rounded mx-2 px-2 py-1.5 text-sm transition-colors hover:bg-accent cursor-pointer",
-          isSelected && "bg-accent text-accent-foreground",
+          (isSelected || isContextMenuOpen) && "bg-accent text-accent-foreground",
           isDragging && "bg-muted/60 shadow-sm opacity-80",
         )}
         onClick={onSelect}
@@ -248,6 +252,7 @@ export function KnowledgeBaseList({
   const [dialogOpen, setDialogOpen] = useState(false);
   const [editingItemId, setEditingItemId] = useState<number | null>(null);
   const [knowledgeBases, setKnowledgeBases] = useState<KnowledgeBase[]>([]);
+  const [contextMenuKnowledgeBaseId, setContextMenuKnowledgeBaseId] = useState<number | null>(null);
   const statusOptions = useMemo(() => getStatusOptions(t), [t]);
 
   const sensors = useSensors(
@@ -487,7 +492,11 @@ export function KnowledgeBaseList({
                     key={item.id}
                     item={item}
                     isSelected={selectedKnowledgeBaseId === item.id}
+                    isContextMenuOpen={contextMenuKnowledgeBaseId === item.id}
                     disabled={loading || sorting}
+                    onContextMenuOpenChange={(open) =>
+                      setContextMenuKnowledgeBaseId(open ? item.id : null)
+                    }
                     onSelect={() => onSelectKnowledgeBase(item)}
                     onEdit={() => openEditDialog(item)}
                     onDelete={() => void handleDelete(item)}