build.py

# @file        build.py
#              Main PythonMonkey build automation script. Run with `poetry build`.
# @author      Hamada Gasmallah, hamada@distributive.network
# @date        April 2023
# @copyright Copyright (c) 2023 Distributive Corp.
#

import subprocess
import os
import sys
import platform
from typing import Optional

TOP_DIR = os.path.abspath(os.path.dirname(__file__))
BUILD_DIR = os.path.join(TOP_DIR, "build")

# Get number of CPU cores
CPUS = os.getenv('CPUS') or os.cpu_count() or 1

BUILD_TYPE = os.environ["BUILD_TYPE"].title() if "BUILD_TYPE" in os.environ else "Release"
BUILD_DOCS = "ON" if "BUILD_DOCS" in os.environ and os.environ["BUILD_DOCS"] in ("1", "ON", "on") else "OFF"


def execute(cmd: str, cwd: Optional[str] = None):
  popen = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
                           shell=True, text=True, cwd=cwd)
  for stdout_line in iter(popen.stdout.readline, ""):
    sys.stdout.write(stdout_line)
    sys.stdout.flush()

  popen.stdout.close()
  return_code = popen.wait()
  if return_code:
    raise subprocess.CalledProcessError(return_code, cmd)


def ensure_spidermonkey():
  # Check if SpiderMonkey libs already exist
  spidermonkey_lib_exist = os.path.exists(os.path.join(TOP_DIR, "_spidermonkey_install/lib"))
  if spidermonkey_lib_exist:
    return

  # Build SpiderMonkey
  execute("bash ./setup.sh", cwd=TOP_DIR)


def run_cmake_build():
  os.makedirs(BUILD_DIR, exist_ok=True)  # mkdir -p

  if platform.system() == "Windows":
    # use Clang/LLVM toolset for Visual Studio
    execute(f"cmake -DBUILD_DOCS={BUILD_DOCS} -DPM_BUILD_TYPE={BUILD_TYPE} .. -T ClangCL", cwd=BUILD_DIR)
  else:
    execute(f"cmake -DBUILD_DOCS={BUILD_DOCS} -DPM_BUILD_TYPE={BUILD_TYPE} ..", cwd=BUILD_DIR)
  execute(f"cmake --build . -j{CPUS} --config Release", cwd=BUILD_DIR)


def copy_artifacts():

  if platform.system() == "Windows":
    execute("cp ./build/src/*/pythonmonkey.pyd ./python/pythonmonkey/", cwd=TOP_DIR)  # Release or Debug build
    execute("cp ./_spidermonkey_install/lib/mozjs-*.dll ./python/pythonmonkey/", cwd=TOP_DIR)
  else:
    execute("cp ./build/src/pythonmonkey.so ./python/pythonmonkey/", cwd=TOP_DIR)
    execute("cp ./_spidermonkey_install/lib/libmozjs* ./python/pythonmonkey/", cwd=TOP_DIR)


def build():
  if BUILD_TYPE != "None":  # do not build SpiderMonkey if we are not compiling
    ensure_spidermonkey()
  run_cmake_build()
  if BUILD_TYPE != "None":  # do not copy artifacts if we did not build them
    copy_artifacts()


if __name__ == "__main__":
  build()