Modified Pandoc Converter Sublime Text Plugin

PandocConvertor.py

# A Sublime Text plugin to convert a Pandoc document into multiple formats:
#   - html
#   - docx
#   - pdf
#   - beamer
#
# The plug-in also deal with options embedded in the file to add more functions
#  such as a table of content, a bibliography or enumerate sections.
#
# Different bibliography styles and templates could be chosen.
#
# G.T. Vallet -- Lyon2 University
# 2012/06/24  -- GPLv3
# 2012/07/10  -- v.02 -- Always launch associated application and better syntax
# 2012/07/20  -- v.02.5 -- Fix the bug when pictures are included, convert PDF
#                           doesn't run. Need to change the path of the figures
# 2012/08/24  --  v.2.6 -- Search for paths before looking for default folders
# 2014/11/06  --  v.3.0 -- Convert the script to Python 3 / Sublime Text 3
#
# Inspired and cannibalized from Pandoc Renderer plugin
#  https://github.com/jclement/SublimePandoc

import sublime
import sublime_plugin
import subprocess
import webbrowser
import tempfile
import time
import sys
import os
import re


class PandocConvertorCommand(sublime_plugin.TextCommand):
    """ Convert a Pandoc file to HTML or Docx fileformat
        Options allow to add a table of contents, a bibliography,
          or to enumerate the sections.
    """
    def is_visible(self):
        return True

    def is_enabled(self):
        if self.view.score_selector(0, "text.pandoc") > 0 or \
            self.view.score_selector(0, "text.html.markdown") > 0 or \
            self.view.score_selector(0, "text.html.markdown.multimarkdown") > 0:
            return True

    def getTemplatePath(self, filename):
        path = os.path.join(sublime.packages_path(), 'PandocAcademic',
                                'Styles', filename)
        if os.path.isfile(path):
            print("Template file found in the default template folder")
        else:
            raise Exception(filename + " file not found!")
        return path

    def grabContent(self):
        """ A simple function to grab the content of the current buffer
        """
        region = sublime.Region(0, self.view.size())
        return self.view.substr(region)

    def testPath(self, style, target):
        """ Test if the provided style exists
              if not, try to search in the package Styles folder
        """
        if target != "beamer" and target != "pdf":
            default_path = self.getTemplatePath(style + "." + target)
            return default_path
        else:
            default_path = style
            if os.path.exists(style):
                style_path = style
            elif os.path.exists(default_path):
                style_path = default_path
            else:
                if target != "beamer":
                    sublime.error_message("Unable to find the style {0}.\
                            \n\nPlease check your path or name.".format(style))
                style_path = None
            return style_path

    def template(self, cmd, target, contents):
        """ A function to select the appropriate template
        """
        regex_docstyle = re.compile(r'\[\[DOCSTYLE=(\S+)\]\]')
        if regex_docstyle.search(contents):
            style = regex_docstyle.search(contents).groups()[0]
            style_path = self.testPath(style, target)
            if style_path:
                if target == 'html' or target == 'pdf':
                    cmd.append('--template=' + style_path)
                elif target == 'docx':
                    cmd.append('--reference-docx='\
                       + self.getTemplatePath(style_path))
            if target == 'beamer':
                cmd.append('-V')
                cmd.append('theme:' + style)
        else:
            if target == 'html':
              cmd.append('--template=' + self.getTemplatePath("template.html"))
            elif target == 'docx':
                cmd.append('--reference-docx='\
                        + self.getTemplatePath("apa.docx"))
        return cmd

    def opt(self, cmd, target, dir_path):
        """ A function to search for options embedded in the Pandoc file
              adding table of contents, bibliography and/or section number
        """
        # Grab the content of the current buffer
        contents = self.grabContent()

        # Check if a custom path is provided
        regex_path = re.compile(r'\[\[PATH=(.+)\]\]')
        regex_class = re.compile(r'\[\[CLASS=(\S+)\]\]')
        if regex_path.search(contents) != None:
            pandoc_path = regex_path.search(contents).groups()[0]
            cmd[0] = pandoc_path
        # Options embedded in the documents to turn on features in the output.
        if '[[TOC]]' in contents:
            cmd.append("--toc")
        if '[[NUM]]' in contents:
            cmd.append("-N")
        if '[[MATH]]' in contents:
            cmd.append("-m")
        if regex_class.search(contents):
            tex_class = regex_class.search(contents).groups()[0]
            cmd.append('-V')
            cmd.append('documentclass:' + tex_class)
        if '[[BIB]]' in contents:
            cmd.append("--bibliography")
            cmd.append(os.path.splitext(self.view.file_name())[0] + ".bib")
            # Style options
            regex_bibstyle = re.compile(r'\[\[BIBSTYLE=(\w+)\]\]')
            if regex_bibstyle.search(contents) != None:
                bibstyle = regex_bibstyle.search(contents).groups()[0]
                cmd.append("--csl")
                cmd.append(self.getTemplatePath(bibstyle + ".csl"))
        if '[[HEADER]]' in contents:
            cmd.append("--include-in-header="+os.path.split(self.view.file_name())[0] + "/header")
        if '[[NORENDER]]' in contents:
            openAfter = False
        else:
            # By default, launch the associated program of the file
            openAfter = True
        # Check for templates and styles options
        cmd = self.template(cmd, target, contents)
        # Adapt the path of the figures included (if any)
        re_img = "!\[.*\]\((.+)\)"  # Regex to find the figures
        for match in re.finditer(re_img, contents):
            contents = contents.replace(match.group(1), os.path.join(dir_path, match.group(1)))
        return cmd, openAfter, contents

    def buildCommand(self, target):
        """ A function to build the final Pandoc command to run
            Take the target and return the command and the output file
        """
        # Extract the filename
        file_name = self.view.file_name()
        filepath = os.path.splitext(file_name)[0]
        dir_path = os.path.split(file_name)[0]
        if not self.view.file_name(): raise Exception("Buffer must be saved!")
        if target == 'beamer':
            output_file = filepath + ".pdf"
        else:
            output_file = filepath + "." + target
        # Adding separated blocks of text to run the command in sublime text
        cmd = ['C:\\users\\James\\AppData\\Local\\Pandoc\\pandoc.exe', '--smart', '--standalone']
        # Check for options in the Pandoc file
        cmd, openAfter, contents = self.opt(cmd, target, dir_path)
        # Create a temporary file to handle the contents
        tmp_md = tempfile.NamedTemporaryFile(delete=False, suffix=".md")
        tmp_md.write(bytes(contents,'UTF-8'))
        tmp_md.close()
        # Complete the command function
        if target != 'pdf':
            cmd.append('-t')
            cmd.append(target)
        cmd.append(tmp_md.name)
        cmd.append("-o")
        cmd.append(output_file)
        return cmd, output_file, openAfter, dir_path

    def status(self, filename):
        """ Update the status monitor of sublime text
        """
        if os.path.exists(filename):
            self.view.set_status('pandoc', 'File converted to ' + filename)
            print("File converted to:", filename)
        else:
            self.view.set_status('pandoc', 'Error in the conversion!')
            print("Unable to convert the file! Please check your options")
        time.sleep(2)
        self.view.erase_status('pandoc')

    def run(self, edit, target="html"):
        # Call the build command function to construct the final pandoc command
        cmd, output_filename, openAfter, dir_path = self.buildCommand(target)
        # Run the main command to convert the file
        try:
            subprocess.call(cmd)
            subprocess.check_output(['DocxMerge', '-i', dir_path + '\\..\\pandoc_cover_template.docx', output_filename, '-o', dir_path + '\\output.docx', '-f'])
            subprocess.check_output(['mv', dir_path + '\\output.docx', '-f', output_filename])
        except Exception as e:
            sublime.error_message("Unable to execute Pandoc.\
                                    \n\nDetails: {0}".format(e))
        self.status(output_filename)

        # View the converted file in the associated program
        if openAfter:
            if target == "html":
                webbrowser.open_new_tab(output_filename)
            elif target != "html" and sys.platform == "win32":
                os.startfile(output_filename)
            elif target != "html" and sys.platform == "mac":
                subprocess.call(["open", output_filename])
            else:
                subprocess.call(["xdg-open", output_filename])