build.py

#!/usr/bin/python3

import json
from os import system
import os
from pathlib import Path
import subprocess
import sys
import shutil
import glob
from urllib import request
import requests

if os.path.exists("version"):
    with open("version") as fp:
        VERSION = fp.read().split("VERSION=")[1]

RELEASES_URL = f"https://api.github.com/repos/Princess-Org/Princess/releases"

if sys.platform == "win32":
    ARGS = ["-llibffi", "--link-flag", "/FORCE:UNRESOLVED", "--link-flag", "/STACK:67108864", "--clang=-gcodeview", "--link-directory", os.environ["LIBRARY_PATH"]]
elif sys.platform == "darwin":
    ARGS = ["-lbfd", "-lelf", "-L/opt/homebrew/lib", "-L/opt/homebrew/opt/libffi/lib", "-L/opt/homebrew/opt/binutils/lib"]
else:
    ARGS = ["-lbfd"]

def exe_file(file: str) -> str:
    if sys.platform == "win32":
        return file + ".exe"
    return file

def release():
    FOLDER=Path(f"princess-{VERSION}")
    
    print("First compilation step")
    build([])

    print("Second compilation step")
    args = [exe_file("bin/princess2"), "--no-incremental", "-d", "-Isrc", "--buildfolder=build", "--outfile", exe_file("bin/princess3"), "src/main.pr"]
    args += ARGS
    subprocess.check_call(args)
    
    print("Creating archive")
    FOLDER.mkdir(exist_ok=True)
    (FOLDER / "bin").mkdir(exist_ok=True)
    (FOLDER / "include").mkdir(exist_ok=True)

    shutil.copy(Path(exe_file("bin/princess2")), FOLDER / exe_file("bin/princess"))
    if sys.platform == "win32":
        shutil.copy(Path("bin/princess2.pdb"), FOLDER / "bin/princess.pdb")
    shutil.copy(Path("version"), FOLDER)

    for path in glob.glob("include/*.h"):
        shutil.copy(path, FOLDER / "include")

    if sys.platform == "win32":
        shutil.copy(Path("bin/libffi.dll"), FOLDER / "bin")
        shutil.copy(Path("bin/libffi.lib"), FOLDER / "bin")

        (FOLDER / "include/windows").mkdir(exist_ok=True)
        for path in glob.glob("include/windows/*.pr"):
            shutil.copy(path, FOLDER / "include/windows")

        shutil.copy(Path("include/windows/ffi.h"), FOLDER / "include/windows")
        shutil.copy(Path("include/windows/ffitarget.h"), FOLDER / "include/windows")
    else: 
        if sys.platform == "darwin":
            (FOLDER / "include/macos").mkdir(exist_ok=True)
            for path in glob.glob("include/macos/*.pr"):
                shutil.copy(path, FOLDER / "include/macos")

        (FOLDER / "include/linux").mkdir(exist_ok=True)
        for path in glob.glob("include/linux/*.pr"):
            shutil.copy(path, FOLDER / "include/linux")

    shutil.copy(Path("include/preload.pr"), FOLDER / "include")
    shutil.copy(Path("include/symbol.pr"), FOLDER / "include")
    shutil.copy(Path("include/gencstd.py"), FOLDER / "include")
    shutil.copy(Path("include/cstd.c"), FOLDER / "include")
    shutil.copytree(Path("std"), FOLDER / "std")

    if sys.platform == "win32":
        shutil.make_archive(f"princess-win32-{VERSION}", "zip", FOLDER)
    elif sys.platform == "darwin":
        shutil.make_archive(f"princess-macos-{VERSION}", "zip", FOLDER)
    else:
        shutil.make_archive(f"princess-{VERSION}", "gztar", FOLDER)

    shutil.rmtree(FOLDER)

def testrunner(extra):
    args = [exe_file("bin/princess"), "--no-incremental", "--outfile", exe_file("bin/testrunner"), "src/testrunner.pr"]
    args += ARGS
    subprocess.check_call(args + extra)

def clean():
    shutil.rmtree(Path("build"))
    for path in glob.glob("princess*.tar.gz") + glob.glob("princess*.zip"):
        Path(path).unlink()

def download():
    print("Downloading Princess compiler from Github...")
    releases = json.loads(requests.get(RELEASES_URL).text)

    asset = None
    release_index = 0
    while not asset and release_index < len(releases):
        latest_version = releases[release_index]
        for a in latest_version["assets"]:
            if sys.platform == "win32":
                if "win32" in a["name"]:
                    asset = a
                    break
            elif sys.platform == "darwin":
                if "macos" in a["name"]:
                    asset = a
                    break
            else:
                if a["name"].endswith(".tar.gz"):
                    asset = a
                    break
        release_index += 1
    
    if not asset:
        print("Couldn't find suitable asset, maybe github is down or something.")
        exit(1)
    
    archive = "princess.zip" if sys.platform == "win32" else "princess.tar.gz"
    with open(archive, "wb") as fp:
        fp.write(requests.get(asset["browser_download_url"]).content)

    shutil.unpack_archive(archive, "princess")
    if sys.platform == "win32":
        shutil.copy(Path("princess") / "bin/princess.exe", "bin")
        shutil.copy(Path("princess") / "bin/princess.pdb", "bin")
    else:
        shutil.copy(Path("princess") / "bin/princess", "bin")
    
    shutil.rmtree(Path("princess"))
    Path(archive).unlink()

def build(extra):
    args = [exe_file("bin/princess"), "-d", "--no-incremental", "-Isrc", "--buildfolder=build", "--outfile", exe_file("bin/princess2"), "src/main.pr"]
    args += ARGS
        
    subprocess.check_call(args + extra)

def main():
    Path("bin/").mkdir(exist_ok = True)
    Path("build/").mkdir(exist_ok = True)
    downloaded = False
    if not Path(exe_file("bin/princess")).exists():
        download()
        downloaded = True

    args = sys.argv
    if len(args) == 1:
        build(args[1:])
    elif args[1] == "release":
        release()
    elif args[1] == "test":
        testrunner(args[2:])
    elif args[1] == "clean":
        clean()
    elif args[1] == "download":
        if not downloaded:
            download()
    else:
        build(args[1:])

if __name__ == "__main__":
    main()