createStream.js 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. 'use strict';
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. var _lodash = require('lodash');
  6. var _lodash2 = _interopRequireDefault(_lodash);
  7. var _makeStreamConfig = require('./makeStreamConfig');
  8. var _makeStreamConfig2 = _interopRequireDefault(_makeStreamConfig);
  9. var _drawRow = require('./drawRow');
  10. var _drawRow2 = _interopRequireDefault(_drawRow);
  11. var _drawBorder = require('./drawBorder');
  12. var _stringifyTableData = require('./stringifyTableData');
  13. var _stringifyTableData2 = _interopRequireDefault(_stringifyTableData);
  14. var _truncateTableData = require('./truncateTableData');
  15. var _truncateTableData2 = _interopRequireDefault(_truncateTableData);
  16. var _mapDataUsingRowHeightIndex = require('./mapDataUsingRowHeightIndex');
  17. var _mapDataUsingRowHeightIndex2 = _interopRequireDefault(_mapDataUsingRowHeightIndex);
  18. var _alignTableData = require('./alignTableData');
  19. var _alignTableData2 = _interopRequireDefault(_alignTableData);
  20. var _padTableData = require('./padTableData');
  21. var _padTableData2 = _interopRequireDefault(_padTableData);
  22. var _calculateRowHeightIndex = require('./calculateRowHeightIndex');
  23. var _calculateRowHeightIndex2 = _interopRequireDefault(_calculateRowHeightIndex);
  24. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  25. /**
  26. * @param {Array} data
  27. * @param {Object} config
  28. * @returns {Array}
  29. */
  30. const prepareData = (data, config) => {
  31. let rows;
  32. rows = (0, _stringifyTableData2.default)(data);
  33. rows = (0, _truncateTableData2.default)(data, config);
  34. const rowHeightIndex = (0, _calculateRowHeightIndex2.default)(rows, config);
  35. rows = (0, _mapDataUsingRowHeightIndex2.default)(rows, rowHeightIndex, config);
  36. rows = (0, _alignTableData2.default)(rows, config);
  37. rows = (0, _padTableData2.default)(rows, config);
  38. return rows;
  39. };
  40. /**
  41. * @param {string[]} row
  42. * @param {number[]} columnWidthIndex
  43. * @param {Object} config
  44. * @returns {undefined}
  45. */
  46. const create = (row, columnWidthIndex, config) => {
  47. const rows = prepareData([row], config);
  48. const body = _lodash2.default.map(rows, literalRow => {
  49. return (0, _drawRow2.default)(literalRow, config.border);
  50. }).join('');
  51. let output;
  52. output = '';
  53. output += (0, _drawBorder.drawBorderTop)(columnWidthIndex, config.border);
  54. output += body;
  55. output += (0, _drawBorder.drawBorderBottom)(columnWidthIndex, config.border);
  56. output = _lodash2.default.trimEnd(output);
  57. process.stdout.write(output);
  58. };
  59. /**
  60. * @param {string[]} row
  61. * @param {number[]} columnWidthIndex
  62. * @param {Object} config
  63. * @returns {undefined}
  64. */
  65. const append = (row, columnWidthIndex, config) => {
  66. const rows = prepareData([row], config);
  67. const body = _lodash2.default.map(rows, literalRow => {
  68. return (0, _drawRow2.default)(literalRow, config.border);
  69. }).join('');
  70. let output;
  71. output = '\r\u001B[K';
  72. output += (0, _drawBorder.drawBorderJoin)(columnWidthIndex, config.border);
  73. output += body;
  74. output += (0, _drawBorder.drawBorderBottom)(columnWidthIndex, config.border);
  75. output = _lodash2.default.trimEnd(output);
  76. process.stdout.write(output);
  77. };
  78. /**
  79. * @param {Object} userConfig
  80. * @returns {Object}
  81. */
  82. exports.default = function () {
  83. let userConfig = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
  84. const config = (0, _makeStreamConfig2.default)(userConfig);
  85. const columnWidthIndex = _lodash2.default.mapValues(config.columns, column => {
  86. return column.width + column.paddingLeft + column.paddingRight;
  87. });
  88. let empty;
  89. empty = true;
  90. return {
  91. /**
  92. * @param {string[]} row
  93. * @returns {undefined}
  94. */
  95. write: row => {
  96. if (row.length !== config.columnCount) {
  97. throw new Error('Row cell count does not match the config.columnCount.');
  98. }
  99. if (empty) {
  100. empty = false;
  101. return create(row, columnWidthIndex, config);
  102. } else {
  103. return append(row, columnWidthIndex, config);
  104. }
  105. }
  106. };
  107. };