Main.py

# -*- coding: UTF-8 -*-

import Recognizer
import Translator
from sys import exit

fileName = "test.md"

headers = (
    "\\documentclass{ctexart}",
    "",
    "\\usepackage[hmargin=1.25in,vmargin=1in]{geometry}",
    "\\usepackage{graphicx}",
    "\\usepackage{float}",
    "\\usepackage{CJKulem}",
    ""
)

docHeaders = (
    "\\maketitle",
    "",
    "\\tableofcontents",
    "",
    "\\thispagestyle{empty}",
    "\\newpage",
    "\\pagestyle{plain}",
    "\\setcounter{page}{1}",
    ""
)

title = ""
author = ""


def makeLine(line):
    for i in range(line):
        print()
        latexFile.write('\n')


def fPrintln(*args):
    word = ''.join(map(str, args))
    word = (max(len(stack) - 1, 0) + sectionLevel) * "    " + word
    print(word)
    latexFile.write(word)
    latexFile.write('\n')


class MDFile:
    data = ""
    point = -1

    def back(self, line=1):
        self.point -= line

    def remain(self):
        if self.point + 1 >= len(self.data):
            return False
        else:
            return True

    def fReadln(self):
        self.point += 1
        if self.point >= len(self.data):
            return ""
        return self.data[self.point]

    def fReadNextln(self):
        self.point += 1
        if self.point >= len(self.data):
            return ""
        while self.data[self.point] == "":
            self.point += 1
            if self.point >= len(self.data):
                return ""
        return self.data[self.point]


stack = []
sectionLevel = 0


def newEnv(env, pro=""):
    if pro == "":
        fPrintln("\\begin{", env, '}')
    else:
        fPrintln("\\begin{", env, "}", pro)
    stack.append(env)


def endEnv(end=""):
    if end == "":
        fPrintln("\\end{", stack.pop(), '}')
        makeLine(1)
    elif end != stack.pop():
        print("Error")
    else:
        fPrintln("\\end{", end, '}')
        makeLine(1)


def compositorPrinter(LaTeX_SP):
    for line in LaTeX_SP:
        if line[0] == "Cmd":
            fPrintln(line[1])
        elif line[0] == "Env":
            if len(line[1]) == 1:
                endEnv()
            else:
                newEnv(line[1][0], line[1][1])
        elif line[0] == "Text":
            fPrintln(line[1])


def process():
    def specialProcesser(tokens, preEnv):
        buffer = [tokens]
        while buffer[len(buffer) - 1] != "":
            buffer.append(md.fReadln())
        if preEnv == "HTML":
            parse = Recognizer.parser(buffer, preEnv)
            LaTeX_ST = Translator.trans(parse)
            compositorPrinter(LaTeX_ST)

    temp = md.fReadln()
    if temp == "":
        makeLine(1)
        return
    env = Recognizer.envRecognizer(temp.split())
    if env[0] == "Special":
        specialProcesser(temp, env[1])
    elif env[0] == "Section":
        global sectionLevel
        sectionLevel = env[1] - 1
        if env[1] == 1:
            fPrintln("\\section{", ' '.join(temp.split()[1:]), "}")
        elif env[1] == 2:
            fPrintln("\\subsection{", ' '.join(temp.split()[1:]), "}")
        elif env[1] == 3:
            fPrintln("\\subsubsection{", ' '.join(temp.split()[1:]), "}")
        sectionLevel += 1
    elif env[0] == "PlainText":
        fPrintln(Translator.textRenderer(temp))
    elif env[0] == "Part":
        sectionLevel = 0
        fPrintln("\\part{", ' '.join(temp.split()[1:]), "}")


if __name__ == '__main__':
    if fileName == "":
        fileName = input()
    if fileName[len(fileName) - 3:] != '.md':
        print("请提供一个.md文件")
    else:
        fileName = fileName[:len(fileName) - 3]
    try:
        mdFile = open(fileName + '.md', "r", encoding="utf-8")
        mdData = mdFile.read().splitlines()
        mdFile.close()
        md = MDFile()
        md.data = mdData
    except IOError:
        print("找不到文件")
        exit()
    latexFile = open(fileName + '.tex', "w", encoding="utf-8")
    for ele in headers:
        fPrintln(ele)
    title = md.fReadln().split()
    if title[0] == '#':
        title = ' '.join(title[1:len(title)])
    else:
        title = ' '.join(title)
    author = md.fReadNextln()
    fPrintln("\\title{", title, "}")
    fPrintln("\\author{", author, '}')
    makeLine(1)
    fPrintln("\\date{\\today}")
    makeLine(2)

    newEnv("document")
    for ele in docHeaders:
        fPrintln(ele)

    while md.remain():
        process()

    sectionLevel = 0
    while stack:
        endEnv()
    latexFile.close()