# -*- Python -*-

# Configuration file for the 'lit' test runner.

import os
import sys
import re
import platform

import lit.util
import lit.formats

# name: The name of this test suite.
config.name = 'STP query-files'

# Choose between lit's internal shell pipeline runner and a real shell.  If
# LIT_USE_INTERNAL_SHELL is in the environment, we use that as an override.
use_lit_shell = os.environ.get("LIT_USE_INTERNAL_SHELL")
if use_lit_shell:
    # 0 is external, "" is default, and everything else is internal.
    execute_external = (use_lit_shell == "0")
else:
    # Otherwise we default to internal on Windows and external elsewhere, as
    # bash on Windows is usually very slow.
    execute_external = (not sys.platform in ['win32'])

# testFormat: The test format to use to interpret tests.
config.test_format = lit.formats.ShTest(execute_external)

# suffixes: A list of file extensions to treat as test files. This is overriden
# by individual lit.local.cfg files in the test subdirectories.
config.suffixes = ['.smt', '.smt2', '.cvc']

# excludes: A list of directories to exclude from the testsuite. The 'Inputs'
# subdirectories contain auxiliary inputs for various tests in their parent
# directories.
config.excludes = []

# test_source_root: The root path where tests are located.
config.test_source_root = os.path.dirname(os.path.abspath(__file__))

# test_exec_root: The root path where tests should be run.
stp_obj_root = getattr(config, 'stp_obj_root', None)
if stp_obj_root is not None:
    config.test_exec_root = os.path.join(stp_obj_root, 'tests', 'query-files')

# Tweak the PATH to include the tools dir.
#if llvm_obj_root is not None:
#    llvm_tools_dir = getattr(config, 'llvm_tools_dir', None)
#    if not llvm_tools_dir:
#        lit_config.fatal('No LLVM tools dir set!')
#    path = os.path.pathsep.join((llvm_tools_dir, config.environment['PATH']))
#    config.environment['PATH'] = path

# Propagate 'HOME' through the environment.
if 'HOME' in os.environ:
    config.environment['HOME'] = os.environ['HOME']

# Propagate 'INCLUDE' through the environment.
if 'INCLUDE' in os.environ:
    config.environment['INCLUDE'] = os.environ['INCLUDE']

# Propagate 'LIB' through the environment.
if 'LIB' in os.environ:
    config.environment['LIB'] = os.environ['LIB']

# Propagate the temp directory. Windows requires this because it uses \Windows\
# if none of these are present.
if 'TMP' in os.environ:
    config.environment['TMP'] = os.environ['TMP']
if 'TEMP' in os.environ:
    config.environment['TEMP'] = os.environ['TEMP']

# Check Python executable
pythonExec = getattr(config, 'python_executable')
if sys.executable != os.path.realpath(pythonExec):
    lit_config.warning('Python executable mis-match: {} != {}'.format(sys.executable, os.path.realpath(pythonExec)))

# Propagate PYTHON_EXECUTABLE into the environment
config.environment['PYTHON_EXECUTABLE'] = pythonExec

###

import os

# Check that the object root is known.
if config.test_exec_root is None:
    lit_config.fatal('Could not determine execution root for tests!')

### Add STP specific substitutions

# Allow user to override the solver
solverExecutable = lit_config.params.get('solver', config.stp_executable)
if not lit.util.which(solverExecutable):
    lit_config.fatal('Cannot find solver:{solver}\n'.format(solver=solverExecutable))
else:
    solverExecutable = lit.util.which( solverExecutable )

# Allow user to provide extra arguments to solver
solverParams = lit_config.params.get('solver_params','')
if len(solverParams) > 0:
    solverExecutable = solverExecutable + ' ' + solverParams

# Inform user what solver is being used
lit_config.note('Using solver: {solver}\n'.format(solver=solverExecutable))

config.substitutions.append( ('%solver', solverExecutable) )

# Find OutputCheck
OutputCheckTool = os.path.join( os.path.dirname( os.path.dirname( config.test_source_root ) ),
                                'utils',
                                'OutputCheck',
                                'bin',
                                'OutputCheck'
                              )
if not os.path.exists(OutputCheckTool):
    lit_config.fatal('Cannot find OutputCheck executable: {OutputCheck}'.format(OutputCheck=OutputCheckTool))

# Add OutputCheck bin directory to PATH (for not utility)
path = os.path.pathsep.join(( os.path.dirname(OutputCheckTool), config.environment['PATH']))
config.environment['PATH'] = path


# Allow user to specify extra flags to OutputCheck (e.g. -l debug)
OutputCheckFlags = lit_config.params.get('outputcheck_params','')
if len(OutputCheckFlags) > 0:
    OutputCheckTool += ' ' + OutputCheckFlags
config.substitutions.append( ('%OutputCheck', pythonExec + ' ' +OutputCheckTool) )

### Features

# Shell execution
if execute_external:
    config.available_features.add('shell')
