dev: add generic mcp_source.py helper to switch MCP package source (upstream/remote/local)

Author: dsarno

mcp_source.py

@@ -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("[email protected]:"):
+        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()