wscript

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from subprocess import Popen, PIPE, PIPE, check_call
from os import path
import os
import webbrowser
from itertools import chain
import sys
import ws.test


WAF_TOOLS = ["compiler_cxx", "boost"]


def options(ctx):
    ctx.load(" ".join(WAF_TOOLS))

    ctx.add_option(
        '--with-llvm-config',
        action='store',
        default='llvm-config'
    )

    ctx.add_option(
        '--release',
        action='store_true',
        default=False
    )


def configure(ctx):
    ctx.load(" ".join(WAF_TOOLS))

    ctx.check(lib='pthread', mandatory=True, uselib_store='PTHREAD')
    ctx.check(lib='dl', mandatory=True, uselib_store='DL')
    ctx.check(lib='tinfo', mandatory=True, uselib_store='TINFO')
    ctx.check(lib='z', mandatory=True, uselib_store='Z')
    ctx.check(lib='gmp', mandatory=True, uselib_store='GMP')
    ctx.check(lib='ffi', mandatory=True, uselib_store='FFI')

    ctx.check_boost(lib="system program_options filesystem")

    ctx.check_cfg(
        path=ctx.options.with_llvm_config, package='',
        args='--ldflags --cflags --libs all',
        uselib_store='LLVM'
    )

    if ctx.env["CXX_NAME"] in ("gcc", "clang"):
        ctx.env.append_unique("CXXFLAGS", "-std=gnu++1y")

        if ctx.options.release:
            ctx.env.append_unique("CXXFLAGS", "-Ofast")

        else:
            ctx.env.append_unique("CXXFLAGS", "-g")
            ctx.env.append_unique("CXXFLAGS", "-O0")
            ctx.env.append_unique("CXXFLAGS", "--coverage")

            ctx.env.append_unique("LINKFLAGS", "--coverage")

        ctx.env.append_unique("CXXFLAGS", "-Wall")
        ctx.env.append_unique("CXXFLAGS", "-Wextra")
        ctx.env.append_unique("CXXFLAGS", "-Wfatal-errors")

        # ctx.env.append_unique("CXXFLAGS", "-Wsuggest-final-types")
        # ctx.env.append_unique("CXXFLAGS", "-Wsuggest-final-methods")
        # ctx.env.append_unique("CXXFLAGS", "-Wsuggest-override")
        # ctx.env.append_unique("CXXFLAGS", "-Wsuggest-attribute=pure")
        # ctx.env.append_unique("CXXFLAGS", "-Wsuggest-attribute=const")
        # ctx.env.append_unique("CXXFLAGS", "-Wsuggest-attribute=noreturn")
        # ctx.env.append_unique("CXXFLAGS", "-Weffc++")
        # ctx.env.append_unique("CXXFLAGS", "-Wpedantic")

        ctx.env.append_unique("CXXFLAGS", "-Woverloaded-virtual")
        ctx.env.append_unique("CXXFLAGS", "-Wno-unused-value")
        ctx.env.append_unique("CXXFLAGS", "-Wno-unused-parameter")

        ctx.env.append_unique("CXXFLAGS", "-rdynamic")
        ctx.env.append_unique("LINKFLAGS", "-rdynamic")


def build(ctx):
    start_dir = ctx.path.find_dir("lib")
    ctx.install_files(
        "${PREFIX}/lib/arrow", start_dir.ant_glob('**/*.as'),
        cwd=start_dir,
        relative_trick=True)

    ctx.program(source=ctx.path.ant_glob("src/**/*.cpp"),
                includes=[
                    "include",
                    "vendor",
                ],
                target="arrow",
                use=["BOOST", "LLVM", "PTHREAD", "DL", "TINFO", "Z", "GMP",
                     "FFI"])


def test(ctx):
    result = ws.test.run(ctx)
    if not result:
        sys.exit(1)


def generate(ctx):
    ws.test.generate(ctx)


def coverage(ctx):
    with open(os.devnull) as nil:
        # Generate *.gcov files
        object_files = ctx.path.ant_glob("build/**/*.o")
        for object_file in object_files:
            check_call([
                "gcov", "-r", "-b", "-c", object_file.abspath()
            ], cwd="./build", stdout=nil, stderr=nil)

        # Process *.gcov files and create a coverage.info file
        check_call([
            "lcov",
            "--capture",
            "--directory", "./build",
            "--output-file", "coverage.info"
        ], stdout=nil, stderr=nil)

        # Remove external files
        check_call([
            "lcov",
            "--remove", "coverage.info",
            "/usr/*",
            "--output-file", "coverage.info"
        ], stdout=nil, stderr=nil)

        check_call([
            "lcov",
            "--remove", "coverage.info",
            "vendor/*",
            "--output-file", "coverage.info"
        ], stdout=nil, stderr=nil)

        # Generate coverage report
        check_call([
            "genhtml",
            "coverage.info",
            "--output-directory", "coverage"
        ], stdout=nil, stderr=nil)

        # Print the coverage report
        check_call([
            "lcov",
            "--summary", "coverage.info",
        ])