You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							299 lines
						
					
					
						
							9.3 KiB
						
					
					
				
			
		
		
	
	
							299 lines
						
					
					
						
							9.3 KiB
						
					
					
				"use strict";
 | 
						|
 | 
						|
function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
 | 
						|
 | 
						|
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
 | 
						|
 | 
						|
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
 | 
						|
 | 
						|
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
 | 
						|
 | 
						|
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
 | 
						|
 | 
						|
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
 | 
						|
 | 
						|
var fs = require('fs');
 | 
						|
 | 
						|
var fastCsv = require('fast-csv');
 | 
						|
 | 
						|
var customParseFormat = require('dayjs/plugin/customParseFormat');
 | 
						|
 | 
						|
var utc = require('dayjs/plugin/utc');
 | 
						|
 | 
						|
var dayjs = require('dayjs').extend(customParseFormat).extend(utc);
 | 
						|
 | 
						|
var StreamBuf = require('../utils/stream-buf');
 | 
						|
 | 
						|
var _require = require('../utils/utils'),
 | 
						|
    exists = _require.fs.exists;
 | 
						|
/* eslint-disable quote-props */
 | 
						|
 | 
						|
 | 
						|
var SpecialValues = {
 | 
						|
  true: true,
 | 
						|
  false: false,
 | 
						|
  '#N/A': {
 | 
						|
    error: '#N/A'
 | 
						|
  },
 | 
						|
  '#REF!': {
 | 
						|
    error: '#REF!'
 | 
						|
  },
 | 
						|
  '#NAME?': {
 | 
						|
    error: '#NAME?'
 | 
						|
  },
 | 
						|
  '#DIV/0!': {
 | 
						|
    error: '#DIV/0!'
 | 
						|
  },
 | 
						|
  '#NULL!': {
 | 
						|
    error: '#NULL!'
 | 
						|
  },
 | 
						|
  '#VALUE!': {
 | 
						|
    error: '#VALUE!'
 | 
						|
  },
 | 
						|
  '#NUM!': {
 | 
						|
    error: '#NUM!'
 | 
						|
  }
 | 
						|
};
 | 
						|
/* eslint-ensable quote-props */
 | 
						|
 | 
						|
var CSV = /*#__PURE__*/function () {
 | 
						|
  function CSV(workbook) {
 | 
						|
    _classCallCheck(this, CSV);
 | 
						|
 | 
						|
    this.workbook = workbook;
 | 
						|
    this.worksheet = null;
 | 
						|
  }
 | 
						|
 | 
						|
  _createClass(CSV, [{
 | 
						|
    key: "readFile",
 | 
						|
    value: function () {
 | 
						|
      var _readFile = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(filename, options) {
 | 
						|
        var stream, worksheet;
 | 
						|
        return regeneratorRuntime.wrap(function _callee$(_context) {
 | 
						|
          while (1) {
 | 
						|
            switch (_context.prev = _context.next) {
 | 
						|
              case 0:
 | 
						|
                options = options || {};
 | 
						|
                _context.next = 3;
 | 
						|
                return exists(filename);
 | 
						|
 | 
						|
              case 3:
 | 
						|
                if (_context.sent) {
 | 
						|
                  _context.next = 5;
 | 
						|
                  break;
 | 
						|
                }
 | 
						|
 | 
						|
                throw new Error("File not found: ".concat(filename));
 | 
						|
 | 
						|
              case 5:
 | 
						|
                stream = fs.createReadStream(filename);
 | 
						|
                _context.next = 8;
 | 
						|
                return this.read(stream, options);
 | 
						|
 | 
						|
              case 8:
 | 
						|
                worksheet = _context.sent;
 | 
						|
                stream.close();
 | 
						|
                return _context.abrupt("return", worksheet);
 | 
						|
 | 
						|
              case 11:
 | 
						|
              case "end":
 | 
						|
                return _context.stop();
 | 
						|
            }
 | 
						|
          }
 | 
						|
        }, _callee, this);
 | 
						|
      }));
 | 
						|
 | 
						|
      function readFile(_x, _x2) {
 | 
						|
        return _readFile.apply(this, arguments);
 | 
						|
      }
 | 
						|
 | 
						|
      return readFile;
 | 
						|
    }()
 | 
						|
  }, {
 | 
						|
    key: "read",
 | 
						|
    value: function read(stream, options) {
 | 
						|
      var _this = this;
 | 
						|
 | 
						|
      options = options || {};
 | 
						|
      return new Promise(function (resolve, reject) {
 | 
						|
        var worksheet = _this.workbook.addWorksheet(options.sheetName);
 | 
						|
 | 
						|
        var dateFormats = options.dateFormats || ['YYYY-MM-DD[T]HH:mm:ssZ', 'YYYY-MM-DD[T]HH:mm:ss', 'MM-DD-YYYY', 'YYYY-MM-DD'];
 | 
						|
 | 
						|
        var map = options.map || function (datum) {
 | 
						|
          if (datum === '') {
 | 
						|
            return null;
 | 
						|
          }
 | 
						|
 | 
						|
          var datumNumber = Number(datum);
 | 
						|
 | 
						|
          if (!Number.isNaN(datumNumber) && datumNumber !== Infinity) {
 | 
						|
            return datumNumber;
 | 
						|
          }
 | 
						|
 | 
						|
          var dt = dateFormats.reduce(function (matchingDate, currentDateFormat) {
 | 
						|
            if (matchingDate) {
 | 
						|
              return matchingDate;
 | 
						|
            }
 | 
						|
 | 
						|
            var dayjsObj = dayjs(datum, currentDateFormat, true);
 | 
						|
 | 
						|
            if (dayjsObj.isValid()) {
 | 
						|
              return dayjsObj;
 | 
						|
            }
 | 
						|
 | 
						|
            return null;
 | 
						|
          }, null);
 | 
						|
 | 
						|
          if (dt) {
 | 
						|
            return new Date(dt.valueOf());
 | 
						|
          }
 | 
						|
 | 
						|
          var special = SpecialValues[datum];
 | 
						|
 | 
						|
          if (special !== undefined) {
 | 
						|
            return special;
 | 
						|
          }
 | 
						|
 | 
						|
          return datum;
 | 
						|
        };
 | 
						|
 | 
						|
        var csvStream = fastCsv.parse(options.parserOptions).on('data', function (data) {
 | 
						|
          worksheet.addRow(data.map(map));
 | 
						|
        }).on('end', function () {
 | 
						|
          csvStream.emit('worksheet', worksheet);
 | 
						|
        });
 | 
						|
        csvStream.on('worksheet', resolve).on('error', reject);
 | 
						|
        stream.pipe(csvStream);
 | 
						|
      });
 | 
						|
    }
 | 
						|
    /**
 | 
						|
     * @deprecated since version 4.0. You should use `CSV#read` instead. Please follow upgrade instruction: https://github.com/exceljs/exceljs/blob/master/UPGRADE-4.0.md
 | 
						|
     */
 | 
						|
 | 
						|
  }, {
 | 
						|
    key: "createInputStream",
 | 
						|
    value: function createInputStream() {
 | 
						|
      throw new Error('`CSV#createInputStream` is deprecated. You should use `CSV#read` instead. This method will be removed in version 5.0. Please follow upgrade instruction: https://github.com/exceljs/exceljs/blob/master/UPGRADE-4.0.md');
 | 
						|
    }
 | 
						|
  }, {
 | 
						|
    key: "write",
 | 
						|
    value: function write(stream, options) {
 | 
						|
      var _this2 = this;
 | 
						|
 | 
						|
      return new Promise(function (resolve, reject) {
 | 
						|
        options = options || {}; // const encoding = options.encoding || 'utf8';
 | 
						|
        // const separator = options.separator || ',';
 | 
						|
        // const quoteChar = options.quoteChar || '\'';
 | 
						|
 | 
						|
        var worksheet = _this2.workbook.getWorksheet(options.sheetName || options.sheetId);
 | 
						|
 | 
						|
        var csvStream = fastCsv.format(options.formatterOptions);
 | 
						|
        stream.on('finish', function () {
 | 
						|
          resolve();
 | 
						|
        });
 | 
						|
        csvStream.on('error', reject);
 | 
						|
        csvStream.pipe(stream);
 | 
						|
        var _options = options,
 | 
						|
            dateFormat = _options.dateFormat,
 | 
						|
            dateUTC = _options.dateUTC;
 | 
						|
 | 
						|
        var map = options.map || function (value) {
 | 
						|
          if (value) {
 | 
						|
            if (value.text || value.hyperlink) {
 | 
						|
              return value.hyperlink || value.text || '';
 | 
						|
            }
 | 
						|
 | 
						|
            if (value.formula || value.result) {
 | 
						|
              return value.result || '';
 | 
						|
            }
 | 
						|
 | 
						|
            if (value instanceof Date) {
 | 
						|
              if (dateFormat) {
 | 
						|
                return dateUTC ? dayjs.utc(value).format(dateFormat) : dayjs(value).format(dateFormat);
 | 
						|
              }
 | 
						|
 | 
						|
              return dateUTC ? dayjs.utc(value).format() : dayjs(value).format();
 | 
						|
            }
 | 
						|
 | 
						|
            if (value.error) {
 | 
						|
              return value.error;
 | 
						|
            }
 | 
						|
 | 
						|
            if (_typeof(value) === 'object') {
 | 
						|
              return JSON.stringify(value);
 | 
						|
            }
 | 
						|
          }
 | 
						|
 | 
						|
          return value;
 | 
						|
        };
 | 
						|
 | 
						|
        var includeEmptyRows = options.includeEmptyRows === undefined || options.includeEmptyRows;
 | 
						|
        var lastRow = 1;
 | 
						|
 | 
						|
        if (worksheet) {
 | 
						|
          worksheet.eachRow(function (row, rowNumber) {
 | 
						|
            if (includeEmptyRows) {
 | 
						|
              while (lastRow++ < rowNumber - 1) {
 | 
						|
                csvStream.write([]);
 | 
						|
              }
 | 
						|
            }
 | 
						|
 | 
						|
            var values = row.values;
 | 
						|
            values.shift();
 | 
						|
            csvStream.write(values.map(map));
 | 
						|
            lastRow = rowNumber;
 | 
						|
          });
 | 
						|
        }
 | 
						|
 | 
						|
        csvStream.end();
 | 
						|
      });
 | 
						|
    }
 | 
						|
  }, {
 | 
						|
    key: "writeFile",
 | 
						|
    value: function writeFile(filename, options) {
 | 
						|
      options = options || {};
 | 
						|
      var streamOptions = {
 | 
						|
        encoding: options.encoding || 'utf8'
 | 
						|
      };
 | 
						|
      var stream = fs.createWriteStream(filename, streamOptions);
 | 
						|
      return this.write(stream, options);
 | 
						|
    }
 | 
						|
  }, {
 | 
						|
    key: "writeBuffer",
 | 
						|
    value: function () {
 | 
						|
      var _writeBuffer = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee2(options) {
 | 
						|
        var stream;
 | 
						|
        return regeneratorRuntime.wrap(function _callee2$(_context2) {
 | 
						|
          while (1) {
 | 
						|
            switch (_context2.prev = _context2.next) {
 | 
						|
              case 0:
 | 
						|
                stream = new StreamBuf();
 | 
						|
                _context2.next = 3;
 | 
						|
                return this.write(stream, options);
 | 
						|
 | 
						|
              case 3:
 | 
						|
                return _context2.abrupt("return", stream.read());
 | 
						|
 | 
						|
              case 4:
 | 
						|
              case "end":
 | 
						|
                return _context2.stop();
 | 
						|
            }
 | 
						|
          }
 | 
						|
        }, _callee2, this);
 | 
						|
      }));
 | 
						|
 | 
						|
      function writeBuffer(_x3) {
 | 
						|
        return _writeBuffer.apply(this, arguments);
 | 
						|
      }
 | 
						|
 | 
						|
      return writeBuffer;
 | 
						|
    }()
 | 
						|
  }]);
 | 
						|
 | 
						|
  return CSV;
 | 
						|
}();
 | 
						|
 | 
						|
module.exports = CSV;
 | 
						|
//# sourceMappingURL=csv.js.map
 |