forked from zhurui/management
157 lines
3.9 KiB
JavaScript
157 lines
3.9 KiB
JavaScript
'use strict';
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
|
|
var _lodash = require('lodash');
|
|
|
|
var _lodash2 = _interopRequireDefault(_lodash);
|
|
|
|
var _makeStreamConfig = require('./makeStreamConfig');
|
|
|
|
var _makeStreamConfig2 = _interopRequireDefault(_makeStreamConfig);
|
|
|
|
var _drawRow = require('./drawRow');
|
|
|
|
var _drawRow2 = _interopRequireDefault(_drawRow);
|
|
|
|
var _drawBorder = require('./drawBorder');
|
|
|
|
var _stringifyTableData = require('./stringifyTableData');
|
|
|
|
var _stringifyTableData2 = _interopRequireDefault(_stringifyTableData);
|
|
|
|
var _truncateTableData = require('./truncateTableData');
|
|
|
|
var _truncateTableData2 = _interopRequireDefault(_truncateTableData);
|
|
|
|
var _mapDataUsingRowHeightIndex = require('./mapDataUsingRowHeightIndex');
|
|
|
|
var _mapDataUsingRowHeightIndex2 = _interopRequireDefault(_mapDataUsingRowHeightIndex);
|
|
|
|
var _alignTableData = require('./alignTableData');
|
|
|
|
var _alignTableData2 = _interopRequireDefault(_alignTableData);
|
|
|
|
var _padTableData = require('./padTableData');
|
|
|
|
var _padTableData2 = _interopRequireDefault(_padTableData);
|
|
|
|
var _calculateRowHeightIndex = require('./calculateRowHeightIndex');
|
|
|
|
var _calculateRowHeightIndex2 = _interopRequireDefault(_calculateRowHeightIndex);
|
|
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
|
|
|
/**
|
|
* @param {Array} data
|
|
* @param {Object} config
|
|
* @returns {Array}
|
|
*/
|
|
const prepareData = (data, config) => {
|
|
let rows;
|
|
|
|
rows = (0, _stringifyTableData2.default)(data);
|
|
|
|
rows = (0, _truncateTableData2.default)(data, config);
|
|
|
|
const rowHeightIndex = (0, _calculateRowHeightIndex2.default)(rows, config);
|
|
|
|
rows = (0, _mapDataUsingRowHeightIndex2.default)(rows, rowHeightIndex, config);
|
|
rows = (0, _alignTableData2.default)(rows, config);
|
|
rows = (0, _padTableData2.default)(rows, config);
|
|
|
|
return rows;
|
|
};
|
|
|
|
/**
|
|
* @param {string[]} row
|
|
* @param {number[]} columnWidthIndex
|
|
* @param {Object} config
|
|
* @returns {undefined}
|
|
*/
|
|
const create = (row, columnWidthIndex, config) => {
|
|
const rows = prepareData([row], config);
|
|
|
|
const body = _lodash2.default.map(rows, literalRow => {
|
|
return (0, _drawRow2.default)(literalRow, config.border);
|
|
}).join('');
|
|
|
|
let output;
|
|
|
|
output = '';
|
|
|
|
output += (0, _drawBorder.drawBorderTop)(columnWidthIndex, config.border);
|
|
output += body;
|
|
output += (0, _drawBorder.drawBorderBottom)(columnWidthIndex, config.border);
|
|
|
|
output = _lodash2.default.trimEnd(output);
|
|
|
|
process.stdout.write(output);
|
|
};
|
|
|
|
/**
|
|
* @param {string[]} row
|
|
* @param {number[]} columnWidthIndex
|
|
* @param {Object} config
|
|
* @returns {undefined}
|
|
*/
|
|
const append = (row, columnWidthIndex, config) => {
|
|
const rows = prepareData([row], config);
|
|
|
|
const body = _lodash2.default.map(rows, literalRow => {
|
|
return (0, _drawRow2.default)(literalRow, config.border);
|
|
}).join('');
|
|
|
|
let output;
|
|
|
|
output = '\r\u001B[K';
|
|
|
|
output += (0, _drawBorder.drawBorderJoin)(columnWidthIndex, config.border);
|
|
output += body;
|
|
output += (0, _drawBorder.drawBorderBottom)(columnWidthIndex, config.border);
|
|
|
|
output = _lodash2.default.trimEnd(output);
|
|
|
|
process.stdout.write(output);
|
|
};
|
|
|
|
/**
|
|
* @param {Object} userConfig
|
|
* @returns {Object}
|
|
*/
|
|
|
|
exports.default = function () {
|
|
let userConfig = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
|
|
|
|
const config = (0, _makeStreamConfig2.default)(userConfig);
|
|
|
|
const columnWidthIndex = _lodash2.default.mapValues(config.columns, column => {
|
|
return column.width + column.paddingLeft + column.paddingRight;
|
|
});
|
|
|
|
let empty;
|
|
|
|
empty = true;
|
|
|
|
return {
|
|
/**
|
|
* @param {string[]} row
|
|
* @returns {undefined}
|
|
*/
|
|
write: row => {
|
|
if (row.length !== config.columnCount) {
|
|
throw new Error('Row cell count does not match the config.columnCount.');
|
|
}
|
|
|
|
if (empty) {
|
|
empty = false;
|
|
|
|
return create(row, columnWidthIndex, config);
|
|
} else {
|
|
return append(row, columnWidthIndex, config);
|
|
}
|
|
}
|
|
};
|
|
}; |