#!/c/python27/python

import os

from utils import *

def cli_cpp(parms):
    return os.path.join(parms['OVPN3'], "core", "test", "ovpncli", "cli.cpp")

def src_fn(parms, srcfile):
    # Get source file name
    if srcfile:
        if '.' not in os.path.basename(srcfile):
            srcfile += ".cpp"
    else:
        srcfile = cli_cpp(parms)
    return srcfile

def is_unit_test(argv):
    unit_test = False
    if len(argv) >= 2:
        unit_test = argv[1] == "unittest"
    return unit_test

def src_fn_argv(parms, argv):
    srcfile = None
    if len(argv) >= 1:
        srcfile = argv[0]
    return src_fn(parms, srcfile)

def build(parms, srcfile, unit_test=False):
    # Debug?
    if parms['DEBUG']:
        dbg_rel_flags = "/Zi"
    else:
        dbg_rel_flags = "/O2"

    # Dictionary we will use to substitute parameters
    # onto VC command line.
    options = {
        "ovpn3"   : parms['OVPN3'],
        "tap" : os.path.join(build_dir(parms), "tap-windows", "src"),
        "tap_component_id" : parms['TAP_WIN_COMPONENT_ID'],
        "asio"    : os.path.join(build_dir(parms), "asio"),
        "mbedtls" : os.path.join(build_dir(parms), "mbedtls"),
        "openssl" : os.path.join(build_dir(parms), "openssl"),
        "lz4" : os.path.join(build_dir(parms), "lz4", "lib"),
        "srcfile" : srcfile,
        "extra_defs" : parms['CPP_EXTRA'],
        "extra_inc" : "",
        "extra_lib_path" : "",
        "extra_lib" : "",
    }

    vc_parms(parms, options)

    # Do we need to support XP and Win 2003?
    arch = os.environ.get("ARCH", parms['ARCH'])
    if arch == "x86_xp":
        options['extra_defs'] += " /D_WIN32_WINNT=0x0501"  # pre-Vista
    else:
        options['extra_defs'] += " /D_WIN32_WINNT=0x0600"  # Vista and later
        options['extra_lib'] += " fwpuclnt.lib"

    # Add jsoncpp (optional)
    if parms.get('USE_JSONCPP'):
        options["jsoncpp"] = os.path.join(build_dir(parms), "jsoncpp")
        options['extra_inc'] += " /DHAVE_JSONCPP /I %(jsoncpp)s/dist" % options
        options['extra_lib_path'] += " /LIBPATH:%(jsoncpp)s/dist" % options
        options['extra_lib'] += " jsoncpp.lib"

    if unit_test:
        options['extra_lib'] += " gtest.lib"
        options['extra_inc'] += " /I %s" % os.path.join(parms["GTEST_ROOT"], "googletest", "include")
        options['extra_lib_path'] += " /LIBPATH:%s" % os.path.join(parms["GTEST_ROOT"], "googlemock", "gtest", "Debug")

    # Build OpenVPN Connect
    if parms.get("CONNECT"):
        options['extra_inc'] += " /I " + os.path.join(parms['OVPN3'], "common")

    if parms.get("USE_OPENSSL"):
        options['extra_inc'] += ' /DUSE_OPENSSL /I %s' % os.path.join(options['openssl'], 'include')
        options['extra_lib_path'] += ' /LIBPATH:%s' % options['openssl']
        options['extra_lib'] += ' libssl.lib libcrypto.lib'
    else:
        options['extra_inc'] += ' /DUSE_MBEDTLS /I %s' % os.path.join(options['mbedtls'], 'include')
        options['extra_lib_path'] += ' /LIBPATH:%s' % os.path.join(options['mbedtls'], 'library')
        options['extra_lib'] += ' mbedtls.lib'

    # build it
    vc_cmd(parms, r"cl %(extra_defs)s /DNOMINMAX /bigobj /D_CRT_SECURE_NO_WARNINGS /DUSE_ASIO /DASIO_STANDALONE /DASIO_NO_DEPRECATED /I %(asio)s\asio\include /DHAVE_LZ4 /I %(lz4)s %(extra_inc)s -DTAP_WIN_COMPONENT_ID=%(tap_component_id)s /I %(tap)s /I %(ovpn3)s\core /EHsc %(link_static_dynamic_flags)s /W0 %(dbg_rel_flags)s /nologo %(srcfile)s /link /LIBPATH:%(lz4)s%(extra_lib_path)s lz4.lib%(extra_lib)s ws2_32.lib crypt32.lib iphlpapi.lib winmm.lib user32.lib gdi32.lib advapi32.lib wininet.lib shell32.lib ole32.lib rpcrt4.lib Wtsapi32.lib Setupapi.lib Cfgmgr32.lib" % options, arch=os.environ.get("ARCH"))

if __name__ == "__main__":
    import sys

    params = read_params()

    src = src_fn_argv(params, sys.argv[1:])
    unit_test = is_unit_test(sys.argv[1:])
    build(params, src, unit_test)
