##
# 'googletest' script to check if Google Unit Test library is installed.
# If not, get it and install it
#
##

import os

Import('env')

gtest_env = env.Clone()
target_os = gtest_env.get('TARGET_OS')
src_dir = gtest_env.get('SRC_DIR')

targets_need_gtest = ['darwin','linux']
gtest_dir      = os.path.join(src_dir, 'extlibs', 'gtest', 'gtest-1.7.0')
gtest_lib_dir = os.path.join(gtest_dir,'lib')
gtest_dotlib_dir =  os.path.join(gtest_lib_dir, '.libs')
gtest_zip_file = os.path.join(src_dir, 'extlibs', 'gtest', 'gtest-1.7.0.zip')
gtest_url      = 'https://googletest.googlecode.com/files/gtest-1.7.0.zip'

if target_os in targets_need_gtest:
	print '*** Checking for installation of google unit test 1.7.0 ***'
	if not os.path.exists(os.path.join(gtest_dir, 'configure')):
		# If the gtest zip file is not already present, download it
		if not os.path.exists(gtest_zip_file):
			gtest_zip = gtest_env.Download(gtest_zip_file, gtest_url)
		else:
			gtest_zip = gtest_zip_file
		print 'Unzipping to : ' + gtest_dir
		gtest_env.UnpackAll(gtest_dir, gtest_zip)

if target_os == 'darwin':
	if os.path.exists(gtest_dir):
		# Build gtest and store it at a temporary directory

		if not os.path.exists(gtest_lib_dir):
#			print 'Create a directory'
#			os.mkdir(gtest_lib_dir)


			print 'Invoke cmake command to generate appropriate make files'
			gtest_env.Configure(gtest_dir, './configure')

			# Run make on gtest
			print 'Making google unit test'
			gtest_env.Configure(gtest_dir, 'make')

#			print 'Create a directory'
#			os.mkdir(gtest_dotlib_dir)

			print 'Change to a directory'
			os.chdir(gtest_dotlib_dir)

##			print 'Change to a directory'
#			os.chdir(gtest_lib_dir)

			print 'Create hard links pointing to gtest libraries'
			os.link('libgtest.a', gtest_lib_dir + 'libgtest.a')
			os.link('libgtest_main.a', gtest_lib_dir +  'libgtest_main.a')
			print 'Create hard links pointing to gtest libraries - DONE'


elif target_os in ['linux']:
	if os.path.exists(gtest_dir):
		if not os.path.exists(gtest_lib_dir):
			# Run configure on gtest
			print 'Configuring google unit test'
			if env.get('CROSS_COMPILE'):
				env.Configure(gtest_dir, './configure --disable-shared --host=' + env['CROSS_COMPILE'])
			else:
				env.Configure(gtest_dir, './configure --disable-shared')
			# Run make on gtest
			print 'Making google unit test'
			gtest_env.Configure(gtest_dir, 'make')


# Export flags once for all
if target_os in targets_need_gtest:
	gtest_env.AppendUnique(LIBPATH = [gtest_dotlib_dir])
	gtest_env.PrependUnique(CPPPATH = [os.path.join(gtest_dir, 'include')])
	gtest_env.AppendENVPath('LD_LIBRARY_PATH', gtest_dotlib_dir)
	if 'g++' in gtest_env.get('CXX'):
		gtest_env.AppendUnique(CXXFLAGS = ['-std=c++0x'])
		gtest_env.AppendUnique(CXXFLAGS = ['-Wall'])
# Note: 'pthread' for android is in bionic 
# On other platform, if use new gcc(>4.9?) it isn't required, otherwise, it's required
		if target_os not in ['android']:
			gtest_env.AppendUnique(CXXFLAGS = ['-pthread'])
			gtest_env.PrependUnique(LIBS = ['pthread'])
	gtest_env.PrependUnique(LIBS = ['gtest', 'gtest_main'])

Return('gtest_env')