highlight.py
94 lines
| 2.8 KiB
| text/x-python
|
PythonLexer
/ hgext / highlight.py
Adam Hupp
|
r5532 | """ | ||
This is Mercurial extension for syntax highlighting in the file | ||||
revision view of hgweb. | ||||
It depends on the pygments syntax highlighting library: | ||||
http://pygments.org/ | ||||
To enable the extension add this to hgrc: | ||||
[extensions] | ||||
hgext.highlight = | ||||
There is a single configuration option: | ||||
[web] | ||||
pygments_style = <style> | ||||
The default is 'colorful'. If this is changed the corresponding CSS | ||||
file should be re-generated by running | ||||
# pygmentize -f html -S <newstyle> | ||||
-- Adam Hupp <adam@hupp.org> | ||||
""" | ||||
from mercurial import demandimport | ||||
Dirkjan Ochtman
|
r6394 | demandimport.ignore.extend(['pkgutil', 'pkg_resources', '__main__',]) | ||
Adam Hupp
|
r5532 | |||
Dirkjan Ochtman
|
r6393 | from mercurial.hgweb import webcommands, webutil | ||
Adam Hupp
|
r5532 | from mercurial import util | ||
Brendan Cully
|
r6193 | from mercurial.templatefilters import filters | ||
Adam Hupp
|
r5532 | |||
from pygments import highlight | ||||
from pygments.util import ClassNotFound | ||||
Brendan Cully
|
r6199 | from pygments.lexers import guess_lexer, guess_lexer_for_filename, TextLexer | ||
Adam Hupp
|
r5532 | from pygments.formatters import HtmlFormatter | ||
Bryan O'Sullivan
|
r5533 | SYNTAX_CSS = ('\n<link rel="stylesheet" href="#staticurl#highlight.css" ' | ||
'type="text/css" />') | ||||
Adam Hupp
|
r5532 | |||
Dirkjan Ochtman
|
r6394 | def pygmentize(field, fctx, style, tmpl): | ||
Adam Hupp
|
r5532 | # append a <link ...> to the syntax highlighting css | ||
Christian Ebert
|
r5616 | old_header = ''.join(tmpl('header')) | ||
Adam Hupp
|
r5532 | if SYNTAX_CSS not in old_header: | ||
new_header = old_header + SYNTAX_CSS | ||||
Christian Ebert
|
r5616 | tmpl.cache['header'] = new_header | ||
Adam Hupp
|
r5532 | |||
Brendan Cully
|
r6194 | text = fctx.data() | ||
if util.binary(text): | ||||
return | ||||
Brendan Cully
|
r6193 | # To get multi-line strings right, we can't format line-by-line | ||
try: | ||||
lexer = guess_lexer_for_filename(fctx.path(), text, | ||||
encoding=util._encoding) | ||||
except ClassNotFound: | ||||
Brendan Cully
|
r6198 | try: | ||
lexer = guess_lexer(text, encoding=util._encoding) | ||||
except ClassNotFound: | ||||
lexer = TextLexer(encoding=util._encoding) | ||||
Brendan Cully
|
r6193 | |||
formatter = HtmlFormatter(style=style, encoding=util._encoding) | ||||
Adam Hupp
|
r5532 | |||
Brendan Cully
|
r6193 | colorized = highlight(text, lexer, formatter) | ||
# strip wrapping div | ||||
colorized = colorized[:colorized.find('\n</pre>')] | ||||
Brendan Cully
|
r6197 | colorized = colorized[colorized.find('<pre>')+5:] | ||
Brendan Cully
|
r6196 | coloriter = iter(colorized.splitlines()) | ||
Brendan Cully
|
r6193 | |||
filters['colorize'] = lambda x: coloriter.next() | ||||
oldl = tmpl.cache[field] | ||||
newl = oldl.replace('line|escape', 'line|colorize') | ||||
tmpl.cache[field] = newl | ||||
Dirkjan Ochtman
|
r6393 | web_filerevision = webcommands._filerevision | ||
web_annotate = webcommands.annotate | ||||
Brendan Cully
|
r6193 | |||
Dirkjan Ochtman
|
r6393 | def filerevision_highlight(web, tmpl, fctx): | ||
Dirkjan Ochtman
|
r6394 | style = web.config('web', 'pygments_style', 'colorful') | ||
pygmentize('fileline', fctx, style, tmpl) | ||||
Dirkjan Ochtman
|
r6393 | return web_filerevision(web, tmpl, fctx) | ||
Brendan Cully
|
r6193 | |||
Dirkjan Ochtman
|
r6393 | def annotate_highlight(web, req, tmpl): | ||
fctx = webutil.filectx(web.repo, req) | ||||
Dirkjan Ochtman
|
r6394 | style = web.config('web', 'pygments_style', 'colorful') | ||
pygmentize('annotateline', fctx, style, tmpl) | ||||
Dirkjan Ochtman
|
r6393 | return web_annotate(web, req, tmpl) | ||
Bryan O'Sullivan
|
r5533 | |||
Adam Hupp
|
r5532 | # monkeypatch in the new version | ||
Dirkjan Ochtman
|
r6393 | |||
webcommands._filerevision = filerevision_highlight | ||||
webcommands.annotate = annotate_highlight | ||||