console-toolkit/src/charts/columns/plain.js

Toolkit to produce a fancy console output (boxes, tables, charts, colors).

import style from '../../style.js';
import Box from '../../box.js';
import {optimize} from '../../ansi/sgr-state.js';
import {allocateSizes} from '../utils.js';
import drawStackedChart from './draw-stacked.js';
import {hBlocks8th} from '../../symbols.js';
import defaultTheme from '../themes/default.js';

// data = [datum]
// datum = {value, colorState, symbol, state}

const defaultSymbol = hBlocks8th[7];

export const drawColumn = (data, width, maxValue, options = {}) => {
  const {reverse, rectSize = 1, initState = {}, theme = defaultTheme} = options,
    {symbol = ' ', state = null, colorState} = theme?.empty || {},
    sizes = allocateSizes(data, maxValue, width),
    blocks = data.map((datum, i) => {
      if (!datum) return Box.makeBlank(rectSize, 0);
      const boxStyle = style.addState(initState).addState(datum.colorState).addState(datum.state);
      return new Box(new Array(sizes[i]).fill(boxStyle.text(datum.symbol || defaultSymbol)), true);
    });
  if (sizes[sizes.length - 1] > 0) {
    blocks.push(
      new Box(
        new Array(sizes[sizes.length - 1]).fill(
          style.addState(initState).addState(colorState).addState(state).text(symbol.repeat(rectSize))
        ),
        true
      )
    );
  }
  const result = (reverse ? blocks : blocks.reverse()).map(block => block.box).flat(1);
  return result.map(line => optimize(line));
};

export const drawChart = drawStackedChart(drawColumn);

export default drawChart;