/*
┌──────────────────────────────────────────────────────────────────┐
│  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 System.Threading.Tasks;
using com.IvanMurzak.McpPlugin;
using com.IvanMurzak.McpPlugin.Common.Model;
using com.IvanMurzak.ReflectorNet.Utils;
using com.IvanMurzak.Unity.MCP.Editor.Utils;
using UnityEditor.PackageManager;

namespace com.IvanMurzak.Unity.MCP.Editor.API
{
    public partial class Tool_Package
    {
        public const string PackageRemoveToolId = "package-remove";
        [AiTool
        (
            PackageRemoveToolId,
            Title = "Package Manager / Remove",
            DestructiveHint = true,
            Enabled = false
        )]
        [AiSkillDescription("Uninstall a UPM package from the Unity project. Modifies `manifest.json` and may " +
            "trigger a domain reload — the final result is delivered after the reload via the request's `requestId`. " +
            "Built-in packages and packages that are dependencies of others cannot be removed. " +
            "Use '" + PackageListToolId + "' to list installed packages first.")]
        [AiSkillBody("Remove (uninstall) a package from the Unity project. " +
            "This removes the package from the project's manifest.json and triggers package resolution. " +
            "Note: Built-in packages and packages that are dependencies of other installed packages cannot be removed. " +
            "Note: Package removal may trigger a domain reload. The result will be sent after the reload completes. " +
            "Use '" + PackageListToolId + "' tool to list installed packages first.\n\n" +
            "## Inputs\n\n" +
            "- `packageId` — package name (e.g. `com.unity.textmeshpro`). A trailing `@<version>` is stripped automatically " +
            "before being passed to `Client.Remove`, so accidental versioned IDs still work.\n\n" +
            "## Behavior\n\n" +
            "First verifies the package is installed via an offline `Client.List` — returns a clear `PackageNotFound` " +
            "error if not. On removal failure, surfaces Unity's error message. On success, schedules a post-domain-reload " +
            "notification so the response is delivered after Unity finishes the resolution.")]
        [Description("Remove (uninstall) a package from the Unity project. " +
            "This removes the package from the project's manifest.json and triggers package resolution. " +
            "Note: Built-in packages and packages that are dependencies of other installed packages cannot be removed. " +
            "Note: Package removal may trigger a domain reload. The result will be sent after the reload completes. " +
            "Use '" + PackageListToolId + "' tool to list installed packages first.")]
        public static ResponseCallTool Remove
        (
            [Description("The ID of the package to remove. Example: 'com.unity.textmeshpro'. Do not include version number.")]
            string packageId,
            [RequestID]
            string? requestId = null
        )
        {
            if (requestId == null || string.IsNullOrWhiteSpace(requestId))
                return ResponseCallTool.Error("[Error] Original request with valid RequestID must be provided.");

            if (string.IsNullOrWhiteSpace(packageId))
                return ResponseCallTool.Error(Error.PackageNameIsEmpty()).SetRequestID(requestId);

            // Remove version suffix if accidentally included
            var cleanPackageId = packageId.Contains("@", StringComparison.OrdinalIgnoreCase)
                ? packageId.Substring(0, packageId.IndexOf('@', StringComparison.OrdinalIgnoreCase))
                : packageId;

            MainThread.Instance.RunAsync(async () =>
            {
                await Task.Yield();

                // First verify the package is installed
                var listRequest = Client.List(offlineMode: true);
                while (!listRequest.IsCompleted)
                    await Task.Yield();

                if (listRequest.Status == StatusCode.Success)
                {
                    var isInstalled = false;
                    foreach (var pkg in listRequest.Result)
                    {
                        if (pkg.name.Equals(cleanPackageId, StringComparison.OrdinalIgnoreCase))
                        {
                            isInstalled = true;
                            break;
                        }
                    }

                    if (!isInstalled)
                    {
                        _ = UnityMcpPluginEditor.NotifyToolRequestCompleted(new RequestToolCompletedData
                        {
                            RequestId = requestId,
                            Result = ResponseCallTool.Error(Error.PackageNotFound(cleanPackageId)).SetRequestID(requestId)
                        });
                        return;
                    }
                }

                var removeRequest = Client.Remove(cleanPackageId);

                while (!removeRequest.IsCompleted)
                    await Task.Yield();

                if (removeRequest.Status == StatusCode.Failure)
                {
                    var errorMessage = Error.PackageOperationFailed("remove", cleanPackageId, removeRequest.Error?.message ?? "Unknown error");
                    _ = UnityMcpPluginEditor.NotifyToolRequestCompleted(new RequestToolCompletedData
                    {
                        RequestId = requestId,
                        Result = ResponseCallTool.Error(errorMessage).SetRequestID(requestId)
                    });
                    return;
                }

                // Schedule notification to be sent after domain reload completes
                PackageUtils.SchedulePostDomainReloadNotification(
                    requestId,
                    cleanPackageId,
                    "remove",
                    expectedResult: true
                );
            });

            return ResponseCallTool.Processing($"Removing package '{cleanPackageId}'. Waiting for package resolution and potential domain reload...").SetRequestID(requestId);
        }
    }
}