aboutsummaryrefslogtreecommitdiffstats
path: root/pysite/views/staff/render.py
blob: 00c9a9f33d131e1d31824ab868d6652cf3b6a4ac (plain) (blame)
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
import re

from docutils.utils import SystemMessage
from flask import jsonify
from schema import Schema

from pysite.base_route import APIView
from pysite.constants import EDITOR_ROLES, ValidationTypes
from pysite.decorators import api_params, csrf, require_roles
from pysite.rst import render

SCHEMA = Schema([{
    "data": str
}])

MESSAGE_REGEX = re.compile(r"<string>:(\d+): \([A-Z]+/\d\) (.*)")


class RenderView(APIView):
    path = "/render"  # "path" means that it accepts slashes
    name = "render"

    @csrf
    @require_roles(*EDITOR_ROLES)
    @api_params(schema=SCHEMA, validation_type=ValidationTypes.json)
    def post(self, data):
        if not len(data):
            return jsonify({"error": "No data!"})

        data = data[0]["data"]
        try:
            html = render(data, link_headers=False)["html"]

            return jsonify({"data": html})
        except SystemMessage as e:
            lines = str(e)
            data = {
                "error": lines,
                "error_lines": []
            }

            if "\n" in lines:
                lines = lines.split("\n")
            else:
                lines = [lines]

            for message in lines:
                match = MESSAGE_REGEX.match(message)

                if match:
                    data["error_lines"].append(
                        {
                            "row": int(match.group(1)) - 3,
                            "column": 0,
                            "type": "error",
                            "text": match.group(2)
                        }
                    )

            print(data)
            return jsonify(data)
        except Exception as e:
            return jsonify({"error": str(e)})