code-graph-rag/build_binary.py at main · khushil/code-graph-rag · 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
91
92
93
94
95
96
97
98
99
100
101
#!/usr/bin/env python3
"""
Build script for creating Graph-Code binaries using PyInstaller.
This handles the complex dependencies and package metadata issues.
"""

import os
import platform
import subprocess
import sys
from pathlib import Path


def build_binary() -> bool:
    """Build the Graph-Code binary using PyInstaller."""

    # Get platform info for naming
    system = platform.system().lower()
    machine = platform.machine().lower()
    if machine == "x86_64":
        machine = "amd64"
    elif machine in ["aarch64", "arm64"]:
        machine = "arm64"

    binary_name = f"graph-code-{system}-{machine}"

    # PyInstaller command with all necessary options
    cmd = [
        "pyinstaller",
        "--name",
        binary_name,
        "--onefile",
        "--noconfirm",
        "--clean",
        # Include all tree-sitter languages
        "--collect-all",
        "tree_sitter_python",
        "--collect-all",
        "tree_sitter_javascript",
        "--collect-all",
        "tree_sitter_typescript",
        "--collect-all",
        "tree_sitter_rust",
        "--collect-all",
        "tree_sitter_go",
        "--collect-all",
        "tree_sitter_scala",
        "--collect-all",
        "tree_sitter_java",
        "--collect-all",
        "tree_sitter_cpp",
        # Include pydantic-ai and dependencies
        "--collect-all",
        "pydantic_ai",
        "--collect-data",
        "pydantic_ai",
        "--hidden-import",
        "pydantic_ai_slim",
        # Include other critical dependencies
        "--collect-all",
        "rich",
        "--collect-all",
        "typer",
        "--collect-all",
        "loguru",
        "--collect-all",
        "toml",
        # Entry point
        "main.py",
    ]

    print(f"Building binary: {binary_name}")
    print("This may take a few minutes...")

    try:
        subprocess.run(cmd, check=True, capture_output=True, text=True)
        print("Binary built successfully!")

        # Show binary info
        binary_path = Path("dist") / binary_name
        if binary_path.exists():
            size_mb = binary_path.stat().st_size / (1024 * 1024)
            print(f"Binary: {binary_path}")
            print(f"Size: {size_mb:.1f} MB")

            # Make sure it's executable
            os.chmod(binary_path, 0o755)
            print("Binary is ready for distribution!")

        return True

    except subprocess.CalledProcessError as e:
        print(f"Build failed: {e}")
        print("STDOUT:", e.stdout)
        print("STDERR:", e.stderr)
        return False


if __name__ == "__main__":
    success = build_binary()
    sys.exit(0 if success else 1)