From ac307b72ad25d51dfb84757282cf50c8c593f1df Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=81=95=E7=91=9E=E7=8E=9B=E7=9A=84=E7=9A=87=E5=B8=9D?=
 <2075125282@qq.com>
Date: Fri, 2 May 2025 19:08:37 +0800
Subject: [PATCH] feat(components): refactoring button configuration,
 supporting the merging and sorting of custom and preset buttons.

---
 .../advanced/table-header-operation.vue       | 249 +++++++++++++++---
 1 file changed, 211 insertions(+), 38 deletions(-)

diff --git a/src/components/advanced/table-header-operation.vue b/src/components/advanced/table-header-operation.vue
index 7ca287fd..c6b5f9a8 100644
--- a/src/components/advanced/table-header-operation.vue
+++ b/src/components/advanced/table-header-operation.vue
@@ -1,40 +1,193 @@
 <script setup lang="ts">
+import { computed } from 'vue';
 import { $t } from '@/locales';
 
 defineOptions({
   name: 'TableHeaderOperation'
 });
 
-interface Props {
-  itemAlign?: NaiveUI.Align;
-  disabledDelete?: boolean;
-  loading?: boolean;
+/** button item config type */
+export interface TableButtonItem {
+  /** button unique identifier */
+  key: string;
+  /** button text, if not passed, use default internationalized text */
+  text?: string;
+  /** button icon */
+  icon: string;
+  /** button type */
+  type?: 'default' | 'primary' | 'info' | 'success' | 'warning' | 'error';
+  /** whether to be ghost button */
+  ghost?: boolean;
+  /** whether to disable button */
+  disabled?: boolean;
+  /** button order, the smaller the value, the closer to the front */
+  order?: number;
+  /** whether to need confirm */
+  needConfirm?: boolean;
+  /** confirm text */
+  confirmText?: string;
+  /** button extra props */
+  buttonProps?: Record<string, any>;
 }
 
-defineProps<Props>();
+/** preset button type */
+export type PresetButtonKey = 'add' | 'delete' | 'refresh';
 
+/** button type constant */
+const ButtonType = {
+  DEFAULT: 'default',
+  PRIMARY: 'primary',
+  INFO: 'info',
+  SUCCESS: 'success',
+  WARNING: 'warning',
+  ERROR: 'error'
+} as const;
+
+/** component props */
+export interface TableHeaderOperationProps {
+  /** internal element alignment */
+  itemAlign?: 'start' | 'end' | 'center' | 'baseline' | 'stretch';
+  /** loading status */
+  loading?: boolean;
+  /** button config list */
+  buttons?: TableButtonItem[];
+  /** whether to show preset buttons */
+  showPresetButtons?: boolean;
+  /** disabled preset buttons */
+  disabledPresetButtons?: PresetButtonKey[];
+  /** hidden preset buttons */
+  hiddenPresetButtons?: PresetButtonKey[];
+}
+
+const props = withDefaults(defineProps<TableHeaderOperationProps>(), {
+  itemAlign: 'center',
+  loading: false,
+  showPresetButtons: true,
+  disabledPresetButtons: () => [],
+  hiddenPresetButtons: () => [],
+  buttons: () => []
+});
+
+/** component events */
 interface Emits {
+  /** add button click event */
   (e: 'add'): void;
+  /** delete button click event */
   (e: 'delete'): void;
+  /** refresh button click event */
   (e: 'refresh'): void;
+  /** custom button click event */
+  (e: 'click', key: string): void;
 }
 
 const emit = defineEmits<Emits>();
 
-const columns = defineModel<NaiveUI.TableColumnCheck[]>('columns', {
+const columns = defineModel<any[]>('columns', {
   default: () => []
 });
 
-function add() {
-  emit('add');
+/** get preset button config */
+const getPresetButtons = (): TableButtonItem[] => {
+  return [
+    {
+      key: 'add',
+      text: $t('common.add'),
+      icon: 'ic-round-plus',
+      type: ButtonType.PRIMARY,
+      ghost: true,
+      order: 10,
+      disabled: props.disabledPresetButtons.includes('add')
+    },
+    {
+      key: 'delete',
+      text: $t('common.batchDelete'),
+      icon: 'ic-round-delete',
+      type: ButtonType.ERROR,
+      ghost: true,
+      order: 20,
+      disabled: props.disabledPresetButtons.includes('delete'),
+      needConfirm: true,
+      confirmText: $t('common.confirmDelete')
+    },
+    {
+      key: 'refresh',
+      text: $t('common.refresh'),
+      icon: 'mdi-refresh',
+      type: ButtonType.DEFAULT,
+      order: 30,
+      disabled: props.disabledPresetButtons.includes('refresh')
+    }
+  ].filter(btn => !props.hiddenPresetButtons.includes(btn.key as PresetButtonKey));
+};
+
+/** merge button config */
+const mergedButtons = computed(() => {
+  const result: TableButtonItem[] = [];
+
+  // add preset buttons
+  if (props.showPresetButtons) {
+    result.push(...getPresetButtons());
+  }
+
+  // add custom buttons
+  if (props.buttons?.length) {
+    // merge custom buttons with preset buttons
+    props.buttons.forEach(customBtn => {
+      const existingIndex = result.findIndex(btn => btn.key === customBtn.key);
+
+      if (existingIndex >= 0) {
+        // override preset buttons
+        result[existingIndex] = {
+          ...result[existingIndex],
+          ...customBtn
+        };
+      } else {
+        // add new button
+        result.push({
+          order: 100, // custom button default order value
+          ...customBtn
+        });
+      }
+    });
+  }
+
+  // sort by order
+  return result.sort((a, b) => (a.order || 0) - (b.order || 0));
+});
+
+/** handle button click event */
+function handleButtonClick(key: string) {
+  switch (key) {
+    case 'add':
+      emit('add');
+      break;
+    case 'delete':
+      emit('delete');
+      break;
+    case 'refresh':
+      emit('refresh');
+      break;
+    default:
+      emit('click', key);
+      break;
+  }
 }
 
-function batchDelete() {
-  emit('delete');
-}
+/** get button text */
+function getButtonText(button: TableButtonItem): string {
+  if (button.text) return button.text;
 
-function refresh() {
-  emit('refresh');
+  // use default internationalized text
+  switch (button.key) {
+    case 'add':
+      return $t('common.add');
+    case 'delete':
+      return $t('common.batchDelete');
+    case 'refresh':
+      return $t('common.refresh');
+    default:
+      return button.key;
+  }
 }
 </script>
 
@@ -42,33 +195,53 @@ function refresh() {
   <NSpace :align="itemAlign" wrap justify="end" class="lt-sm:w-200px">
     <slot name="prefix"></slot>
     <slot name="default">
-      <NButton size="small" ghost type="primary" @click="add">
-        <template #icon>
-          <icon-ic-round-plus class="text-icon" />
-        </template>
-        {{ $t('common.add') }}
-      </NButton>
-      <NPopconfirm @positive-click="batchDelete">
-        <template #trigger>
-          <NButton size="small" ghost type="error" :disabled="disabledDelete">
-            <template #icon>
-              <icon-ic-round-delete class="text-icon" />
-            </template>
-            {{ $t('common.batchDelete') }}
-          </NButton>
-        </template>
-        {{ $t('common.confirmDelete') }}
-      </NPopconfirm>
-    </slot>
-    <NButton size="small" @click="refresh">
-      <template #icon>
-        <icon-mdi-refresh class="text-icon" :class="{ 'animate-spin': loading }" />
+      <template v-for="button in mergedButtons" :key="button.key">
+        <!-- button that needs confirmation -->
+        <NPopconfirm v-if="button.needConfirm" @positive-click="handleButtonClick(button.key)">
+          <template #trigger>
+            <NButton
+              size="small"
+              :ghost="button.ghost"
+              :type="button.type"
+              :disabled="button.disabled"
+              v-bind="button.buttonProps"
+            >
+              <template #icon>
+                <SvgIcon
+                  :icon="button.icon"
+                  class="text-icon"
+                  :class="{ 'animate-spin': loading && button.key === 'refresh' }"
+                />
+              </template>
+              {{ getButtonText(button) }}
+            </NButton>
+          </template>
+          {{ button.confirmText }}
+        </NPopconfirm>
+
+        <!-- normal button -->
+        <NButton
+          v-else
+          size="small"
+          :ghost="button.ghost"
+          :type="button.type"
+          :disabled="button.disabled"
+          v-bind="button.buttonProps"
+          @click="handleButtonClick(button.key)"
+        >
+          <template #icon>
+            <SvgIcon
+              :icon="button.icon"
+              class="text-icon"
+              :class="{ 'animate-spin': loading && button.key === 'refresh' }"
+            />
+          </template>
+          {{ getButtonText(button) }}
+        </NButton>
       </template>
-      {{ $t('common.refresh') }}
-    </NButton>
+    </slot>
+
     <TableColumnSetting v-model:columns="columns" />
     <slot name="suffix"></slot>
   </NSpace>
 </template>
-
-<style scoped></style>