forked from extern/httpie-cli
32 lines
851 B
Python
32 lines
851 B
Python
import re
|
|
|
|
from pygments.lexer import bygroups, using, RegexLexer
|
|
from pygments.lexers.data import JsonLexer
|
|
from pygments.token import Token
|
|
|
|
PREFIX_TOKEN = Token.Error
|
|
PREFIX_REGEX = r'[^{\["]+'
|
|
|
|
|
|
class EnhancedJsonLexer(RegexLexer):
|
|
"""
|
|
Enhanced JSON lexer for Pygments.
|
|
|
|
It adds support for eventual data prefixing the actual JSON body.
|
|
|
|
"""
|
|
name = 'JSON'
|
|
flags = re.IGNORECASE | re.DOTALL
|
|
tokens = {
|
|
'root': [
|
|
# Eventual non-JSON data prefix followed by actual JSON body.
|
|
# FIX: data prefix + number (integer or float) is not correctly handled.
|
|
(
|
|
fr'({PREFIX_REGEX})' + r'((?:[{\["]|true|false|null).+)',
|
|
bygroups(PREFIX_TOKEN, using(JsonLexer))
|
|
),
|
|
# JSON body.
|
|
(r'.+', using(JsonLexer)),
|
|
],
|
|
}
|