import * as XLSX from "xlsx"
import { ElMessage, ElMessageBox } from 'element-plus'

export const tableRemoveRow = (params: any, callback: (res: boolean) => void) => {
    if (!params.data) params.data = []
    if (Array.isArray(params.data)) {
        if (params.data.length < 1) {
            ElMessageBox.alert('请至少选择一条数据进行删除', '系统提醒', { type: 'warning' })
            return callback(false)
        }
        ElMessageBox.confirm(
            params.batchRemoveTip || '是否确认删除选中的' + params.data.length + '条数据？此操作将永久删除',
            '系统提醒',
            {
                confirmButtonText: '确定',
                cancelButtonText: '取消',
                type: 'warning',
            }
        ).then(() => {
            ElMessage.success('删除成功！')
            callback(true)
        }).catch(() => {
            callback(false)
        })
    } else {
        ElMessageBox.confirm(
            params.removeTip || '此操作将永久删除',
            '系统提醒',
            {
                confirmButtonText: '确定',
                cancelButtonText: '取消',
                type: 'warning',
            }
        ).then(() => {
            ElMessage.success('删除成功！')
            callback(true)
        }).catch(() => {
            callback(false)
        })
    }
}

export const exportData = (fileName: string, data: Array<any>) => {
    const wb = XLSX.utils.book_new();
    XLSX.utils.book_append_sheet(
        wb,
        XLSX.utils.json_to_sheet(data)
    );
    XLSX.writeFile(wb, fileName + '.xls'); // 导出Excel
}