#!/usr/bin/env python
#
# runtests.py - wrapper script for running SCons tests
#
# This script mainly exists to set PYTHONPATH to the right list of
# directories to test the SCons modules.
#
# By default, it directly uses the modules in the local tree:
# ./src/ (source files we ship) and ./etc/ (other modules we don't)
#
# When "-b aegis" is specified, it assumes it's in a directory
# in which an Aegis build (aeb) has been performed, and sets
# PYTHONPATH so that it *only* references the modules that have
# unpacked from the built packages, to test whether the packages
# are good.
#
# Options:
#
#	-a		Run all tests; does a virtual 'find' for
#			all SCons tests under the current directory.
#
#	-b system	Assume you're in the specified built system.
#			'aegis' is the only one currently defined.
#
#	-d		Debug.  Runs the script under the Python
#			debugger (pdb.py) so you don't have to
#			muck with PYTHONPATH yourself.
#
#	-q		Quiet.  By default, runtest.py prints the
#			command line it will execute before
#			executing it.  This suppresses that print.
#
#	-v		Version.  Specifies the version number to
#			be used for Aegis interaction.
#

import getopt
import os
import os.path
import re
import string
import sys

all = 0
build = None
debug = ''
tests = []
printcmd = 1
version = None

opts, tests = getopt.getopt(sys.argv[1:], "ab:dqv:",
			    ['all','build=','debug','quiet','version='])

for o, a in opts:
    if o == '-a' or o == '--all': all = 1
    elif o == '-b' or o == '--build': build = a
    elif o == '-d' or o == '--debug': debug = os.path.join(
					sys.exec_prefix,
					"lib",
					"python" + sys.version[0:3],
					"pdb.py")
    elif o == '-q' or o == '--quiet': printcmd = 0
    elif o == '-v' or o == '--version': version = a

cwd = os.getcwd()

if tests:
    map(os.path.abspath, tests)
elif all:
    def find_Test_py(arg, dirname, names):
	global tests
        n = filter(lambda n: n[-8:] == "Tests.py", names)
	n = map(lambda x,d=dirname: os.path.join(d, x), n)
	tests = tests + n
    os.path.walk('src', find_Test_py, 0)

    def find_py(arg, dirname, names):
	global tests
        n = filter(lambda n: n[-3:] == ".py", names)
	n = map(lambda x,d=dirname: os.path.join(d, x), n)
	tests = tests + n
    os.path.walk('test', find_py, 0)

if build == 'aegis':
    if not version:
	version = os.popen("aesub '$version'").read()[:-1]

    match = re.compile(r'^[CD]0*')

    def aegis_to_version(aever):
	arr = string.split(aever, '.')
	end = max(len(arr) - 1, 2)
	arr = map(lambda e: match.sub('', e), arr[:end])
	def rep(e):
	    if len(e) == 1:
		e = '0' + e
	    return e
	arr[1:] = map(rep, arr[1:])
	return string.join(arr, '.')

    version = aegis_to_version(version)

    scons_dir = os.path.join(cwd, 'build', 'test', 'bin')

    os.environ['PYTHONPATH'] = os.path.join(cwd,
					    'build',
					    'test',
					    'lib',
					    'python' + sys.version[0:3],
					    'site-packages')

else:

    scons_dir = os.path.join(cwd, 'src', 'script')

    os.environ['PYTHONPATH'] = string.join([os.path.join(cwd, 'src', 'engine'),
					    os.path.join(cwd, 'etc')],
					   os.pathsep)

os.chdir(scons_dir)

fail = []

for path in tests:
    if os.path.isabs(path):
	abs = path
    else:
	abs = os.path.join(cwd, path)
    cmd = string.join(["python", debug, abs], " ")
    if printcmd:
	print cmd
    if os.system(cmd):
	fail.append(path)

sys.exit(len(fail))