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
 |