highlight.py
101 lines
| 3.1 KiB
| text/x-python
|
PythonLexer
|
r8251 | # highlight.py - highlight extension implementation file | ||
# | ||||
# Copyright 2007-2009 Adam Hupp <adam@hupp.org> and others | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
|
r10263 | # GNU General Public License version 2 or any later version. | ||
|
r6938 | # | ||
# The original module was split in an interface and an implementation | ||||
# file to defer pygments loading and speedup extension setup. | ||||
|
r29485 | from __future__ import absolute_import | ||
|
r6938 | from mercurial import demandimport | ||
|
r43346 | |||
|
r43347 | demandimport.IGNORES.update([b'pkgutil', b'pkg_resources', b'__main__']) | ||
|
r29485 | |||
from mercurial import ( | ||||
encoding, | ||||
|
r43193 | pycompat, | ||
|
r37102 | ) | ||
|
r43346 | from mercurial.utils import stringutil | ||
|
r6938 | |||
|
r32908 | with demandimport.deactivated(): | ||
import pygments | ||||
import pygments.formatters | ||||
import pygments.lexers | ||||
|
r35330 | import pygments.plugin | ||
|
r32908 | import pygments.util | ||
|
r35330 | for unused in pygments.plugin.find_plugin_lexers(): | ||
pass | ||||
|
r29485 | highlight = pygments.highlight | ||
ClassNotFound = pygments.util.ClassNotFound | ||||
guess_lexer = pygments.lexers.guess_lexer | ||||
guess_lexer_for_filename = pygments.lexers.guess_lexer_for_filename | ||||
TextLexer = pygments.lexers.TextLexer | ||||
HtmlFormatter = pygments.formatters.HtmlFormatter | ||||
|
r6938 | |||
|
r43346 | SYNTAX_CSS = ( | ||
|
r43347 | b'\n<link rel="stylesheet" href="{url}highlightcss" ' b'type="text/css" />' | ||
|
r43346 | ) | ||
|
r6938 | |||
|
r26680 | def pygmentize(field, fctx, style, tmpl, guessfilenameonly=False): | ||
|
r6938 | |||
# append a <link ...> to the syntax highlighting css | ||||
|
r43347 | tmpl.load(b'header') | ||
old_header = tmpl.cache[b'header'] | ||||
|
r6938 | if SYNTAX_CSS not in old_header: | ||
|
r27637 | new_header = old_header + SYNTAX_CSS | ||
|
r43347 | tmpl.cache[b'header'] = new_header | ||
|
r6938 | |||
text = fctx.data() | ||||
|
r37102 | if stringutil.binary(text): | ||
|
r6938 | return | ||
|
r23613 | # str.splitlines() != unicode.splitlines() because "reasons" | ||
|
r43347 | for c in b"\x0c\x1c\x1d\x1e": | ||
|
r23613 | if c in text: | ||
|
r43347 | text = text.replace(c, b'') | ||
|
r23613 | |||
|
r9424 | # Pygments is best used with Unicode strings: | ||
# <http://pygments.org/docs/unicode/> | ||||
|
r43193 | text = text.decode(pycompat.sysstr(encoding.encoding), 'replace') | ||
|
r7120 | |||
|
r6938 | # To get multi-line strings right, we can't format line-by-line | ||
try: | ||||
|
r43193 | path = pycompat.sysstr(fctx.path()) | ||
|
r43346 | lexer = guess_lexer_for_filename(path, text[:1024], stripnl=False) | ||
|
r6938 | except (ClassNotFound, ValueError): | ||
|
r26680 | # guess_lexer will return a lexer if *any* lexer matches. There is | ||
# no way to specify a minimum match score. This can give a high rate of | ||||
# false positives on files with an unknown filename pattern. | ||||
if guessfilenameonly: | ||||
return | ||||
|
r6938 | try: | ||
|
r19169 | lexer = guess_lexer(text[:1024], stripnl=False) | ||
|
r6938 | except (ClassNotFound, ValueError): | ||
r25899 | # Don't highlight unknown files | |||
return | ||||
# Don't highlight text files | ||||
if isinstance(lexer, TextLexer): | ||||
return | ||||
|
r6938 | |||
|
r43193 | formatter = HtmlFormatter(nowrap=True, style=pycompat.sysstr(style)) | ||
|
r6938 | |||
colorized = highlight(text, lexer, formatter) | ||||
|
r43346 | coloriter = ( | ||
s.encode(pycompat.sysstr(encoding.encoding), 'replace') | ||||
for s in colorized.splitlines() | ||||
) | ||||
|
r6938 | |||
|
r43347 | tmpl._filters[b'colorize'] = lambda x: next(coloriter) | ||
|
r6938 | |||
oldl = tmpl.cache[field] | ||||
|
r43347 | newl = oldl.replace(b'line|escape', b'line|colorize') | ||
|
r6938 | tmpl.cache[field] = newl | ||