Compass/fetch_repos.py at main · CardputerZero/Compass · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
import json
import os
import subprocess
from pathlib import Path


def clone_or_update_repo(
    repo_url, path, ref=None, with_submodules=False, patch_path=None, update_existing=True
):
    path = str(path)
    if not Path(path).exists():
        subprocess.run(["git", "clone", repo_url, path], check=True)
    elif update_existing:
        subprocess.run(["git", "-C", path, "fetch"], check=True)

    if ref:
        subprocess.run(["git", "-C", path, "checkout", ref], check=True)

    if with_submodules:
        subprocess.run(
            ["git", "-C", path, "submodule", "update", "--init", "--recursive"],
            check=True,
        )

    if patch_path:
        patch_full_path = (
            patch_path
            if os.path.isabs(patch_path)
            else str(Path.cwd() / patch_path)
        )
        check_result = subprocess.run(
            ["git", "-C", path, "apply", "--check", patch_full_path]
        )
        if check_result.returncode == 0:
            subprocess.run(["git", "-C", path, "apply", patch_full_path], check=True)
            print(f"Applied patch {patch_path} to {path}")
        else:
            print(f"Patch {patch_path} cannot be applied cleanly to {path}, skipped.")


def load_repos():
    script_dir = Path(__file__).resolve().parent
    with open(script_dir / "repos.json") as f:
        return json.load(f)


def repo_path(repo):
    return Path(__file__).resolve().parent / repo["path"]


def is_repo_fetched(repo):
    path = repo_path(repo)
    return path.exists() and (path / ".git").exists()


def fetch_dependencies(repos=None, update_existing=True):
    repos = repos or load_repos()

    for repo in repos:
        path = repo_path(repo)
        patch = repo.get("patch")
        if patch:
            patch = Path(patch)
            if not patch.is_absolute():
                patch = Path(__file__).resolve().parent / patch

        clone_or_update_repo(
            repo["url"],
            path,
            repo.get("branch"),
            repo.get("with_submodules", False),
            patch,
            update_existing=update_existing,
        )


def ensure_dependencies():
    repos = load_repos()
    missing = [repo for repo in repos if not is_repo_fetched(repo)]

    if not missing:
        return

    names = ", ".join(repo["path"] for repo in missing)
    print(f"Compass: fetching missing dependencies: {names}")
    fetch_dependencies(missing, update_existing=False)


if __name__ == "__main__":
    fetch_dependencies()