boards/mdx_neboard.py

from django.core.urlresolvers import reverse
import markdown
from markdown.inlinepatterns import Pattern
from markdown.util import etree
import boards

__author__ = 'neko259'


AUTOLINK_PATTERN = r'(https?://\S+)'
QUOTE_PATTERN = r'^(?<!>)(>[^>].+)$'
REFLINK_PATTERN = r'((>>)(\d+))'
SPOILER_PATTERN = r'%%(.+)%%'
COMMENT_PATTERN = r'^(//(.+))'

class AutolinkPattern(Pattern):
    def handleMatch(self, m):
        link_element = etree.Element('a')
        href = m.group(2)
        link_element.set('href', href)
        link_element.text = href

        return link_element


class QuotePattern(Pattern):
    def handleMatch(self, m):
        quote_element = etree.Element('span')
        quote_element.set('class', 'quote')
        quote_element.text = m.group(2)

        return quote_element


class ReflinkPattern(Pattern):
    def handleMatch(self, m):
        post_id = m.group(4)

        posts = boards.models.Post.objects.filter(id=post_id)
        if posts.count() > 0:
            ref_element = etree.Element('a')

	    post = posts[0]
            if post.thread:
                link = reverse(boards.views.thread, kwargs={'post_id': post.thread.id}) \
                                               + '#' + post_id
            else:
		link = reverse(boards.views.thread, kwargs={'post_id': post_id})

            ref_element.set('href', link)
            ref_element.text = m.group(2)

        return ref_element


class SpoilerPattern(Pattern):
    def handleMatch(self, m):
        quote_element = etree.Element('span')
        quote_element.set('class', 'spoiler')
        quote_element.text = m.group(2)

        return quote_element


class CommentPattern(Pattern):
    def handleMatch(self, m):
        quote_element = etree.Element('span')
        quote_element.set('class', 'comment')
        quote_element.text = '//' + m.group(3)

        return quote_element


class NeboardMarkdown(markdown.Extension):
    def extendMarkdown(self, md, md_globals):
        self._add_neboard_patterns(md)
        self._delete_patterns(md)

    def _delete_patterns(self, md):
        del md.parser.blockprocessors['quote']

        del md.inlinePatterns['image_link']
        del md.inlinePatterns['image_reference']

    def _add_neboard_patterns(self, md):
        autolink = AutolinkPattern(AUTOLINK_PATTERN, md)
        quote = QuotePattern(QUOTE_PATTERN, md)
        reflink = ReflinkPattern(REFLINK_PATTERN, md)
        spoiler = SpoilerPattern(SPOILER_PATTERN, md)
        comment = CommentPattern(COMMENT_PATTERN, md)

        md.inlinePatterns[u'autolink_ext'] = autolink
        md.inlinePatterns[u'spoiler'] = spoiler
        md.inlinePatterns[u'comment'] = comment
        md.inlinePatterns[u'reflink'] = reflink
        md.inlinePatterns[u'quote'] = quote


def makeExtension(configs=None):
    return NeboardMarkdown(configs=configs)

neboard_extension = makeExtension()


def markdown_extended(markup):
    return markdown.markdown(markup, [neboard_extension], safe_mode=True)