ffc-pay-etl-framework/dist/cjs/loaders/csvloader.js

A framework for creating ETL pipelines in node

// @ts-nocheck
const fs = require("fs");
const { Transform } = require("stream");
const { parse } = require("csv-parse");
/**
 *
 * @param {Object} options
 * @param {String} [options.path]
 * @param {ReadableStream} [options.stream]
 * @param {Number} [options.startingLine]
 * @param {Boolean} [options.relax]
 * @returns StreamReader
 */
function CSVLoader(options) {
    var _a, _b;
    let csvLoader;
    if (options.path) {
        csvLoader = fs.createReadStream(options.path);
    }
    else {
        csvLoader = options.stream;
    }
    let lineCount = 1;
    const fromLine = (_a = options.startingLine) !== null && _a !== void 0 ? _a : 2;
    const relaxQuotes = (_b = options.relax) !== null && _b !== void 0 ? _b : false;
    csvLoader._columns = options.columns;
    csvLoader.pump = (csvLoader) => {
        const parser = parse({ delimiter: ",", from_line: fromLine, relax_quotes: relaxQuotes });
        const transformer = new Transform({
            readableObjectMode: true,
            writableObjectMode: true,
            emitClose: true,
            transform(chunk, _, callback) {
                chunk["_columns"] = options.columns;
                chunk["_linecount"] = lineCount;
                lineCount += 1;
                // remove non-printable characters
                options.columns.forEach((_column, index) => {
                    if (chunk[index]) {
                        chunk[index] = chunk[index].replace(/[\x00-\x1F\x7F-\x9F]/g, '');
                    }
                });
                callback(null, chunk);
            }
        });
        return csvLoader
            .pipe(parser)
            .pipe(transformer);
    };
    return csvLoader;
}
module.exports = {
    CSVLoader
};