wscript

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

from waflib.Tools import waf_unit_test
from waflib.Build import BuildContext
import os

VERSION = '0.1'
APPNAME = 'cxxjson'

top = '.'
out = 'build'


def init(ctx):
    ctx.load('build_logs')


def options(ctx):
    ctx.load('compiler_cxx')
    ctx.load('gnu_dirs')
    ctx.load('waf_unit_test')


def configure(ctx):
    ctx.env.DEFINES = os.environ.get('DEFINES', '').split()
    ctx.load('compiler_cxx')
    ctx.load('gnu_dirs')
    ctx.load('clang_compilation_database')
    ctx.load('waf_unit_test')
    ctx.msg('Used CXXFLAGS', ' '.join(ctx.env.CXXFLAGS))
    ctx.msg('Used LINKFLAGS', ' '.join(ctx.env.LINKFLAGS))
    ctx.msg('Used DEFINES', ' '.join(ctx.env.DEFINES))
    ctx.env.INCLUDES = [ctx.path.abspath(), './inc']


def cxx(ctx):
    incs = ctx.path.find_dir('inc')
    ctx.install_files(
        '${INCLUDEDIR}',
        incs.ant_glob('**/*.(h|hpp)'),
        cwd=incs,
        relative_trick=True
    )

    ctx.objects(
        source=ctx.path.ant_glob(['src/**/*.(c|cpp)']),
        target='SRCLIB',
        features='cxx',
    )

    ctx(
        target=APPNAME,
        features='cxx cxxshlib',
        vnum=VERSION,
        use=['SRCLIB'],
        install_path='${LIBDIR}',
    )


def build(ctx):
    cxx(ctx)

    ctx(
        source=ctx.path.ant_glob(['test/test_main.cpp']),
        target='test_main',
        features='cxx',
        includes=['test'],
    )

    for suite in ctx.path.ant_glob(
        ['test/**/*.(c|cpp)'],
            excl=['test/test_main.cpp']):
        target, _ = suite.name.split('.')
        ctx(
            source=suite,
            target='ut_' + target,
            features='cxx cxxprogram test',
            use=['test_main', 'SRCLIB'],
            lib=['pthread'],
            install_path=None,
        )

    ctx.add_post_fun(waf_unit_test.summary)
    ctx.add_post_fun(waf_unit_test.set_exit_code)


def benchmark(ctx):
    cxx(ctx)

    for suite in ctx.path.ant_glob(
            ['benchmark/**/*.(c|cpp)']):
        target, _ = suite.name.split('.')
        ctx(
            source=suite,
            target='benchmark_' + target,
            features='cxx cxxprogram test',
            use=['SRCLIB'],
            lib=['pthread', 'benchmark', 'benchmark_main'],
            install_path=None
        )

    def summary(bld):
        from waflib import Logs
        lst = getattr(bld, 'utest_results', [])
        for(f, code, out, err)in (lst or []):
            Logs.pprint('CYAN', '<<=== %s ===>>' % f)
            out = out.decode('utf-8')
            Logs.pprint('CYAN', str(out))
            Logs.pprint('CYAN', '')

    ctx.add_post_fun(summary)
    ctx.add_post_fun(waf_unit_test.set_exit_code)


def fuzzer(ctx):
    ctx.env.CXXFLAGS.extend(['-g', '-fsanitize=fuzzer,address,undefined'])
    ctx.env.LINKFLAGS.extend(['-g', '-fsanitize=fuzzer,address,undefined'])
    cxx(ctx)

    for suite in ctx.path.ant_glob(
            ['fuzzer/**/*.(c|cpp)']):
        target, _ = suite.name.split('.')
        ctx(
            source=suite,
            target='fuzzer_' + target,
            features='cxx cxxprogram',
            use=['SRCLIB'],
            install_path=None
        )


class Benchmark(BuildContext):
    cmd = 'benchmark'
    fun = 'benchmark'


class Fuzzer(BuildContext):
    cmd = 'fuzzer'
    fun = 'fuzzer'