#!/usr/bin/env python

# Copyright (C) 2016 g10 Code GmbH
#
# This file is part of GPGME.
#
# GPGME is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# GPGME is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
# or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
# Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this program; if not, see <http://www.gnu.org/licenses/>.

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

import argparse
import glob
import os
import subprocess
import sys

class SplitAndAccumulate(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        current = getattr(namespace, self.dest, list())
        current.extend(values.split())
        setattr(namespace, self.dest, current)

parser = argparse.ArgumentParser(description='Run tests.')
parser.add_argument('tests', metavar='TEST', type=str, nargs='+',
                    help='A test to run')
parser.add_argument('-v', '--verbose', action="store_true", default=False,
                    help='Be verbose.')
parser.add_argument('-q', '--quiet', action="store_true", default=False,
                    help='Be quiet.')
parser.add_argument('--interpreters', metavar='PYTHON', type=str,
                    default=[], action=SplitAndAccumulate,
                    help='Use these interpreters to run the tests, ' +
                    'separated by spaces.')
parser.add_argument('--srcdir', type=str,
                    default=os.environ.get("srcdir", ""),
                    help='Location of the tests.')
parser.add_argument('--builddir', type=str,
                    default=os.environ.get("abs_builddir", ""),
                    help='Location of the tests.')
parser.add_argument('--python-libdir', type=str,
                    default=None,
                    help='Optional location of the in-tree module lib directory.')
parser.add_argument('--parallel', action="store_true", default=False,
                    help='Ignored.  For compatibility with run-tests.scm.')

args = parser.parse_args()
if not args.interpreters:
    args.interpreters = [sys.executable]

out = sys.stdout if args.verbose else None
err = sys.stderr if args.verbose else None

def status_to_str(code):
    return {0: "PASS", 77: "SKIP", 99: "ERROR"}.get(code, "FAIL")

results = list()
for interpreter in args.interpreters:
    version = subprocess.check_output(
        [interpreter, "-c", "import sys; print('{0}.{1}'.format(sys.version_info[0], sys.version_info[1]))"]).strip().decode()

    if args.python_libdir:
        python_libdir = args.python_libdir
    else:
        pattern = os.path.join(args.builddir, "..",
                               "{0}-gpg".format(os.path.basename(interpreter)),
                               "lib*")
        libdirs = glob.glob(pattern)
        if len(libdirs) == 0:
            sys.exit("Build directory matching {0!r} not found.".format(pattern))
        elif len(libdirs) > 1:
            sys.exit("Multiple build directories matching {0!r} found: {1}".format(
                pattern, libdirs))
        python_libdir = libdirs[0]

    env = dict(os.environ)
    env["PYTHONPATH"] = python_libdir

    if not args.quiet:
        print("Running tests using {0} ({1})...".format(interpreter, version))

    for test in args.tests:
        status = subprocess.call(
            [interpreter, os.path.join(args.srcdir, test)],
            env=env, stdout=out, stderr=err)
        if not args.quiet:
            print("{0}: {1}".format(status_to_str(status), test))
        results.append(status)

def count(status):
    return len(list(filter(lambda x: x == status, results)))
def failed():
    return len(list(filter(lambda x: x not in (0, 77, 99), results)))

if not args.quiet:
    print("{0} tests run, {1} succeeded, {2} failed, {3} skipped.".format(
        len(results), count(0), failed(), count(77)))
    sys.exit(len(results) - count(0) - count(77))
sys.exit(results[0])