from os.path import join, split, dirname
import os
import sys
from distutils.dep_util import newer
from numpy.distutils.misc_util import msvc_runtime_library

def msvc_version():
    """Return the msvc version used to build the running python, None if not
    built with MSVC."""
    msc_pos = sys.version.find('MSC v.')
    if msc_pos != -1:
        return sys.version[msc_pos+6:msc_pos+10]
    return None

def msvcrt_to_hex(msvc):
    major = msvc / 10
    minor = msvc - major * 10
    return hex(major * 256 + minor)

def configuration(parent_package='',top_path=None):
    from numpy.distutils.misc_util import Configuration, get_mathlibs
    config = Configuration('random',parent_package,top_path)

    def generate_libraries(ext, build_dir):
        config_cmd = config.get_config_cmd()
        libs = get_mathlibs()
        tc = testcode_wincrypt()
        if config_cmd.try_run(tc):
            libs.append('Advapi32')
        ext.libraries.extend(libs)
        return None

    def generate_config_h(ext, build_dir):
        defs = []
        target = join(build_dir, "mtrand", 'config.h')
        dir = dirname(target)
        if not os.path.exists(dir):
            os.makedirs(dir)

        msv = msvc_version()
        if msv and msv >= 1400:
            msvcrt = msvc_runtime_library()
            if msvcrt is None:
                raise ValueError("Discrepancy between " \
                                 "msvc_runtime_library " \
                                 "and our msvc detection scheme ?")
            hmsvc = msvcrt_to_hex(int(msvcrt[5:]))
            defs.append("NPY_NEEDS_MINGW_TIME_WORKAROUND")
            defs.append(("NPY_MSVCRT_VERSION", str(hmsvc)))

        if newer(__file__, target):
            target_f = open(target, 'a')
            for d in defs:
                if isinstance(d, str):
                    target_f.write('#define %s\n' % (d))
            target_f.close()

    libs = []
    # Configure mtrand
    config.add_extension('mtrand',
                         sources=[join('mtrand', x) for x in
                                  ['mtrand.c', 'randomkit.c', 'initarray.c',
                                   'distributions.c']]+[generate_libraries]
                                   + [generate_config_h],
                         libraries=libs,
                         depends = [join('mtrand','*.h'),
                                    join('mtrand','*.pyx'),
                                    join('mtrand','*.pxi'),
                                    ]
                        )

    config.add_data_files(('.', join('mtrand', 'randomkit.h')))
    config.add_data_dir('tests')

    return config

def testcode_wincrypt():
    return """\
/* check to see if _WIN32 is defined */
int main(int argc, char *argv[])
{
#ifdef _WIN32
    return 0;
#else
    return 1;
#endif
}
"""

if __name__ == '__main__':
    from numpy.distutils.core import setup
    setup(configuration=configuration)