dev: add generic mcp_source.py helper to switch MCP package source (upstream/remote/local)
David Sarno
parent
b6b8d47dfe
commit
cd707284d7
|
|
@ -0,0 +1,241 @@
|
|||
#!/usr/bin/env python3
|
||||
"""
|
||||
Generic helper to switch the Unity MCP package source in a Unity project's
|
||||
Packages/manifest.json without embedding any personal paths.
|
||||
|
||||
Usage:
|
||||
python mcp_source.py [--manifest /abs/path/to/manifest.json] [--repo /abs/path/to/unity-mcp] [--choice 1|2|3]
|
||||
|
||||
Choices:
|
||||
1) Upstream main (CoplayDev/unity-mcp)
|
||||
2) Your remote current branch (derived from `origin` and current branch)
|
||||
3) Local repo workspace (file: URL to UnityMcpBridge in your checkout)
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import os
|
||||
import pathlib
|
||||
import re
|
||||
import subprocess
|
||||
import sys
|
||||
from typing import Optional
|
||||
|
||||
PKG_NAME = "com.coplaydev.unity-mcp"
|
||||
BRIDGE_SUBPATH = "UnityMcpBridge"
|
||||
|
||||
|
||||
def run_git(repo: pathlib.Path, *args: str) -> str:
|
||||
result = subprocess.run([
|
||||
"git", "-C", str(repo), *args
|
||||
], capture_output=True, text=True)
|
||||
if result.returncode != 0:
|
||||
raise RuntimeError(result.stderr.strip() or f"git {' '.join(args)} failed")
|
||||
return result.stdout.strip()
|
||||
|
||||
|
||||
def normalize_origin_to_https(url: str) -> str:
|
||||
"""Map common SSH origin forms to https for Unity's git URL scheme."""
|
||||
if url.startswith("git@github.com:"):
|
||||
owner_repo = url.split(":", 1)[1]
|
||||
if owner_repo.endswith(".git"):
|
||||
owner_repo = owner_repo[:-4]
|
||||
return f"https://github.com/{owner_repo}.git"
|
||||
# already https or file: etc.
|
||||
return url
|
||||
|
||||
|
||||
def detect_repo_root(explicit: Optional[str]) -> pathlib.Path:
|
||||
if explicit:
|
||||
return pathlib.Path(explicit).resolve()
|
||||
# Prefer the git toplevel from the script's directory
|
||||
here = pathlib.Path(__file__).resolve().parent
|
||||
try:
|
||||
top = run_git(here, "rev-parse", "--show-toplevel")
|
||||
return pathlib.Path(top)
|
||||
except Exception:
|
||||
return here
|
||||
|
||||
|
||||
def detect_branch(repo: pathlib.Path) -> str:
|
||||
return run_git(repo, "rev-parse", "--abbrev-ref", "HEAD")
|
||||
|
||||
|
||||
def detect_origin(repo: pathlib.Path) -> str:
|
||||
url = run_git(repo, "remote", "get-url", "origin")
|
||||
return normalize_origin_to_https(url)
|
||||
|
||||
|
||||
def find_manifest(explicit: Optional[str]) -> pathlib.Path:
|
||||
if explicit:
|
||||
return pathlib.Path(explicit).resolve()
|
||||
# Walk up from CWD looking for Packages/manifest.json
|
||||
cur = pathlib.Path.cwd().resolve()
|
||||
for parent in [cur, *cur.parents]:
|
||||
candidate = parent / "Packages" / "manifest.json"
|
||||
if candidate.exists():
|
||||
return candidate
|
||||
raise FileNotFoundError("Could not find Packages/manifest.json from current directory. Use --manifest to specify a path.")
|
||||
|
||||
|
||||
def read_json(path: pathlib.Path) -> dict:
|
||||
with path.open("r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
def write_json(path: pathlib.Path, data: dict) -> None:
|
||||
with path.open("w", encoding="utf-8") as f:
|
||||
json.dump(data, f, indent=2)
|
||||
f.write("\n")
|
||||
|
||||
|
||||
def build_options(repo_root: pathlib.Path, branch: str, origin_https: str):
|
||||
upstream = "https://github.com/CoplayDev/unity-mcp.git?path=/UnityMcpBridge"
|
||||
# Ensure origin is https
|
||||
origin = origin_https
|
||||
# If origin is a local file path or non-https, try to coerce to https github if possible
|
||||
if origin.startswith("file:"):
|
||||
# Not meaningful for remote option; keep upstream
|
||||
origin_remote = upstream
|
||||
else:
|
||||
origin_remote = origin
|
||||
return [
|
||||
("[1] Upstream main", upstream),
|
||||
("[2] Remote current branch", f"{origin_remote}?path=/{BRIDGE_SUBPATH}#{branch}"),
|
||||
("[3] Local workspace", f"file:{(repo_root / BRIDGE_SUBPATH).as_posix()}"),
|
||||
]
|
||||
|
||||
|
||||
def parse_args() -> argparse.Namespace:
|
||||
p = argparse.ArgumentParser(description="Switch Unity MCP package source")
|
||||
p.add_argument("--manifest", help="Path to Packages/manifest.json")
|
||||
p.add_argument("--repo", help="Path to unity-mcp repo root (for local file option)")
|
||||
p.add_argument("--choice", choices=["1", "2", "3"], help="Pick option non-interactively")
|
||||
return p.parse_args()
|
||||
|
||||
|
||||
def main() -> None:
|
||||
args = parse_args()
|
||||
try:
|
||||
repo_root = detect_repo_root(args.repo)
|
||||
branch = detect_branch(repo_root)
|
||||
origin = detect_origin(repo_root)
|
||||
except Exception as e:
|
||||
print(f"Error: {e}", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
options = build_options(repo_root, branch, origin)
|
||||
|
||||
try:
|
||||
manifest_path = find_manifest(args.manifest)
|
||||
except Exception as e:
|
||||
print(f"Error: {e}", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
print("Select MCP package source by number:")
|
||||
for label, _ in options:
|
||||
print(label)
|
||||
|
||||
if args.choice:
|
||||
choice = args.choice
|
||||
else:
|
||||
choice = input("Enter 1-3: ").strip()
|
||||
|
||||
if choice not in {"1", "2", "3"}:
|
||||
print("Invalid selection.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
idx = int(choice) - 1
|
||||
_, chosen = options[idx]
|
||||
|
||||
data = read_json(manifest_path)
|
||||
deps = data.get("dependencies", {})
|
||||
if PKG_NAME not in deps:
|
||||
print(f"Error: '{PKG_NAME}' not found in manifest dependencies.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
print(f"\nUpdating {PKG_NAME} → {chosen}")
|
||||
deps[PKG_NAME] = chosen
|
||||
data["dependencies"] = deps
|
||||
write_json(manifest_path, data)
|
||||
print(f"Done. Wrote to: {manifest_path}")
|
||||
print("Tip: In Unity, open Package Manager and Refresh to re-resolve packages.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
#!/usr/bin/env python3
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
|
||||
# Defaults for your environment
|
||||
UNITY_PROJECT = "/Users/davidsarno/ramble" # change if needed
|
||||
MANIFEST = os.path.join(UNITY_PROJECT, "Packages", "manifest.json")
|
||||
LOCAL_REPO = "/Users/davidsarno/unity-mcp" # local repo root
|
||||
PKG_NAME = "com.coplaydev.unity-mcp"
|
||||
|
||||
def get_current_branch(repo_path: str) -> str:
|
||||
result = subprocess.run(
|
||||
["git", "-C", repo_path, "rev-parse", "--abbrev-ref", "HEAD"],
|
||||
capture_output=True, text=True
|
||||
)
|
||||
if result.returncode != 0:
|
||||
print("Error: unable to detect current branch from local repo.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
return result.stdout.strip()
|
||||
|
||||
def build_options(branch: str):
|
||||
return [
|
||||
("[1] Upstream main", "https://github.com/CoplayDev/unity-mcp.git?path=/UnityMcpBridge"),
|
||||
(f"[2] Remote {branch}", f"https://github.com/dsarno/unity-mcp.git?path=/UnityMcpBridge#{branch}"),
|
||||
(f"[3] Local {branch}", f"file:{os.path.join(LOCAL_REPO, 'UnityMcpBridge')}"),
|
||||
]
|
||||
|
||||
def read_manifest(path: str):
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
|
||||
def write_manifest(path: str, data: dict):
|
||||
with open(path, "w", encoding="utf-8") as f:
|
||||
json.dump(data, f, indent=2)
|
||||
f.write("\n")
|
||||
|
||||
def main():
|
||||
# Allow overrides via args: python mcp_source.py [manifest.json] [local_repo]
|
||||
manifest_path = MANIFEST if len(sys.argv) < 2 else sys.argv[1]
|
||||
repo_path = LOCAL_REPO if len(sys.argv) < 3 else sys.argv[2]
|
||||
|
||||
branch = get_current_branch(repo_path)
|
||||
options = build_options(branch)
|
||||
|
||||
print("Select MCP package source by number:")
|
||||
for label, _ in options:
|
||||
print(label)
|
||||
choice = input("Enter 1-3: ").strip()
|
||||
if choice not in {"1", "2", "3"}:
|
||||
print("Invalid selection.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
idx = int(choice) - 1
|
||||
_, chosen = options[idx]
|
||||
|
||||
data = read_manifest(manifest_path)
|
||||
deps = data.get("dependencies", {})
|
||||
if PKG_NAME not in deps:
|
||||
print(f"Error: '{PKG_NAME}' not found in manifest dependencies.", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
print(f"\nUpdating {PKG_NAME} → {chosen}")
|
||||
deps[PKG_NAME] = chosen
|
||||
data["dependencies"] = deps
|
||||
write_manifest(manifest_path, data)
|
||||
print(f"Done. Wrote to: {manifest_path}")
|
||||
print("Tip: In Unity, open Package Manager and Refresh to re-resolve packages.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Reference in New Issue