#!/usr/bin/env python
# encoding: utf-8

from glob import glob
import os
import re
import sys

###################
# inputs, outputs #
###################
cpp_pairs = (
    ("../lua_root.cc", "source/wl.rst"),
    ("../lua_bases.cc", "source/wl_bases.rst"),
    ("../lua_editor.cc", "source/wl_editor.rst"),
    ("../lua_map.cc", "source/wl_map.rst"),
    ("../lua_game.cc", "source/wl_game.rst"),
    ("../lua_ui.cc", "source/wl_ui.rst"),
    ("../lua_globals.cc", "source/globals.rst"),
)

def extract_rst_from_cpp(inname, outname=None):
    """
    Searches for /* RST comments in the given filename, strips the lines
    and prints them out on stdout or writes them to outname.
    """
    data = open(inname, "r").read()

    res = re.findall(r"\s*/\* RST\s(.*?)\*/", data, re.M | re.DOTALL)

    output = ""
    for r in res:
        r = r.expandtabs(4)
        output += r + '\n'

    if output.strip():
        out = sys.stdout if not outname else open(outname, "w")
        out.write(output)

def extract_rst_from_lua(inname):
    """
    Searches for /* RST comments in the given filename, strips the lines
    and prints them out on stdout or writes them to outname.
    """
    data = open(inname, "r").read()

    res = re.findall(r"-- RST\s(.*?)(?:^(?!--))", data, re.M | re.DOTALL)

    outname = "source/aux_%s.rst" %os.path.basename(os.path.splitext(inname)[0])

    output = ""
    for r in res:
        r = re.subn(re.compile(r'^-- ?', re.M | re.DOTALL), "", r)[0]
        output += r + '\n'

    if output.strip():
        out = sys.stdout if not outname else open(outname, "w")
        out.write(output)
        return os.path.basename(outname)

def replace_auxilary_toc(aux_fns):
    aux_in = open("source/auxiliary.rst.in", "r").read()
    aux_in = aux_in.replace("REPLACE_ME",
                '\n'.join('   ' + fn for fn in aux_fns))
    open("source/auxiliary.rst", "w").write(aux_in)

if __name__ == '__main__':
    def main():
        for inf, outf in cpp_pairs:
            extract_rst_from_cpp(inf, outf)

        replace_auxilary_toc(
            filter(lambda a: a, [extract_rst_from_lua(i) for i in glob("../../../scripting/*.lua")])
        )

    main()