feat: usage

2026-04-22 13:16:48 +08:00
parent b057d52499
commit 7d75714662
37 changed files with 2199 additions and 288 deletions
@@ -0,0 +1,76 @@
+import {
+  apiKeysTableDataAtom,
+  modelsTableDataAtom,
+  usersTableDataAtom
+} from '@/atoms/usage';
+import { exportJsonToExcel } from '@/utils/excel-reader';
+import { useStore } from 'jotai';
+import useAPIKeysColumns from './use-apikeys-columns';
+import useModelsColumns from './use-models-columns';
+import useUsersColumns from './use-users-columns';
+
+const useExportTable = () => {
+  const store = useStore();
+  const modelsColumns = useModelsColumns();
+  const apiKeysColumns = useAPIKeysColumns();
+  const usersColumns = useUsersColumns();
+
+  const exportTable = () => {
+    const usersTableData = store.get(usersTableDataAtom);
+    const apiKeysTableData = store.get(apiKeysTableDataAtom);
+    const modelsTableData = store.get(modelsTableDataAtom);
+    exportJsonToExcel({
+      fileName: 'table_data.xlsx',
+      sheets: [
+        {
+          jsonData: modelsTableData.dataList || [],
+          sheetName: 'models',
+          fields: modelsColumns
+            .map((col: any) => col.dataIndex)
+            .filter(Boolean) as string[],
+          fieldLabels: modelsColumns.reduce(
+            (map, col) => {
+              map[col.dataIndex] = col.title;
+              return map;
+            },
+            {} as Record<string, any>
+          ),
+          formatMap: {}
+        },
+        {
+          jsonData: apiKeysTableData.dataList || [],
+          sheetName: 'api_keys',
+          fields: apiKeysColumns
+            .map((col: any) => col.dataIndex)
+            .filter(Boolean) as string[],
+          fieldLabels: apiKeysColumns.reduce(
+            (map, col) => {
+              map[col.dataIndex] = col.title;
+              return map;
+            },
+            {} as Record<string, any>
+          ),
+          formatMap: {}
+        },
+        {
+          jsonData: usersTableData.dataList || [],
+          sheetName: 'users',
+          fields: usersColumns
+            .map((col: any) => col.dataIndex)
+            .filter(Boolean) as string[],
+          fieldLabels: usersColumns.reduce(
+            (map, col) => {
+              map[col.dataIndex] = col.title;
+              return map;
+            },
+            {} as Record<string, any>
+          ),
+          formatMap: {}
+        }
+      ]
+    });
+  };
+  return { exportTable };
+};
+
+export default useExportTable;