#!/usr/bin/env python3
import os.path
import shlex
import sys
try:
    from setuptools import setup, Extension
except ImportError:
    from distutils.core import setup, Extension
try:
    from distutils import sysconfig
except ImportError:
    import sysconfig


# C++ is only supported on Python 3.6 and newer
TEST_CXX = (sys.version_info >= (3, 6))

SRC_DIR = os.path.normpath(os.path.join(os.path.dirname(__file__), '..'))

# Windows uses MSVC compiler
MSVC = (os.name == "nt")

COMMON_FLAGS = [
    '-I' + SRC_DIR,
]
if not MSVC:
    # C compiler flags for GCC and clang
    COMMON_FLAGS.extend((
        # Treat warnings as error
        '-Werror',
        # Enable all warnings
        '-Wall', '-Wextra',
        # Extra warnings
        '-Wconversion',
        # Formatting checks
        '-Wformat',
        '-Wformat-nonliteral',
        '-Wformat-security',
    ))
    CFLAGS = COMMON_FLAGS
else:
    # C compiler flags for MSVC
    COMMON_FLAGS.extend((
        # Treat all compiler warnings as compiler errors
        '/WX',
    ))
    # Python 3.11 and older emits C4100 "unreferenced parameter" warnings
    # on Py_UNUSED() parameters. Py_UNUSED() was modified in Python 3.12
    # to support MSVC.
    if sys.version_info >= (3, 12):
        COMMON_FLAGS.extend((
            # Display warnings level 1 to 4
            '/W4',
        ))
    CFLAGS = list(COMMON_FLAGS)
CXXFLAGS = list(COMMON_FLAGS)

if not MSVC:
    C_VERSIONS = ('c99', 'c11')
else:
    # MSVC doesn't support /std:c99 flag
    C_VERSIONS = ('c11',)

if not MSVC:
    CXX_VERSIONS = [
        ('test_pythoncapi_compat_cpp03ext', ['-std=c++03']),
        ('test_pythoncapi_compat_cpp11ext', ['-std=c++11']),
        ('test_pythoncapi_compat_cpp14ext', ['-std=c++14']),
        ('test_pythoncapi_compat_cpp17ext', ['-std=c++17']),
        ('test_pythoncapi_compat_cpp20ext', ['-std=c++20']),
    ]
else:
    # MSVC doesn't support /std:c++11
    CXX_VERSIONS = [
        ('test_pythoncapi_compat_cppext', None),
        ('test_pythoncapi_compat_cpp14ext', ['/std:c++14', '/Zc:__cplusplus']),
    ]


def main():
    # gh-105776: When "gcc -std=11" is used as the C++ compiler, -std=c11
    # option emits a C++ compiler warning. Remove "-std11" option from the
    # CC command.
    cmd = (sysconfig.get_config_var('CC') or '')
    if cmd:
        cmd = shlex.split(cmd)
        cmd = [arg for arg in cmd if not arg.startswith('-std=')]
        if (sys.version_info >= (3, 8)):
            cmd = shlex.join(cmd)
        elif (sys.version_info >= (3, 3)):
            cmd = ' '.join(shlex.quote(arg) for arg in cmd)
        else:
            # Python 2.7
            import pipes
            cmd = ' '.join(pipes.quote(arg) for arg in cmd)
        # CC env var overrides sysconfig CC variable in setuptools
        os.environ['CC'] = cmd

    # C extension
    extensions = []
    for std in C_VERSIONS:
        if not MSVC:
            cflags = CFLAGS + ['-std=%s' % std]
        else:
            cflags = CFLAGS + ['/std:%s' % std]
        c_ext = Extension(
            'test_pythoncapi_compat_cext_%s' % std,
            sources=['test_pythoncapi_compat_cext.c'],
            extra_compile_args=cflags)
        extensions.append(c_ext)

    if TEST_CXX:
        # C++ extension
        for name, std_flags in CXX_VERSIONS:
            flags = list(CXXFLAGS)
            if std_flags is not None:
                flags.extend(std_flags)
            cpp_ext = Extension(
                name,
                sources=['test_pythoncapi_compat_cppext.cpp'],
                extra_compile_args=flags,
                language='c++')
            extensions.append(cpp_ext)

    setup(name="test_pythoncapi_compat",
          ext_modules=extensions)


if __name__ == "__main__":
    main()