UNPKG

@beenotung/tslib

Version:
158 lines (157 loc) 5.55 kB
"use strict"; 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'; }