mdrelink.py

import sublime, sublime_plugin, math

def removeOphange(self, edit):
    links = []
    regions = self.view.find_all("(\[\&\d+\])[:]", sublime.IGNORECASE, "\\1", links)

    for i, r in enumerate(links):
        find = self.view.find_all(r, sublime.LITERAL)
        if len(find) == 1:
            line = self.view.line(regions[i])
            self.view.erase(edit, line)


def cleanup(self, edit):
    redundantNewLine = self.view.find_all("\[\&.+?\]\: [http|\/|\.\/|\~\/].+?\n{2,}\[.+?\]\: [http|\/|\.\/|\~\/].+?\n")

    trimNewLine = []
    self.view.find_all("(\[\&.+?\]\: [http|\/|\.\/|\~\/].+?\n)\n+(\[\&.+?\]\: [http|\/|\.\/|\~\/].+?\n)", sublime.IGNORECASE, "\\1\\2", trimNewLine)

    for i, r in enumerate(redundantNewLine):
        self.view.replace(edit, r, trimNewLine[i])

def reorderReferences(self, edit):
    references = []
    self.view.find_all("(\[\&\d+\])[^:]", sublime.IGNORECASE, "\\1", references)
    for i, ref in enumerate(references):
        regions = self.view.find_all(ref, sublime.LITERAL)
        regions.reverse()
        pad = int(math.floor(math.log(len(references), 10)))+1
        for r in regions:
            self.view.replace(edit, r, "[#{:0{padding}d}#]".format(i+1, padding = pad))

    # remove #
    hashedReferences = []
    hashedRegion = self.view.find_all("\[\#(\d+)\#\]", sublime.IGNORECASE, "[&\\1]", hashedReferences)
    hashedReferences.reverse()
    hashedRegion.reverse()
    for i, r in enumerate(hashedRegion):
        self.view.replace(edit, r, hashedReferences[i])

    # sort
    texts = []
    newRegions = self.view.find_all("^(\[\&\d+\]:.+?\n)", sublime.IGNORECASE, "\\1", texts)
    newRegions.reverse()
    for i, r in enumerate(newRegions):
        self.view.erase(edit, r)

    for address in sorted(texts):
        self.view.insert(edit, self.view.size(), address)

def formattedLinkReference(i, size):
    pad = int(math.floor(math.log(size, 10)))+1
    return "[&{:0{padding}d}]".format(i, padding = pad)


class mdrelinkCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        removeOphange(self, edit)

        oldlinks = []
        self.view.find_all("^\s*(\[\&\d+\]):", sublime.IGNORECASE, "\\1", oldlinks)

        # get all link location
        links = self.view.find_all("\[.+?\](\([http|\/|\.\/|\~\/].+?\))")

        # get link's text
        texts = []
        self.view.find_all("(\[.+?\])\([http|\/|\.\/|\~\/].+?\)", sublime.IGNORECASE, "\\1", texts)

        # get link's addresses
        addresses = []
        self.view.find_all("\[.+?\]\(([http|\/|\.\/|\~\/].+?)\)", sublime.IGNORECASE, "\\1", addresses)

        addresses.reverse()

        if len(links) > 0:
            self.view.insert(edit, self.view.size(), '\n\n')

        # large enough to prevent same link id
        counter = len(oldlinks)*100 + 1
        newnumbers = []

        for i, r in enumerate(links):
            formatRefLink = formattedLinkReference(counter, len(oldlinks))
            newnumbers.append(formatRefLink)
            line = formatRefLink + ': '+ addresses.pop() + '\n'
            self.view.insert(edit, self.view.size(), line)
            counter += 1

        for r in reversed(links):
            self.view.replace(edit, r, texts.pop() + newnumbers.pop())
            # self.view.replace(edit, r, texts.pop())

        # Cleanup
        cleanup(self, edit)
        reorderReferences(self, edit)