From 602366a54425acdb87971af1ed736749230101b2 Mon Sep 17 00:00:00 2001
From: David Wu <133224895+David-Wu1119@users.noreply.github.com>
Date: Mon, 7 Sep 2026 22:18:05 -0700
Subject: [PATCH] Type the private_ble_device coordinator with a HassKey
 (#181099)

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>
---
 .../components/private_ble_device/const.py         | 11 +++++++++++
 .../components/private_ble_device/coordinator.py   | 14 +++++---------
 2 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/homeassistant/components/private_ble_device/const.py b/homeassistant/components/private_ble_device/const.py
index 4eaf862ca755..01400294243e 100644
--- a/homeassistant/components/private_ble_device/const.py
+++ b/homeassistant/components/private_ble_device/const.py
@@ -1,3 +1,14 @@
 """Constants for Private BLE Device."""
 
+from typing import TYPE_CHECKING
+
+from homeassistant.util.hass_dict import HassKey
+
+if TYPE_CHECKING:
+    from .coordinator import PrivateDevicesCoordinator
+
 DOMAIN = "private_ble_device"
+
+# Resolving a private MAC against an IRK costs AES work, so a single
+# coordinator does it once for every config entry rather than per entry.
+PRIVATE_BLE_DEVICE_DATA: HassKey[PrivateDevicesCoordinator] = HassKey(DOMAIN)
diff --git a/homeassistant/components/private_ble_device/coordinator.py b/homeassistant/components/private_ble_device/coordinator.py
index 6dd60337e26d..7ba64cfa609e 100644
--- a/homeassistant/components/private_ble_device/coordinator.py
+++ b/homeassistant/components/private_ble_device/coordinator.py
@@ -1,9 +1,7 @@
 """Central manager for tracking devices with random but resolvable MAC addresses."""
-# pylint: disable=home-assistant-use-runtime-data  # Uses legacy hass.data[DOMAIN] pattern
 
 from collections.abc import Callable
 import logging
-from typing import cast
 
 from bluetooth_data_tools import get_cipher_for_irk, resolve_private_address
 from cryptography.hazmat.primitives.ciphers import Cipher
@@ -12,7 +10,7 @@ from homeassistant.components import bluetooth
 from homeassistant.components.bluetooth.match import BluetoothCallbackMatcher
 from homeassistant.core import HomeAssistant
 
-from .const import DOMAIN
+from .const import PRIVATE_BLE_DEVICE_DATA
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -241,11 +239,9 @@ def async_get_coordinator(hass: HomeAssistant) -> PrivateDevicesCoordinator:
     mac addresses with an IRK involves AES operations. We don't want to
     duplicate that work.
     """
-    if existing := hass.data.get(DOMAIN):
-        return cast(PrivateDevicesCoordinator, existing)
+    if (existing := hass.data.get(PRIVATE_BLE_DEVICE_DATA)) is not None:
+        return existing
 
-    # Uses legacy hass.data[DOMAIN] pattern
-    # pylint: disable-next=home-assistant-use-runtime-data
-    pdm = hass.data[DOMAIN] = PrivateDevicesCoordinator(hass)
+    coordinator = hass.data[PRIVATE_BLE_DEVICE_DATA] = PrivateDevicesCoordinator(hass)
 
-    return pdm
+    return coordinator