# It's important that we use a function here, as any global variables we make in this script will
# also be available in the real entrypoint run by runpy._run_module_as_main.
def _soong_main():
  import os
  import runpy
  import shutil
  import sys
  import tempfile
  import zipfile

  from pathlib import PurePath

  sys.argv[0] = __loader__.archive

  # Set sys.executable to None. The real executable is available as
  # sys.argv[0], and too many things assume sys.executable is a regular Python
  # binary, which isn't available. By setting it to None we get clear errors
  # when people try to use it.
  sys.executable = None

  # Extract the shared libraries from the zip file into a temporary directory.
  # This works around the limitations of dynamic linker.  Some Python libraries
  # reference the .so files relatively and so extracting only the .so files
  # does not work, so we extract the entire parent directory of the .so files to a
  # tempdir and then add that to sys.path.
  tempdir = None
  with zipfile.ZipFile(__loader__.archive) as z:
    # any root so files or root directories that contain so files will be
    # extracted to the tempdir so the linker load them, this minimizes the
    # number of files that need to be extracted to a tempdir
    extract_paths = {}
    for member in z.infolist():
      if member.filename.endswith('.so'):
        extract_paths[PurePath(member.filename).parts[0]] = member.filename
    if extract_paths:
      tempdir = tempfile.mkdtemp()
      for member in z.infolist():
        if not PurePath(member.filename).parts[0] in extract_paths.keys():
          continue
        if member.is_dir():
          os.makedirs(os.path.join(tempdir, member.filename))
        else:
          z.extract(member, tempdir)
      sys.path.insert(0, tempdir)
  try:
    runpy._run_module_as_main("ENTRY_POINT", alter_argv=False)
  finally:
    if tempdir is not None:
      shutil.rmtree(tempdir)

if __name__ == "__main__":
  _soong_main()