bnash/src/colors.js

A command-line editor for bash or compatible shells, aiming to provide great UX and productivity

const chalkModule = require('chalk')

const { ucfirst } = require('./utils')
const { getOption } = require('./startup')


let chalk = null

function getChalk() {
    if (chalk) return chalk
    let trueColor = getOption('colors.trueColor')
    if (trueColor)
        chalk = new chalkModule.Instance({ level: 3 })
    else
        chalk = chalkModule
    return chalk
}

function colorizer(cname) {
    if (!cname) return str => str
    let chalk = getChalk()
    let colors = cname.trim().split(' ')
    let colorfunc = chalk
    for (let color of colors) {
        if (color.startsWith('/')) {
            color = color.substr(1)
            if (color.startsWith('#'))
                colorfunc = colorfunc.bgHex(color)
            else
                colorfunc = colorfunc['bg' + ucfirst(color)]
        }
        else {
            if (color.startsWith('#'))
                colorfunc = colorfunc.hex(color)
            else
                colorfunc = colorfunc[color]
        }
    }
    return colorfunc
}

function colorize(cname, str) {
    if (str == ''|| cname == '') return str
    let colorfunc = colorizer(cname)
    return colorfunc(str)
}


module.exports = {
    colorizer,
    colorize
}