"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.getImportationTemplate = exports.exportEntity = exports.importEntity = exports.clearPorts = exports.registerPorts = void 0;
const assert_1 = require("oak-domain/lib/utils/assert");
const xlsx_1 = require("xlsx");
const Importations = {};
const Exportations = {};
function registerPorts(importations, exportations) {
    for (const i of importations) {
        (0, assert_1.assert)(!Importations.hasOwnProperty(i.id), `Importation中的id【${i.id}】重复了`);
        Object.assign(Importations, {
            [i.id]: i,
        });
    }
    for (const e of exportations) {
        (0, assert_1.assert)(!Exportations.hasOwnProperty(e.id), `Exportation中的id【${e.id}】重复了`);
        Object.assign(Exportations, {
            [e.id]: e,
        });
    }
}
exports.registerPorts = registerPorts;
function clearPorts() {
    for (const i in Importations) {
        delete Importations[i];
    }
    for (const e in Exportations) {
        delete Exportations[e];
    }
}
exports.clearPorts = clearPorts;
function getImportation(id) {
    (0, assert_1.assert)(Importations.hasOwnProperty(id), `id为[${id}]的importation不存在`);
    return Importations[id];
}
function getExportation(id) {
    (0, assert_1.assert)(Exportations.hasOwnProperty(id), `id为[${id}]的exportation不存在`);
    return Exportations[id];
}
async function importEntity(params, context) {
    const entity = params.entity;
    const file = params.file;
    const id = params.id;
    const option = JSON.parse(params.option);
    const importation = getImportation(id);
    if (!importation) {
        throw new Error('尚不支持此数据的导入');
    }
    const { fn } = importation;
    // const arrayBuffer = await file.arrayBuffer();
    const workbook = (0, xlsx_1.readFile)(file.filepath, { type: 'buffer' });
    const { SheetNames, Sheets } = workbook;
    const errorSheets = [];
    for (const sheetName of SheetNames) {
        const sheet = Sheets[sheetName];
        const dataList = xlsx_1.utils.sheet_to_json(sheet);
        const errorMessageList = await fn(dataList, context, option);
        if (errorMessageList.length > 0) {
            errorSheets.push({
                sheetName,
                worksheet: xlsx_1.utils.json_to_sheet(errorMessageList),
            });
        }
    }
    if (errorSheets.length > 0) {
        const errorWorkbook = xlsx_1.utils.book_new();
        for (const sheetData of errorSheets) {
            xlsx_1.utils.book_append_sheet(errorWorkbook, sheetData.worksheet, sheetData.sheetName);
        }
        return await (0, xlsx_1.write)(errorWorkbook, { type: 'buffer' });
    }
    // throw new Error('not implement yet');
}
exports.importEntity = importEntity;
async function exportEntity(params, context) {
    const id = params.id;
    const filter = params.filter;
    const properties = params.properties;
    const exportation = getExportation(id);
    if (!exportation) {
        throw new Error('尚不支持此数据的导出');
    }
    const { projection, headers, fn, entity, makeHeaders } = exportation;
    const dataList = await context.select(entity, {
        filter,
        data: projection,
    }, {});
    const headers2 = makeHeaders ? makeHeaders(dataList) : headers;
    (0, assert_1.assert)(headers2 && headers2.length > 0, '导出未传入表头');
    const fittedDatalist = [];
    for (const data of dataList) {
        fittedDatalist.push(await fn(data, context, properties));
    }
    const exportSheet = xlsx_1.utils.json_to_sheet(fittedDatalist, {
        header: headers2,
    });
    const exportBook = xlsx_1.utils.book_new();
    xlsx_1.utils.book_append_sheet(exportBook, exportSheet);
    return await (0, xlsx_1.write)(exportBook, { type: 'buffer' });
    // throw new Error('export not implement yet');
}
exports.exportEntity = exportEntity;
async function getImportationTemplate(params, context) {
    const id = params.id;
    const importation = getImportation(id);
    const { headers } = importation;
    if (!importation) {
        throw new Error('未找到对应的模板');
    }
    const exportSheet = xlsx_1.utils.json_to_sheet([], { header: headers });
    const widthList = headers.map((ele) => {
        return {
            width: ele.length * 2.2,
        };
    });
    exportSheet['!cols'] = widthList;
    const exportBook = xlsx_1.utils.book_new();
    xlsx_1.utils.book_append_sheet(exportBook, exportSheet);
    return await (0, xlsx_1.write)(exportBook, { type: 'buffer' });
    // throw new Error('not implement yet');
}
exports.getImportationTemplate = getImportationTemplate;