# vim: set sts=2 ts=8 sw=2 tw=99 et ft=python:

 import os, sys

 class SDK(object):

   def __init__(self, sdk, ext, aDef, name, platform, dir):

     if dir == 'ep1':

       folder = 'hl2sdk'

     else:

       folder = 'hl2sdk-' + dir

     self.envvar = sdk

     self.ext = ext

     self.code = aDef

     self.define = name

     self.platform = platform

     self.folder = folder # Default folder name.

     self.name = dir

     self.path = None # Actual path

 WinOnly = ['windows']

 WinLinux = ['windows', 'linux']

 WinLinuxMac = ['windows', 'linux', 'mac']

 PossibleSDKs = {

   'ep1':  SDK('HL2SDK', '1.ep1', '1', 'EPISODEONE', WinLinux, 'ep1'),

   'ep2':  SDK('HL2SDKOB', '2.ep2', '3', 'ORANGEBOX', WinLinux, 'ob'),

   'css':  SDK('HL2SDKCSS', '2.css', '6', 'CSS', WinLinuxMac, 'css'),

   'hl2dm':  SDK('HL2SDKHL2DM', '2.hl2dm', '7', 'HL2DM', WinLinuxMac, 'hl2dm'),

   'dods': SDK('HL2SDKDODS', '2.dods', '8', 'DODS', WinLinuxMac, 'dods'),

   'sdk2013': SDK('HL2SDK2013', '2.sdk2013', '9', 'SDK2013', WinLinuxMac, '2013'),

   'tf2':  SDK('HL2SDKTF2', '2.tf2', '10', 'TF2', WinLinuxMac, 'tf2'),

   'l4d':  SDK('HL2SDKL4D', '2.l4d', '11', 'LEFT4DEAD', WinLinuxMac, 'l4d'),

   'nd': SDK('HL2SDKND', '2.nd', '12', 'NUCLEARDAWN', WinLinuxMac, 'nd'),

   'l4d2': SDK('HL2SDKL4D2', '2.l4d2', '13', 'LEFT4DEAD2', WinLinuxMac, 'l4d2'),

   'darkm':  SDK('HL2SDK-DARKM', '2.darkm', '2', 'DARKMESSIAH', WinOnly, 'darkm'),

   'swarm':  SDK('HL2SDK-SWARM', '2.swarm', '14', 'ALIENSWARM', WinOnly, 'swarm'),

   'bgt':  SDK('HL2SDK-BGT', '2.bgt', '4', 'BLOODYGOODTIME', WinOnly, 'bgt'),

   'eye':  SDK('HL2SDK-EYE', '2.eye', '5', 'EYE', WinOnly, 'eye'),

   'csgo': SDK('HL2SDKCSGO', '2.csgo', '18', 'CSGO', WinLinuxMac, 'csgo'),

   'dota': SDK('HL2SDKDOTA', '2.dota', '19', 'DOTA', WinOnly, 'dota'),

   'portal2':  SDK('HL2SDKPORTAL2', '2.portal2', '15', 'PORTAL2', [], 'portal2'),

   'blade':  SDK('HL2SDKBLADE', '2.blade', '16', 'BLADE', WinLinux, 'blade'),

   'insurgency':  SDK('HL2SDKINSURGENCY', '2.insurgency', '17', 'INSURGENCY', WinLinuxMac, 'insurgency'),

 }

 def ResolveEnvPath(env, folder):

   if env in os.environ:

     path = os.environ[env]

     if os.path.isdir(path):

       return path

   else:

     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

 class SMConfig(object):

   def __init__(self):

     self.sdks = {}

     self.binaries = []

     self.mms_root = None

     self.mysql_root = None

   def detectSDKs(self):

     sdk_list = builder.options.sdks.split(',')

     use_all = sdk_list[0] == 'all'

     use_present = sdk_list[0] == 'present'

     for sdk_name in PossibleSDKs:

       sdk = PossibleSDKs[sdk_name]

       if builder.target_platform in sdk.platform:

         sdk_path = ResolveEnvPath(sdk.envvar, sdk.folder)

         if sdk_path is None:

           if use_all or sdk_name in sdk_list:

             raise Exception('Could not find a valid path for {0}'.format(sdk.envvar))

           continue

         if use_all or use_present or sdk_name in sdk_list:

           sdk.path = sdk_path

           self.sdks[sdk_name] = sdk

     if len(self.sdks) < 1:

       raise Exception('At least one SDK must be available.')

     self.mms_root = ResolveEnvPath('MMSOURCE110', 'mmsource-1.10')

     if not self.mms_root:

       self.mms_root = ResolveEnvPath('MMSOURCE_DEV', 'mmsource-central')

       if not self.mms_root:

         raise Exception('Could not find a source copy of Metamod:Source')

     if builder.options.hasMySql:

       for i in range(7):

         self.mysql_root = ResolveEnvPath('MYSQL5', 'mysql-5.' + str(i))

         if self.mysql_root:

           break

       if not self.mysql_root:

         raise Exception('Could not find a path to MySQL!')

   def configure(self):

     cfg = builder.DetectCompilers()

     cxx = cfg.cxx

     if cxx.behavior is 'gcc':

       cfg.defines += [

         'stricmp=strcasecmp',

         '_stricmp=strcasecmp',

         '_snprintf=snprintf',

         '_vsnprintf=vsnprintf',

         'HAVE_STDINT_H',

         'GNUC',

       ]

       cfg.cflags += [

         '-pipe',

         '-fno-strict-aliasing',

         '-Wall',

         '-Werror',

         '-Wno-uninitialized',

         '-Wno-unused',

         '-Wno-switch',

         '-msse',

         '-m32',

         '-g3',

       ]

       have_gcc = cxx.name is 'gcc'

       have_clang = cxx.name is 'clang'

       if have_clang or (have_gcc and cxx.majorVersion >= 4):

         cfg.cflags += ['-fvisibility=hidden']

         cfg.cxxflags += ['-fvisibility-inlines-hidden']

         if have_clang or (have_gcc and cxx.minorVersion >= 6):

           cfg.cflags += ['-Wno-narrowing']

         if (have_gcc and cxx.minorVersion >= 7) or (have_clang and cxx.majorVersion >= 4):

           cfg.cxxflags += ['-Wno-delete-non-virtual-dtor']

       cfg.linkflags += ['-m32']

       cfg.cxxflags += [

         '-fno-exceptions',

         '-fno-threadsafe-statics',

         '-Wno-non-virtual-dtor',

         '-Wno-overloaded-virtual',

       ]

       if have_gcc:

         cfg.cflags += ['-mfpmath=sse']

     elif cxx.name == 'msvc':

       if builder.options.debug == '1':

         cfg.cflags += ['/MTd']

         cfg.linkflags += ['/NODEFAULTLIB:libcmt']

       else:

         cfg.cflags += ['/MT']

       cfg.defines += [

         '_CRT_SECURE_NO_DEPRECATE',

         '_CRT_SECURE_NO_WARNINGS',

         '_CRT_NONSTDC_NO_DEPRECATE',

         '_ITERATOR_DEBUG_LEVEL=0',

       ]

       cfg.cflags += [

         '/W3',

         '/Zi',

       ]

       cfg.cxxflags += [

         '/EHsc',

         '/GR-',

         '/TP',

       ]

       cfg.linkflags += [

         '/DEBUG',

         '/MACHINE:X86',

         '/SUBSYSTEM:WINDOWS',

         'kernel32.lib',

         'user32.lib',

         'gdi32.lib',

         'winspool.lib',

         'comdlg32.lib',

         'advapi32.lib',

         'shell32.lib',

         'ole32.lib',

         'oleaut32.lib',

         'uuid.lib',

         'odbc32.lib',

         'odbccp32.lib',

       ] 

     # Optimization

     if builder.options.opt == '1':

       cfg.defines += ['NDEBUG']

       if cxx.behavior == 'gcc':

         cfg.cflags += ['-O3']

       elif cxx.behavior == 'msvc':

         cfg.cflags += ['/Ox']

         cfg.linkflags += ['/OPT:ICF', '/OPT:REF']

     # Debugging

     if builder.options.debug == '1':

       cfg.defines += ['DEBUG', '_DEBUG']

       if cxx.behavior == 'gcc':

         cfg.cflags += ['-g3']

       elif cxx.behavior == 'msvc':

         cfg.cflags += ['/Od', '/RTC1']

     # This needs to be after our optimization flags which could otherwise disable it.

     if cxx.name == 'msvc':

       # Don't omit the frame pointer.

       cfg.cflags += ['/Oy-']

     # Platform-specifics

     if builder.target_platform == 'linux':

       cfg.defines += ['_LINUX', 'POSIX']

       if cxx.name == 'gcc':

         cfg.linkflags += ['-static-libgcc']

       elif cxx.name == 'clang':

         cfg.linkflags += ['-lgcc_eh']

     elif builder.target_platform == 'mac':

       cfg.defines += ['OSX', '_OSX', 'POSIX']

       cfg.linkflags += [

         '-mmacosx-version-min=10.5',

         '-arch', 'i386',

         '-lstdc++',

       ]

     elif builder.target_platform == 'windows':

       cfg.defines += ['WIN32', '_WINDOWS']

     # Finish up.

     cfg.defines += [

       'SOURCEMOD_BUILD',

     ]

     cfg.includes += [os.path.join(builder.buildPath, 'includes')]

   def LibraryBuilder(self, compiler, name):

     binary = compiler.Library(name)

     return binary

   def ProgramBuilder(self, compiler, name):

     binary = compiler.Program(name)

     return binary

   def Library(self, context, name):

     compiler = context.compiler.clone()

     return self.LibraryBuilder(compiler, name)

   def Program(self, context, name):

     compiler = context.compiler.clone()

     return self.ProgramBuilder(compiler, name)

   def ExtCompiler(self, context):

     compiler = context.compiler.clone()

     compiler.cxxincludes += [

       os.path.join(context.currentSourcePath),

       os.path.join(context.currentSourcePath, 'sdk'),

       os.path.join(builder.sourcePath, 'public'),

       os.path.join(builder.sourcePath, 'public', 'extensions'),

       os.path.join(builder.sourcePath, 'public', 'sourcepawn'),

       os.path.join(builder.sourcePath, 'public', 'amtl'),

     ]

     return compiler

   def HL2Compiler(self, context, sdk):

     compiler = self.ExtCompiler(context)

     if sdk.name == 'ep1':

       mms_path = os.path.join(self.mms_root, 'core-legacy')

     else:

       mms_path = os.path.join(self.mms_root, 'core')

     compiler.cxxincludes += [

       os.path.join(mms_path),

       os.path.join(mms_path, 'sourcehook'),

     ]

     defines = ['SE_' + PossibleSDKs[i].define + '=' + PossibleSDKs[i].code for i in PossibleSDKs]

     compiler.defines += defines

     paths = [

       ['public'],

       ['public', 'engine'],

       ['public', 'mathlib'],

       ['public', 'vstdlib'],

       ['public', 'tier0'],

       ['public', 'tier1']

     ]

     if sdk.name == 'ep1' or sdk.name == 'darkm':

       paths.append(['public', 'dlls'])

       paths.append(['game_shared'])

     else:

       paths.append(['public', 'game', 'server'])

       paths.append(['public', 'toolframework'])

       paths.append(['game', 'shared'])

       paths.append(['common'])

     compiler.defines += ['SOURCE_ENGINE=' + sdk.code]

     if sdk.name == '2013' and compiler.cxx.behavior == 'gcc':

       # The 2013 SDK already has these in public/tier0/basetypes.h

       compiler.defines.remove('stricmp=strcasecmp')

       compiler.defines.remove('_stricmp=strcasecmp')

       compiler.defines.remove('_snprintf=snprintf')

       compiler.defines.remove('_vsnprintf=vsnprintf')

     if sdk.name in ['swarm', 'blade', 'insurgency', 'csgo', 'dota']:

       if compiler.cc.behavior == 'msvc':

         compiler.defines += ['COMPILER_MSVC', 'COMPILER_MSVC32']

       else:

         compiler.defines += ['COMPILER_GCC']

     if sdk.name in ['css', 'hl2dm', 'dods', '2013', 'tf2', 'l4d2']:

       if builder.target_platform in ['linux', 'mac']:

         compiler.defines += ['NO_MALLOC_OVERRIDE']

     for path in paths:

       compiler.cxxincludes += [os.path.join(sdk.path, *path)]

     return compiler

   def ExtLibrary(self, context, name):

     compiler = self.ExtCompiler(context)

     return self.LibraryBuilder(compiler, name)

   def HL2Library(self, context, name, sdk):

     compiler = self.HL2Compiler(context, sdk)

     if builder.target_platform is 'linux':

       if sdk.name is 'ep1':

         lib_folder = os.path.join(sdk.path, 'linux_sdk')

       elif sdk.name is '2013':

         lib_folder = os.path.join(sdk.path, 'lib', 'public', 'linux32')

       else:

         lib_folder = os.path.join(sdk.path, 'lib', 'linux')

     elif builder.target_platform == 'mac':

       if sdk.name is '2013':

         lib_folder = os.path.join(sdk.path, 'lib', 'public', 'osx32')

       else:

         lib_folder = os.path.join(sdk.path, 'lib', 'mac')

     if builder.target_platform in ['linux', 'mac']:

       if sdk.name is '2013':

         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.name in ['blade', 'insurgency', 'csgo', 'dota']:

         compiler.postlink += [compiler.Dep(os.path.join(lib_folder, 'interfaces_i486.a'))]

     binary = self.LibraryBuilder(compiler, name)

     dynamic_libs = []

     if builder.target_platform == 'linux':

       compiler.linkflags[0:0] = ['-lm']

       if sdk.name in ['css', 'hl2dm', 'dods', 'tf2', '2013', 'l4d2']:

         dynamic_libs = ['libtier0_srv.so', 'libvstdlib_srv.so']

       elif sdk.name in ['l4d', 'nd', 'blade', 'insurgency', 'csgo']:

         dynamic_libs = ['libtier0.so', 'libvstdlib.so']

       else:

         dynamic_libs = ['tier0_i486.so', 'vstdlib_i486.so']

     elif builder.target_platform == 'mac':

       binary.compiler.linkflags.append('-liconv')

       dynamic_libs = ['libtier0.dylib', 'libvstdlib.dylib']

     elif builder.target_platform == 'windows':

       libs = ['tier0', 'tier1', 'vstdlib', 'mathlib']

       if sdk.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'

         binary.compiler.linkflags.append(binary.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)

       binary.compiler.linkflags[0:0] = [binary.Dep(library, linker)]

     return binary

 SM = SMConfig()

 SM.detectSDKs()

 SM.configure()

 builder.RunBuildScripts(

   [

     'loader/AMBuilder',

     'core/AMBuilder',

     'core/logic/AMBuilder',

     'extensions/bintools/AMBuilder',

     'extensions/clientprefs/AMBuilder',

     'extensions/cstrike/AMBuilder',

     'extensions/geoip/AMBuilder',

     'extensions/mysql/AMBuilder',

     'extensions/regex/AMBuilder',

     'extensions/sdkhooks/AMBuilder',

     'extensions/sdktools/AMBuilder',

     'extensions/sqlite/AMBuilder',

     'extensions/tf2/AMBuilder',

     'extensions/topmenus/AMBuilder',

     'extensions/updater/AMBuilder',

     'sourcepawn/compiler/AMBuilder',

     'sourcepawn/jit/AMBuilder',

   ],

   {

     'SM': SM

   }

 )