gfw-api/gfw-ogr-api

const logger = require('logger');
const ogr2ogr = require('ogr2ogr').default;
const XLSX = require('xlsx');
const fs = require('fs');
const path = require('path');
const unzipper = require('unzipper');

class OGRConverter {

    static async convert(ctx) {
        logger.info('Converting file...', ctx.request.body);
        logger.debug(`[OGRRouterV2 - convertV2] request: ${JSON.stringify(ctx.request)}`);
        ctx.assert(ctx.request.files && ctx.request.files.file, 400, 'File required');
        logger.debug(`[OGRRouterV2 - convertV2] file data: ${JSON.stringify(ctx.request.files.file)}`);

        let result;
        logger.info(`[OGRRouterV2 - convertV2] file type: ${ctx.request.files.file.type}`);
        logger.debug(`[OGRRouterV2 - convertV2] file size: ${ctx.request.files.file.size}`);

        if (path.extname(ctx.request.files.file.filepath) === '.xlsx') {
            logger.info('[OGRRouterV2 - convertV2] It is an excel file');
            const xslxFile = XLSX.readFile(ctx.request.files.file.filepath);
            const csvPAth = path.parse(ctx.request.files.file.filepath);
            csvPAth.ext = '.csv';
            csvPAth.base = csvPAth.name + csvPAth.ext;
            XLSX.writeFile(xslxFile, path.format(csvPAth), { type: 'file', bookType: 'csv' });
            ctx.request.files.file.filepath = path.format(csvPAth);
            const options = [
                '-s_srs', 'EPSG:4326', '-t_srs', 'EPSG:4326',
                '-oo', 'GEOM_POSSIBLE_NAMES=*geom*',
                '-oo', 'CSVLINE=YES',
                '-oo', 'X_POSSIBLE_NAMES=lon*,Lon*',
                '-oo', 'Y_POSSIBLE_NAMES=lat*,Lon*',
                '-oo', 'KEEP_GEOM_COLUMNS=YES'
            ];
            result = await ogr2ogr(ctx.request.files.file.filepath, { options, timeout: 60000 });
        } else {
            logger.info('[OGRRouterV2 - convertV2] Not an excel file');

            let options;
            let inputPath = ctx.request.files.file.filepath;

            if (ctx.request.files.file.mimetype === 'text/csv' || ctx.request.files.file.mimetype === 'application/vnd.ms-excel') {
                logger.debug('csv transforming ...');
                // @TODO
                options = [
                    '-s_srs', 'EPSG:4326', '-t_srs', 'EPSG:4326',
                    '-oo', 'GEOM_POSSIBLE_NAMES=*geom*',
                    '-oo', 'CSVLINE=YES',
                    '-oo', 'X_POSSIBLE_NAMES=lon*,Lon*',
                    '-oo', 'Y_POSSIBLE_NAMES=lat*,Lon*',
                    '-oo', 'KEEP_GEOM_COLUMNS=YES'
                ];
            } else {
                options = ['-t_srs', 'EPSG:4326', '-dim', '2'];
                if (ctx.request.files.file.mimetype === 'application/zip') {
                    const readStream = fs.createReadStream(inputPath);
                    await new Promise((resolve) => {
                        readStream.pipe(unzipper.Parse())
                            .on('entry', (entry) => {
                                if (entry.type === 'Directory') {
                                    inputPath = `/vsizip/${inputPath}/${entry.path}`;
                                }
                                // Continue processing other entries
                                entry.autodrain();
                            })
                            .on('finish', () => {
                                logger.info('Processing zip file finished');
                                resolve();
                            });
                    });
                }
            }
            result = await ogr2ogr(inputPath, { options, timeout: 60000 });
        }

        return result;

    }

}

module.exports = OGRConverter;