Unity × Claude MCP構築 (実践編: 読み書き・自律エージェント化)

using UnityEngine;
using UnityEditor;
using System.IO;
using System.Text;
using System.Collections.Generic;

public class McpExporter : MonoBehaviour
{
    // 保存先
    private static string FilePath => Path.Combine(Directory.GetParent(Application.dataPath).FullName, "mcp_hierarchy.json");

    // 【重要】シーン全部だと重すぎるので「選択したものだけ」書き出す仕様に変更
    [MenuItem("MCP/Export Selected Objects")]
    public static void ExportSelected()
    {
        var selectedObjects = Selection.gameObjects;

        if (selectedObjects.Length == 0)
        {
            Debug.LogWarning("[MCP] オブジェクトが選択されていません。Hierarchyで対象を選択してください。");
            return;
        }

        var sb = new StringBuilder();
        sb.AppendLine("{ \"mode\": \"SelectedObjects\", \"objects\": [");

        for (int i = 0; i < selectedObjects.Length; i++)
        {
            Traverse(selectedObjects[i].transform, sb, 1);
            if (i < selectedObjects.Length - 1) sb.AppendLine(",");
        }

        sb.AppendLine("\n] }");

        // UTF-8 (BOMなし) で保存
        File.WriteAllText(FilePath, sb.ToString(), new UTF8Encoding(false));
        Debug.Log($"[MCP] 選択された {selectedObjects.Length} 個の情報を書き出しました: {FilePath}");
    }

    // 再帰的に子要素とコンポーネントを探索
    private static void Traverse(Transform tf, StringBuilder sb, int indentLevel)
    {
        string indent = new string(' ', indentLevel * 2);
        
        // 座標は桁数を丸めてトークン節約
        string posStr = $"({tf.position.x:F2}, {tf.position.y:F2}, {tf.position.z:F2})";

        sb.Append($"{indent}{{ \"name\": \"{EscapeJson(tf.name)}\", ");
        sb.Append($"\"active\": {tf.gameObject.activeInHierarchy.ToString().ToLower()}, ");
        sb.Append($"\"position\": \"{posStr}\", ");

        // コンポーネント一覧を取得（Transform以外）
        var components = tf.GetComponents<Component>();
        List<string> compNames = new List<string>();
        foreach (var c in components)
        {
            if (c != null && !(c is Transform)) compNames.Add(c.GetType().Name);
        }
        sb.Append($"\"components\": [{string.Join(", ", compNames.ConvertAll(c => $"\"{c}\""))}], ");

        // 子オブジェクトがあれば再帰処理
        if (tf.childCount > 0)
        {
            sb.AppendLine($"\"children\": [");
            for (int i = 0; i < tf.childCount; i++)
            {
                Traverse(tf.GetChild(i), sb, indentLevel + 1);
                if (i < tf.childCount - 1) sb.AppendLine(",");
            }
            sb.Append($"{indent}]");
        }
        else
        {
            sb.Append($"\"children\": []");
        }

        sb.Append(" }");
    }

    private static string EscapeJson(string str)
    {
        if (string.IsNullOrEmpty(str)) return "";
        return str.Replace("\\", "\\\\").Replace("\"", "\\\"");
    }
}

from fastmcp import FastMCP
import os

# サーバー定義
mcp = FastMCP("UnityContext")

# パス設定（server.pyの位置を基準に絶対パス生成）
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
ASSETS_DIR = os.path.join(BASE_DIR, "Assets")
HIERARCHY_FILE = os.path.join(BASE_DIR, "mcp_hierarchy.json")

# --- Resource: シーン情報の提供 ---
@mcp.resource("unity://hierarchy")
def get_hierarchy() -> str:
    """Unityの現在のシーン階層構造（Hierarchy）を読み取ります。"""
    if not os.path.exists(HIERARCHY_FILE):
        return "エラー: データなし。Unityで書き出しを実行してください。"
    try:
        with open(HIERARCHY_FILE, "r", encoding="utf-8") as f:
            return f.read()
    except Exception as e:
        return f"エラー: {str(e)}"

# --- Tool 1: ファイル一覧取得 ---
@mcp.tool()
def list_scripts() -> str:
    """
    Assetsフォルダ内にある C# スクリプト(.cs) の一覧を取得します。
    ファイル構成を把握し、既存のクラス名を知るために使用します。
    """
    if not os.path.exists(ASSETS_DIR):
        return "エラー: Assetsフォルダが見つかりません。"

    file_list = []
    for root, dirs, files in os.walk(ASSETS_DIR):
        for file in files:
            if file.endswith(".cs"):
                # フルパスからAssets以下の相対パスに変換
                full_path = os.path.join(root, file)
                rel_path = os.path.relpath(full_path, BASE_DIR)
                file_list.append(rel_path)
    
    return "\n".join(file_list) if file_list else "スクリプトが見つかりませんでした。"

# --- Tool 2: ファイル読み込み ---
@mcp.tool()
def read_file(path: str) -> str:
    """
    指定されたパスのファイルの中身を読み取ります。
    コードの修正や確認を行う前に必ず使用してください。
    path: Assets/ から始まるファイルパス
    """
    target_path = os.path.join(BASE_DIR, path)
    if not os.path.exists(target_path): return "エラー: ファイルが見つかりません。"
    if "Assets" not in os.path.abspath(target_path): return "エラー: Assets外へのアクセス禁止。"

    try:
        with open(target_path, "r", encoding="utf-8") as f:
            return f.read()
    except Exception as e:
        return f"読み込みエラー: {str(e)}"

# --- Tool 3: ファイル書き込み（作成・修正） ---
@mcp.tool()
def write_file(path: str, content: str) -> str:
    """
    指定されたパスにファイルを作成、または上書き保存します。
    新規スクリプトの作成や、コードの修正に使用します。
    
    path: Assets/ から始まるファイルパス (例: Assets/Scripts/New.cs)
    content: 書き込む完全なテキスト内容
    """
    target_path = os.path.join(BASE_DIR, path)
    if "Assets" not in os.path.abspath(target_path): return "エラー: Assets外への書き込み禁止。"

    # フォルダがなければ自動作成
    os.makedirs(os.path.dirname(target_path), exist_ok=True)

    try:
        with open(target_path, "w", encoding="utf-8") as f:
            f.write(content)
        return f"成功: 保存しました: {path}"
    except Exception as e:
        return f"書き込みエラー: {str(e)}"

if __name__ == "__main__":
    mcp.run()