From 207e8e63ad144a47f70be3a8f7fbb488a9f2b4ff Mon Sep 17 00:00:00 2001
From: t8y2 <1156263951@qq.com>
Date: Sat, 23 May 2026 21:52:40 +0800
Subject: [PATCH] feat(grid): add copy column name on header right-click
 context menu

---
 apps/desktop/src/components/grid/DataGrid.vue | 21 ++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/apps/desktop/src/components/grid/DataGrid.vue b/apps/desktop/src/components/grid/DataGrid.vue
index 2e2129d8a..566c8ce04 100644
--- a/apps/desktop/src/components/grid/DataGrid.vue
+++ b/apps/desktop/src/components/grid/DataGrid.vue
@@ -297,6 +297,7 @@ function typeColorClass(t: string): string {
   return "text-muted-foreground";
 }
 const contextCell = ref<{ rowId: number; rowIndex: number; col: number } | null>(null);
+const contextHeaderColumn = ref<string | null>(null);
 const detailCell = ref<{ rowIndex: number; col: number } | null>(null);
 const hoveredDetailCell = ref<{ rowIndex: number; col: number } | null>(null);
 const showCellDetail = ref(false);
@@ -2779,7 +2780,18 @@ watch(
 );
 
 // --- Context menu handlers ---
+function onHeaderContext(col: string) {
+  contextCell.value = null;
+  clearCellSelection();
+  clearRowSelection();
+  contextHeaderColumn.value = col;
+}
+async function copyHeaderColumn() {
+  if (!contextHeaderColumn.value) return;
+  await copyText(contextHeaderColumn.value);
+}
 function onCellContext(rowId: number, rowIndex: number, colIdx: number, visibleColIdx: number) {
+  contextHeaderColumn.value = null;
   contextCell.value = { rowId, rowIndex, col: colIdx };
   if (hasRowSelection.value && isRowSelected(rowId)) {
     void prefetchCopyStatements();
@@ -2793,6 +2805,7 @@ function onCellContext(rowId: number, rowIndex: number, colIdx: number, visibleC
 }
 
 function onRowContext(rowId: number, rowIndex: number) {
+  contextHeaderColumn.value = null;
   contextCell.value = { rowId, rowIndex, col: -1 };
   if (!isRowSelected(rowId)) {
     clearCellSelection();
@@ -3669,6 +3682,7 @@ defineExpose({
                           }"
                           :style="{ width: `var(--col-w-${colIdx})` }"
                           :data-grid-column-index="actualColumnIndex(colIdx)"
+                          @contextmenu="onHeaderContext(col)"
                         >
                           <span class="flex min-w-0 items-center gap-1 overflow-hidden">
                             <span
@@ -4674,6 +4688,11 @@ defineExpose({
       </ContextMenuTrigger>
 
       <ContextMenuContent class="w-max max-w-[min(80vw,20rem)]">
+        <template v-if="contextHeaderColumn">
+          <ContextMenuItem @click="copyHeaderColumn">
+            <Copy class="w-3.5 h-3.5 mr-2" /> {{ t("grid.copyColumnName") }}
+          </ContextMenuItem>
+        </template>
         <template v-if="contextColumn">
           <ContextMenuItem @click="applyContextSort('asc')">
             <ArrowUp class="w-3.5 h-3.5 mr-2" /> {{ t("grid.sortAscending") }}
@@ -4708,7 +4727,7 @@ defineExpose({
           </ContextMenuSub>
           <ContextMenuSeparator />
         </template>
-        <ContextMenuSub>
+        <ContextMenuSub v-if="!contextHeaderColumn">
           <ContextMenuSubTrigger> <Copy class="w-3.5 h-3.5 mr-2" /> {{ t("grid.copy") }} </ContextMenuSubTrigger>
           <ContextMenuSubContent class="w-max max-w-[min(80vw,18rem)]">
             <ContextMenuItem v-if="contextColumn" @click="copyCell">{{ t("grid.copyCell") }}</ContextMenuItem>