# This file is auto-generated by the CDP protocol generator.
# Do not edit this file manually as your changes will be overwritten.
# Generated from Chrome DevTools Protocol specifications.

"""CDP LayerTree Domain Event Registration"""

from typing import Callable, Optional

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from ..registry import EventRegistry
    from .events import LayerPaintedEvent, LayerTreeDidChangeEvent


class LayerTreeRegistration:
    """Event registration interface for LayerTree domain."""

    def __init__(self, registry: "EventRegistry"):
        self._registry = registry
        self._domain = "LayerTree"

    def layerPainted(
        self,
        callback: Callable[["LayerPaintedEvent", Optional[str]], None],
    ) -> None:
        """
        Register a callback for layerPainted events.

        Args:
            callback: Function to call when event occurs.
                     Receives (event_data, session_id) as parameters.
        """
        self._registry.register("LayerTree.layerPainted", callback)

    def layerTreeDidChange(
        self,
        callback: Callable[["LayerTreeDidChangeEvent", Optional[str]], None],
    ) -> None:
        """
        Register a callback for layerTreeDidChange events.

        Args:
            callback: Function to call when event occurs.
                     Receives (event_data, session_id) as parameters.
        """
        self._registry.register("LayerTree.layerTreeDidChange", callback)