rj1
about | log | files | refs | license

highlight.py (880B) - raw

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#!/usr/bin/env python3
import pygments
from pygments import highlight
from pygments.formatters import HtmlFormatter
from pygments_cache import guess_lexer_for_filename
from pygments.lexers import guess_lexer

from sys import stdin, stderr

filename = stdin.readline().strip()
contents = stdin.read()

lexer=None

try:
    lexer = guess_lexer_for_filename(filename, contents)
except pygments.util.ClassNotFound:
    try:
        lexer = guess_lexer(contents)
    except pygments.util.ClassNotFound:
        pass

if lexer is None:
    from pygments.lexers import TextLexer
    lexer = TextLexer()

FORMAT = HtmlFormatter(
    style='one-dark',
    cssclass='highlight',
    linenos='table',
    lineanchors='loc',
    anchorlinenos=True)

print('<div id="blob">')
print(highlight(contents, lexer, FORMAT))
print('</div>')

print(f"processing {filename} with {lexer}", file=stderr)