"""Spook - Not your homie."""
from __future__ import annotations

from typing import TYPE_CHECKING

import voluptuous as vol

from homeassistant.components.input_number import DOMAIN, InputNumber

from ....services import AbstractSpookEntityComponentService, ReplaceExistingService

if TYPE_CHECKING:
    from homeassistant.core import ServiceCall


class SpookService(AbstractSpookEntityComponentService, ReplaceExistingService):
    """Input number entity service, increase value by a single step.

    It override the built-in increment service to allow for a custom amount.
    """

    domain = DOMAIN
    service = "increment"
    schema = {vol.Optional("amount"): vol.Coerce(float)}

    async def async_handle_service(
        self,
        entity: InputNumber,
        call: ServiceCall,
    ) -> None:
        """Handle the service call."""
        # pylint: disable=protected-access
        if (
            amount := call.data.get("amount", entity._step)  # noqa: SLF001
        ) % entity._step != 0:  # noqa: SLF001
            msg = (
                f"Amount {amount} not valid for {entity.entity_id}, "
                f"it needs to be a multiple of {entity._step}",  # noqa: SLF001
            )
            raise ValueError(msg)

        await entity.async_set_value(
            min(
                entity._current_value + amount,  # noqa: SLF001
                entity._maximum,  # noqa: SLF001
            ),
        )
