const hljs = require('highlight.js');

function highlight(code, lang) {
    const language = hljs.getLanguage(lang) ? lang : 'plaintext';

    const highlighted = hljs.highlight(code, { language }).value;
    return highlighted;
}

module.exports = {
    highlight
};