#!/usr/bin/env python
import glob
import os
import shutil
import subprocess
import waflib.Logs as Logs
import waflib.Options as Options
import waflib.extras.autowaf as autowaf

# Library and package version (UNIX style major, minor, micro)
# major increment <=> incompatible changes
# minor increment <=> compatible changes (additions)
# micro increment <=> no interface changes
SERD_VERSION       = '0.18.3'
SERD_MAJOR_VERSION = '0'

# Mandatory waf variables
APPNAME = 'serd'        # Package name for waf dist
VERSION = SERD_VERSION  # Package version for waf dist
top     = '.'           # Source directory
out     = 'build'       # Build directory

def options(opt):
    opt.load('compiler_c')
    autowaf.set_options(opt)
    opt.add_option('--no-utils', action='store_true', dest='no_utils',
                   help='Do not build command line utilities')
    opt.add_option('--test', action='store_true', dest='build_tests',
                   help='Build unit tests')
    opt.add_option('--stack-check', action='store_true', dest='stack_check',
                   help='Include runtime stack sanity checks')
    opt.add_option('--static', action='store_true', dest='static',
                   help='Build static library')
    opt.add_option('--no-shared', action='store_true', dest='no_shared',
                   help='Do not build shared library')
    opt.add_option('--static-progs', action='store_true', dest='static_progs',
                   help='Build programs as static binaries')
    opt.add_option('--largefile', action='store_true', dest='largefile',
                   help='Build with large file support on 32-bit systems')

def configure(conf):
    conf.load('compiler_c')
    autowaf.configure(conf)
    autowaf.set_c99_mode(conf)
    autowaf.display_header('Serd Configuration')

    conf.env.BUILD_TESTS  = Options.options.build_tests
    conf.env.BUILD_UTILS  = not Options.options.no_utils
    conf.env.BUILD_SHARED = not Options.options.no_shared
    conf.env.STATIC_PROGS = Options.options.static_progs
    conf.env.BUILD_STATIC = (Options.options.static or
                             Options.options.static_progs)

    if not conf.env.BUILD_SHARED and not conf.env.BUILD_STATIC:
        conf.fatal('Neither a shared nor a static build requested')

    if Options.options.stack_check:
        autowaf.define(conf, 'SERD_STACK_CHECK', SERD_VERSION)

    if Options.options.largefile:
        conf.env.append_unique('DEFINES', ['_FILE_OFFSET_BITS=64'])

    if conf.env.BUILD_TESTS:
        conf.check(lib         = 'gcov',
                   define_name = 'HAVE_GCOV',
                   mandatory   = False)

    conf.check(function_name = 'fmax',
               header_name   = 'math.h',
               define_name   = 'HAVE_FMAX',
               lib           = ['m'],
               mandatory     = False)

    conf.check(function_name = 'posix_memalign',
               header_name   = 'stdlib.h',
               define_name   = 'HAVE_POSIX_MEMALIGN',
               defines       = ['_POSIX_C_SOURCE=201112L'],
               mandatory     = False)

    conf.check(function_name = 'posix_fadvise',
               header_name   = 'fcntl.h',
               define_name   = 'HAVE_POSIX_FADVISE',
               defines       = ['_POSIX_C_SOURCE=201112L'],
               mandatory     = False)

    conf.check(function_name = 'fileno',
               header_name   = 'stdio.h',
               define_name   = 'HAVE_FILENO',
               defines       = ['_POSIX_C_SOURCE=201112L'],
               mandatory     = False)

    autowaf.define(conf, 'SERD_VERSION', SERD_VERSION)
    autowaf.set_lib_env(conf, 'serd', SERD_VERSION)
    conf.write_config_header('serd_config.h', remove=False)

    autowaf.display_msg(conf, 'Utilities', str(conf.env.BUILD_UTILS))
    autowaf.display_msg(conf, 'Unit tests', str(conf.env.BUILD_TESTS))
    print('')

lib_source = [
    'src/env.c',
    'src/node.c',
    'src/reader.c',
    'src/string.c',
    'src/uri.c',
    'src/writer.c',
]

def build(bld):
    # C Headers
    includedir = '${INCLUDEDIR}/serd-%s/serd' % SERD_MAJOR_VERSION
    bld.install_files(includedir, bld.path.ant_glob('serd/*.h'))

    # Pkgconfig file
    autowaf.build_pc(bld, 'SERD', SERD_VERSION, SERD_MAJOR_VERSION, [],
                     {'SERD_MAJOR_VERSION' : SERD_MAJOR_VERSION})

    libflags = ['-fvisibility=hidden']
    libs     = ['m']
    defines  = []
    if bld.env.MSVC_COMPILER:
        libflags = []
        libs     = []
        defines  = ['snprintf=_snprintf']

    # Shared Library
    if bld.env.BUILD_SHARED:
        bld(features        = 'c cshlib',
            export_includes = ['.'],
            source          = lib_source,
            includes        = ['.', './src'],
            lib             = libs,
            name            = 'libserd',
            target          = 'serd-%s' % SERD_MAJOR_VERSION,
            vnum            = SERD_VERSION,
            install_path    = '${LIBDIR}',
            defines         = defines + ['SERD_SHARED', 'SERD_INTERNAL'],
            cflags          = libflags)

    # Static library
    if bld.env.BUILD_STATIC:
        bld(features        = 'c cstlib',
            export_includes = ['.'],
            source          = lib_source,
            includes        = ['.', './src'],
            lib             = libs,
            name            = 'libserd_static',
            target          = 'serd-%s' % SERD_MAJOR_VERSION,
            vnum            = SERD_VERSION,
            install_path    = '${LIBDIR}',
            defines         = defines + ['SERD_INTERNAL'])

    if bld.env.BUILD_TESTS:
        test_libs   = libs
        test_cflags = ['']
        if bld.is_defined('HAVE_GCOV'):
            test_libs   += ['gcov']
            test_cflags += ['-fprofile-arcs', '-ftest-coverage']

        # Profiled static library for test coverage
        bld(features     = 'c cstlib',
            source       = lib_source,
            includes     = ['.', './src'],
            lib          = test_libs,
            name         = 'libserd_profiled',
            target       = 'serd_profiled',
            install_path = '',
            defines      = defines + ['SERD_INTERNAL'],
            cflags       = test_cflags)

        # Static profiled serdi for tests
        bld(features     = 'c cprogram',
            source       = 'src/serdi.c',
            includes     = ['.', './src'],
            use          = 'libserd_profiled',
            lib          = test_libs,
            target       = 'serdi_static',
            install_path = '',
            defines      = defines,
            cflags       = test_cflags)

        # Unit test program
        bld(features     = 'c cprogram',
            source       = 'tests/serd_test.c',
            includes     = ['.', './src'],
            use          = 'libserd_profiled',
            lib          = test_libs,
            target       = 'serd_test',
            install_path = '',
            defines      = defines,
            cflags       = test_cflags)

    # Utilities
    if bld.env.BUILD_UTILS:
        obj = bld(features     = 'c cprogram',
                  source       = 'src/serdi.c',
                  target       = 'serdi',
                  includes     = ['.', './src'],
                  use          = 'libserd',
                  lib          = libs,
                  install_path = '${BINDIR}')
        if not bld.env.BUILD_SHARED or bld.env.STATIC_PROGS:
            obj.use = 'libserd_static'
        if bld.env.STATIC_PROGS:
            obj.env.SHLIB_MARKER = obj.env.STLIB_MARKER
            obj.linkflags        = ['-static']

    # Documentation
    autowaf.build_dox(bld, 'SERD', SERD_VERSION, top, out)

    # Man page
    bld.install_files('${MANDIR}/man1', 'doc/serdi.1')

    bld.add_post_fun(autowaf.run_ldconfig)
    if bld.env.DOCS:
        bld.add_post_fun(fix_docs)

def lint(ctx):
    subprocess.call('cpplint.py --filter=+whitespace/comments,-whitespace/tab,-whitespace/braces,-whitespace/labels,-build/header_guard,-readability/casting,-readability/todo,-build/include src/* serd/*', shell=True)

def amalgamate(ctx):
    shutil.copy('serd/serd.h', 'build/serd.h')
    amalgamation = open('build/serd.c', 'w')

    serd_internal_h = open('src/serd_internal.h')
    for l in serd_internal_h:
        if l == '#include "serd/serd.h"\n':
            amalgamation.write('#include "serd.h"\n')
        else:
            amalgamation.write(l)
    serd_internal_h.close()

    for f in lib_source:
        fd = open(f)
        amalgamation.write('\n/**\n   @file %s\n*/' % f)
        header = True
        for l in fd:
            if header:
                if l == '*/\n':
                    header = False
            else:
                if l != '#include "serd_internal.h"\n':
                    amalgamation.write(l)
        fd.close()
    amalgamation.close()

    for i in ['c', 'h']:
        Logs.info('Wrote build/serd.%s' % i)

def fix_docs(ctx):
    if ctx.cmd == 'build':
        autowaf.make_simple_dox(APPNAME)

def upload_docs(ctx):
    os.system('rsync -ravz --delete -e ssh build/doc/html/ drobilla@drobilla.net:~/drobilla.net/docs/serd/')

def file_equals(patha, pathb, subst_from='', subst_to=''):
    fa = open(patha, 'rU')
    fb = open(pathb, 'rU')
    for line in fa:
        if line.replace(subst_from, subst_to) != fb.readline().replace(subst_from, subst_to):
            return False
    fa.close()
    fb.close()
    return True

def earl_assertion(test, passed, asserter):
    import datetime

    return '''
[]
	a earl:Assertion ;%s
	earl:subject <http://drobilla.net/sw/serd> ;
	earl:test <%s> ;
	earl:result [
		a earl:TestResult ;
		earl:outcome %s ;
		dc:date "%s"^^xsd:dateTime
	] .
''' % (('\n\tearl:assertedBy <%s> ;' % asserter) if asserter else '',
       test,
       "earl:passed" if passed else "earl:failed",
       datetime.datetime.now().replace(microsecond=0).isoformat())

def test_thru(ctx, base, path, check_filename, flags):
    in_filename = os.path.join(ctx.path.abspath(), path);
    out_filename = path + '.thru'

    command = ('%s %s -i turtle -o turtle -p foo "%s" "%s" | '
               '%s -i turtle -o ntriples -c foo - "%s" > %s') % (
        'serdi_static', flags.ljust(5),
        in_filename, base,
        'serdi_static', base, out_filename)

    passed = autowaf.run_test(ctx, APPNAME, command, 0, name=out_filename)
    if not passed:
        Logs.pprint('RED', '** Failed command: %s' % command)
        return False

    if not os.access(out_filename, os.F_OK):
        Logs.pprint('RED', 'FAIL: %s is missing' % out_filename)
    elif not file_equals(check_filename, out_filename, '_:docid', '_:genid'):
        Logs.pprint('RED', 'FAIL: %s != %s' % (out_filename, check_filename))
    else:
        #Logs.pprint('GREEN', '** Pass %s == %s' % (out_filename, check_filename))
        return True

    return False

def test_manifest(ctx, srcdir, testdir, report, test_base, parse_base):
    import rdflib
    import urlparse

    rdf  = rdflib.Namespace('http://www.w3.org/1999/02/22-rdf-syntax-ns#')
    rdfs = rdflib.Namespace('http://www.w3.org/2000/01/rdf-schema#')
    mf   = rdflib.Namespace('http://www.w3.org/2001/sw/DataAccess/tests/test-manifest#')
    rdft = rdflib.Namespace('http://www.w3.org/ns/rdftest#')
    earl = rdflib.Namespace('http://www.w3.org/ns/earl#')

    model = rdflib.ConjunctiveGraph()
    model.parse(os.path.join(srcdir, 'tests', testdir, 'manifest.ttl'),
                rdflib.URIRef(test_base + 'manifest.ttl'),
                format='n3')

    is_drobilla = (os.getenv('USER') == 'drobilla')
    asserter    = 'http://drobilla.net/drobilla#me' if is_drobilla else ''

    def run_test(action_node, expected_return):
        output  = os.path.join('tests', testdir, action_node + '.out')
        action  = os.path.join(srcdir, 'tests', testdir, action_node)
        rel     = os.path.relpath(action, os.path.join(srcdir, 'tests', testdir))
        command = 'serdi_static -f "%s" "%s" > %s' % (action, parse_base + rel, output)

        return autowaf.run_test(ctx, APPNAME, command, expected_return, name=name)

    for i in sorted(model.triples([None, rdf.type, rdft.TestTurtlePositiveSyntax])):
        test        = i[0]
        name        = model.value(test, mf.name, None)
        action_node = model.value(test, mf.action, None)[len(test_base):]

        passed = run_test(action_node, 0)
        report.write(earl_assertion(test, passed, asserter))

    for i in sorted(model.triples([None, rdf.type, rdft.TestTurtleNegativeSyntax])):
        test        = i[0]
        name        = model.value(test, mf.name, None)
        action_node = model.value(test, mf.action, None)[len(test_base):]

        passed = run_test(action_node, 1)
        report.write(earl_assertion(test, passed, asserter))

    for i in sorted(model.triples([None, rdf.type, rdft.TestTurtleNegativeEval])):
        test        = i[0]
        name        = model.value(test, mf.name, None)
        action_node = model.value(test, mf.action, None)[len(test_base):]

        passed = run_test(action_node, 1)
        report.write(earl_assertion(test, passed, asserter))

    for i in sorted(model.triples([None, rdf.type, rdft.TestTurtleEval])):
        test        = i[0]
        name        = model.value(test, mf.name, None)
        action_node = model.value(test, mf.action, None)[len(test_base):]
        result_node = model.value(test, mf.result, None)[len(test_base):]

        passed = run_test(action_node, 0)

        if passed:
            action = os.path.join('tests', testdir, action_node)
            output = action + '.out'
            result = os.path.join(srcdir, 'tests', testdir, result_node)

            if not os.access(output, os.F_OK):
                passed = False
                Logs.pprint('RED', 'FAIL: %s output %s is missing' % (name, output))
            elif not file_equals(result, output):
                passed = False
                Logs.pprint('RED', 'FAIL: %s != %s' % (os.path.abspath(output), result))
            else:
                Logs.pprint('GREEN', '** Pass %s' % output)

            test_thru(ctx, parse_base + action_node, action, result, "")

        report.write(earl_assertion(test, passed, asserter))

def test(ctx):
    blddir = autowaf.build_dir(APPNAME, 'tests')
    for i in ['', 'bad', 'good', 'new', 'tests-ttl', 'extra']:
        try:
            os.makedirs(os.path.join(blddir, i))
        except:
            pass

    for i in glob.glob(blddir + '/*.*'):
        os.remove(i)

    srcdir   = ctx.path.abspath()
    orig_dir = os.path.abspath(os.curdir)

    os.chdir(os.path.join(srcdir, 'tests', 'good'))
    old_good_tests = glob.glob('*.ttl')
    old_good_tests.sort()
    old_good_tests.remove('manifest.ttl')
    good_tests = { 'good': old_good_tests }
    os.chdir(orig_dir)

    os.chdir(srcdir)
    bad_tests = glob.glob('tests/bad/*.ttl')
    bad_tests.sort()
    os.chdir(orig_dir)

    autowaf.pre_test(ctx, APPNAME)

    os.environ['PATH'] = '.' + os.pathsep + os.getenv('PATH')

    autowaf.run_tests(ctx, APPNAME, ['serd_test'], dirs=['.'])

    autowaf.run_tests(ctx, APPNAME, [
            'serdi_static -q -o turtle %s/tests/good/base.ttl "base.ttl" > tests/good/base.ttl.out' % srcdir],
                      0, name='base')

    if not file_equals('%s/tests/good/base.ttl' % srcdir, 'tests/good/base.ttl.out'):
        Logs.pprint('RED', 'FAIL: build/tests/base.ttl.out is incorrect')

    nul = os.devnull
    autowaf.run_tests(ctx, APPNAME, [
            'serdi_static file://%s/tests/good/manifest.ttl > %s' % (srcdir, nul),
#            'serdi_static %s/tests/good/UTF-8.ttl > %s' % (srcdir, nul),
            'serdi_static -v > %s' % nul,
            'serdi_static -h > %s' % nul,
            'serdi_static -s "<foo> a <#Thingie> ." > %s' % nul,
            'serdi_static %s > %s' % (nul, nul)],
                      0, name='serdi-cmd-good')

    autowaf.run_tests(ctx, APPNAME, [
            'serdi_static -q file://%s/tests/bad-id-clash.ttl > %s' % (srcdir, nul),
            'serdi_static > %s' % nul,
            'serdi_static ftp://example.org/unsupported.ttl > %s' % nul,
            'serdi_static -i > %s' % nul,
            'serdi_static -o > %s' % nul,
            'serdi_static -z > %s' % nul,
            'serdi_static -p > %s' % nul,
            'serdi_static -c > %s' % nul,
            'serdi_static -r > %s' % nul,
            'serdi_static -i illegal > %s' % nul,
            'serdi_static -o illegal > %s' % nul,
            'serdi_static -i turtle > %s' % nul,
            'serdi_static /no/such/file > %s' % nul],
                      1, name='serdi-cmd-bad')

    def test_base(test):
        return ('http://www.w3.org/2001/sw/DataAccess/df1/tests/'
                + test.replace('\\', '/'))

    # Good tests
    for tdir, tests in good_tests.items():
        commands = []
     
        for test in tests:
            path = os.path.join('tests', tdir, test)
            commands += [ 'serdi_static -f "%s" "%s" > %s.out' % (
                    os.path.join('..', path), test_base(test), path) ]
    
        autowaf.run_tests(ctx, APPNAME, commands, 0, name=tdir)
    
        Logs.pprint('BOLD', '\nVerifying turtle => ntriples')
        for test in tests:
            check_filename = os.path.join(
                srcdir, 'tests', tdir, test.replace('.ttl', '.nt'))
            out_filename = os.path.join('tests', tdir, test + '.out')
            if not os.access(out_filename, os.F_OK):
                Logs.pprint('RED', 'FAIL: %s output is missing' % test)
            elif not file_equals(check_filename, out_filename):
                Logs.pprint('RED', 'FAIL: %s is incorrect' % out_filename)
            else:
                Logs.pprint('GREEN', 'Pass: %s' % test)
    
    # Bad tests
    commands = []
    for test in bad_tests:
        commands += [ 'serdi_static -q "%s" "%s" > %s.out' % (
                os.path.join(srcdir, test), test_base(test), test) ]

    autowaf.run_tests(ctx, APPNAME, commands, 1, name='bad')

    # Don't do a round-trip test for test-id.ttl, IDs have changed
    good_tests['good'].remove('test-id.ttl')

    # Round-trip good tests
    for tdir, tests in good_tests.items():
        thru_tests = tests;
    
        commands = []
        num = 0
        for test in thru_tests:
            num += 1
            flags = ''
            if (num % 2 == 0):
                flags += '-b'
            if (num % 5 == 0):
                flags += ' -f'
            if (num % 3 == 0):
                flags += ' -r http://www.w3.org/'
            if (num % 7 == 0):
                flags += ' -e'

            path  = os.path.join('tests', tdir, test)
            check = os.path.join(srcdir, path.replace('.ttl', '.nt'))
            test_thru(ctx, test_base(test), path, check, flags)

    try:
        report = open('earl.ttl', 'w')
        report.write('''@prefix earl: <http://www.w3.org/ns/earl#> .
@prefix xsd: <http://www.w3.org/2001/XMLSchema#> .\n''')

        serd_ttl = open(os.path.join(srcdir, 'serd.ttl'))
        for line in serd_ttl:
            report.write(line)
        serd_ttl.close()
        rdf_turtle = 'https://dvcs.w3.org/hg/rdf/raw-file/default/rdf-turtle/'
        test_manifest(ctx, srcdir, 'tests-ttl', report,
                      rdf_turtle + 'tests-ttl/', 'http://example/base/')

        report.close()

    except:
        pass
    
    autowaf.post_test(ctx, APPNAME)