import os
import shutil
import re
import pydoc
import sys
import subprocess

def mk_dir(d):
    if not os.path.exists(d):
        os.makedirs(d)

# Eliminate def_API and extra_API directives from file 'inf'.
# The result is stored in 'outf'.
def cleanup_API(inf, outf):
    pat1  = re.compile(".*def_API.*")
    pat2  = re.compile(".*extra_API.*")
    _inf  = open(inf, 'r')
    _outf = open(outf, 'w') 
    for line in _inf:
        if not pat1.match(line) and not pat2.match(line):
            _outf.write(line)

try:
    mk_dir('html')
    cleanup_API('../src/api/z3_api.h', 'z3_api.h')
    print "Removed annotations from z3_api.h."
    DEVNULL = open(os.devnull, 'wb')
    if subprocess.call(['doxygen', 'z3.dox'], stdout=DEVNULL, stderr=DEVNULL) != 0:
        print "ERROR: failed to execute doxygen z3.dox"
        exit(1)
    print "Generated C and .NET API documentation."
    os.remove('z3_api.h')
    print "Removed temporary file z3_api.h."
    shutil.copy('z3.css', 'html/z3.css')
    print "Copied z3.css."
    shutil.copy('z3.png', 'html/z3.png')
    print "Copied z3.png."
    sys.path.append('../src/api/python')
    pydoc.writedoc('z3')
    shutil.move('z3.html', 'html/z3.html')
    print "Generated Python documentation."
    print "Documentation was successfully generated at subdirectory './html'."
except:
    print "ERROR: failed to generate documentation"
    exit(1)