export2Excel.ts
3.4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/* eslint-disable */
import * as XLSX from "xlsx";
export function toExcel(content: string) {
let sheetName = "result"; // 默认表名
let tableContent = "";
// 查找标题内容(以 ** 或 ## 或 ### 开头)
const titleMatch = content.match(/^(#{2,3}|\*\*)\s*(.*)/m);
if (titleMatch) {
sheetName = titleMatch[2].trim().replace(/\s+/g, "_"); // 使用标题作为表名
// 提取表格内容(跳过标题部分)
tableContent = content.substring(titleMatch[0].length).trim();
} else {
tableContent = content;
}
// 查找表格的起始位置(第一行以 | 开头且以 | 结尾)
let tableStartIndex = -1;
const lines = tableContent.split("\n");
for (let i = 0; i < lines.length; i++) {
if (lines[i].trim().startsWith("|") && lines[i].trim().endsWith("|")) {
tableStartIndex = i;
break;
}
}
if (tableStartIndex === -1) {
console.error("表格内容未找到");
return;
}
// 查找表格的结束位置(遇到不以 | 开头或者不以 | 结尾的行)
let tableEndIndex = -1;
for (let i = tableStartIndex; i < lines.length; i++) {
if (!lines[i].trim().startsWith("|") || !lines[i].trim().endsWith("|")) {
tableEndIndex = i;
break;
}
}
if (tableEndIndex === -1) {
tableEndIndex = lines.length;
}
// 提取表格内容
const tableData = lines
.slice(tableStartIndex, tableEndIndex)
.join("\n")
.trim();
// 解析表格内容
const rows = tableData.split("\n");
const data: any[] = [];
let headers: string[] = [];
rows.forEach((row, rowIndex) => {
// 去掉行首尾的 | 符号,并按 | 分割
const cells = row
.replace(/^\s*\|/g, "")
.replace(/\|\s*$/g, "")
.split(/\s*\|\s*/);
// 跳过分隔线(假设分隔线的每个单元格都是由短横线组成)
if (
rowIndex > 0 &&
cells.every(
(cell) => cell.trim().replace(/-/g, "").length === 0, // 检查单元格内容是否只包含短横线
)
) {
return;
}
if (rowIndex === 0) {
// 第一行是表头
headers = cells.map((cell) => cell.trim());
} else {
// 数据行
const rowData: any = {};
cells.forEach((cell, cellIndex) => {
if (cellIndex < headers.length) {
rowData[headers[cellIndex]] = cell.trim();
}
});
data.push(rowData);
}
});
// 创建工作表
const ws = XLSX.utils.json_to_sheet(data);
const wb = XLSX.utils.book_new();
XLSX.utils.book_append_sheet(wb, ws, sheetName);
// 生成文件并下载
XLSX.writeFile(wb, `${sheetName}.xlsx`);
}
export function getExcelData(file: File): Promise<any[][]> {
return new Promise((resolve, reject) => {
const reader = new FileReader();
reader.onload = (e) => {
try {
const data = e.target?.result;
if (!data) {
reject(new Error("Failed to read file data"));
return;
}
const workbook = XLSX.read(data, { type: "array" });
const firstSheetName = workbook.SheetNames[0];
const worksheet = workbook.Sheets[firstSheetName];
// 使用类型断言将 unknown[] 转换为 any[][]
const jsonData = XLSX.utils.sheet_to_json(worksheet, {
header: 1,
}) as any[][];
resolve(jsonData.slice(0, 100));
} catch (error) {
reject(error);
}
};
reader.onerror = (error) => {
reject(error);
};
reader.readAsArrayBuffer(file);
});
}