AMBuildScript

import os
import enum

from typing import List


class Platform(str, enum.Enum):
    """Aliases for target platform from AMB2 builder"""
    WINDOWS = 'win'
    LINUX = 'linux'
    MAC = 'mac'

    def __str__(self):
        return self.name

    @classmethod
    def get(cls, name: str):
        return cls[name.upper()]


class SourceEngine(enum.IntEnum):
    """Source Engine

    Different games uses different Source Engine versions
    If you you want make compatible Extension with more than one SE
    you should use this codes to split engine-specific code

    Example:

    .. code-block:: cpp

        // SOURCE_ENGINE passed builder

        #if SOURCE_ENGINE >= SE_LEFT4DEAD2
            // Here you place code for Left4Dead2 or newer
        #else
            // Here older
        #endif

    """
    TF2 = 11
    LEFT4DEAD = 12
    LEFT4DEAD2 = 15
    CSS = 6
    CSGO = 21

    def make_define(self):
        return f'SE_{self.name}={self.value}'

    @property
    def code(self):
        return self.value

    @classmethod
    def all_defines(cls):
        return [se.make_define() for se in cls.__members__.values()]


class SDK:
    def __init__(
        self,
        short_name: str,
        envvar: str,
        engine: SourceEngine,
        platforms: List[Platform],
    ):
        self.folder = 'hl2sdk-' + short_name
        self.envvar = envvar
        self.engine = engine
        self.platforms = platforms
        self.short_name = short_name
        self.path = None  # Actual path

    def __repr__(self):
        return f'SDK(n:{self.short_name}, p:{[str(p) for p in self.platforms]})'


WinOnly = [Platform.WINDOWS]
WinLinux = [Platform.WINDOWS, Platform.LINUX]
WinLinuxMac = WinLinux + [Platform.MAC]

PossibleSDKs = [
    SDK('tf2', 'HL2SDKTF2', SourceEngine.TF2, WinLinuxMac),
    SDK('l4d', 'HL2SDKL4D', SourceEngine.LEFT4DEAD, WinLinuxMac),
    SDK('l4d2', 'HL2SDKL4D2', SourceEngine.LEFT4DEAD2, WinLinuxMac),
    SDK('csgo', 'HL2SDKCSGO', SourceEngine.CSGO, WinLinuxMac),
    SDK('css', 'HL2SDKCSS', SourceEngine.CSS, WinLinuxMac),
]


def ResolveEnvPath(env, folder):
    if env in os.environ:
        path = os.environ[env]
        if os.path.isdir(path):
            return path
        return None

    head = os.getcwd()
    oldhead = None
    while head != None and head != oldhead:
        path = os.path.join(head, folder)
        if os.path.isdir(path):
            return path
        oldhead = head
        head, tail = os.path.split(head)

    return None


def Normalize(path):
    return os.path.abspath(os.path.normpath(path))


class BuildInfo:
    def __init__(self, builder):
        self.builder = builder
        self.sdks = {}
        self.binaries = []
        self.extensions = []
        self.generated_headers = None
        self.mms_root = None
        self.sm_root = None

    @property
    def tag(self):
        if self.builder.options.debug == '1':
            return 'Debug'
        return 'Release'

    def detectSDKs(self):
        sdk_list = self.builder.options.sdks.split(',')
        use_all = sdk_list[0] == 'all'
        use_present = sdk_list[0] == 'present'

        target_platform = Platform.get(self.builder.target_platform)
        for sdk in PossibleSDKs:
            if target_platform not in sdk.platforms:
                continue

            if self.builder.options.hl2sdk_root:
                sdk_path = os.path.join(self.builder.options.hl2sdk_root, sdk.folder)
            else:
                sdk_path = ResolveEnvPath(sdk.envvar, sdk.folder)

            if sdk_path is None or not os.path.isdir(sdk_path):
                if use_all or sdk.short_name in sdk_list:
                    raise RuntimeError('Could not find a valid path for {0}'.format(sdk.envvar))
                continue

            if use_all or use_present or sdk.short_name in sdk_list:
                sdk.path = Normalize(sdk_path)
                self.sdks[sdk.short_name] = sdk

        if len(self.sdks) < 1:
            raise RuntimeError('At least one SDK must be available.')

        if self.builder.options.sm_path:
            self.sm_root = self.builder.options.sm_path
        else:
            self.sm_root = ResolveEnvPath('SOURCEMOD18', 'sourcemod-1.8')
            if not self.sm_root:
                self.sm_root = ResolveEnvPath('SOURCEMOD', 'sourcemod')
            if not self.sm_root:
                self.sm_root = ResolveEnvPath('SOURCEMOD_DEV', 'sourcemod-central')

        if not self.sm_root or not os.path.isdir(self.sm_root):
            raise Exception('Could not find a source copy of SourceMod')
        self.sm_root = Normalize(self.sm_root)

        if self.builder.options.mms_path:
            self.mms_root = self.builder.options.mms_path
        else:
            self.mms_root = ResolveEnvPath('MMSOURCE110', 'mmsource-1.10')
            if not self.mms_root:
                self.mms_root = ResolveEnvPath('MMSOURCE', 'metamod-source')
            if not self.mms_root:
                self.mms_root = ResolveEnvPath('MMSOURCE_DEV', 'mmsource-central')

        if not self.mms_root or not os.path.isdir(self.mms_root):
            raise Exception('Could not find a source copy of Metamod:Source')
        self.mms_root = Normalize(self.mms_root)

    def configure(self):
        cxx = self.builder.DetectCompilers()

        if cxx.like('gcc'):
            self.configure_gcc(cxx)
        elif cxx.vendor == 'msvc':
            self.configure_msvc(cxx)

        # Optimizaiton
        if self.builder.options.opt == '1':
            cxx.defines += ['NDEBUG']

        # Debugging
        if self.builder.options.debug == '1':
            cxx.defines += ['DEBUG', '_DEBUG']

        # Platform-specifics
        if self.builder.target_platform == 'linux':
            self.configure_linux(cxx)
        elif self.builder.target_platform == 'mac':
            self.configure_mac(cxx)
        elif self.builder.target_platform == 'windows':
            self.configure_windows(cxx)

        # Finish up.
        cxx.includes += [
            os.path.join(self.sm_root, 'public'),
        ]

    def configure_gcc(self, cxx):
        cxx.defines += [
            'stricmp=strcasecmp',
            '_stricmp=strcasecmp',
            '_snprintf=snprintf',
            '_vsnprintf=vsnprintf',
            'HAVE_STDINT_H',
            'HAVE_STRING_H',
            'GNUC',
        ]
        cxx.cflags += [
            '-pipe',
            '-fno-strict-aliasing',
            '-Wall',
            '-Werror',
            '-Wno-unused',
            '-Wno-switch',
            '-Wno-array-bounds',
            '-msse',
            '-m32',
            '-fvisibility=hidden',
        ]
        cxx.cxxflags += [
            '-std=c++14',
            '-fno-exceptions',
            '-fno-threadsafe-statics',
            '-Wno-non-virtual-dtor',
            '-Wno-overloaded-virtual',
            '-fvisibility-inlines-hidden',
        ]
        cxx.linkflags += ['-m32']

        have_gcc = cxx.vendor == 'gcc'
        have_clang = cxx.vendor == 'clang'
        if cxx.version >= 'clang-3.9' or cxx.version >= 'apple-clang-10.0':
            cxx.cxxflags += ['-Wno-expansion-to-defined']
        if cxx.version >= 'clang-3.6':
            cxx.cxxflags += ['-Wno-inconsistent-missing-override']
        if have_clang or (cxx.version >= 'gcc-4.6'):
            cxx.cflags += ['-Wno-narrowing']
        if have_clang or (cxx.version >= 'gcc-4.7'):
            cxx.cxxflags += ['-Wno-delete-non-virtual-dtor']
        if cxx.version >= 'gcc-4.8':
            cxx.cflags += ['-Wno-unused-result']
        if cxx.version >= 'gcc-8.0':
            cxx.cxxflags += ['-Wno-class-memaccess']

        if have_clang:
            cxx.cxxflags += ['-Wno-implicit-exception-spec-mismatch']
            if cxx.version >= 'apple-clang-5.1' or cxx.version >= 'clang-3.4':
                cxx.cxxflags += ['-Wno-deprecated-register']
            else:
                cxx.cxxflags += ['-Wno-deprecated']
            cxx.cflags += ['-Wno-sometimes-uninitialized']

        # Work around SDK warnings.
        if cxx.version >= 'clang-10.0':
            cxx.cflags += [
                '-Wno-implicit-int-float-conversion',
                '-Wno-tautological-overlap-compare',
            ]

        if have_gcc:
            cxx.cflags += ['-mfpmath=sse']

        if self.builder.options.opt == '1':
            cxx.cflags += ['-O3']

        # Don't omit the frame pointer.
        cxx.cflags += ['-fno-omit-frame-pointer']

    def configure_msvc(self, cxx):
        if self.builder.options.debug == '1':
            cxx.cflags += ['/MTd']
            cxx.linkflags += ['/NODEFAULTLIB:libcmt']
        else:
            cxx.cflags += ['/MT']
        cxx.defines += [
            '_CRT_SECURE_NO_DEPRECATE',
            '_CRT_SECURE_NO_WARNINGS',
            '_CRT_NONSTDC_NO_DEPRECATE',
            '_ITERATOR_DEBUG_LEVEL=0',
        ]
        cxx.cflags += [
            '/W3',
        ]
        cxx.cxxflags += [
            '/EHsc',
            '/GR-',
            '/TP',
        ]
        cxx.linkflags += [
            '/MACHINE:X86',
            'kernel32.lib',
            'user32.lib',
            'gdi32.lib',
            'winspool.lib',
            'comdlg32.lib',
            'advapi32.lib',
            'shell32.lib',
            'ole32.lib',
            'oleaut32.lib',
            'uuid.lib',
            'odbc32.lib',
            'odbccp32.lib',
        ]

        if self.builder.options.opt == '1':
            cxx.cflags += ['/Ox', '/Zo']
            cxx.linkflags += ['/OPT:ICF', '/OPT:REF']

        if self.builder.options.debug == '1':
            cxx.cflags += ['/Od', '/RTC1']

        # This needs to be after our optimization flags which could otherwise disable it.
        # Don't omit the frame pointer.
        cxx.cflags += ['/Oy-']

    def configure_linux(self, cxx):
        cxx.defines += ['_LINUX', 'POSIX']
        cxx.linkflags += ['-lm']
        if cxx.vendor == 'gcc':
            cxx.linkflags += ['-static-libgcc']
        elif cxx.vendor == 'clang':
            cxx.linkflags += ['-lgcc_eh']

    def configure_mac(self, cxx):
        cxx.defines += ['OSX', '_OSX', 'POSIX']
        cxx.cflags += ['-mmacosx-version-min=10.5']
        cxx.linkflags += [
            '-mmacosx-version-min=10.5',
            '-arch', 'i386',
            '-lstdc++',
            '-stdlib=libc++',
        ]
        cxx.cxxflags += ['-stdlib=libc++']

    def configure_windows(self, cxx):
        cxx.defines += ['WIN32', '_WINDOWS']

    def ConfigureForExtension(self, context, compiler):
        compiler.cxxincludes += [
            os.path.join(context.currentSourcePath),
            os.path.join(self.sm_root, 'public'),
            os.path.join(self.sm_root, 'public', 'extensions'),
            os.path.join(self.sm_root, 'sourcepawn', 'include'),
            os.path.join(self.sm_root, 'public', 'amtl', 'amtl'),
            os.path.join(self.sm_root, 'public', 'amtl'),
        ]
        return compiler

    def ConfigureForHL2(self, binary, sdk: SDK):
        compiler = binary.compiler

        compiler.cxxincludes += [
            os.path.join(self.mms_root, 'core'),
            os.path.join(self.mms_root, 'core', 'sourcehook'),
        ]

        compiler.defines += SourceEngine.all_defines()

        paths = [
            ['public'],
            ['public', 'engine'],
            ['public', 'mathlib'],
            ['public', 'vstdlib'],
            ['public', 'tier0'],
            ['public', 'tier1'],
            ['public', 'game', 'server'],
            ['public', 'toolframework'],
            ['game', 'shared'],
            ['common']
        ]

        compiler.defines += [f'SOURCE_ENGINE={sdk.engine.code}']

        if compiler.like('msvc'):
            compiler.defines += ['COMPILER_MSVC', 'COMPILER_MSVC32']
        else:
            compiler.defines += ['COMPILER_GCC']

        if sdk.short_name in ['css', 'hl2dm', 'dods', 'sdk2013', 'bms', 'tf2', 'l4d', 'nucleardawn', 'l4d2', 'dota']:
            if self.builder.target_platform in ['linux', 'mac']:
                compiler.defines += ['NO_HOOK_MALLOC', 'NO_MALLOC_OVERRIDE']

        if sdk.short_name == 'csgo' and self.builder.target_platform == 'linux':
            compiler.linkflags += ['-lstdc++']

        # For everything after Swarm, this needs to be defined for entity networking
        # to work properly with sendprop value changes.
        if sdk.short_name in ['blade', 'insurgency', 'doi', 'csgo']:
            compiler.defines += ['NETWORK_VARS_ENABLED']

        for path in paths:
            compiler.cxxincludes += [os.path.join(sdk.path, *path)]

        if self.builder.target_platform == 'linux':
            if sdk.short_name == 'episode1':
                 lib_folder = os.path.join(sdk.path, 'linux_sdk')
            elif sdk.short_name in ['sdk2013', 'bms']:
                 lib_folder = os.path.join(sdk.path, 'lib', 'public', 'linux32')
            else:
                 lib_folder = os.path.join(sdk.path, 'lib', 'linux')
        elif self.builder.target_platform == 'mac':
            if sdk.short_name in ['sdk2013', 'bms']:
                 lib_folder = os.path.join(sdk.path, 'lib', 'public', 'osx32')
            else:
                 lib_folder = os.path.join(sdk.path, 'lib', 'mac')

        if self.builder.target_platform in ['linux', 'mac']:
            if sdk.short_name in ['sdk2013', 'bms']:
                 compiler.postlink += [
                     compiler.Dep(os.path.join(lib_folder, 'tier1.a')),
                     compiler.Dep(os.path.join(lib_folder, 'mathlib.a'))
                 ]
            else:
                 compiler.postlink += [
                     compiler.Dep(os.path.join(lib_folder, 'tier1_i486.a')),
                     compiler.Dep(os.path.join(lib_folder, 'mathlib_i486.a'))
                 ]

            if sdk.short_name in ['blade', 'insurgency', 'csgo', 'dota']:
                compiler.postlink += [compiler.Dep(os.path.join(lib_folder, 'interfaces_i486.a'))]

        dynamic_libs = []
        if self.builder.target_platform == 'linux':
            if sdk.short_name in ['css', 'hl2dm', 'dods', 'tf2', 'sdk2013', 'bms', 'nucleardawn', 'l4d2', 'insurgency']:
                dynamic_libs = ['libtier0_srv.so', 'libvstdlib_srv.so']
            elif sdk.short_name in ['l4d', 'blade', 'insurgency', 'csgo', 'dota']:
                dynamic_libs = ['libtier0.so', 'libvstdlib.so']
            else:
                dynamic_libs = ['tier0_i486.so', 'vstdlib_i486.so']

        elif self.builder.target_platform == 'mac':
            compiler.linkflags.append('-liconv')
            dynamic_libs = ['libtier0.dylib', 'libvstdlib.dylib']

        elif self.builder.target_platform == 'windows':
            libs = ['tier0', 'tier1', 'vstdlib', 'mathlib']
            if sdk.short_name in ['swarm', 'blade', 'insurgency', 'csgo', 'dota']:
                libs.append('interfaces')

            for lib in libs:
                lib_path = os.path.join(sdk.path, 'lib', 'public', lib) + '.lib'
                compiler.linkflags.append(compiler.Dep(lib_path))

        for library in dynamic_libs:
            source_path = os.path.join(lib_folder, library)
            output_path = os.path.join(binary.localFolder, library)

            def make_linker(source_path, output_path):
                def link(context, binary):
                    cmd_node, (output,) = context.AddSymlink(source_path, output_path)
                    return output
                return link

            linker = make_linker(source_path, output_path)
            compiler.linkflags.append(compiler.Dep(library, linker))

        return binary

    def HL2Library(self, context, name, sdk):
        binary = context.compiler.Library(name)
        self.ConfigureForExtension(context, binary.compiler)
        return self.ConfigureForHL2(binary, sdk)

    def HL2Project(self, context, name):
        project = context.compiler.LibraryProject(name)
        self.ConfigureForExtension(context, project.compiler)
        return project

    def HL2Config(self, project, name, sdk):
        binary = project.Configure(name, '{0} - {1}'.format(self.tag, sdk.short_name))
        return self.ConfigureForHL2(binary, sdk)


build_info = BuildInfo(builder)
build_info.detectSDKs()
build_info.configure()

# Add additional buildscripts here
BuildScripts = [
    'AMBuilder',
]

if builder.backend == 'amb2':
    BuildScripts += [
        'PackageScript',
    ]

builder.RunBuildScripts(BuildScripts, { 'build_info': build_info })