import * as XLSX from "xlsx";
// 导入excel文件显示到前端页面
const getTableFileData = async (file, tableColumns) => {
    const data = await readFile(file.raw);
    // 使用 xlsx 解析
    let workbook = XLSX.read(data, {
        type: 'array',
        cellDates: true,
        cellStyles: true
    });
    const sheetName = workbook.SheetNames[0];
    const sheet = workbook.Sheets[sheetName];
    const rawData = XLSX.utils.sheet_to_json(sheet, {
        header: 1,
        defval: ""
    });
    let list = [];
    const [head, ...fileData] = rawData;
    fileData && fileData.map(item => {
        let obj = {};
        tableColumns.map(column => {
            let index = head.indexOf(column.label);
            let def = column.type === "number" ? null : "";
            obj[column.prop] = index != -1 ? item[index] || def : def;
        })
        list.push(obj)
    })
    return list;
};
const readFile = (file) => {
    return new Promise((resolve, reject) => {
        const reader = new FileReader();
        reader.onload = (e) => resolve(e.target.result);
        reader.onerror = reject;
        reader.readAsArrayBuffer(file);
    });
};
// 导出页面数据，生成excel文件
const exportTableFile = (tableData, tableColumns, title, fileName) => {
    const wb = XLSX.utils.book_new()
        
    // 2. 准备数据
    let wsData = title ? [[title]] : []
    
    // 添加表头
    if (tableColumns && tableColumns.length > 0) {
        const headerRow = tableColumns.map(h => h.label || h)
        wsData.push(headerRow)
    }
    
    // 添加数据行
    tableData.forEach(item => {
        if (tableColumns) {
            // 按表头配置提取数据
            const row = tableColumns.map(header => {
                const value = getNestedValue(item, header.prop || header)
                return formatCellValue(value, header.formatter)
            })
            wsData.push(row)
        } else {
            // 如果没有表头配置，直接使用对象值
            const row = Object.values(item)
            wsData.push(row)
        }
    })
    
    // 3. 创建工作表
    const ws = XLSX.utils.aoa_to_sheet(wsData)
    
    // 4. 设置列宽（可选）
    if (tableColumns) {
        const colWidths = tableColumns.map(header => ({
            wch: header.width || Math.max(
                (header.label || header).length,
                ...tableData.map(item => {
                    const value = getNestedValue(item, header.prop || header)
                    return String(value || '').length
                })
            ) * 2
        }))
        ws['!cols'] = colWidths
    }
    // 合并标题单元格
    if (title) {
        ws['!merges'] = [
            XLSX.utils.decode_range('A1:Q1') // 合并第一行的 A-E 列
        ]
    }
    // 5. 将工作表添加到工作簿
    XLSX.utils.book_append_sheet(wb, ws, 'Sheet1')
    
    // 6. 生成 Excel 文件并下载
    XLSX.writeFile(wb, fileName + ".xlsx");
};
// 辅助函数：获取嵌套对象的值
const getNestedValue = (obj, path) => {
    return path.split('.').reduce((prev, curr) => {
        return prev ? prev[curr] : null
    }, obj)
}

// 辅助函数：格式化单元格值
const formatCellValue = (value, formatter) => {
    if (formatter && typeof formatter === 'function') {
        return formatter(value)
    }
    return value
}
export{
    getTableFileData,
    exportTableFile
}