IronFox/scripts/get_sources.py

#!/usr/bin/env python

import os
from config import GITHUB_REF_DOWNLOADS, GITHUB_REF_URL, GeckoPaths
from utils import download, query_yes_no, rmdirrec, zipextract_rmtoplevel

def get_sources(paths: GeckoPaths):
    print("Downloading sources...")

    for repo_name, repo, ref in GITHUB_REF_DOWNLOADS:
        repo_zip = paths.builddir / ("{}.zip".format(repo_name))
        repo_path = paths.rootdir / repo_name
        download(GITHUB_REF_URL.format(repo, ref), repo_zip)

        if not repo_zip.exists():
            raise RuntimeError("Source archive for Fennec does not exist.")

        if repo_path.exists():
            if len(os.listdir(repo_path)) > 0:
                print(f"{repo_path} already exists and is not an empty directory.")
                if query_yes_no("Do you want to overwrite its contents?"):
                    print(f"Deleting {repo_path}...")
                    rmdirrec(repo_path)
                else:
                    return

        if not repo_path.exists():
            repo_path.mkdir(parents=True)

        print(f"Extracting {repo_zip}")
        zipextract_rmtoplevel(repo_zip, repo_path)
        print("\n")


if __name__ == "__main__":
    paths = GeckoPaths()

    if not paths.builddir.exists():
        paths.builddir.mkdir()

    get_sources(paths)

    with open(paths.rootdir / "scripts/paths_local.sh") as f:
        f.write(
            f"""
readonly paths_source="true"
readonly patches={paths.patchdir}
readonly android_components={paths.android_components}
readonly application_services={paths.appservicesdir}
readonly glean={paths.gleandir}
readonly fenix={paths.fenix}
readonly mozilla_release={paths.geckodir}
readonly gmscore={paths.gmscoredir}
                """
        )