@hybridlabor-api/aos 4.4.0 → 4.4.2-beta.0

package/mcps/davinci-resolve-mcp/src/granular/media_storage.py

@@ -1,179 +0,0 @@
-"""MediaStorage tools."""
-
-from src.granular.common import *  # noqa: F401,F403
-
-resolve = ResolveProxy()
-
-@mcp.tool()
-def get_mounted_volumes() -> Dict[str, Any]:
-    """Get list of mounted volumes displayed in Resolve's Media Storage."""
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-    volumes = ms.GetMountedVolumeList()
-    return {"volumes": volumes if volumes else []}
-
-
-@mcp.tool()
-def get_media_storage_subfolders(folder_path: str) -> Dict[str, Any]:
-    """Get subfolders in a given absolute folder path from Media Storage.
-
-    Args:
-        folder_path: Absolute path to the folder to list subfolders for.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-    subfolders = ms.GetSubFolderList(folder_path)
-    return {"folder_path": folder_path, "subfolders": subfolders if subfolders else []}
-
-
-@mcp.tool()
-def get_media_storage_files(folder_path: str) -> Dict[str, Any]:
-    """Get media and file listings in a given absolute folder path from Media Storage.
-
-    Args:
-        folder_path: Absolute path to the folder to list files for.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-    files = ms.GetFileList(folder_path)
-    return {"folder_path": folder_path, "files": files if files else []}
-
-
-@mcp.tool()
-def reveal_in_media_storage(file_path: str) -> Dict[str, Any]:
-    """Reveal a file path in Resolve's Media Storage browser.
-
-    Args:
-        file_path: Absolute path to the file to reveal.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-    result = ms.RevealInStorage(file_path)
-    return {"success": bool(result), "file_path": file_path}
-
-
-@mcp.tool()
-def add_items_to_media_pool_from_storage(
-    file_paths: Optional[List[str]] = None,
-    item_infos: Optional[List[Dict[str, Any]]] = None,
-) -> Dict[str, Any]:
-    """Add specified file/folder paths from Media Storage into current Media Pool folder.
-
-    Args:
-        file_paths: Simple form — list of absolute file/folder paths.
-        item_infos: Positioned form — list of dicts with keys media (required),
-            startFrame, endFrame. Mirrors
-            MediaStorage.AddItemListToMediaPool([{itemInfo}, ...]) per docs line 210.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-    if item_infos is not None:
-        if not isinstance(item_infos, list) or not item_infos:
-            return {"error": "item_infos must be a non-empty list"}
-        for i, info in enumerate(item_infos):
-            if not isinstance(info, dict):
-                return {"error": f"item_infos[{i}] must be an object"}
-            if not info.get("media"):
-                return {"error": f"item_infos[{i}] requires media (file path)"}
-        clips = ms.AddItemListToMediaPool(item_infos)
-    else:
-        if not file_paths:
-            return {"error": "Provide file_paths (simple) or item_infos (positioned)"}
-        clips = ms.AddItemListToMediaPool(file_paths)
-    if clips:
-        return {"success": True, "clips_added": len(clips)}
-    return {"success": False, "error": "Failed to add items to Media Pool"}
-
-
-@mcp.tool()
-def add_clip_mattes_to_media_pool(media_pool_item_id: str, matte_paths: List[str]) -> Dict[str, Any]:
-    """Add clip mattes from Media Storage to a MediaPoolItem.
-
-    Args:
-        media_pool_item_id: The unique ID of the MediaPoolItem.
-        matte_paths: List of absolute file paths for the matte files.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-
-    # Find the media pool item by ID
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project currently open"}
-    mp = project.GetMediaPool()
-    root = mp.GetRootFolder()
-
-    # Search for clip by ID
-    def find_clip_by_id(folder, target_id):
-        for clip in (folder.GetClipList() or []):
-            if clip.GetUniqueId() == target_id:
-                return clip
-        for sub in (folder.GetSubFolderList() or []):
-            found = find_clip_by_id(sub, target_id)
-            if found:
-                return found
-        return None
-
-    clip = find_clip_by_id(root, media_pool_item_id)
-    if not clip:
-        return {"error": f"MediaPoolItem with ID {media_pool_item_id} not found"}
-
-    result = ms.AddClipMattesToMediaPool(clip, matte_paths, root)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def add_timeline_mattes_to_media_pool(timeline_item_index: int, matte_paths: List[str], track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Add timeline mattes from Media Storage to a timeline item.
-
-    Args:
-        timeline_item_index: 0-based index of the item in the track.
-        matte_paths: List of absolute file paths for the matte files.
-        track_type: Track type ('video' or 'audio'). Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    ms = resolve.GetMediaStorage()
-    if not ms:
-        return {"error": "Failed to get MediaStorage"}
-
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project currently open"}
-    timeline = project.GetCurrentTimeline()
-    if not timeline:
-        return {"error": "No current timeline"}
-
-    items = timeline.GetItemListInTrack(track_type, track_index)
-    if not items or timeline_item_index < 0 or timeline_item_index >= len(items):  # reject negatives (EX5)
-        return {"error": f"Timeline item at index {timeline_item_index} not found"}
-
-    item = items[timeline_item_index]
-    result = ms.AddTimelineMattesToMediaPool(item, matte_paths)
-    return {"success": bool(result)}