using ModelContextProtocol;
using ModelContextProtocol.Server;
using System.ComponentModel;
using System.Globalization;
using System.Text.Json;

namespace AspNetCoreMcpServer.Tools;

[McpServerToolType]
public sealed class IotDeviceTools
{
// 模拟设备状态存储
    private static readonly Dictionary<string, DeviceState> _deviceStates = new();
    
    public IotDeviceTools()
    {
        // 初始化一些默认设备
        InitializeDefaultDevices();
    }
    
    private void InitializeDefaultDevices()
    {
        _deviceStates["living_room_light"] = new DeviceState 
        { 
            Id = "living_room_light",
            Name = "客厅灯",
            Type = "light",
            IsOn = false,
            Brightness = 100,
            Temperature = 4000
        };
        
        _deviceStates["bedroom_curtain"] = new DeviceState 
        { 
            Id = "bedroom_curtain",
            Name = "卧室窗帘",
            Type = "curtain",
            IsOn = false,
            Position = 0
        };
        
        _deviceStates["living_room_ac"] = new DeviceState 
        { 
            Id = "living_room_ac",
            Name = "客厅空调",
            Type = "air_conditioner",
            IsOn = false,
            Temperature = 26,
            Mode = "cool",
            FanSpeed = "medium"
        };
    }

    // ==================== 灯光控制 ====================
    
    [McpServerTool, Description("控制灯光的开关状态")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "light")]
    public async Task<string> ControlLight(
        [Description("设备ID，例如: living_room_light, bedroom_light")] string deviceId,
        [Description("开关状态: on 或 off")] string action)
    {
        await Task.Delay(100); // 模拟网络延迟
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "light")
        {
            return $"错误: {deviceId} 不是灯光设备";
        }

        bool turnOn = action.ToLower() == "on";
        device.IsOn = turnOn;
        
        return $"""
                设备控制成功
                设备: {device.Name} ({deviceId})
                状态: {(turnOn ? "已开启" : "已关闭")}
                亮度: {device.Brightness}%
                色温: {device.Temperature}K
                """;
    }

    [McpServerTool, Description("调节灯光亮度")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "light")]
    public async Task<string> SetLightBrightness(
        [Description("设备ID")] string deviceId,
        [Description("亮度值 (0-100)")] int brightness)
    {
        await Task.Delay(100);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "light")
        {
            return $"错误: {deviceId} 不是灯光设备";
        }

        if (brightness < 0 || brightness > 100)
        {
            return "错误: 亮度值必须在 0-100 之间";
        }

        device.Brightness = brightness;
        if (brightness > 0 && !device.IsOn)
        {
            device.IsOn = true;
        }
        
        return $"""
                亮度调节成功
                设备: {device.Name}
                当前亮度: {brightness}%
                状态: {(device.IsOn ? "开启" : "关闭")}
                """;
    }

    [McpServerTool, Description("调节灯光色温")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "light")]
    public async Task<string> SetLightTemperature(
        [Description("设备ID")] string deviceId,
        [Description("色温值 (2700-6500K)")] int temperature)
    {
        await Task.Delay(100);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "light")
        {
            return $"错误: {deviceId} 不是灯光设备";
        }

        if (temperature < 2700 || temperature > 6500)
        {
            return "错误: 色温值必须在 2700-6500K 之间";
        }

        device.Temperature = temperature;
        
        string tempDesc = temperature < 3500 ? "暖光" : temperature < 5000 ? "自然光" : "冷光";
        
        return $"""
                色温调节成功
                设备: {device.Name}
                当前色温: {temperature}K ({tempDesc})
                亮度: {device.Brightness}%
                """;
    }

    // ==================== 窗帘控制 ====================
    
    [McpServerTool, Description("控制窗帘的开关")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "curtain")]
    public async Task<string> ControlCurtain(
        [Description("设备ID，例如: bedroom_curtain")] string deviceId,
        [Description("操作: open(打开), close(关闭), stop(停止)")] string action)
    {
        await Task.Delay(200); // 模拟窗帘运行时间
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "curtain")
        {
            return $"错误: {deviceId} 不是窗帘设备";
        }

        string result = action.ToLower() switch
        {
            "open" => HandleCurtainOpen(device),
            "close" => HandleCurtainClose(device),
            "stop" => HandleCurtainStop(device),
            _ => "错误: 无效的操作，请使用 open, close 或 stop"
        };
        
        return result;
    }

    [McpServerTool, Description("设置窗帘打开的位置")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "curtain")]
    public async Task<string> SetCurtainPosition(
        [Description("设备ID")] string deviceId,
        [Description("位置百分比 (0-100)，0表示完全关闭，100表示完全打开")] int position)
    {
        await Task.Delay(150);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "curtain")
        {
            return $"错误: {deviceId} 不是窗帘设备";
        }

        if (position < 0 || position > 100)
        {
            return "错误: 位置值必须在 0-100 之间";
        }

        device.Position = position;
        device.IsOn = position > 0;
        
        return $"""
                窗帘位置调节成功
                设备: {device.Name}
                当前位置: {position}%
                状态: {(position == 0 ? "完全关闭" : position == 100 ? "完全打开" : "部分打开")}
                """;
    }

    // ==================== 空调控制 ====================
    
    [McpServerTool, Description("控制空调的开关")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "air_conditioner")]
    public async Task<string> ControlAirConditioner(
        [Description("设备ID，例如: living_room_ac")] string deviceId,
        [Description("开关状态: on 或 off")] string action)
    {
        await Task.Delay(100);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "air_conditioner")
        {
            return $"错误: {deviceId} 不是空调设备";
        }

        bool turnOn = action.ToLower() == "on";
        device.IsOn = turnOn;
        
        return $"""
                空调控制成功
                设备: {device.Name}
                状态: {(turnOn ? "已开启" : "已关闭")}
                {(turnOn ? $"温度: {device.Temperature}°C\n模式: {GetModeDescription(device.Mode)}\n风速: {GetFanSpeedDescription(device.FanSpeed)}" : "")}
                """;
    }

    [McpServerTool, Description("设置空调温度")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "air_conditioner")]
    public async Task<string> SetAirConditionerTemperature(
        [Description("设备ID")] string deviceId,
        [Description("目标温度 (16-30°C)")] int temperature)
    {
        await Task.Delay(100);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "air_conditioner")
        {
            return $"错误: {deviceId} 不是空调设备";
        }

        if (temperature < 16 || temperature > 30)
        {
            return "错误: 温度必须在 16-30°C 之间";
        }

        device.Temperature = temperature;
        if (!device.IsOn)
        {
            device.IsOn = true;
        }
        
        return $"""
                空调温度设置成功
                设备: {device.Name}
                目标温度: {temperature}°C
                当前模式: {GetModeDescription(device.Mode)}
                风速: {GetFanSpeedDescription(device.FanSpeed)}
                """;
    }

    [McpServerTool, Description("设置空调模式")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "air_conditioner")]
    public async Task<string> SetAirConditionerMode(
        [Description("设备ID")] string deviceId,
        [Description("模式: cool(制冷), heat(制热), fan(送风), dry(除湿), auto(自动)")] string mode)
    {
        await Task.Delay(100);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "air_conditioner")
        {
            return $"错误: {deviceId} 不是空调设备";
        }

        var validModes = new[] { "cool", "heat", "fan", "dry", "auto" };
        if (!validModes.Contains(mode.ToLower()))
        {
            return $"错误: 无效的模式。有效模式: {string.Join(", ", validModes)}";
        }

        device.Mode = mode.ToLower();
        
        return $"""
                空调模式设置成功
                设备: {device.Name}
                当前模式: {GetModeDescription(device.Mode)}
                温度: {device.Temperature}°C
                风速: {GetFanSpeedDescription(device.FanSpeed)}
                """;
    }

    [McpServerTool, Description("设置空调风速")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "air_conditioner")]
    public async Task<string> SetAirConditionerFanSpeed(
        [Description("设备ID")] string deviceId,
        [Description("风速: low(低), medium(中), high(高), auto(自动)")] string fanSpeed)
    {
        await Task.Delay(100);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}";
        }

        var device = _deviceStates[deviceId];
        if (device.Type != "air_conditioner")
        {
            return $"错误: {deviceId} 不是空调设备";
        }

        var validSpeeds = new[] { "low", "medium", "high", "auto" };
        if (!validSpeeds.Contains(fanSpeed.ToLower()))
        {
            return $"错误: 无效的风速。有效风速: {string.Join(", ", validSpeeds)}";
        }

        device.FanSpeed = fanSpeed.ToLower();
        
        return $"""
                空调风速设置成功
                设备: {device.Name}
                当前风速: {GetFanSpeedDescription(device.FanSpeed)}
                模式: {GetModeDescription(device.Mode)}
                温度: {device.Temperature}°C
                """;
    }

    // ==================== 通用查询 ====================
    
    [McpServerTool, Description("获取所有IoT设备的状态")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "all")]
    public async Task<string> GetAllDevicesStatus()
    {
        await Task.Delay(50);
        
        var statusList = _deviceStates.Values.Select(device => 
        {
            string status = device.Type switch
            {
                "light" => $"开关: {(device.IsOn ? "开" : "关")}, 亮度: {device.Brightness}%, 色温: {device.Temperature}K",
                "curtain" => $"位置: {device.Position}%, 状态: {(device.Position == 0 ? "关闭" : device.Position == 100 ? "打开" : "部分打开")}",
                "air_conditioner" => $"开关: {(device.IsOn ? "开" : "关")}, 温度: {device.Temperature}°C, 模式: {GetModeDescription(device.Mode)}, 风速: {GetFanSpeedDescription(device.FanSpeed)}",
                _ => "未知设备类型"
            };
            
            return $"""
                    [{device.Type.ToUpper()}] {device.Name} ({device.Id})
                    {status}
                    """;
        });
        
        return $"当前设备总数: {_deviceStates.Count}\n\n" + string.Join("\n---\n", statusList);
    }

    [McpServerTool, Description("获取指定设备的详细状态")]
    [McpMeta("category", "iot")]
    [McpMeta("device", "all")]
    public async Task<string> GetDeviceStatus(
        [Description("设备ID")] string deviceId)
    {
        await Task.Delay(50);
        
        if (!_deviceStates.ContainsKey(deviceId))
        {
            return $"错误: 找不到设备 {deviceId}\n\n可用设备: {string.Join(", ", _deviceStates.Keys)}";
        }

        var device = _deviceStates[deviceId];
        
        string details = device.Type switch
        {
            "light" => $"""
                        开关状态: {(device.IsOn ? "开启" : "关闭")}
                        亮度: {device.Brightness}%
                        色温: {device.Temperature}K ({(device.Temperature < 3500 ? "暖光" : device.Temperature < 5000 ? "自然光" : "冷光")})
                        """,
            "curtain" => $"""
                         位置: {device.Position}%
                         状态: {(device.Position == 0 ? "完全关闭" : device.Position == 100 ? "完全打开" : "部分打开")}
                         """,
            "air_conditioner" => $"""
                                  开关状态: {(device.IsOn ? "开启" : "关闭")}
                                  温度设置: {device.Temperature}°C
                                  运行模式: {GetModeDescription(device.Mode)}
                                  风速: {GetFanSpeedDescription(device.FanSpeed)}
                                  """,
            _ => "未知设备类型"
        };
        
        return $"""
                设备信息
                名称: {device.Name}
                ID: {device.Id}
                类型: {device.Type}
                
                {details}
                """;
    }

    // ==================== 辅助方法 ====================
    
    private string HandleCurtainOpen(DeviceState device)
    {
        device.Position = 100;
        device.IsOn = true;
        return $"""
                窗帘正在打开...
                设备: {device.Name}
                目标位置: 100% (完全打开)
                预计时间: 5秒
                """;
    }

    private string HandleCurtainClose(DeviceState device)
    {
        device.Position = 0;
        device.IsOn = false;
        return $"""
                窗帘正在关闭...
                设备: {device.Name}
                目标位置: 0% (完全关闭)
                预计时间: 5秒
                """;
    }

    private string HandleCurtainStop(DeviceState device)
    {
        return $"""
                窗帘已停止
                设备: {device.Name}
                当前位置: {device.Position}%
                """;
    }

    private static string GetModeDescription(string? mode) => mode switch
    {
        "cool" => "制冷",
        "heat" => "制热",
        "fan" => "送风",
        "dry" => "除湿",
        "auto" => "自动",
        _ => "未知"
    };

    private static string GetFanSpeedDescription(string? speed) => speed switch
    {
        "low" => "低速",
        "medium" => "中速",
        "high" => "高速",
        "auto" => "自动",
        _ => "未知"
    };
}

// 设备状态数据模型
internal class DeviceState
{
    public string Id { get; set; } = "";
    public string Name { get; set; } = "";
    public string Type { get; set; } = "";
    public bool IsOn { get; set; }
    
    // 灯光属性
    public int Brightness { get; set; }
    public int Temperature { get; set; }
    
    // 窗帘属性
    public int Position { get; set; }
    
    // 空调属性
    public string? Mode { get; set; }
    public string? FanSpeed { get; set; }
}