feat: job bundle browser/sharing

src/deadline/_mcp/registry.py

@@ -5,7 +5,7 @@
 from typing import Any, Callable, List, Optional, TypedDict, Dict
 
 from ..client import api
-from .tools import job, logs
+from .tools import job, logs, bundles
 
 
 class ToolDefinition(TypedDict):
@@ -115,4 +115,17 @@ def get_all_tool_names() -> List[str]:
             "item_offset",
         ],
     },
+    # Bundle sharing tools
+    "list_shared_bundles": {
+        "func": bundles.list_shared_bundles,
+        "param_names": ["farm_id", "queue_id", "show_hidden"],
+    },
+    "upload_bundle": {
+        "func": bundles.upload_bundle,
+        "param_names": ["job_bundle", "name", "farm_id", "queue_id"],
+    },
+    "download_bundle": {
+        "func": bundles.download_bundle,
+        "param_names": ["bundle_name", "output_dir", "farm_id", "queue_id"],
+    },
 }

src/deadline/_mcp/tools/bundles.py

@@ -0,0 +1,100 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+
+"""
+Deadline Cloud Bundle sharing tools for MCP.
+"""
+
+import json
+from typing import Any, Dict, Optional
+
+from click.testing import CliRunner
+
+from ...client.cli import main
+
+
+def list_shared_bundles(
+    farm_id: Optional[str] = None,
+    queue_id: Optional[str] = None,
+    show_hidden: bool = False,
+) -> Dict[str, Any]:
+    """List job bundles shared on the queue.
+
+    Returns a list of bundles with their name and format.
+    Hidden bundles are excluded by default unless show_hidden is True.
+    """
+    args = ["bundle", "list", "--queue", "--output", "json"]
+    if farm_id:
+        args.extend(["--farm-id", farm_id])
+    if queue_id:
+        args.extend(["--queue-id", queue_id])
+    if show_hidden:
+        args.append("--show-hidden")
+
+    runner = CliRunner()
+    result = runner.invoke(main, args)
+
+    if result.exit_code != 0:
+        return {"success": False, "error": result.output.strip()}
+    return {"bundles": json.loads(result.output)}
+
+
+def upload_bundle(
+    job_bundle: str,
+    name: Optional[str] = None,
+    farm_id: Optional[str] = None,
+    queue_id: Optional[str] = None,
+) -> Dict[str, Any]:
+    """Upload a local job bundle to the queue as a shared .ojd archive.
+
+    Args:
+        job_bundle: Path to a job bundle directory or .ojd archive file.
+        name: Override the bundle name (defaults to directory/file name).
+        farm_id: The farm ID (uses default if not specified).
+        queue_id: The queue ID (uses default if not specified).
+    """
+    args = ["bundle", "upload", job_bundle]
+    if name:
+        args.extend(["--name", name])
+    if farm_id:
+        args.extend(["--farm-id", farm_id])
+    if queue_id:
+        args.extend(["--queue-id", queue_id])
+
+    runner = CliRunner()
+    result = runner.invoke(main, args, input="y\n")
+
+    if result.exit_code != 0:
+        return {"success": False, "error": result.output.strip()}
+    return {"success": True, "message": result.output.strip()}
+
+
+def download_bundle(
+    bundle_name: str,
+    output_dir: Optional[str] = None,
+    farm_id: Optional[str] = None,
+    queue_id: Optional[str] = None,
+) -> Dict[str, Any]:
+    """Download a shared bundle from the queue to a local directory.
+
+    Args:
+        bundle_name: Name of the bundle to download.
+        output_dir: Local directory to download to (uses cache if not specified).
+        farm_id: The farm ID (uses default if not specified).
+        queue_id: The queue ID (uses default if not specified).
+    """
+    args = ["bundle", "download", bundle_name, "--output", "json"]
+    if output_dir:
+        args.extend(["-o", output_dir])
+    if farm_id:
+        args.extend(["--farm-id", farm_id])
+    if queue_id:
+        args.extend(["--queue-id", queue_id])
+
+    runner = CliRunner()
+    result = runner.invoke(main, args)
+
+    if result.exit_code != 0:
+        return {"success": False, "error": result.output.strip()}
+
+    data = json.loads(result.output)
+    return {"success": True, "path": data["path"]}