PackageScript

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

# This is where the files will be output to
# package is the default
builder.SetBuildFolder('package' + PackageDirSuffix)

# Add any folders you need to this list
folder_list = [
  'addons/sourcemod/extensions',
  'addons/sourcemod/extensions/x64',
#  'addons/sourcemod/scripting',
#  'addons/sourcemod/scripting/include',
  'addons/sourcemod/gamedata',
  'addons/sourcemod/gamedata/sigsegv',
  'addons/sourcemod/translations',
  'addons/sourcemod/translations/zho',
  'addons/sourcemod/translations/chi',
#  'addons/sourcemod/configs',
  'cfg',
]
if not Extension.optimize_mods_only:
  folder_list += [
    'custom/sigsegv/resource',
    'custom/sigsegv/scripts'
  ]
  if not 'mods_mvm' in Extension.exclude:
    folder_list += [
      'custom/sigsegv/scripts/population'
    ]

if 'x86_64' in Extension.target_archs:
  folder_list.extend([
    'addons/sourcemod/extensions/x64',
  ])

# Create the distribution folder hierarchy.
folder_map = {}
for folder in folder_list:
  norm_folder = os.path.normpath(folder)
  folder_map[folder] = builder.AddFolder(norm_folder)

# Do all straight-up file copies from the source tree.
def CopyFiles(src, dest, files):
  if not dest:
    dest = src
  dest_entry = folder_map[dest]
  for source_file in files:
    source_path = os.path.join(builder.sourcePath, src, source_file)
    builder.AddCopy(source_path, dest_entry)

# Include files
#CopyFiles('include', 'addons/sourcemod/scripting/include',
#  [
#    'sample.inc',
#  ]
#)

# GameData files
CopyFiles('gamedata/sigsegv', 'addons/sourcemod/gamedata/sigsegv',
  [
    'client.txt',
    'convars.txt',
    'custom_attributes.txt',
    'datamaps.txt',
    'debugoverlay.txt',
    'entities.txt',
    'globals.txt',
    'misc.txt',
    'nav.txt',
    'NextBotBehavior.txt',
    'NextBotBodyInterface.txt',
    'NextBotChasePath.txt',
    'NextBotContextualQueryInterface.txt',
    'NextBotEventResponderInterface.txt',
    'NextBotInterface.txt',
    'NextBotIntentionInterface.txt',
    'NextBotKnownEntity.txt',
    'NextBotLocomotionInterface.txt',
    'NextBotManager.txt',
    'NextBotPlayer.txt',
    'NextBotPlayerBody.txt',
    'NextBotPlayerLocomotion.txt',
    'NextBotPath.txt',
    'NextBotPathFollow.txt',
    'NextBotVisionInterface.txt',
    'population.txt',
    'tfplayer.txt',
    'tfbot.txt',
    'tfbot_body.txt',
    'tfbot_locomotion.txt',
    'tfbot_vision.txt',
    'tfbot_behavior.txt',
    'fugue.txt',
  ]
)

# Config files
#CopyFiles('configs', 'addons/sourcemod/configs',
#  [
#    'configfile.cfg',
#  ]
#)
if not Extension.optimize_mods_only:
  # Resource files
  CopyFiles('resource', 'custom/sigsegv/resource',
    [
      'sigsegv_events.res',
    ]
  )

  # Script files
  CopyFiles('scripts', 'custom/sigsegv/scripts',
    [
      'sigsegv_sound_overrides.txt',
      'globals.lua',
      'library.lua'
    ]
  )

# Translation files
CopyFiles('translations', 'addons/sourcemod/translations',
  [
    'sigsegv.phrases.txt',
    'sigsegvattributes.phrases.txt'
  ]
)
CopyFiles('translations/zho', 'addons/sourcemod/translations/zho',
  [
    'sigsegv.phrases.txt',
    'sigsegvattributes.phrases.txt'
  ]
)
CopyFiles('translations/chi', 'addons/sourcemod/translations/chi',
  [
    'sigsegv.phrases.txt',
    'sigsegvattributes.phrases.txt'
  ]
)

if Extension.optimize_mods_only:
  builder.AddCopy('cfg/sigsegv_convars_optimize_only.cfg', 'cfg/sigsegv_convars.cfg')
elif 'mods_mvm' in Extension.exclude:
  builder.AddCopy('cfg/sigsegv_convars_no_mvm.cfg', 'cfg/sigsegv_convars.cfg')
else:
  builder.AddCopy('cfg/sigsegv_convars.cfg', folder_map['cfg'])
  builder.AddCopy('cfg/sigsegv_convars_pvp.cfg', folder_map['cfg'])
  builder.AddCopy('cfg/sigsegv_convars_vanilla.cfg', folder_map['cfg'])
  builder.AddCopy('scripts/mvm_bigrock_sigdemo.pop', folder_map['custom/sigsegv/scripts/population'])
  builder.AddCopy('scripts/mvm_bigrock_luademo.pop', folder_map['custom/sigsegv/scripts/population'])

builder.AddCopy('sigsegv.autoload', 'addons/sourcemod/extensions/' + Extension.name + '.autoload')

l4dlinks = [
  "csgo",
  "pvkii",
  "blade",
]

# Copy binaries.
for cxx_task in Extension.extensions:
  src = os.path.join('..', cxx_task.binary.path)
  # binary marked as using 'other' engine is the "default engine"
  name = os.path.basename(cxx_task.binary.path).replace(".2.sdk2013", "")

  dst = ''
  if cxx_task.target.arch == 'x86_64':
    dst = os.path.join('addons', 'sourcemod', 'extensions', 'x64', name)
  else:
    dst = os.path.join('addons', 'sourcemod', 'extensions', name)

  builder.AddCommand(
    inputs  = [ cxx_task.binary ],
    argv    = [ 'cp', '--remove-destination', src, dst ],
    outputs = [ dst ],
  )
  
  if ".2.l4d" in cxx_task.binary.path:
    for l4dlink in l4dlinks:
      dst_link = os.path.join('addons', 'sourcemod', 'extensions', name.replace(".2.l4d", ".2." + l4dlink))
      builder.AddCommand(
        inputs  = [ cxx_task.binary ],
        argv    = [ 'ln', '-s', './'+os.path.basename(cxx_task.binary.path), dst_link ],
        outputs = [ dst_link ],
      )
#  builder.AddCopy(cxx_task.binary, folder_map['addons/sourcemod/extensions'])

# Copy libstrcompat (Linux only)
# if builder.target.platform == 'linux':
#   src = os.path.join(builder.sourcePath, 'libs', 'strcompat', 'libstrcompat.so')
#   dst = os.path.join('addons', 'sourcemod', 'extensions', 'libstrcompat.so')
#   builder.AddCommand(
#     inputs  = [ src ],
#     argv    = [ 'cp', '--remove-destination', src, dst ],
#     outputs = [ dst ],
#   )