/*
┌──────────────────────────────────────────────────────────────────┐
│  Author: Ivan Murzak (https://github.com/IvanMurzak)             │
│  Repository: GitHub (https://github.com/IvanMurzak/Unity-MCP)    │
│  Copyright (c) 2025 Ivan Murzak                                  │
│  Licensed under the Apache License, Version 2.0.                 │
│  See the LICENSE file in the project root for more information.  │
└──────────────────────────────────────────────────────────────────┘
*/

#nullable enable
using System;
using System.ComponentModel;
using com.IvanMurzak.McpPlugin;
using com.IvanMurzak.ReflectorNet.Utils;
using com.IvanMurzak.Unity.MCP.Editor.Utils;
using AIGD;
using com.IvanMurzak.Unity.MCP.Runtime.Extensions;
using UnityEditor;

namespace com.IvanMurzak.Unity.MCP.Editor.API
{
    public partial class Tool_Assets_Prefab
    {
        public const string AssetsPrefabOpenToolId = "assets-prefab-open";
        [AiTool
        (
            AssetsPrefabOpenToolId,
            Title = "Assets / Prefab / Open"
        )]
        [AiSkillDescription("Open the prefab edit stage for a prefab instance or prefab asset GameObject. " +
            "Modifications inside the edit stage propagate to all instances. " +
            "Pair with '" + AssetsPrefabCloseToolId + "' to exit the stage when done.")]
        [AiSkillBody("Open prefab edit mode for a specific GameObject. " +
            "In the Edit mode you can modify the prefab. " +
            "The modification will be applied to all instances of the prefab across the project. " +
            "Note: Please use '" + AssetsPrefabCloseToolId + "' tool later to exit prefab editing mode.\n\n" +
            "## Inputs\n\n" +
            "- `gameObjectRef` — reference to a scene prefab instance OR a prefab asset GameObject. The tool resolves " +
            "the prefab asset path via `PrefabUtility.GetPrefabAssetPathOfNearestInstanceRoot` and opens the appropriate " +
            "prefab stage.\n\n" +
            "## Behavior\n\n" +
            "Asset-side GameObjects open via the simple `OpenPrefab(path)` overload. Scene-instance GameObjects open " +
            "via `OpenPrefab(path, gameObject)` so the editor remembers which instance prompted the edit. Editor windows " +
            "are repainted before returning. Throws when the GameObject cannot be resolved or the stage fails to open.")]
        [Description("Open prefab edit mode for a specific GameObject. " +
            "In the Edit mode you can modify the prefab. " +
            "The modification will be applied to all instances of the prefab across the project. " +
            "Note: Please use '" + AssetsPrefabCloseToolId + "' tool later to exit prefab editing mode.")]
        public void Open
        (
            [Description("GameObject that represents prefab instance of an original prefab GameObject.")]
            GameObjectRef gameObjectRef
        )
        {
            if (gameObjectRef?.IsValid(out var validationError) == false)
                throw new ArgumentException(validationError, nameof(gameObjectRef));

            MainThread.Instance.Run(() =>
            {
                var prefabStage = UnityEditor.SceneManagement.PrefabStageUtility.GetCurrentPrefabStage();
                var gameObject = gameObjectRef.FindGameObject();

                if (gameObject == null)
                    throw new Exception("GameObject not found. Provide a reference to existed GameObject.");

                var prefabAssetPath = PrefabUtility.GetPrefabAssetPathOfNearestInstanceRoot(gameObject);

                prefabStage = gameObject.IsAsset()
                    ? UnityEditor.SceneManagement.PrefabStageUtility.OpenPrefab(prefabAssetPath)
                    : UnityEditor.SceneManagement.PrefabStageUtility.OpenPrefab(prefabAssetPath, gameObject);

                if (prefabStage == null)
                    throw new Exception("Failed to open prefab edit mode for the provided GameObject.");

                EditorUtils.RepaintAllEditorWindows();
            });
        }
    }
}