unity-editor-mcp/src/handlers/asset/ModifyPrefabToolHandler.js

MCP server for Unity Editor integration - enables AI assistants to control Unity Editor

import { BaseToolHandler } from '../base/BaseToolHandler.js';

export class ModifyPrefabToolHandler extends BaseToolHandler {
    constructor(unityConnection) {
        super(
            'modify_prefab',
            'Modify properties of an existing prefab',
            {
                type: 'object',
                properties: {
                    prefabPath: {
                        type: 'string',
                        description: 'Asset path to the prefab (must start with Assets/ and end with .prefab)'
                    },
                    modifications: {
                        type: 'object',
                        description: 'Object containing properties to modify'
                    },
                    applyToInstances: {
                        type: 'boolean',
                        default: true,
                        description: 'Apply changes to scene instances of the prefab'
                    }
                },
                required: ['prefabPath', 'modifications']
            }
        );
        this.unityConnection = unityConnection;
    }

    validate(params) {
        // Call parent validation for required fields
        super.validate(params);

        const { prefabPath, modifications } = params;

        // Validate prefabPath format
        if (!prefabPath.startsWith('Assets/') || !prefabPath.endsWith('.prefab')) {
            throw new Error('prefabPath must start with Assets/ and end with .prefab');
        }

        // Validate modifications is object
        if (typeof modifications !== 'object' || modifications === null || Array.isArray(modifications)) {
            throw new Error('modifications must be an object');
        }

        // Validate modifications is not empty
        if (Object.keys(modifications).length === 0) {
            throw new Error('modifications cannot be empty');
        }
    }

    async execute(params) {
        const {
            prefabPath,
            modifications,
            applyToInstances = true
        } = params;

        // Ensure connected
        if (!this.unityConnection.isConnected()) {
            await this.unityConnection.connect();
        }

        const result = await this.unityConnection.sendCommand('modify_prefab', {
            prefabPath,
            modifications,
            applyToInstances
        });

        return result;
    }
}