k3down2/syntax_highlight.py at master · pykit3/k3down2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
from pygments import highlight
from pygments.lexers import get_lexer_by_name
from pygments.formatters import HtmlFormatter
from pygments.style import Style
from pygments.token import (
    Keyword,
    Name,
    Comment,
    String,
    Error,
    Text,
    Number,
    Operator,
    Punctuation,
    Literal,
)


base00 = "#263238"
base01 = "#2e3c43"
base02 = "#314549"
base03 = "#546e7a"
base04 = "#b2ccd6"
base05 = "#eeffff"
base06 = "#eeffff"
base07 = "#ffffff"
base08 = "#f07178"
base09 = "#f78c6c"
base0a = "#ffcb6b"
base0b = "#c3e88d"
base0c = "#89ddff"
base0d = "#82aaff"
base0e = "#c792ea"
base0f = "#ff5370"


class Base16Style(Style):
    background_color = base00
    highlight_color = base02
    default_style = base05

    styles = {
        Text: base05,
        Error: base08,
        Comment: base03,
        Keyword: base0e,
        Keyword.Type: base0a,
        Keyword.Constant: base0d,
        Keyword.Namespace: base0d,
        Name: base05,
        Name.Builtin: base0a,
        Name.Function: base0d,
        Name.Class: base0d,
        Name.Decorator: base0e,
        Name.Exception: base08,
        Number: base09,
        Operator: base0d,
        Operator.Word: base0d,
        Punctuation: base05,
        Literal: base0b,
        String: base0b,
    }


def escape(s, quote=True):
    s = s.replace("&", "&amp;")
    s = s.replace("<", "&lt;")
    s = s.replace(">", "&gt;")
    if quote:
        s = s.replace('"', "&quot;")
    return s


def code_to_html(text):
    lines = text.strip().split("\n")

    # extract code language: ```go
    lang = lines[0][3:].strip()
    text = "\n".join(lines[1:-1])

    linenos = False
    style = Base16Style

    lineheight = "1.8"

    if lang == "":
        # For illustration text graph, use lower line height.
        # Becuase in such case it prefers an easy to read digram than a
        # comforable line height.
        lineheight = "1.3"

    prestyles = (
        r"line-height: {} !important;"
        " margin: 0 !important;"
        " padding: 1em;"
        " white-space: pre-wrap;"
        " background: {};"
        " color: {};"
    ).format(
        lineheight,
        style.background_color,
        style.default_style,
    )

    if lang == "":
        text = text.strip()
        return '<pre style="%s"><code>%s</code></pre>\n' % (prestyles, escape(text))

    try:
        lexer = get_lexer_by_name(lang, stripall=True)
        formatter = HtmlFormatter(noclasses=True, linenos=linenos, style=style, prestyles=prestyles)
        code = highlight(text, lexer, formatter)
        if linenos:
            return '<div class="highlight-wrapper">%s</div>\n' % code
        return code
    except Exception:
        return '<pre class="%s"><code>%s</code></pre>\n' % (lang, escape(text))