89 lines
2.3 KiB
TypeScript
89 lines
2.3 KiB
TypeScript
import { JSON2SheetOpts, read, utils, write } from 'xlsx';
|
|
|
|
// 静态资源路径
|
|
const baseUrl = import.meta.env.VITE_HISTORY_BASE_URL;
|
|
export const xlsxUrl = `${
|
|
baseUrl.length === 1 && baseUrl.indexOf('/') === 0
|
|
? ''
|
|
: baseUrl.indexOf('/') === -1
|
|
? '/' + baseUrl
|
|
: baseUrl
|
|
}/alarmHelp`;
|
|
|
|
/**
|
|
* 读取本地文件
|
|
* @param id 表格ID
|
|
* @returns 数据数组
|
|
* @example
|
|
* readLoalXlsx('20001').then(res=>{
|
|
* console.log(res)
|
|
* });
|
|
*/
|
|
export async function readLoalXlsx(
|
|
lang: string,
|
|
id: string
|
|
): Promise<Record<string, any>[]> {
|
|
let result = await fetch(`${xlsxUrl}/${lang}/${id}.xlsx`);
|
|
let fileBuffer = await result.arrayBuffer();
|
|
// 判断是否xlsx文件
|
|
const data = new Uint8Array(fileBuffer);
|
|
const isXlsxFile =
|
|
data[0] === 0x50 &&
|
|
data[1] === 0x4b &&
|
|
data[2] === 0x03 &&
|
|
data[3] === 0x04;
|
|
if (!isXlsxFile) {
|
|
result = await fetch(`${xlsxUrl}/${lang}/all.xlsx`);
|
|
fileBuffer = await result.arrayBuffer();
|
|
}
|
|
return readSheet(fileBuffer, 0);
|
|
}
|
|
|
|
/**
|
|
* 读取表格数据 工作表
|
|
* @param fileBolb 文件对象
|
|
* @param index 文件保存路径
|
|
* @return 表格对象列表
|
|
*/
|
|
export async function readSheet(
|
|
fileBolb: Blob | ArrayBuffer,
|
|
index: number = 0
|
|
): Promise<Record<string, string>[]> {
|
|
const workBook = read(fileBolb);
|
|
let workSheet = workBook.Sheets[workBook.SheetNames[index]];
|
|
return utils.sheet_to_json<Record<string, string>>(workSheet);
|
|
}
|
|
|
|
/**
|
|
* 写入表格数据，一般用于导出
|
|
* @param filePath 文件路径
|
|
* @param sheetName 工作表名称
|
|
* @return xlsx文件流， 使用saveAs函数保存
|
|
* @example
|
|
* writeSheet(res.data, from.logType).then(fileBlob =>
|
|
* saveAs(fileBlob, `${from.logType}_${Date.now()}.xlsx`)
|
|
* );
|
|
*
|
|
*/
|
|
export async function writeSheet(
|
|
data: any[],
|
|
sheetName: string,
|
|
opts?: JSON2SheetOpts
|
|
) {
|
|
if (data.length === 0) {
|
|
return new Blob([], { type: 'application/octet-stream' });
|
|
}
|
|
|
|
const workSheet = utils.json_to_sheet(data, opts);
|
|
console.log(workSheet);
|
|
|
|
// 设置列宽度，单位厘米
|
|
workSheet['!cols'] = Object.keys(data[0]).map(() => {
|
|
return { wch: 20 };
|
|
});
|
|
const workBook = utils.book_new();
|
|
utils.book_append_sheet(workBook, workSheet, sheetName);
|
|
const excelBuffer = write(workBook, { type: 'array', bookType: 'xlsx' });
|
|
return new Blob([excelBuffer], { type: 'application/octet-stream' });
|
|
}
|