"""
Entity Rename E2E Tests

Tests for the ha_set_entity tool which changes entity_ids via
the config/entity_registry/update WebSocket API.

Key test scenarios:
- Rename helper entity successfully
- Validate domain preservation (cannot change domain)
- Validate entity_id format
- Handle non-existent entities
- Update name and icon along with rename
- Voice assistant exposure migration on rename
- Rename entity and device together (convenience wrapper)
"""

import logging

import pytest

from ...utilities.assertions import safe_call_tool
from ...utilities.wait_helpers import wait_for_entity_registration

logger = logging.getLogger(__name__)


@pytest.mark.registry
@pytest.mark.cleanup
class TestEntityRename:
    """Test entity renaming via ha_set_entity tool."""

    async def test_rename_helper_entity(self, mcp_client, cleanup_tracker):
        """
        Test: Create helper -> Rename entity_id -> Verify new entity works

        This is the primary use case for entity renaming.
        """
        original_name = "test_rename_original"
        new_name = "test_rename_new"
        logger.info(f"Testing entity rename: {original_name} -> {new_name}")

        # 1. CREATE: Helper entity to rename
        create_data = await safe_call_tool(
            mcp_client,
            "ha_config_set_helper",
            {
                "helper_type": "input_boolean",
                "name": original_name,
                "icon": "mdi:toggle-switch",
            },
        )

        assert create_data.get("success"), f"Failed to create helper: {create_data}"

        original_entity_id = f"input_boolean.{original_name}"
        new_entity_id = f"input_boolean.{new_name}"
        cleanup_tracker.track("input_boolean", new_entity_id)
        logger.info(f"Created helper: {original_entity_id}")

        # 2. VERIFY: Original entity exists
        entity_ready = await wait_for_entity_registration(
            mcp_client, original_entity_id
        )
        assert entity_ready, (
            f"Original entity not found after waiting: {original_entity_id}"
        )
        logger.info(f"Verified original entity exists: {original_entity_id}")

        # 3. RENAME: Change entity_id
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "new_entity_id": new_entity_id,
            },
        )

        assert rename_data.get("success"), f"Failed to rename entity: {rename_data}"
        assert rename_data.get("old_entity_id") == original_entity_id
        assert rename_data.get("entity_id") == new_entity_id
        logger.info(f"Renamed entity: {original_entity_id} -> {new_entity_id}")

        # 4. VERIFY: New entity exists and works
        new_entity_ready = await wait_for_entity_registration(mcp_client, new_entity_id)
        assert new_entity_ready, (
            f"New entity not accessible after waiting: {new_entity_id}"
        )
        logger.info(f"Verified new entity exists: {new_entity_id}")

        # 5. VERIFY: Old entity_id no longer exists
        old_state_data = await safe_call_tool(
            mcp_client, "ha_get_state", {"entity_id": original_entity_id}
        )
        # Should fail or return empty/unavailable
        old_exists = (
            "data" in old_state_data
            and old_state_data["data"].get("state")
            and old_state_data["data"]["state"] != "unavailable"
        )
        assert not old_exists, f"Old entity should not exist: {old_state_data}"
        logger.info(f"Verified old entity no longer exists: {original_entity_id}")

        # 6. CLEANUP: Delete renamed entity
        delete_data = await safe_call_tool(
            mcp_client,
            "ha_remove_helpers_integrations",
            {
                "helper_type": "input_boolean",
                "target": new_name,
                "confirm": True,
            },
        )
        assert delete_data.get("success"), f"Failed to delete helper: {delete_data}"
        logger.info("Cleanup completed")

    async def test_rename_with_name_and_icon(self, mcp_client, cleanup_tracker):
        """
        Test: Rename entity while also updating friendly name and icon
        """
        original_name = "test_rename_full"
        new_name = "test_rename_full_new"
        logger.info("Testing rename with name and icon update")

        # 1. CREATE: Helper entity
        create_data = await safe_call_tool(
            mcp_client,
            "ha_config_set_helper",
            {
                "helper_type": "input_boolean",
                "name": original_name,
                "icon": "mdi:toggle-switch",
            },
        )

        assert create_data.get("success"), f"Failed to create helper: {create_data}"

        original_entity_id = f"input_boolean.{original_name}"
        new_entity_id = f"input_boolean.{new_name}"
        cleanup_tracker.track("input_boolean", new_entity_id)

        assert await wait_for_entity_registration(mcp_client, original_entity_id), (
            f"Entity not registered: {original_entity_id}"
        )
        # 2. RENAME: With name and icon updates
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "new_entity_id": new_entity_id,
                "name": "My Renamed Toggle",
                "icon": "mdi:lightbulb",
            },
        )

        assert rename_data.get("success"), f"Failed to rename entity: {rename_data}"
        logger.info("Renamed entity with name and icon update")

        # 3. VERIFY: New entity has updated attributes
        assert await wait_for_entity_registration(mcp_client, new_entity_id), (
            f"Entity not registered: {new_entity_id}"
        )
        state_data = await safe_call_tool(
            mcp_client, "ha_get_state", {"entity_id": new_entity_id}
        )
        assert "data" in state_data, f"Failed to get new entity state: {state_data}"

        # Note: The friendly_name might be set in registry, actual display may vary
        logger.info(f"New entity state: {state_data}")

        # 4. CLEANUP
        delete_data = await safe_call_tool(
            mcp_client,
            "ha_remove_helpers_integrations",
            {
                "helper_type": "input_boolean",
                "target": new_name,
                "confirm": True,
            },
        )
        assert delete_data.get("success"), f"Failed to delete helper: {delete_data}"
        logger.info("Cleanup completed")

    async def test_rename_domain_mismatch_rejected(self, mcp_client):
        """
        Test: Attempting to change domain should fail

        Entity renaming cannot change the domain (e.g., light -> switch).
        """
        logger.info("Testing domain mismatch rejection")

        # Attempt to rename with domain change
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": "input_boolean.some_entity",
                "new_entity_id": "input_number.some_entity",
            },
        )

        assert not rename_data.get("success"), "Domain change should be rejected"
        # Contract: error must be a dict with code and message keys
        error = rename_data.get("error")
        assert isinstance(error, dict), (
            f"error should be dict, got {type(error).__name__}: {error!r}"
        )
        assert error.get("code") == "VALIDATION_INVALID_PARAMETER", (
            f"Expected VALIDATION_INVALID_PARAMETER, got: {error}"
        )
        assert "domain" in error.get("message", "").lower(), (
            f"Error message should indicate domain mismatch, got: {error}"
        )
        logger.info("Domain mismatch correctly rejected")

    async def test_rename_invalid_format_rejected(self, mcp_client):
        """
        Test: Invalid entity_id formats should be rejected
        """
        logger.info("Testing invalid entity_id format rejection")

        # Test with invalid new_entity_id format
        invalid_formats = [
            "invalid_format",  # Missing domain
            "Domain.Upper",  # Uppercase not allowed
            "light.has spaces",  # Spaces not allowed
            "light.special!chars",  # Special chars not allowed
        ]

        for invalid_id in invalid_formats:
            rename_data = await safe_call_tool(
                mcp_client,
                "ha_set_entity",
                {
                    "entity_id": "input_boolean.test",
                    "new_entity_id": invalid_id,
                },
            )

            assert not rename_data.get("success"), (
                f"Invalid format should be rejected: {invalid_id}"
            )
            # Contract: error must be a dict with code and message keys
            error = rename_data.get("error")
            assert isinstance(error, dict), (
                f"error should be dict, got {type(error).__name__}: {error!r}"
            )
            assert error.get("code") == "VALIDATION_INVALID_PARAMETER", (
                f"Expected VALIDATION_INVALID_PARAMETER for {invalid_id}, got: {error}"
            )
            error_msg = error.get("message", "")
            assert invalid_id in error_msg, (
                f"Error message should identify rejected input '{invalid_id}', got: {error_msg}"
            )
            logger.info(f"Invalid format correctly rejected: {invalid_id}")

    async def test_rename_nonexistent_entity(self, mcp_client):
        """
        Test: Renaming non-existent entity should fail gracefully
        """
        logger.info("Testing non-existent entity rename")

        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": "input_boolean.definitely_does_not_exist_12345",
                "new_entity_id": "input_boolean.new_name_12345",
            },
        )

        assert not rename_data.get("success"), "Non-existent entity rename should fail"
        assert rename_data["error"]["code"] == "SERVICE_CALL_FAILED"
        logger.info(
            f"Non-existent entity correctly rejected: {rename_data.get('error')}"
        )


@pytest.mark.registry
async def test_rename_entity_basic(mcp_client, cleanup_tracker):
    """
    Quick test: Basic entity rename functionality

    Simple test that creates, renames, and cleans up a helper entity.
    """
    logger.info("Running basic entity rename test")

    # Create helper
    create_data = await safe_call_tool(
        mcp_client,
        "ha_config_set_helper",
        {
            "helper_type": "input_button",
            "name": "test_quick_rename",
            "icon": "mdi:button-pointer",
        },
    )
    assert create_data.get("success"), f"Failed to create helper: {create_data}"

    original_id = "input_button.test_quick_rename"
    new_id = "input_button.test_quick_renamed"
    cleanup_tracker.track("input_button", new_id)

    assert await wait_for_entity_registration(mcp_client, original_id), (
        f"Entity not registered: {original_id}"
    )
    # Rename
    rename_data = await safe_call_tool(
        mcp_client,
        "ha_set_entity",
        {
            "entity_id": original_id,
            "new_entity_id": new_id,
        },
    )
    assert rename_data.get("success"), f"Failed to rename: {rename_data}"

    assert await wait_for_entity_registration(mcp_client, new_id), (
        f"Entity not registered: {new_id}"
    )
    # Cleanup
    delete_data = await safe_call_tool(
        mcp_client,
        "ha_remove_helpers_integrations",
        {
            "helper_type": "input_button",
            "target": "test_quick_renamed",
            "confirm": True,
        },
    )
    assert delete_data.get("success"), f"Failed to cleanup: {delete_data}"

    logger.info("Basic entity rename test completed")


@pytest.mark.registry
@pytest.mark.cleanup
class TestEntityRenameVoiceExposure:
    """Test that HA Core preserves voice exposure automatically during rename."""

    async def test_rename_preserves_voice_exposure(self, mcp_client, cleanup_tracker):
        """
        Test: Rename entity and verify voice exposure is preserved by HA Core.

        HA Core stores voice exposure in RegistryEntry.options, which is
        preserved via attr.evolve during entity_id renames. No manual
        migration is needed.

        1. Create entity
        2. Expose it to conversation assistant
        3. Rename entity via ha_set_entity(new_entity_id=...)
        4. Verify exposure is preserved on new entity_id
        """
        original_name = "test_rename_expose"
        new_name = "test_rename_expose_new"
        logger.info("Testing rename preserves voice exposure")

        # 1. CREATE: Helper entity
        create_data = await safe_call_tool(
            mcp_client,
            "ha_config_set_helper",
            {
                "helper_type": "input_boolean",
                "name": original_name,
            },
        )
        assert create_data.get("success"), f"Failed to create helper: {create_data}"

        original_entity_id = f"input_boolean.{original_name}"
        new_entity_id = f"input_boolean.{new_name}"
        cleanup_tracker.track("input_boolean", new_entity_id)

        assert await wait_for_entity_registration(mcp_client, original_entity_id), (
            f"Entity not registered: {original_entity_id}"
        )
        # 2. EXPOSE: Entity to conversation assistant
        expose_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "expose_to": {"conversation": True},
            },
        )
        assert expose_data.get("success"), f"Failed to expose entity: {expose_data}"
        logger.info(f"Exposed {original_entity_id} to conversation")

        # 3. RENAME: Entity via ha_set_entity (no separate migration needed)
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "new_entity_id": new_entity_id,
            },
        )
        assert rename_data.get("success"), f"Failed to rename entity: {rename_data}"

        assert await wait_for_entity_registration(mcp_client, new_entity_id), (
            f"Entity not registered: {new_entity_id}"
        )
        # 4. VERIFY: New entity still has exposure settings (preserved by HA Core)
        check_data = await safe_call_tool(
            mcp_client,
            "ha_get_entity_exposure",
            {"entity_id": new_entity_id},
        )
        assert check_data.get("success"), f"Failed to check exposure: {check_data}"
        logger.info(f"New entity exposure: {check_data.get('exposed_to')}")

        # 5. CLEANUP
        delete_data = await safe_call_tool(
            mcp_client,
            "ha_remove_helpers_integrations",
            {"helper_type": "input_boolean", "target": new_name, "confirm": True},
        )
        assert delete_data.get("success"), f"Failed to cleanup: {delete_data}"
        logger.info("Cleanup completed")


@pytest.mark.registry
@pytest.mark.cleanup
class TestRenameEntityWithDevice:
    """Test ha_set_entity with new_device_name parameter (entity+device rename)."""

    async def test_rename_entity_and_device_helper(self, mcp_client, cleanup_tracker):
        """
        Test: Rename helper entity (no device) using convenience wrapper

        Helper entities don't have associated devices, so only entity rename should occur.
        """
        original_name = "test_combo_rename"
        new_name = "test_combo_rename_new"
        logger.info("Testing ha_set_entity with new_device_name with helper entity")

        # 1. CREATE: Helper entity
        create_data = await safe_call_tool(
            mcp_client,
            "ha_config_set_helper",
            {
                "helper_type": "input_boolean",
                "name": original_name,
            },
        )
        assert create_data.get("success"), f"Failed to create helper: {create_data}"

        original_entity_id = f"input_boolean.{original_name}"
        new_entity_id = f"input_boolean.{new_name}"
        cleanup_tracker.track("input_boolean", new_entity_id)

        assert await wait_for_entity_registration(mcp_client, original_entity_id), (
            f"Entity not registered: {original_entity_id}"
        )
        # 2. RENAME: Using convenience wrapper
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "new_entity_id": new_entity_id,
                "new_device_name": "Test Device Name",  # Won't apply - no device
            },
        )

        assert rename_data.get("success"), f"Failed to rename: {rename_data}"
        assert rename_data.get("old_entity_id") == original_entity_id
        assert rename_data.get("entity_id") == new_entity_id

        # Check device rename was skipped (no device for helper)
        device_result = rename_data.get("device_rename", {})
        warnings = device_result.get("warnings")
        assert isinstance(warnings, list) and warnings, (
            f"Device rename should have warnings for helper entity (no device): {device_result}"
        )
        assert any("no associated device" in w for w in warnings), (
            f"Expected no-associated-device warning content; got: {warnings!r}"
        )
        logger.info(f"Device rename result: {device_result}")

        assert await wait_for_entity_registration(mcp_client, new_entity_id), (
            f"Entity not registered: {new_entity_id}"
        )
        # 3. VERIFY: New entity exists
        state_data = await safe_call_tool(
            mcp_client, "ha_get_state", {"entity_id": new_entity_id}
        )
        assert "data" in state_data, f"New entity not found: {state_data}"

        # 4. CLEANUP
        delete_data = await safe_call_tool(
            mcp_client,
            "ha_remove_helpers_integrations",
            {"helper_type": "input_boolean", "target": new_name, "confirm": True},
        )
        assert delete_data.get("success"), f"Failed to cleanup: {delete_data}"
        logger.info("Cleanup completed")

    async def test_rename_entity_without_device_name(self, mcp_client, cleanup_tracker):
        """
        Test: Rename entity without providing new_device_name

        Should rename entity and skip device rename.
        """
        original_name = "test_entity_only_rename"
        new_name = "test_entity_only_rename_new"
        logger.info("Testing ha_set_entity with new_device_name without device name")

        # 1. CREATE: Helper entity
        create_data = await safe_call_tool(
            mcp_client,
            "ha_config_set_helper",
            {
                "helper_type": "input_boolean",
                "name": original_name,
            },
        )
        assert create_data.get("success"), f"Failed to create helper: {create_data}"

        original_entity_id = f"input_boolean.{original_name}"
        new_entity_id = f"input_boolean.{new_name}"
        cleanup_tracker.track("input_boolean", new_entity_id)

        assert await wait_for_entity_registration(mcp_client, original_entity_id), (
            f"Entity not registered: {original_entity_id}"
        )
        # 2. RENAME: Without new_device_name
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "new_entity_id": new_entity_id,
            },
        )

        assert rename_data.get("success"), f"Failed to rename: {rename_data}"

        logger.info(f"Rename result: {rename_data.get('message')}")

        # 3. CLEANUP
        delete_data = await safe_call_tool(
            mcp_client,
            "ha_remove_helpers_integrations",
            {"helper_type": "input_boolean", "target": new_name, "confirm": True},
        )
        assert delete_data.get("success"), f"Failed to cleanup: {delete_data}"

    async def test_rename_entity_and_device_with_friendly_name(
        self, mcp_client, cleanup_tracker
    ):
        """
        Test: Rename entity with new friendly name using convenience wrapper
        """
        original_name = "test_friendly_rename"
        new_name = "test_friendly_rename_new"
        logger.info("Testing ha_set_entity with new_device_name with friendly name")

        # 1. CREATE: Helper entity
        create_data = await safe_call_tool(
            mcp_client,
            "ha_config_set_helper",
            {
                "helper_type": "input_boolean",
                "name": original_name,
            },
        )
        assert create_data.get("success"), f"Failed to create helper: {create_data}"

        original_entity_id = f"input_boolean.{original_name}"
        new_entity_id = f"input_boolean.{new_name}"
        cleanup_tracker.track("input_boolean", new_entity_id)

        assert await wait_for_entity_registration(mcp_client, original_entity_id), (
            f"Entity not registered: {original_entity_id}"
        )
        # 2. RENAME: With new entity friendly name
        rename_data = await safe_call_tool(
            mcp_client,
            "ha_set_entity",
            {
                "entity_id": original_entity_id,
                "new_entity_id": new_entity_id,
                "name": "My Friendly Test Entity",
            },
        )

        assert rename_data.get("success"), f"Failed to rename: {rename_data}"

        logger.info(f"Rename result: {rename_data}")

        # 3. CLEANUP
        delete_data = await safe_call_tool(
            mcp_client,
            "ha_remove_helpers_integrations",
            {"helper_type": "input_boolean", "target": new_name, "confirm": True},
        )
        assert delete_data.get("success"), f"Failed to cleanup: {delete_data}"


@pytest.mark.registry
async def test_rename_entity_with_device_basic(mcp_client, cleanup_tracker):
    """
    Quick test: Basic ha_set_entity with new_device_name parameter
    """
    logger.info("Running basic entity and device rename test")

    # Create helper
    create_data = await safe_call_tool(
        mcp_client,
        "ha_config_set_helper",
        {
            "helper_type": "input_button",
            "name": "test_combo_quick",
        },
    )
    assert create_data.get("success"), f"Failed to create helper: {create_data}"

    original_id = "input_button.test_combo_quick"
    new_id = "input_button.test_combo_quick_new"
    cleanup_tracker.track("input_button", new_id)

    assert await wait_for_entity_registration(mcp_client, original_id), (
        f"Entity not registered: {original_id}"
    )
    # Rename using convenience wrapper
    rename_data = await safe_call_tool(
        mcp_client,
        "ha_set_entity",
        {
            "entity_id": original_id,
            "new_entity_id": new_id,
        },
    )
    assert rename_data.get("success"), f"Failed to rename: {rename_data}"

    assert await wait_for_entity_registration(mcp_client, new_id), (
        f"Entity not registered: {new_id}"
    )
    # Cleanup
    delete_data = await safe_call_tool(
        mcp_client,
        "ha_remove_helpers_integrations",
        {
            "helper_type": "input_button",
            "target": "test_combo_quick_new",
            "confirm": True,
        },
    )
    assert delete_data.get("success"), f"Failed to cleanup: {delete_data}"

    logger.info("Basic entity and device rename test completed")
