@forzalabs/remora
Version:
A powerful CLI tool for seamless data translation.
88 lines (87 loc) • 5.1 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __asyncValues = (this && this.__asyncValues) || function (o) {
if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
var m = o[Symbol.asyncIterator], i;
return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const DriverHelper_1 = __importDefault(require("../../drivers/DriverHelper"));
const Affirm_1 = __importDefault(require("../../core/Affirm"));
const xlsx_1 = __importDefault(require("xlsx"));
class XLSParserClass {
constructor() {
this.getStreamXls = (filePath, sheetName) => __awaiter(this, void 0, void 0, function* () {
const fileXls = xlsx_1.default.readFile(filePath);
(0, Affirm_1.default)(fileXls.SheetNames.length > 0, 'The Excel file has no sheets.');
const targetSheetName = yield this.getTargetSheetName(fileXls, sheetName);
const workSheet = fileXls.Sheets[targetSheetName];
const xlsStream = xlsx_1.default.stream.to_csv(workSheet);
return xlsStream;
});
this.getHeaderXls = (filePath, sheetName) => __awaiter(this, void 0, void 0, function* () {
const header = yield DriverHelper_1.default.quickReadStream(yield this.getStreamXls(filePath, sheetName), 2);
return header[0];
});
this.getHeaderXlsFromStream = (stream, sheetName) => __awaiter(this, void 0, void 0, function* () {
var _a, e_1, _b, _c;
let header;
const workbook = xlsx_1.default.read(stream);
const targetSheetName = yield this.getTargetSheetName(workbook, sheetName);
const xlsxStream = xlsx_1.default.stream.to_csv(workbook.Sheets[targetSheetName]);
try {
for (var _d = true, xlsxStream_1 = __asyncValues(xlsxStream), xlsxStream_1_1; xlsxStream_1_1 = yield xlsxStream_1.next(), _a = xlsxStream_1_1.done, !_a; _d = true) {
_c = xlsxStream_1_1.value;
_d = false;
const line = _c;
if (line != '') {
header = line;
break;
}
}
}
catch (e_1_1) { e_1 = { error: e_1_1 }; }
finally {
try {
if (!_d && !_a && (_b = xlsxStream_1.return)) yield _b.call(xlsxStream_1);
}
finally { if (e_1) throw e_1.error; }
}
(0, Affirm_1.default)(header, `the file received from the stream is empty`);
return header;
});
this.getTargetSheetName = (workbook, sheetName) => __awaiter(this, void 0, void 0, function* () {
let targetSheetName = sheetName ? sheetName : workbook.SheetNames[0];
if (!targetSheetName) {
(0, Affirm_1.default)(workbook.SheetNames.length > 0, 'The Excel file has no sheets.');
targetSheetName = workbook.SheetNames[0];
}
else {
(0, Affirm_1.default)(workbook.SheetNames.includes(targetSheetName), `The sheet "${targetSheetName}" doesn't exist in the excel (available: ${workbook.SheetNames.join(', ')})`);
}
return targetSheetName;
});
this.parseXLSStream = (xlsStream, sheetName) => __awaiter(this, void 0, void 0, function* () {
const workbook = xlsx_1.default.read(xlsStream);
const targetSheetName = yield this.getTargetSheetName(workbook, sheetName);
const workSheet = workbook.Sheets[targetSheetName];
const csvStream = xlsx_1.default.stream.to_csv(workSheet);
return csvStream;
});
}
}
const XLSParser = new XLSParserClass();
exports.default = XLSParser;