mcpforunityserver 9.3.0b20260129104751__py3-none-any.whl

cli/commands/material.py

@@ -0,0 +1,239 @@
+"""Material CLI commands."""
+
+import sys
+import json
+import click
+from typing import Optional, Any, Tuple
+
+from cli.utils.config import get_config
+from cli.utils.output import format_output, print_error, print_success
+from cli.utils.connection import run_command, handle_unity_errors
+from cli.utils.parsers import parse_value_safe, parse_json_dict_or_exit
+from cli.utils.constants import SEARCH_METHOD_CHOICE_RENDERER
+
+
+@click.group()
+def material():
+    """Material operations - create, modify, assign materials."""
+    pass
+
+
+@material.command("info")
+@click.argument("path")
+@handle_unity_errors
+def info(path: str):
+    """Get information about a material.
+
+    \b
+    Examples:
+        unity-mcp material info "Assets/Materials/Red.mat"
+    """
+    config = get_config()
+
+    result = run_command("manage_material", {
+        "action": "get_material_info",
+        "materialPath": path,
+    }, config)
+    click.echo(format_output(result, config.format))
+
+
+@material.command("create")
+@click.argument("path")
+@click.option(
+    "--shader", "-s",
+    default="Standard",
+    help="Shader to use (default: Standard)."
+)
+@click.option(
+    "--properties", "-p",
+    default=None,
+    help='Initial properties as JSON.'
+)
+@handle_unity_errors
+def create(path: str, shader: str, properties: Optional[str]):
+    """Create a new material.
+
+    \b
+    Examples:
+        unity-mcp material create "Assets/Materials/NewMat.mat"
+        unity-mcp material create "Assets/Materials/Red.mat" --shader "Universal Render Pipeline/Lit"
+        unity-mcp material create "Assets/Materials/Blue.mat" --properties '{"_Color": [0,0,1,1]}'
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "create",
+        "materialPath": path,
+        "shader": shader,
+    }
+
+    if properties:
+        params["properties"] = parse_json_dict_or_exit(properties, "properties")
+
+    result = run_command("manage_material", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Created material: {path}")
+
+
+@material.command("set-color")
+@click.argument("path")
+@click.argument("r", type=float)
+@click.argument("g", type=float)
+@click.argument("b", type=float)
+@click.argument("a", type=float, default=1.0)
+@click.option(
+    "--property", "-p",
+    default="_Color",
+    help="Color property name (default: _Color)."
+)
+@handle_unity_errors
+def set_color(path: str, r: float, g: float, b: float, a: float, property: str):
+    """Set a material's color.
+
+    \b
+    Examples:
+        unity-mcp material set-color "Assets/Materials/Red.mat" 1 0 0
+        unity-mcp material set-color "Assets/Materials/Blue.mat" 0 0 1 0.5
+        unity-mcp material set-color "Assets/Materials/Mat.mat" 1 1 0 --property "_BaseColor"
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "set_material_color",
+        "materialPath": path,
+        "property": property,
+        "color": [r, g, b, a],
+    }
+
+    result = run_command("manage_material", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Set color on: {path}")
+
+
+@material.command("set-property")
+@click.argument("path")
+@click.argument("property_name")
+@click.argument("value")
+@handle_unity_errors
+def set_property(path: str, property_name: str, value: str):
+    """Set a shader property on a material.
+
+    \b
+    Examples:
+        unity-mcp material set-property "Assets/Materials/Mat.mat" _Metallic 0.5
+        unity-mcp material set-property "Assets/Materials/Mat.mat" _Smoothness 0.8
+        unity-mcp material set-property "Assets/Materials/Mat.mat" _MainTex "Assets/Textures/Tex.png"
+    """
+    config = get_config()
+
+    # Try to parse value as JSON for complex types
+    parsed_value = parse_value_safe(value)
+
+    params: dict[str, Any] = {
+        "action": "set_material_shader_property",
+        "materialPath": path,
+        "property": property_name,
+        "value": parsed_value,
+    }
+
+    result = run_command("manage_material", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Set {property_name} on: {path}")
+
+
+@material.command("assign")
+@click.argument("material_path")
+@click.argument("target")
+@click.option(
+    "--search-method",
+    type=SEARCH_METHOD_CHOICE_RENDERER,
+    default=None,
+    help="How to find the target GameObject."
+)
+@click.option(
+    "--slot", "-s",
+    default=0,
+    type=int,
+    help="Material slot index (default: 0)."
+)
+@click.option(
+    "--mode", "-m",
+    type=click.Choice(["shared", "instance", "property_block"]),
+    default="shared",
+    help="Assignment mode."
+)
+@handle_unity_errors
+def assign(material_path: str, target: str, search_method: Optional[str], slot: int, mode: str):
+    """Assign a material to a GameObject's renderer.
+
+    \b
+    Examples:
+        unity-mcp material assign "Assets/Materials/Red.mat" "Cube"
+        unity-mcp material assign "Assets/Materials/Blue.mat" "Player" --mode instance
+        unity-mcp material assign "Assets/Materials/Mat.mat" "-81840" --search-method by_id --slot 1
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "assign_material_to_renderer",
+        "materialPath": material_path,
+        "target": target,
+        "slot": slot,
+        "mode": mode,
+    }
+
+    if search_method:
+        params["searchMethod"] = search_method
+
+    result = run_command("manage_material", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Assigned material to: {target}")
+
+
+@material.command("set-renderer-color")
+@click.argument("target")
+@click.argument("r", type=float)
+@click.argument("g", type=float)
+@click.argument("b", type=float)
+@click.argument("a", type=float, default=1.0)
+@click.option(
+    "--search-method",
+    type=SEARCH_METHOD_CHOICE_RENDERER,
+    default=None,
+    help="How to find the target GameObject."
+)
+@click.option(
+    "--mode", "-m",
+    type=click.Choice(["shared", "instance", "property_block"]),
+    default="property_block",
+    help="Modification mode."
+)
+@handle_unity_errors
+def set_renderer_color(target: str, r: float, g: float, b: float, a: float, search_method: Optional[str], mode: str):
+    """Set a renderer's material color directly.
+
+    \b
+    Examples:
+        unity-mcp material set-renderer-color "Cube" 1 0 0
+        unity-mcp material set-renderer-color "Player" 0 1 0 --mode instance
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "set_renderer_color",
+        "target": target,
+        "color": [r, g, b, a],
+        "mode": mode,
+    }
+
+    if search_method:
+        params["searchMethod"] = search_method
+
+    result = run_command("manage_material", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Set renderer color on: {target}")

cli/commands/prefab.py

@@ -0,0 +1,248 @@
+"""Prefab CLI commands."""
+
+import sys
+import click
+from typing import Optional, Any
+
+from cli.utils.config import get_config
+from cli.utils.output import format_output, print_error, print_success
+from cli.utils.connection import run_command, handle_unity_errors
+
+
+@click.group()
+def prefab():
+    """Prefab operations - info, hierarchy, open, save, close, create prefabs."""
+    pass
+
+
+@prefab.command("open")
+@click.argument("path")
+@handle_unity_errors
+def open_stage(path: str):
+    """Open a prefab in the prefab stage for editing.
+
+    \b
+    Examples:
+        unity-mcp prefab open "Assets/Prefabs/Player.prefab"
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "open_stage",
+        "prefabPath": path,
+    }
+
+    result = run_command("manage_prefabs", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Opened prefab: {path}")
+
+
+@prefab.command("close")
+@click.option(
+    "--save", "-s",
+    is_flag=True,
+    help="Save the prefab before closing."
+)
+@handle_unity_errors
+def close_stage(save: bool):
+    """Close the current prefab stage.
+
+    \b
+    Examples:
+        unity-mcp prefab close
+        unity-mcp prefab close --save
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "close_stage",
+    }
+    if save:
+        params["saveBeforeClose"] = True
+
+    result = run_command("manage_prefabs", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success("Closed prefab stage")
+
+
+@prefab.command("save")
+@click.option(
+    "--force", "-f",
+    is_flag=True,
+    help="Force save even if no changes detected. Useful for automated workflows."
+)
+@handle_unity_errors
+def save_stage(force: bool):
+    """Save the currently open prefab stage.
+
+    \b
+    Examples:
+        unity-mcp prefab save
+        unity-mcp prefab save --force
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "save_open_stage",
+    }
+    if force:
+        params["force"] = True
+
+    result = run_command("manage_prefabs", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success("Saved prefab")
+
+
+@prefab.command("info")
+@click.argument("path")
+@click.option(
+    "--compact", "-c",
+    is_flag=True,
+    help="Show compact output (key values only)."
+)
+@handle_unity_errors
+def info(path: str, compact: bool):
+    """Get information about a prefab asset.
+
+    \b
+    Examples:
+        unity-mcp prefab info "Assets/Prefabs/Player.prefab"
+        unity-mcp prefab info "Assets/Prefabs/UI.prefab" --compact
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "get_info",
+        "prefabPath": path,
+    }
+
+    result = run_command("manage_prefabs", params, config)
+    # Get the actual response data from the wrapped result structure
+    response_data = result.get("result", result)
+    if compact and response_data.get("success") and response_data.get("data"):
+        data = response_data["data"]
+        click.echo(f"Prefab: {data.get('assetPath', path)}")
+        click.echo(f"  Type: {data.get('prefabType', 'Unknown')}")
+        click.echo(f"  Root: {data.get('rootObjectName', 'N/A')}")
+        click.echo(f"  GUID: {data.get('guid', 'N/A')}")
+        click.echo(
+            f"  Components: {len(data.get('rootComponentTypes', []))}")
+        click.echo(f"  Children: {data.get('childCount', 0)}")
+        if data.get('isVariant'):
+            click.echo(f"  Variant of: {data.get('parentPrefab', 'N/A')}")
+    else:
+        click.echo(format_output(result, config.format))
+
+
+@prefab.command("hierarchy")
+@click.argument("path")
+@click.option(
+    "--compact", "-c",
+    is_flag=True,
+    help="Show compact output (names and paths only)."
+)
+@click.option(
+    "--show-prefab-info", "-p",
+    is_flag=True,
+    help="Show prefab nesting information."
+)
+@handle_unity_errors
+def hierarchy(path: str, compact: bool, show_prefab_info: bool):
+    """Get the hierarchical structure of a prefab.
+
+    \b
+    Examples:
+        unity-mcp prefab hierarchy "Assets/Prefabs/Player.prefab"
+        unity-mcp prefab hierarchy "Assets/Prefabs/UI.prefab" --compact
+        unity-mcp prefab hierarchy "Assets/Prefabs/Complex.prefab" --show-prefab-info
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "get_hierarchy",
+        "prefabPath": path,
+    }
+
+    result = run_command("manage_prefabs", params, config)
+    # Get the actual response data from the wrapped result structure
+    response_data = result.get("result", result)
+    if compact and response_data.get("success") and response_data.get("data"):
+        data = response_data["data"]
+        items = data.get("items", [])
+        for item in items:
+            indent = "  " * item.get("path", "").count("/")
+            prefab_info = ""
+            if show_prefab_info and item.get("prefab", {}).get("isNestedRoot"):
+                prefab_info = f" [nested: {item['prefab']['assetPath']}]"
+            click.echo(f"{indent}{item.get('name')}{prefab_info}")
+        click.echo(f"\nTotal: {data.get('total', 0)} objects")
+    elif show_prefab_info:
+        # Show prefab info in readable format
+        if response_data.get("success") and response_data.get("data"):
+            data = response_data["data"]
+            items = data.get("items", [])
+            for item in items:
+                prefab = item.get("prefab", {})
+                prefab_info = ""
+                if prefab.get("isRoot"):
+                    prefab_info = " [root]"
+                elif prefab.get("isNestedRoot"):
+                    prefab_info = f" [nested: {prefab.get('nestingDepth', 0)}]"
+                click.echo(f"{item.get('path')}{prefab_info}")
+            click.echo(f"\nTotal: {data.get('total', 0)} objects")
+        else:
+            click.echo(format_output(result, config.format))
+    else:
+        click.echo(format_output(result, config.format))
+
+
+@prefab.command("create")
+@click.argument("target")
+@click.argument("path")
+@click.option(
+    "--overwrite",
+    is_flag=True,
+    help="Overwrite existing prefab at path."
+)
+@click.option(
+    "--include-inactive",
+    is_flag=True,
+    help="Include inactive objects when finding target."
+)
+@click.option(
+    "--unlink-if-instance",
+    is_flag=True,
+    help="Unlink from existing prefab before creating new one."
+)
+@handle_unity_errors
+def create(target: str, path: str, overwrite: bool, include_inactive: bool, unlink_if_instance: bool):
+    """Create a prefab from a scene GameObject.
+
+    \b
+    Examples:
+        unity-mcp prefab create "Player" "Assets/Prefabs/Player.prefab"
+        unity-mcp prefab create "Enemy" "Assets/Prefabs/Enemy.prefab" --overwrite
+        unity-mcp prefab create "EnemyInstance" "Assets/Prefabs/BossEnemy.prefab" --unlink-if-instance
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "create_from_gameobject",
+        "target": target,
+        "prefabPath": path,
+    }
+
+    if overwrite:
+        params["allowOverwrite"] = True
+    if include_inactive:
+        params["searchInactive"] = True
+    if unlink_if_instance:
+        params["unlinkIfInstance"] = True
+
+    result = run_command("manage_prefabs", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Created prefab: {path}")