patternplate-client/source/application/components/common/code/highlight.js

Universal javascript client application for patternplate

import {includes} from 'lodash';
import low from 'lowlight/lib/core';
import {pd as pretty} from 'pretty-data';

import css from 'highlight.js/lib/languages/css.js';
import less from 'highlight.js/lib/languages/less.js';
import scss from 'highlight.js/lib/languages/scss.js';
import stylus from 'highlight.js/lib/languages/stylus.js';

import js from 'highlight.js/lib/languages/javascript.js';
import ts from 'highlight.js/lib/languages/typescript.js';
import json from 'highlight.js/lib/languages/json.js';

import xml from 'highlight.js/lib/languages/xml.js';
import md from 'highlight.js/lib/languages/markdown.js';

import bash from 'highlight.js/lib/languages/bash.js';

// CSS and friends
low.registerLanguage('css', css);
low.registerLanguage('less', less);
low.registerLanguage('scss', scss);
low.registerLanguage('stylus', stylus);

// JS and friends
low.registerLanguage('js', js);
low.registerLanguage('javascript', js);
low.registerLanguage('jsx', js);
low.registerLanguage('ts', ts);
low.registerLanguage('tsx', ts);
low.registerLanguage('typescript', ts);
low.registerLanguage('json', json);

// HTML and friends
low.registerLanguage('html', xml);
low.registerLanguage('xml', xml);
low.registerLanguage('md', md);
low.registerLanguage('markdown', md);

// (s)hell(ish)s
low.registerLanguage('bash', bash);
// low.registerLanguage('shell', bash);

const languages = [
	'css', 'less', 'scss', 'stylus', 'js', 'javascript', 'jsx', 'ts', 'tsx',
	'typescript', 'json', 'html', 'xml', 'md', 'markdown', 'bash'
];

const prettyPrinted = ['xml', 'html'];

export default function highlight(language, source) {
	if (!includes(languages, language)) {
		return source;
	}
	const code = includes(prettyPrinted, language) ? pretty.xml(source) : source;
	const {value: children} = low.highlight(language, code);
	return children;
}