import platform

import sys

try:

# Import setuptools so that it monkey-patch overrides distutils

import setuptools # noqa: F401

except ImportError:

pass

if sys.version_info >= (3, 12):

# From setuptools' monkeypatch

from distutils import ccompiler # type: ignore[import-not-found]

else:

from distutils import ccompiler

EXTRA_FLAGS_PER_COMPILER_TYPE_PER_PATH_COMPONENT = {

"unix": {

"base64/arch/ssse3": ["-mssse3"],

"base64/arch/sse41": ["-msse4.1"],

"base64/arch/sse42": ["-msse4.2"],

"base64/arch/avx2": ["-mavx2"],

"base64/arch/avx": ["-mavx"],

},

"msvc": {

"base64/arch/sse42": ["/arch:SSE4.2"],

"base64/arch/avx2": ["/arch:AVX2"],

"base64/arch/avx": ["/arch:AVX"],

},

}

ccompiler.CCompiler.__spawn = ccompiler.CCompiler.spawn # type: ignore[attr-defined]

X86_64 = platform.machine() in ("x86_64", "AMD64", "amd64")

def spawn(self, cmd, **kwargs) -> None: # type: ignore[no-untyped-def]

compiler_type: str = self.compiler_type

extra_options = EXTRA_FLAGS_PER_COMPILER_TYPE_PER_PATH_COMPONENT[compiler_type]

new_cmd = list(cmd)

if X86_64 and extra_options is not None:

# filenames are closer to the end of command line

for argument in reversed(new_cmd):

# Check if the matching argument contains a source filename.

if not str(argument).endswith(".c"):

continue

for path in extra_options.keys():

if path in str(argument):

if compiler_type == "bcpp":

compiler = new_cmd.pop()

# Borland accepts a source file name at the end,

# insert the options before it

new_cmd.extend(extra_options[path])

new_cmd.append(compiler)

else:

new_cmd.extend(extra_options[path])

# path component is found, no need to search any further

break

self.__spawn(new_cmd, **kwargs)

ccompiler.CCompiler.spawn = spawn # type: ignore[method-assign]