Website/article/adapt.py

import re


def adapt(text, title, layout):
    text = correct_assets(text, layout)
    text = add_load_static(text)
    text = add_meta_data(text)
    text = correct_title(text, title)
    text = add_icons(text)
    text = add_back_to_articles(text)
    # text = correct_links(text)
    # text = correct_scripts(text)
    text = add_mathjax(text)
    text = add_plantuml(text)
    text = add_back_to_top_button(text)
    text = add_scroll_support(text)
    text = correct_check(text)
    return text


def add_load_static(text):
    """Add load static on top of the file."""
    return "{% load static %}" + text


def add_back_to_articles(text):
    text = text.replace(
        '<div id="toc-inner">',
        '<div id="toc-inner">\n\
        <a class="back" class="container" href="{% url \'article\' %}"><p>Back to articles</p></a><div class="fade_rule"></div>',
    )
    return text


def correct_title(text, title):
    """Put the true title."""
    pattern = re.compile(r"<title>(.*)</title>")
    wrong_title = re.findall(pattern, text)[0]
    text = text.replace(
        f"<title>{wrong_title}</title>", f"<title>{title.capitalize()}</title>"
    )
    return text


def add_icons(text):
    """Add orchid icons in tabs."""
    text = text.replace(
        "</title>",
        '</title>\n\
        <link href="{% static \'home/assets/img/white-orchid.svg\' %}" rel="icon">\n\
        <link href="{% static \'home/assets/img/black-orchid.svg\' %}" rel="apple-touch-icon">\n\
        <link href="{% static \'article/assets/css/main.css\' %}" rel="stylesheet">',
    )
    return text


def correct_links(text):
    """Correct import links."""
    pattern = re.compile(r"href=.assets\/(.*\.css)")
    for file in re.findall(pattern, text):
        print(file)
        text = text.replace(
            f'href="assets/{file}"',
            "href=\"{% static 'article/assets/" + file + "' %}\"",
        )
    return text


def correct_scripts(text):
    """Correct import scripts."""
    pattern = re.compile(r'src="assets\/(.*\.js)"')
    for file in re.findall(pattern, text):
        text = text.replace(
            "</body>",
            "<script src=\"{% static 'article/assets/"
            + file
            + "' %}\"></script>\n\
        </body>",
        )
    return text


def correct_assets(text, layout):
    """Correct assets imports."""
    # pattern = re.compile(r'([src|href].?=.?[\'|"]assets\/.*\.[js|css].?[\'|"])')
    pattern = re.compile(r"(assets/.*\.[jc]ss?)")
    for match in re.findall(pattern, text):
        substitute = match.replace("assets", layout)
        substitute = "{% static 'article/" + substitute + "' %}"
        text = text.replace(match, substitute)
    return text


def add_mathjax(text):
    """Add mathjax support for latex formulaes rendering."""
    text = text.replace(
        "</head>",
        '<script src="https://cdn.mathjax.org/mathjax/latest/MathJax.js?config=TeX-MML-AM_CHTML"></script>\n\
        </head>',
    )
    return text


def add_plantuml(text):
    """Render plantuml diagrams."""
    text = text.replace(
        "</head>",
        '<script src="https://cdn.jsdelivr.net/npm/node-plantuml@0.9.0/index.min.js"></script>\n\
        </head>',
    )
    return text


def add_scroll_support(text):
    """Add scroll support."""
    text = text.replace(
        "</body>",
        "<script src=\"{% static 'home/assets/vendor/jquery/jquery.min.js' %}\"></script>\n\
        <script src=\"{% static 'home/assets/vendor/waypoints/jquery.waypoints.min.js' %}\"></script>\n\
        <script src=\"{% static 'article/assets/js/main.js' %}\"></script>\n\
    </body>",
    )
    return text


def correct_check(text):
    """Replace markdown check to checked buttons."""
    text = text.replace("[ ]", '<input type="checkbox">')
    text = text.replace("[x]", '<input type="checkbox" checked="">')
    return text


def add_back_to_top_button(text):
    """Add back to top button."""
    text = text.replace(
        "</body>",
        '</body>\n<a href="#" class="back-to-top" style="display: inline; \
        position:fixed; right:15px; bottom:15px; z-index:99999;"> \
        <i style="display: flex; align-items: center; justify-content: center; \
        font-size: 24px; width: 40px; height: 40px; border-radius: 4px; \
        background: #7cc576; color: #fff; transition: all 0.4s;" \
        class="icofont-simple-up"></i></a>',
    )
    return text


def add_meta_data(text):
    """Add a like button for an article."""
    text = text.replace(
        "<body>",
        """<body>
        <div style="clear: both;">
            <div style="float: left">
                {{ view_count }}
                <span class="iconify" data-icon="carbon:view" data-inline="false"></span>
                {{ like_count }}
                <span class="iconify" data-icon="fluent-thumb-like-16-regular" data-inline="false"></span>
                {{ dislike_count }}
                <span class="iconify" data-icon="fluent:thumb-dislike-24-regular"data-inline="false"></span>
                <script src="https://code.iconify.design/1/1.0.7/iconify.min.js"></script>
            </div>
            <div style="float: right">
                <small style="text-align: right">{{ created }}</small>
                {% if update_count > 0 %}
                <small> V{{ update_count }}</small>
                {% endif %}
            </div>
        </div>
        """,
    )
    # text = text.replace('</body>',
    #     """
    #     <hr>
    #     created: {{ updated }}<br>
    #     </body>
    #     """)
    return text