rhodecode/lib/markdown_ext.py

# -*- coding: utf-8 -*-

# Copyright (C) 2010-2016  RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/

import re

import markdown

from mdx_gfm import GithubFlavoredMarkdownExtension  # noqa


class FlavoredCheckboxExtension(markdown.Extension):

    def extendMarkdown(self, md, md_globals):
        md.preprocessors.add('checklist',
                             FlavoredCheckboxPreprocessor(md), '<reference')
        md.postprocessors.add('checklist',
                              FlavoredCheckboxPostprocessor(md), '>unescape')


class FlavoredCheckboxPreprocessor(markdown.preprocessors.Preprocessor):
    """
    Replaces occurrences of [ ] or [x] to checkbox input
    """

    pattern = re.compile(r'^([*-]) \[([ x])\]')

    def run(self, lines):
        return [self._transform_line(line) for line in lines]

    def _transform_line(self, line):
        return self.pattern.sub(self._replacer, line)

    def _replacer(self, match):
        list_prefix, state = match.groups()
        checked = '' if state == ' ' else ' checked="checked"'
        return '%s <input type="checkbox" disabled="disabled"%s>' % (list_prefix,
                                                                   checked)


class FlavoredCheckboxPostprocessor(markdown.postprocessors.Postprocessor):
    """
    Adds `flavored_checkbox_list` class to list of checkboxes
    """

    pattern = re.compile(r'^([*-]) \[([ x])\]')

    def run(self, html):
        before = '<ul>\n<li><input type="checkbox"'
        after = '<ul class="flavored_checkbox_list">\n<li><input type="checkbox"'
        return html.replace(before, after)


# Global Vars
URLIZE_RE = '(%s)' % '|'.join([
    r'<(?:f|ht)tps?://[^>]*>',
    r'\b(?:f|ht)tps?://[^)<>\s]+[^.,)<>\s]',
    r'\bwww\.[^)<>\s]+[^.,)<>\s]',
    r'[^(<\s]+\.(?:com|net|org)\b',
])


class UrlizePattern(markdown.inlinepatterns.Pattern):
    """ Return a link Element given an autolink (`http://example/com`). """
    def handleMatch(self, m):
        url = m.group(2)

        if url.startswith('<'):
            url = url[1:-1]

        text = url

        if not url.split('://')[0] in ('http','https','ftp'):
            if '@' in url and not '/' in url:
                url = 'mailto:' + url
            else:
                url = 'http://' + url

        el = markdown.util.etree.Element("a")
        el.set('href', url)
        el.text = markdown.util.AtomicString(text)
        return el


class UrlizeExtension(markdown.Extension):
    """ Urlize Extension for Python-Markdown. """

    def extendMarkdown(self, md, md_globals):
        """ Replace autolink with UrlizePattern """
        md.inlinePatterns['autolink'] = UrlizePattern(URLIZE_RE, md)