@beenotung/tslib
Version:
utils library in Typescript
158 lines (157 loc) • 5.55 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.wrap_into_workbook = wrap_into_workbook;
exports.write_xlsx_file = write_xlsx_file;
exports.read_xlsx_file = read_xlsx_file;
exports.get_worksheet_headers = get_worksheet_headers;
exports.fill_by_headers = fill_by_headers;
async function wrap_into_workbook(data) {
const XLSX = await Promise.resolve().then(() => __importStar(require('@e965/xlsx')));
const workbook = XLSX.utils.book_new();
if (Array.isArray(data)) {
const worksheet = XLSX.utils.json_to_sheet(data);
XLSX.utils.book_append_sheet(workbook, worksheet);
}
else {
for (const [sheet_name, rows] of Object.entries(data)) {
const worksheet = XLSX.utils.json_to_sheet(rows);
XLSX.utils.book_append_sheet(workbook, worksheet, sheet_name);
}
}
return wrap_workbook(workbook, XLSX);
}
async function write_xlsx_file(file, data, options = {}) {
const { workbook, XLSX } = await wrap_into_workbook(data);
await new Promise((resolve, reject) => {
XLSX.writeFileAsync(file, workbook, options, function () {
const error = arguments[0];
if (error) {
reject(error);
}
else {
resolve();
}
});
});
}
async function read_xlsx_file(file) {
const XLSX = await Promise.resolve().then(() => __importStar(require('@e965/xlsx')));
const workbook = XLSX.readFile(file);
return wrap_workbook(workbook, XLSX);
}
async function wrap_workbook(workbook, XLSX) {
function get_default_sheet_name() {
const sheet_names = workbook.SheetNames;
if (sheet_names.length === 1) {
return sheet_names[0];
}
if (sheet_names.length === 0) {
throw new Error('No sheets exist in the workbook');
}
throw new Error(`Multiple sheets exist in the workbook, please specify the sheet name: ${sheet_names.join(', ')}`);
}
function get_sheet(name = get_default_sheet_name()) {
const sheet = workbook.Sheets[name];
if (!sheet) {
throw new Error(`Sheet ${name} not found`);
}
return sheet;
}
/** Converts a worksheet object to an array of JSON objects */
function get_sheet_as_json(name, options) {
return XLSX.utils.sheet_to_json(get_sheet(name), options);
}
/** Generate delimited-separated-values output */
function get_sheet_as_csv(name, options) {
return XLSX.utils.sheet_to_csv(get_sheet(name), options);
}
/** Generate UTF16 Formatted Text */
function get_sheet_as_text(name, options) {
return XLSX.utils.sheet_to_txt(get_sheet(name), options);
}
/** Generate HTML */
function get_sheet_as_html(name, options) {
return XLSX.utils.sheet_to_html(get_sheet(name), options);
}
/** Generate a list of the formulae (with value fallbacks) */
function get_sheet_as_formulae(name, options) {
return XLSX.utils.sheet_to_formulae(get_sheet(name), options);
}
return {
workbook,
get_default_sheet_name,
get_sheet,
get_sheet_as_json,
get_sheet_as_csv,
get_sheet_as_text,
get_sheet_as_html,
get_sheet_as_formulae,
XLSX,
};
}
function get_worksheet_headers(worksheet) {
const headers = [];
for (const key of Object.keys(worksheet)) {
const col = parse_column_range(key);
const row = key.slice(col.length);
if (row != '1')
continue;
const cell = worksheet[key];
const value = cell.v;
headers.push({ col, value });
}
return headers;
}
function fill_by_headers(args) {
const { rows, headers } = args;
const default_value = args.default_value || '';
for (const row of rows) {
for (const col of headers) {
row[col] ||= default_value;
}
}
return rows;
}
function parse_column_range(cell) {
// e.g. 'L5' -> 'L'
return cell
.split('')
.filter(c => !is_digit(c))
.join('');
}
function is_digit(c) {
return c >= '0' && c <= '9';
}