redoc/src/utils/highlight.ts

56 lines
1.7 KiB
TypeScript
Raw Normal View History

2017-10-12 00:01:37 +03:00
import * as Prism from 'prismjs';
import 'prismjs/components/prism-actionscript.js';
import 'prismjs/components/prism-c.js';
import 'prismjs/components/prism-cpp.js';
import 'prismjs/components/prism-csharp.js';
import 'prismjs/components/prism-php.js';
import 'prismjs/components/prism-coffeescript.js';
import 'prismjs/components/prism-go.js';
import 'prismjs/components/prism-haskell.js';
import 'prismjs/components/prism-java.js';
import 'prismjs/components/prism-lua.js';
import 'prismjs/components/prism-matlab.js';
import 'prismjs/components/prism-perl.js';
import 'prismjs/components/prism-python.js';
import 'prismjs/components/prism-r.js';
import 'prismjs/components/prism-ruby.js';
import 'prismjs/components/prism-bash.js';
import 'prismjs/components/prism-swift.js';
import 'prismjs/components/prism-objectivec.js';
import 'prismjs/components/prism-scala.js';
import 'prismjs/components/prism-markup.js'; // xml
import 'prismjs/themes/prism-dark.css'; // dark theme
const DEFAULT_LANG = 'clike';
/**
* map language names to Prism.js names
*/
function mapLang(lang: string): string {
return (
{
json: 'js',
'c++': 'cpp',
'c#': 'csharp',
'objective-c': 'objectivec',
shell: 'bash',
viml: 'vim',
}[lang] || DEFAULT_LANG
);
}
/**
* Highlight source code string using Prism.js
* @param source source code to highlight
* @param lang highlight language
* @return highlighted souce code as **html string**
*/
export function highlight(source: string, lang: string): string {
let grammar = Prism.languages[lang];
if (!grammar) {
grammar = Prism.languages[mapLang(lang)];
}
return Prism.highlight(source, grammar);
}