# vim: set sts=2 ts=8 sw=2 tw=99 et ft=python:
import os, sys, shutil

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))

def SetArchFlags(compiler):
  if compiler.behavior == 'gcc':
    if compiler.target.arch == 'x86_64':
      compiler.cflags += ['-fPIC']
  elif compiler.like('msvc'):
    if compiler.target.arch == 'x86_64':
      compiler.defines += ['WIN64']

class ExtensionConfig(object):
  def __init__(self):
    self.sdk_manifests = []
    self.sdks = {}
    self.sdk_targets = []
    self.binaries = []
    self.extensions = []
    self.generated_headers = None
    self.ext_root = None
    self.sdk_helper = None
    self.hl2sdk_manifests_path = None
    self.mms_root = None
    self.sm_root = None
    self.all_targets = []
    self.target_archs = set()
    self.libsafetyhook = {}

    if builder.options.targets:
      target_archs = builder.options.targets.split(',')
    else:
      target_archs = ['x86', 'x86_64']

    for arch in target_archs:
        try:
            cxx = builder.DetectCxx(target_arch = arch)
            self.target_archs.add(cxx.target.arch)
        except Exception as e:
            # Error if archs were manually overridden.
            if builder.options.targets:
                raise
            print('Skipping target {}: {}'.format(arch, e))
            continue
        self.all_targets.append(cxx)

    if not self.all_targets:
        raise Exception('No suitable C/C++ compiler was found.')

  def use_auto_versioning(self):
    if builder.backend != 'amb2':
      return False
    return not getattr(builder.options, 'disable_auto_versioning', False)

  def AddVersioning(self, binary, project):
    if binary.compiler.target.platform == 'windows':
      binary.sources += [
        os.path.join(builder.sourcePath, 'src', 'version.rc')
      ]
      binary.compiler.rcdefines += [
        'PROJECT_NAME="{0}"'.format(project.name),
        'BINARY_NAME="{0}"'.format(binary.outputFile),
        'RC_COMPILE',
      ]
    elif binary.compiler.target.platform == 'mac':
      if binary.type == 'library':
        binary.compiler.postlink += [
          '-compatibility_version', '1.0.0',
          '-current_version', self.productVersion
        ]
    if self.use_auto_versioning():
      binary.compiler.sourcedeps += self.generated_headers
    return binary

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

  def detectProductVersion(self):
    product_version_path = os.path.join(self.ext_root, 'product.version')
    builder.AddConfigureFile(product_version_path)

    # For OS X dylib versioning
    import re
    with open(product_version_path, 'r') as fp:
      productContents = fp.read()
    m = re.match(r'(\\d+)\.(\\d+)\.(\\d+).*', productContents)
    if m == None:
      self.productVersion = '1.0.0'
    else:
      major, minor, release = m.groups()
      self.productVersion = '{0}.{1}.{2}'.format(major, minor, release)

  def findSdkPath(self, sdk_name):
    dir_name = 'hl2sdk-{}'.format(sdk_name)
    if builder.options.hl2sdk_root:
      sdk_path = os.path.join(builder.options.hl2sdk_root, dir_name)
    else:
      sdk_path = ResolveEnvPath('HL2SDK{}'.format(sdk_name.upper()), dir_name)

    if not sdk_path or not os.path.isdir(sdk_path):
      raise Exception('Could not find a source copy of HL2 SDK: {}'.format(sdk_name))
    sdk_path = Normalize(sdk_path)

    return sdk_path

  def shouldIncludeSdk(self, sdk):
    return not sdk.get('source2', False)

  def detectExtensionRoot(self):
    if builder.options.ext_root:
      self.ext_root = os.path.realpath(builder.options.ext_root)

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

  def detectSDKs(self):
    sdk_list = builder.options.sdks.split(' ')

    if builder.options.hl2sdk_manifests_path:
      self.hl2sdk_manifests_path = os.path.realpath(builder.options.hl2sdk_manifests_path)
    else:
      self.hl2sdk_manifests_path = ResolveEnvPath('HL2SDK_MANIFESTS', 'hl2sdk-manifests')

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

    hl2sdk_manifests_ambuild_path = os.path.join(self.hl2sdk_manifests_path, 'SdkHelpers.ambuild')
    if not os.path.exists(hl2sdk_manifests_ambuild_path):
      raise Exception('Could not find SdkHelpers.ambuild in the given HL2SDK Manifest path!')
    hl2sdk_manifests_ambuild_path = Normalize(hl2sdk_manifests_ambuild_path)

    self.sdk_helper = builder.Eval(hl2sdk_manifests_ambuild_path, {
      'Project': 'sm-extension'
    })

    self.sdk_helper.sdk_filter = self.shouldIncludeSdk
    self.sdk_helper.find_sdk_path = self.findSdkPath
    self.sdk_helper.findSdks(builder, self.all_targets, sdk_list)

    self.sdks = self.sdk_helper.sdks
    self.sdk_manifests = self.sdk_helper.sdk_manifests
    self.sdk_targets = self.sdk_helper.sdk_targets

    if len(self.sdks) < 1 and len(sdk_list) and not use_none:
      raise Exception('No applicable SDKs were found, nothing to do')

    if builder.options.sm_path:
      self.sm_root = os.path.realpath(builder.options.sm_path)
    else:
      self.sm_root = ResolveEnvPath('SOURCEMOD112', 'sourcemod-1.12')
      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 builder.options.mms_path:
      self.mms_root = builder.options.mms_path
    else:
      self.mms_root = ResolveEnvPath('MMSOURCE112', 'mmsource-1.12')
      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):
    if not set(self.target_archs).issubset(['x86', 'x86_64']):
      raise Exception('Unknown target architecture: {0}'.format(self.target_archs))

    for cxx in self.all_targets:
        self.configure_cxx(cxx)

  def configure_cxx(self, cxx):
    if cxx.family == 'msvc':
      if cxx.version < 1914 and builder.options.generator != 'vs':
        raise Exception(f'Only MSVC 2017 15.7 and later are supported, full C++17 support is required. ({str(cxx.version)} < 1914)')
    elif cxx.family == 'gcc':
      if cxx.version < 'gcc-9':
        raise Exception('Only GCC versions 9 or later are supported, full C++17 support is required.')
    elif cxx.family == 'clang':
      if cxx.version < 'clang-5':
        raise Exception('Only clang versions 5 or later are supported, full C++17 support is required.')

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

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

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

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

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

    if self.use_auto_versioning():
      cxx.defines += ['SM_GENERATED_BUILD']
      cxx.includes += [
        os.path.join(builder.buildPath, 'includes')
      ]

  def configure_gcc(self, cxx):
    cxx.defines += [
      'stricmp=strcasecmp',
      '_stricmp=strcasecmp',
      '_snprintf=snprintf',
      '_vsnprintf=vsnprintf',
      'HAVE_STDINT_H',
      'GNUC',
    ]
    cxx.cflags += [
      '-pipe',
      '-fno-strict-aliasing',
      '-Wall',
      '-Werror',
      '-Wno-unused',
      '-Wno-switch',
      '-Wno-array-bounds',
      '-msse',
      '-Wno-unknown-pragmas',
      '-Wno-dangling-else',
      '-fvisibility=hidden',
    ]

    cxx.cxxflags += [
      '-std=c++17',
      '-fno-threadsafe-statics',
      '-Wno-non-virtual-dtor',
      '-Wno-overloaded-virtual',
      '-Wno-register',
      '-fvisibility-inlines-hidden',
    ]

    have_gcc = cxx.family == 'gcc'
    have_clang = cxx.family == 'clang'
    if cxx.version >= 'clang-3.9' or cxx.version == 'clang-3.4' or cxx.version > 'apple-clang-6.0':
      cxx.cxxflags += ['-Wno-expansion-to-defined']
    if cxx.version == 'clang-3.9' or cxx.version == 'apple-clang-8.0':
      cxx.cflags += ['-Wno-varargs']
    if cxx.version >= 'clang-3.4' or cxx.version >= 'apple-clang-7.0':
      cxx.cxxflags += ['-Wno-inconsistent-missing-override']
    if cxx.version >= 'clang-2.9' or cxx.version >= 'apple-clang-3.0':
      cxx.cxxflags += ['-Wno-null-dereference']
    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-9.0':
      cxx.cxxflags += ['-Wno-class-memaccess', '-Wno-packed-not-aligned']
    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' or cxx.version >= 'apple-clang-12.0':
      cxx.cflags += [
        '-Wno-implicit-int-float-conversion',
        '-Wno-tautological-overlap-compare',
      ]

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

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

  def configure_msvc(self, cxx):
    if 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',
      '/std:c++17',
    ]
    cxx.linkflags += [
      '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 builder.options.opt == '1':
      cxx.cflags += ['/Ox', '/Zo']
      cxx.linkflags += ['/OPT:ICF', '/OPT:REF']

    if 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', '_LINUX', 'POSIX', '_FILE_OFFSET_BITS=64']
    cxx.linkflags += ['-lm']
    if cxx.family == 'gcc':
      cxx.linkflags += ['-static-libgcc']
    elif cxx.family == 'clang':
      cxx.linkflags += ['-lgcc_eh']

  def configure_mac(self, cxx):
    cxx.defines += ['OSX', '_OSX', 'POSIX', 'KE_ABSOLUTELY_NO_STL']
    cxx.cflags += ['-mmacosx-version-min=10.7']
    cxx.linkflags += [
      '-mmacosx-version-min=10.7',
      '-stdlib=libc++',
      '-lc++',
    ]
    cxx.cxxflags += ['-stdlib=libc++']

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

  def ConfigureForExtension(self, context, binary):
    binary.compiler.cxxincludes += [
      os.path.join(context.sourcePath),
      os.path.join(context.sourcePath, 'src'),
      os.path.join(context.sourcePath, 'sdk'),
      os.path.join(self.ext_root, 'src'),
      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 binary

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

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

    defines = ['RAD_TELEMETRY_DISABLED']

    for other_sdk in self.sdk_manifests:
      defines += ['SE_{}={}'.format(other_sdk['define'], other_sdk['code'])]

    compiler.defines += defines

    self.sdk_helper.configureCxx(context, binary, sdk)
    for i in range(len(compiler.cxxincludes)):
        compiler.cxxincludes[i] = Normalize(compiler.cxxincludes[i])

    return binary

  def AddCDetour(self, binary):
    sm_public_path = os.path.join(self.sm_root, 'public')

    if os.path.exists(os.path.join(sm_public_path, 'safetyhook')):
      binary.sources += [ os.path.join(sm_public_path, 'CDetour', 'detours.cpp') ]
      binary.compiler.cxxincludes += [ os.path.join(builder.sourcePath, 'safetyhook', 'include') ]

      for task in self.libsafetyhook:
        if task.target.arch == binary.compiler.target.arch:
          binary.compiler.linkflags += [task.binary]
          return
      raise Exception('No suitable build of safetyhook was found.')
    else:
      binary.sources += [
        os.path.join(sm_public_path, 'CDetour', 'detours.cpp'),
        os.path.join(sm_public_path, 'asm', 'asm.c'),
      ]
      # sm1.10+
      libudis_folder = os.path.join(sm_public_path, 'libudis86')
      if os.path.isdir(libudis_folder):
        binary.compiler.defines += ['HAVE_STRING_H']
        binary.sources += [
          os.path.join(libudis_folder, 'decode.c'),
          os.path.join(libudis_folder, 'itab.c'),
          os.path.join(libudis_folder, 'syn-att.c'),
          os.path.join(libudis_folder, 'syn-intel.c'),
          os.path.join(libudis_folder, 'syn.c'),
          os.path.join(libudis_folder, 'udis86.c'),
        ]

  def HL2Config(self, project, context, compiler, name, sdk):
    binary = project.Configure(compiler, name,
                               '{0} - {1} {2}'.format(self.tag, sdk['name'], compiler.target.arch))
    self.AddVersioning(binary, project)
    return self.ConfigureForHL2(context, binary, sdk)

  def HL2ExtConfig(self, project, context, compiler, name, sdk):
    binary = project.Configure(compiler, name,
                               '{0} - {1} {2}'.format(self.tag, sdk['name'], compiler.target.arch))
    self.AddVersioning(binary, project)
    self.ConfigureForHL2(context, binary, sdk)
    self.ConfigureForExtension(context, binary)
    return binary

class SafetyHookShim(object):
  def __init__(self):
    self.all_targets = {}
    self.libsafetyhook = {}

if getattr(builder, 'target', None) is not None:
    sys.stderr.write("Your output folder was configured for AMBuild 2.1.\n")
    sys.stderr.write("Please remove your output folder and reconfigure to continue.\n")
    os._exit(1)

Extension = ExtensionConfig()
Extension.detectExtensionRoot()
Extension.detectProductVersion()
Extension.detectSDKs()
Extension.configure()

if Extension.use_auto_versioning():
  Extension.generated_headers = builder.Build(os.path.join('buildbot', 'Versioning'), { 'Extension': Extension })

if os.path.exists(os.path.join(Extension.sm_root, 'public', 'safetyhook')):
    # we need to pull safetyhook in locally because ambuild does not take kindly to outside relpaths
    safetyhook_dest = os.path.join(builder.sourcePath, 'safetyhook')
    shutil.copytree(os.path.join(Extension.sm_root, 'public', 'safetyhook'), safetyhook_dest, dirs_exist_ok=True)

    SafetyHook = SafetyHookShim()
    SafetyHook.all_targets = Extension.all_targets
    builder.Build(os.path.join('safetyhook', 'AMBuilder'), { 'SafetyHook': SafetyHook })
    Extension.libsafetyhook = SafetyHook.libsafetyhook

# This will clone the list and each cxx object as we recurse, preventing child
# scripts from messing up global state.
builder.targets = builder.CloneableList(Extension.all_targets)

# Hack to please ambuild because it does not handle having outside ambuilder
# https://github.com/alliedmodders/ambuild/blob/7736f4b155851fa354d88a6d9103e284c7d7ddc7/ambuild2/frontend/v2_2/cpp/builders.py#L298
# self.outputPath is wrongly built
ext_ambuilder_src_path = os.path.join(Extension.ext_root, 'src', 'AMBuilder')
ext_ambuilder_dest_path = os.path.join(builder.sourcePath, 'src', 'AMBuilder')
if os.path.exists(ext_ambuilder_src_path):
  shutil.copy(ext_ambuilder_src_path, ext_ambuilder_dest_path)

# Add additional buildscripts here
BuildScripts = [
  os.path.join('src', 'AMBuilder'),
  os.path.join('buildbot', 'PackageScript'),
  os.path.join('buildbot', 'BreakpadSymbols'),
]

builder.Build(BuildScripts, { 'Extension': Extension })