import os, sys
import time
import multiprocessing, threading
import logging
import turtle

# current file's path
fpath = os.path.dirname(os.path.abspath(__file__))

def extname(fname):
    fa = os.path.splitext(fname)
    if len(fa) > 1:
        return fa[1].lower()
    else:
        return ""


def filename(fname):
    return os.path.split(fname)[1]


def loadModule(mpath):
    # get module's name
    moduleName = os.path.splitext(os.path.split(mpath)[1])[0]

    # try load module
    try:
        # for Python 3.5+
        import importlib.util
        spec = importlib.util.spec_from_file_location(moduleName, mpath)
        module = importlib.util.module_from_spec(spec)
        spec.loader.exec_module(module)

        return module
    except Exception as e:
        logging.error("Can not load module %s, error msg: %s" % (mpath, str(e)))
        return None


def getScriptList(fp):
    fl = os.listdir(fp)
    fileList = [f for f in fl if extname(f) == ".py"]
    fileList.sort()
    return fileList


class RunScript:
    def __init__(self, fname, txtfile):
        self.p = None
        self.fname = fname
        self.txtfile = txtfile

        self.finished = False

    def runScript_(self):
        mod = loadModule(self.fname)
        if mod:
            # create
            ts = threading.Thread(target=self.saveRes)
            ts.start()

            # call `draw` function
            print(">>> load module: %s" % filename(self.fname))
            f = getattr(mod, "draw")
            f()

            # regist timer function, and do mainloop
            turtle.ontimer(self.drawFinished, 0)
            turtle.mainloop()

    def drawFinished(self):
        print("- draw finished!")
        self.finished = True

    def saveRes(self):
        while not self.finished:
            time.sleep(0.5)

        time.sleep(5)

        # do snapshot
        imgPath = os.path.join(fpath, "images")
        fbaseName = os.path.splitext(os.path.split(self.fname)[1])[0]
        imgName = fbaseName + ".png"
        imgFP = os.path.join(imgPath, imgName)

        cmd = 'shutter -e --min_at_startup --window="Python Turtle Graphics" -o "%s"' % imgFP
        print("cmd: ", cmd)
        os.system(cmd)

        # do markdown text output
        cfname = "codes/" + os.path.split(self.fname)[1]
        imgname = "images/" + imgName
        mt = """code: [%s](%s)\n\n![%s](%s)\n\n\n""" % (fbaseName, cfname, fbaseName, imgname)

        f = open(self.txtfile, "at")
        f.write(mt)
        f.close()

        # close drawing window
        turtle.bye()

        print("- capture finished!")

    def runScript(self):
        self.p = multiprocessing.Process(name="run_script", target=self.runScript_)
        self.p.start()
        self.p.join()

        print("- script run finished!\n\n")

if __name__ == "__main__":
    # code path
    codePath = os.path.join(fpath, "codes")
    mtfile = os.path.join(fpath, "gallery.md")

    # get code list
    if len(sys.argv) > 1:
        cl = sys.argv[1:]
    else:
        cl = getScriptList(codePath)

    print("code list: ", cl)
    print("")

    # run codes
    for c in cl:
        s = RunScript(os.path.join(codePath, c), mtfile)
        s.runScript()