import markdown
from markdown.inlinepatterns import LinkPattern
from markdown.util import etree

IMAGES_RE = r"\!(\s?)\((<.*?>|([^\)]*))\)"


class ShortImagesExtension(markdown.Extension):
    def extendMarkdown(self, md):
        md.registerExtension(self)
        md.inlinePatterns.add(
            "misago_short_images", ShortImagePattern(IMAGES_RE, md), "_end"
        )


class ShortImagePattern(LinkPattern):
    def handleMatch(self, m):
        img_src = m.groups()[2].strip()
        if img_src:
            img = etree.Element("img")
            img.set("src", img_src)
            img.set("alt", img_src)
            return img