mcpforunityserver 9.4.0b20260203025228__py3-none-any.whl

cli/commands/scene.py

@@ -0,0 +1,231 @@
+"""Scene 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 scene():
+    """Scene operations - hierarchy, load, save, create scenes."""
+    pass
+
+
+@scene.command("hierarchy")
+@click.option(
+    "--parent",
+    default=None,
+    help="Parent GameObject to list children of (name, path, or instance ID)."
+)
+@click.option(
+    "--max-depth", "-d",
+    default=None,
+    type=int,
+    help="Maximum depth to traverse."
+)
+@click.option(
+    "--include-transform", "-t",
+    is_flag=True,
+    help="Include transform data for each node."
+)
+@click.option(
+    "--limit", "-l",
+    default=50,
+    type=int,
+    help="Maximum nodes to return."
+)
+@click.option(
+    "--cursor", "-c",
+    default=0,
+    type=int,
+    help="Pagination cursor."
+)
+@handle_unity_errors
+def hierarchy(
+    parent: Optional[str],
+    max_depth: Optional[int],
+    include_transform: bool,
+    limit: int,
+    cursor: int,
+):
+    """Get the scene hierarchy.
+
+    \b
+    Examples:
+        unity-mcp scene hierarchy
+        unity-mcp scene hierarchy --max-depth 3
+        unity-mcp scene hierarchy --parent "Canvas" --include-transform
+        unity-mcp scene hierarchy --format json
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "get_hierarchy",
+        "pageSize": limit,
+        "cursor": cursor,
+    }
+
+    if parent:
+        params["parent"] = parent
+    if max_depth is not None:
+        params["maxDepth"] = max_depth
+    if include_transform:
+        params["includeTransform"] = True
+
+    result = run_command("manage_scene", params, config)
+    click.echo(format_output(result, config.format))
+
+
+@scene.command("active")
+@handle_unity_errors
+def active():
+    """Get information about the active scene."""
+    config = get_config()
+    result = run_command("manage_scene", {"action": "get_active"}, config)
+    click.echo(format_output(result, config.format))
+
+
+@scene.command("load")
+@click.argument("scene")
+@click.option(
+    "--by-index", "-i",
+    is_flag=True,
+    help="Load by build index instead of path/name."
+)
+@handle_unity_errors
+def load(scene: str, by_index: bool):
+    """Load a scene.
+
+    \b
+    Examples:
+        unity-mcp scene load "Assets/Scenes/Main.unity"
+        unity-mcp scene load "MainScene"
+        unity-mcp scene load 0 --by-index
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {"action": "load"}
+
+    if by_index:
+        try:
+            params["buildIndex"] = int(scene)
+        except ValueError:
+            print_error(f"Invalid build index: {scene}")
+            sys.exit(1)
+    else:
+        if scene.endswith(".unity"):
+            params["path"] = scene
+        else:
+            params["name"] = scene
+
+    result = run_command("manage_scene", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Loaded scene: {scene}")
+
+
+@scene.command("save")
+@click.option(
+    "--path",
+    default=None,
+    help="Path to save the scene to (for new scenes)."
+)
+@handle_unity_errors
+def save(path: Optional[str]):
+    """Save the current scene.
+
+    \b
+    Examples:
+        unity-mcp scene save
+        unity-mcp scene save --path "Assets/Scenes/NewScene.unity"
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {"action": "save"}
+    if path:
+        params["path"] = path
+
+    result = run_command("manage_scene", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success("Scene saved")
+
+
+@scene.command("create")
+@click.argument("name")
+@click.option(
+    "--path",
+    default=None,
+    help="Path to create the scene at."
+)
+@handle_unity_errors
+def create(name: str, path: Optional[str]):
+    """Create a new scene.
+
+    \b
+    Examples:
+        unity-mcp scene create "NewLevel"
+        unity-mcp scene create "TestScene" --path "Assets/Scenes/Test"
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "create",
+        "name": name,
+    }
+    if path:
+        params["path"] = path
+
+    result = run_command("manage_scene", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Created scene: {name}")
+
+
+@scene.command("build-settings")
+@handle_unity_errors
+def build_settings():
+    """Get scenes in build settings."""
+    config = get_config()
+    result = run_command("manage_scene", {"action": "get_build_settings"}, config)
+    click.echo(format_output(result, config.format))
+
+
+@scene.command("screenshot")
+@click.option(
+    "--filename", "-f",
+    default=None,
+    help="Output filename (default: timestamp)."
+)
+@click.option(
+    "--supersize", "-s",
+    default=1,
+    type=int,
+    help="Supersize multiplier (1-4)."
+)
+@handle_unity_errors
+def screenshot(filename: Optional[str], supersize: int):
+    """Capture a screenshot of the scene.
+
+    \b
+    Examples:
+        unity-mcp scene screenshot
+        unity-mcp scene screenshot --filename "level_preview"
+        unity-mcp scene screenshot --supersize 2
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {"action": "screenshot"}
+    if filename:
+        params["fileName"] = filename
+    if supersize > 1:
+        params["superSize"] = supersize
+
+    result = run_command("manage_scene", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success("Screenshot captured")

cli/commands/script.py

@@ -0,0 +1,222 @@
+"""Script CLI commands."""
+
+import sys
+import json
+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
+from cli.utils.parsers import parse_json_list_or_exit
+from cli.utils.confirmation import confirm_destructive_action
+
+
+@click.group()
+def script():
+    """Script operations - create, read, edit C# scripts."""
+    pass
+
+
+@script.command("create")
+@click.argument("name")
+@click.option(
+    "--path", "-p",
+    default="Assets/Scripts",
+    help="Directory to create the script in."
+)
+@click.option(
+    "--type", "-t",
+    "script_type",
+    type=click.Choice(["MonoBehaviour", "ScriptableObject",
+                      "Editor", "EditorWindow", "Plain"]),
+    default="MonoBehaviour",
+    help="Type of script to create."
+)
+@click.option(
+    "--namespace", "-n",
+    default=None,
+    help="Namespace for the script."
+)
+@click.option(
+    "--contents", "-c",
+    default=None,
+    help="Full script contents (overrides template)."
+)
+@handle_unity_errors
+def create(name: str, path: str, script_type: str, namespace: Optional[str], contents: Optional[str]):
+    """Create a new C# script.
+
+    \b
+    Examples:
+        unity-mcp script create "PlayerController"
+        unity-mcp script create "GameManager" --path "Assets/Scripts/Managers"
+        unity-mcp script create "EnemyData" --type ScriptableObject
+        unity-mcp script create "CustomEditor" --type Editor --namespace "MyGame.Editor"
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "action": "create",
+        "name": name,
+        "path": path,
+        "scriptType": script_type,
+    }
+
+    if namespace:
+        params["namespace"] = namespace
+    if contents:
+        params["contents"] = contents
+
+    result = run_command("manage_script", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Created script: {name}.cs")
+
+
+@script.command("read")
+@click.argument("path")
+@click.option(
+    "--start-line", "-s",
+    default=None,
+    type=int,
+    help="Starting line number (1-based)."
+)
+@click.option(
+    "--line-count", "-n",
+    default=None,
+    type=int,
+    help="Number of lines to read."
+)
+@handle_unity_errors
+def read(path: str, start_line: Optional[int], line_count: Optional[int]):
+    """Read a C# script file.
+
+    \b
+    Examples:
+        unity-mcp script read "Assets/Scripts/Player.cs"
+        unity-mcp script read "Assets/Scripts/Player.cs" --start-line 10 --line-count 20
+    """
+    config = get_config()
+
+    parts = path.rsplit("/", 1)
+    filename = parts[-1]
+    directory = parts[0] if len(parts) > 1 else "Assets"
+    name = filename[:-3] if filename.endswith(".cs") else filename
+
+    params: dict[str, Any] = {
+        "action": "read",
+        "name": name,
+        "path": directory,
+    }
+
+    if start_line:
+        params["startLine"] = start_line
+    if line_count:
+        params["lineCount"] = line_count
+
+    result = run_command("manage_script", params, config)
+    # For read, just output the content directly
+    if result.get("success") and result.get("data"):
+        data = result.get("data", {})
+        if isinstance(data, dict) and "contents" in data:
+            click.echo(data["contents"])
+        else:
+            click.echo(format_output(result, config.format))
+    else:
+        click.echo(format_output(result, config.format))
+
+
+@script.command("delete")
+@click.argument("path")
+@click.option(
+    "--force", "-f",
+    is_flag=True,
+    help="Skip confirmation prompt."
+)
+@handle_unity_errors
+def delete(path: str, force: bool):
+    """Delete a C# script.
+
+    \b
+    Examples:
+        unity-mcp script delete "Assets/Scripts/OldScript.cs"
+    """
+    config = get_config()
+
+    confirm_destructive_action("Delete", "script", path, force)
+
+    parts = path.rsplit("/", 1)
+    filename = parts[-1]
+    directory = parts[0] if len(parts) > 1 else "Assets"
+    name = filename[:-3] if filename.endswith(".cs") else filename
+
+    params: dict[str, Any] = {
+        "action": "delete",
+        "name": name,
+        "path": directory,
+    }
+
+    result = run_command("manage_script", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Deleted: {path}")
+
+
+@script.command("edit")
+@click.argument("path")
+@click.option(
+    "--edits", "-e",
+    required=True,
+    help='Edits as JSON array of {startLine, startCol, endLine, endCol, newText}.'
+)
+@handle_unity_errors
+def edit(path: str, edits: str):
+    """Apply text edits to a script.
+
+    \b
+    Examples:
+        unity-mcp script edit "Assets/Scripts/Player.cs" --edits '[{"startLine": 10, "startCol": 1, "endLine": 10, "endCol": 20, "newText": "// Modified"}]'
+    """
+    config = get_config()
+
+    edits_list = parse_json_list_or_exit(edits, "edits")
+
+    params: dict[str, Any] = {
+        "uri": path,
+        "edits": edits_list,
+    }
+
+    result = run_command("apply_text_edits", params, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Applied edits to: {path}")
+
+
+@script.command("validate")
+@click.argument("path")
+@click.option(
+    "--level", "-l",
+    type=click.Choice(["basic", "standard"]),
+    default="basic",
+    help="Validation level."
+)
+@handle_unity_errors
+def validate(path: str, level: str):
+    """Validate a C# script for errors.
+
+    \b
+    Examples:
+        unity-mcp script validate "Assets/Scripts/Player.cs"
+        unity-mcp script validate "Assets/Scripts/Player.cs" --level standard
+    """
+    config = get_config()
+
+    params: dict[str, Any] = {
+        "uri": path,
+        "level": level,
+        "include_diagnostics": True,
+    }
+
+    result = run_command("validate_script", params, config)
+    click.echo(format_output(result, config.format))

cli/commands/shader.py

@@ -0,0 +1,226 @@
+"""Shader CLI commands for managing Unity shaders."""
+
+import sys
+import click
+from typing import Optional
+
+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.confirmation import confirm_destructive_action
+
+
+@click.group()
+def shader():
+    """Shader operations - create, read, update, delete shaders."""
+    pass
+
+
+@shader.command("read")
+@click.argument("path")
+@handle_unity_errors
+def read_shader(path: str):
+    """Read a shader file.
+
+    \\b
+    Examples:
+        unity-mcp shader read "Assets/Shaders/MyShader.shader"
+    """
+    config = get_config()
+
+    # Extract name from path
+    import os
+    name = os.path.splitext(os.path.basename(path))[0]
+    directory = os.path.dirname(path)
+
+    result = run_command("manage_shader", {
+        "action": "read",
+        "name": name,
+        "path": directory or "Assets/",
+    }, config)
+
+    # If successful, display the contents nicely
+    if result.get("success") and result.get("data", {}).get("contents"):
+        click.echo(result["data"]["contents"])
+    else:
+        click.echo(format_output(result, config.format))
+
+
+@shader.command("create")
+@click.argument("name")
+@click.option(
+    "--path", "-p",
+    default="Assets/Shaders",
+    help="Directory to create shader in."
+)
+@click.option(
+    "--contents", "-c",
+    default=None,
+    help="Shader code (reads from stdin if not provided)."
+)
+@click.option(
+    "--file", "-f",
+    "file_path",
+    default=None,
+    type=click.Path(exists=True),
+    help="Read shader code from file."
+)
+@handle_unity_errors
+def create_shader(name: str, path: str, contents: Optional[str], file_path: Optional[str]):
+    """Create a new shader.
+
+    \\b
+    Examples:
+        unity-mcp shader create "MyShader" --path "Assets/Shaders"
+        unity-mcp shader create "MyShader" --file local_shader.shader
+        echo "Shader code..." | unity-mcp shader create "MyShader"
+    """
+    config = get_config()
+
+    # Get contents from file, option, or stdin
+    if file_path:
+        with open(file_path, 'r') as f:
+            shader_contents = f.read()
+    elif contents:
+        shader_contents = contents
+    else:
+        # Read from stdin if available
+        import sys
+        if not sys.stdin.isatty():
+            shader_contents = sys.stdin.read()
+        else:
+            # Provide default shader template
+            shader_contents = f'''Shader "Custom/{name}"
+{{
+    Properties
+    {{
+        _Color ("Color", Color) = (1,1,1,1)
+        _MainTex ("Albedo (RGB)", 2D) = "white" {{}}
+    }}
+    SubShader
+    {{
+        Tags {{ "RenderType"="Opaque" }}
+        LOD 200
+        
+        CGPROGRAM
+        #pragma surface surf Standard fullforwardshadows
+        #pragma target 3.0
+        
+        sampler2D _MainTex;
+        fixed4 _Color;
+        
+        struct Input
+        {{
+            float2 uv_MainTex;
+        }};
+        
+        void surf (Input IN, inout SurfaceOutputStandard o)
+        {{
+            fixed4 c = tex2D(_MainTex, IN.uv_MainTex) * _Color;
+            o.Albedo = c.rgb;
+            o.Alpha = c.a;
+        }}
+        ENDCG
+    }}
+    FallBack "Diffuse"
+}}
+'''
+
+    result = run_command("manage_shader", {
+        "action": "create",
+        "name": name,
+        "path": path,
+        "contents": shader_contents,
+    }, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Created shader: {path}/{name}.shader")
+
+
+@shader.command("update")
+@click.argument("path")
+@click.option(
+    "--contents", "-c",
+    default=None,
+    help="New shader code."
+)
+@click.option(
+    "--file", "-f",
+    "file_path",
+    default=None,
+    type=click.Path(exists=True),
+    help="Read shader code from file."
+)
+@handle_unity_errors
+def update_shader(path: str, contents: Optional[str], file_path: Optional[str]):
+    """Update an existing shader.
+
+    \\b
+    Examples:
+        unity-mcp shader update "Assets/Shaders/MyShader.shader" --file updated.shader
+        echo "New shader code" | unity-mcp shader update "Assets/Shaders/MyShader.shader"
+    """
+    config = get_config()
+
+    import os
+    name = os.path.splitext(os.path.basename(path))[0]
+    directory = os.path.dirname(path)
+
+    # Get contents from file, option, or stdin
+    if file_path:
+        with open(file_path, 'r') as f:
+            shader_contents = f.read()
+    elif contents:
+        shader_contents = contents
+    else:
+        import sys
+        if not sys.stdin.isatty():
+            shader_contents = sys.stdin.read()
+        else:
+            print_error(
+                "No shader contents provided. Use --contents, --file, or pipe via stdin.")
+            sys.exit(1)
+
+    result = run_command("manage_shader", {
+        "action": "update",
+        "name": name,
+        "path": directory or "Assets/",
+        "contents": shader_contents,
+    }, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Updated shader: {path}")
+
+
+@shader.command("delete")
+@click.argument("path")
+@click.option(
+    "--force", "-f",
+    is_flag=True,
+    help="Skip confirmation prompt."
+)
+@handle_unity_errors
+def delete_shader(path: str, force: bool):
+    """Delete a shader.
+
+    \\b
+    Examples:
+        unity-mcp shader delete "Assets/Shaders/OldShader.shader"
+        unity-mcp shader delete "Assets/Shaders/OldShader.shader" --force
+    """
+    config = get_config()
+
+    confirm_destructive_action("Delete", "shader", path, force)
+
+    import os
+    name = os.path.splitext(os.path.basename(path))[0]
+    directory = os.path.dirname(path)
+
+    result = run_command("manage_shader", {
+        "action": "delete",
+        "name": name,
+        "path": directory or "Assets/",
+    }, config)
+    click.echo(format_output(result, config.format))
+    if result.get("success"):
+        print_success(f"Deleted shader: {path}")