mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 17:58:27 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
# 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
|
|
# GNU General Public License version 2 or any later version.
|
|
#
|
|
# The original module was split in an interface and an implementation
|
|
# file to defer pygments loading and speedup extension setup.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from edenscm.mercurial import demandimport, encoding, util
|
|
|
|
|
|
demandimport.ignore.extend(["pkgutil", "pkg_resources", "__main__"])
|
|
|
|
|
|
with demandimport.deactivated():
|
|
import pygments
|
|
import pygments.formatters
|
|
import pygments.lexers
|
|
import pygments.util
|
|
|
|
|
|
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
|
|
|
|
SYNTAX_CSS = '\n<link rel="stylesheet" href="{url}highlightcss" ' 'type="text/css" />'
|
|
|
|
|
|
def pygmentize(field, fctx, style, tmpl, guessfilenameonly=False):
|
|
|
|
# append a <link ...> to the syntax highlighting css
|
|
old_header = tmpl.load("header")
|
|
if SYNTAX_CSS not in old_header:
|
|
new_header = old_header + SYNTAX_CSS
|
|
tmpl.cache["header"] = new_header
|
|
|
|
text = fctx.data()
|
|
if util.binary(text):
|
|
return
|
|
|
|
# str.splitlines() != unicode.splitlines() because "reasons"
|
|
for c in "\x0c\x1c\x1d\x1e":
|
|
if c in text:
|
|
text = text.replace(c, "")
|
|
|
|
# Pygments is best used with Unicode strings:
|
|
# <http://pygments.org/docs/unicode/>
|
|
text = text.decode(encoding.encoding, "replace")
|
|
|
|
# To get multi-line strings right, we can't format line-by-line
|
|
try:
|
|
lexer = guess_lexer_for_filename(fctx.path(), text[:1024], stripnl=False)
|
|
except (ClassNotFound, ValueError):
|
|
# 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
|
|
|
|
try:
|
|
lexer = guess_lexer(text[:1024], stripnl=False)
|
|
except (ClassNotFound, ValueError):
|
|
# Don't highlight unknown files
|
|
return
|
|
|
|
# Don't highlight text files
|
|
if isinstance(lexer, TextLexer):
|
|
return
|
|
|
|
formatter = HtmlFormatter(nowrap=True, style=style)
|
|
|
|
colorized = highlight(text, lexer, formatter)
|
|
coloriter = (s.encode(encoding.encoding, "replace") for s in colorized.splitlines())
|
|
|
|
tmpl.filters["colorize"] = lambda x: next(coloriter)
|
|
|
|
oldl = tmpl.cache[field]
|
|
newl = oldl.replace("line|escape", "line|colorize")
|
|
tmpl.cache[field] = newl
|