forked from extern/httpie-cli
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
import json
|
|
from functools import partial
|
|
import pygments
|
|
from pygments.lexers import get_lexer_for_mimetype
|
|
from pygments.formatters.terminal256 import Terminal256Formatter
|
|
from pygments.lexer import RegexLexer, bygroups
|
|
from pygments import token
|
|
|
|
|
|
TYPE_JS = 'application/javascript'
|
|
|
|
|
|
class HTTPLexer(RegexLexer):
|
|
name = 'HTTP'
|
|
aliases = ['http']
|
|
filenames = ['*.http']
|
|
tokens = {
|
|
'root': [
|
|
(r'\s+', token.Text),
|
|
(r'(HTTP/[\d.]+\s+)(\d+)(\s+.+)', bygroups(
|
|
token.Operator, token.Number, token.String)),
|
|
(r'(.*?:)(.+)', bygroups(token.Name, token.String))
|
|
]}
|
|
|
|
|
|
highlight = partial(pygments.highlight,
|
|
formatter=Terminal256Formatter(style='native'))
|
|
highlight_http = partial(highlight, lexer=HTTPLexer())
|
|
|
|
|
|
def prettify_http(headers):
|
|
return highlight_http(headers)[:-1]
|
|
|
|
|
|
def prettify_body(content, content_type):
|
|
content_type = content_type.split(';')[0]
|
|
if 'json' in content_type:
|
|
content_type = TYPE_JS
|
|
try:
|
|
# Indent JSON
|
|
content = json.dumps(json.loads(content),
|
|
sort_keys=True, indent=4)
|
|
except Exception:
|
|
pass
|
|
try:
|
|
lexer = get_lexer_for_mimetype(content_type)
|
|
content = highlight(code=content, lexer=lexer)
|
|
if content:
|
|
content = content[:-1]
|
|
except Exception:
|
|
pass
|
|
return content
|
|
|