#!/usr/bin/python
# Project: Soar <http://soar.googlecode.com>
# Author: Jonathan Voigt <voigtjr@gmail.com>
#

from __future__ import print_function

import os
import sys
import itertools
import subprocess
import SCons.Script

# adapted from http://www.scons.org/wiki/JavaNativeInterface
def ConfigureJNI(env):
    base = os.environ.get('JAVA_HOME')
    if not base:
        if sys.platform == 'darwin':
            # Apple's OS X has its own special java base directory, which has
            # /urs/libexec/java_home will give us
            try:
                p = subprocess.Popen(['/usr/libexec/java_home'], stdout = subprocess.PIPE)
                base = p.stdout.readline().split()[-1].decode()
                p.wait()
            except OSError:
                print('Error running /usr/libexec/java_home to get java home', file=sys.stderr)
                Exit(1)
            if (not os.path.exists(base)):
                base = '/System/Library/Frameworks/JavaVM.framework'
        else:
            # Search for the java compiler
            p = env.WhereIs('javac')
            if not p:
                return None
            jcdir = os.path.dirname(os.path.realpath(p))
            # assuming the compiler found is in some directory like
            # /usr/jdkX.X/bin/javac, java's home directory is /usr/jdkX.X
            base = os.path.split(jcdir)[0]

    if sys.platform == 'darwin':
        print('Looking for Mac headers at ', os.path.join(base, 'include'))
        if os.path.exists(os.path.join(base, 'include')):
            # JDK 7+
            headers = [os.path.join(base, 'include')]
            headers.append(os.path.join(base, 'include/darwin'))
            libs = [os.path.join(base, 'lib')]
        else:
            # JDK 6
            headers = [os.path.join(base, 'Headers')]
            libs = [os.path.join(base, 'Libraries')]
    else:
        headers = [os.path.join(base, 'include')]
        libs = [os.path.join(base, 'lib')]
        if sys.platform == 'win32':
            headers.append(os.path.join(base, 'include', 'win32'))
        elif sys.platform.startswith('linux'):
            headers.append(os.path.join(base, 'include', 'linux'))
        else:
            return None

    return (headers, libs)

def get_headers(d):
    chain = itertools.chain.from_iterable
    return list(chain([os.path.join(p, x) for x in f if x.endswith('.h')] for p, d, f in os.walk(d)))

Import('env', 'compiler')
config = ConfigureJNI(env)
if config == None:
    print("Could not configure Java. If you know where java is on your system,")
    print("set environment variable JAVA_HOME to point to the directory containing")
    print("the Java include, bin, and lib directories.")
    Return()

clone = env.Clone()
clone.Append(
    CPPPATH = config[0],
    LIBPATH = config[1],
)
java_sml_alias = clone['SML_JAVA_ALIAS']

# TODO: why?
if compiler == 'g++':
    if clone['VISHIDDEN']:
        clone['CPPFLAGS'].remove('-fvisibility=hidden')
        clone.Append(CPPFLAGS = ['-fvisibility=default'])
    clone.Append(CPPFLAGS = '-Wno-unused -fno-strict-aliasing'.split())

name = 'Java_sml_ClientInterface'
interface = clone.File(name + '.i')
wrapper = clone.File(name + '_wrap.cpp')
srcdir = clone.Dir('src')
smldir = srcdir.Dir('sml')
classdir = clone.Dir('classes')
jar = clone.File('sml.jar')
java_target = java_sml_alias
fake_java_target = '#fake_java_target'

if sys.platform == 'darwin':
    install_name = os.path.join('@loader_path', name + '.jnilib')
    clone.Append(SHLINKFLAGS = ' -install_name ' + install_name)
    # Specifying the framework prevents linking in OSX on 11/16/20.  Removing seems to fix.
#     clone.Append(SHLINKFLAGS = ' -framework JavaVM')

# This builder will insert extra dependencies during the COMPILATION stage
# adapted from http://scons.org/wiki/DynamicSourceGenerator
def late_java_builder(target, source, env):
    if not os.path.isdir(srcdir.abspath):
        print('source directory does not exist',srcdir.abspath)
        sys.exit(1)

    clone.Java(classdir, srcdir)
    clone['JARCHDIR'] = classdir
    smljar = clone.Jar(jar, classdir)
    env.Depends(smljar, fake_java_target)
    install = env.Install(os.path.join(env['OUT_DIR'], 'java'), smljar)
    Alias(java_target, install)

clone.Append(BUILDERS = {'LateJavaBuilder' : Builder(action=late_java_builder)})

headers = []
for d in 'ClientSML ConnectionSML ElementXML'.split():
    headers += get_headers(env.Dir(os.path.join('#Core', d, 'src')).abspath)

swig_deps = [interface] + headers
incs = ' '.join('-I"%s"' % GetBuildPath(d) for d in clone['CPPPATH'])
nosvs = ''
if GetOption('nosvs'):
    nosvs = '-DNO_SVS'
swig_cmd = 'swig %s -o "%s" -c++ -java -Wall -package sml -outdir "%s" %s "%s"' % (incs, wrapper.abspath, smldir.abspath, nosvs, interface)
if not GetOption('verbose'):
    swig_cmd = '@' + swig_cmd
clone.Command(wrapper, swig_deps, [Mkdir(smldir), swig_cmd])

clone.Command('JavaCallbackByHand.h', clone.File('JavaCallbackByHand.h').srcnode(), Copy("$TARGET", "$SOURCE"))
if sys.platform == 'darwin':
    # have to call it Java_sml_ClientInterface.jnilib instead of the usual .dylib
    name += '.jnilib'
shlib = clone.SharedLibrary(name, wrapper)[:1]  # [:1] to throw away .exp and .lib files in Windows
lib_install = clone.Install(env['OUT_DIR'], shlib)

clone.Depends(jar, [wrapper, fake_java_target])
clone.LateJavaBuilder(fake_java_target, None)
jar_install = clone.Install(os.path.join(env['OUT_DIR'], 'java'), jar)
clone.Clean(jar, [wrapper, srcdir, classdir])

env.Alias(java_sml_alias, [lib_install, jar_install])
