import os

def extraire_caml(nf_tex, nf_ml, nf_texo, tag) :
    nf_tex = nf_tex.replace("file://", "")
    nf_ml = nf_ml.replace("file://", "")
    nf_texo = nf_texo.replace("file://", "")
    output_tex = open(nf_texo, "w")
    output_ml = open(nf_ml, "w")
    incode = False
    nomfonc = ""
    complement = None
    numfoncAnon = 0
    mlref = os.path.basename(nf_ml)
    for l in open(nf_tex) :
        lstrip = l.strip()
        if lstrip.startswith(r"\begin{"+tag+"}") :
            incode = True
            nomfonc = ""
            complement = None
            if "[" in lstrip :
                complement = lstrip.split("[")[1].split("]")[0]
            continue
        if incode :
            if nomfonc == "" :
                if lstrip == "" :
                    continue
                if "let" in lstrip :
                    nomfonc = l[4:]
                    if nomfonc.startswith("rec ") :
                        nomfonc = nomfonc[4:]
                    nomfonc = nomfonc.split()[0]
                else :
                    nomfonc = "anon" + str(numfoncAnon)
                    numfoncAnon += 1
                output_ml.write("(*** " + nomfonc + " ***)\n")
            if lstrip == r"\end{"+tag+"}" :
                output_ml.write("(*** " + nomfonc + "_fin ***)\n\n")
                if complement is not None :
                    output_tex.write(r"\inputcaml["+nomfonc+"]{"+ mlref + "}["+complement+"]\n")
                else :
                    output_tex.write(r"\inputcaml["+nomfonc+"]{"+ mlref + "}\n")
                incode = False
                continue
            output_ml.write(l)
        else :
            output_tex.write(l)

def caml(s, start=0) :
    try :
        pos_let = s.index("\nlet", start)
        fin = s.index(r";;", pos_let)
        print(s[pos_let:fin+3])
        if s[pos_let+4:].startswith(" rec") :
            delta = 5
        else :
            delta = 0
        pos_nom = s.index(" ", pos_let + delta) + 1
        fin_nom = s.index(" ", pos_nom)
        nom = s[pos_nom:fin_nom]
        print(nom)
        news = s[:pos_let] + "(*** " + nom + " ***)" + s[pos_let:fin+3] + "(*** " + nom + "_fin ***)\n" + s[fin+3:]
        return caml(news, fin + 3 + len("(*** " + nom + " ***)")+ len("(*** " + nom + "_fin ***)\n"))
    except ValueError :
        return s