feat: port local scanner support from HA (#9)

Author: bdraco

build_ext.py

@@ -30,7 +30,7 @@ def build(setup_kwargs: Any) -> None:
         setup_kwargs.update(
             {
                 "ext_modules": cythonize(
-                    ["src/habluetooth/base_scanner.py"],
+                    ["src/habluetooth/base_scanner.py", "src/habluetooth/scanner.py"],
                     compiler_directives={"language_level": "3"},  # Python 3
                 ),
                 "cmdclass": {"build_ext": BuildExt},

poetry.lock

@@ -33,6 +33,7 @@ bleak-retry-connector = ">=3.3.0"
 bluetooth-data-tools = ">=1.16.0"
 home-assistant-bluetooth = ">=1.10.4"
 bluetooth-adapters = ">=0.16.1"
+bluetooth-auto-recovery = ">=1.2.3"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.0"

pyproject.toml

@@ -8,8 +8,12 @@
     SCANNER_WATCHDOG_TIMEOUT,
 )
 from .models import HaBluetoothConnector
+from .scanner import BluetoothScanningMode, HaScanner, ScannerStartError
 
 __all__ = [
+    "BluetoothScanningMode",
+    "ScannerStartError",
+    "HaScanner",
     "BaseHaScanner",
     "BaseHaRemoteScanner",
     "HaBluetoothConnector",

src/habluetooth/__init__.py

@@ -15,6 +15,7 @@
 from home_assistant_bluetooth import BluetoothServiceInfoBleak
 
 from .const import (
+    CALLBACK_TYPE,
     CONNECTABLE_FALLBACK_MAXIMUM_STALE_ADVERTISEMENT_SECONDS,
     SCANNER_WATCHDOG_INTERVAL,
     SCANNER_WATCHDOG_TIMEOUT,
@@ -24,7 +25,6 @@
 SCANNER_WATCHDOG_INTERVAL_SECONDS: Final = SCANNER_WATCHDOG_INTERVAL.total_seconds()
 MONOTONIC_TIME: Final = monotonic_time_coarse
 _LOGGER = logging.getLogger(__name__)
-CALLBACK_TYPE = Callable[[], None]
 
 
 _float = float

src/habluetooth/base_scanner.py

@@ -2,7 +2,13 @@
 from __future__ import annotations
 
 from datetime import timedelta
-from typing import Final
+from typing import Callable, Final
+
+CALLBACK_TYPE = Callable[[], None]
+
+SOURCE_LOCAL: Final = "local"
+
+START_TIMEOUT = 15
 
 # The maximum time between advertisements for a device to be considered
 # stale when the advertisement tracker cannot determine the interval.

src/habluetooth/const.py

@@ -3,6 +3,7 @@
 
 from collections.abc import Callable
 from dataclasses import dataclass
+from enum import Enum
 from typing import Final
 
 from bleak import BaseBleakClient
@@ -18,3 +19,10 @@ class HaBluetoothConnector:
     client: type[BaseBleakClient]
     source: str
     can_connect: Callable[[], bool]
+
+
+class BluetoothScanningMode(Enum):
+    """The mode of scanning for bluetooth devices."""
+
+    PASSIVE = "passive"
+    ACTIVE = "active"

src/habluetooth/models.py

@@ -0,0 +1,24 @@
+from .base_scanner cimport BaseHaScanner
+
+
+cdef object NO_RSSI_VALUE
+cdef object BluetoothServiceInfoBleak
+cdef object AdvertisementData
+cdef object BLEDevice
+cdef bint TYPE_CHECKING
+
+
+cdef class HaScanner(BaseHaScanner):
+
+    cdef public object mac_address
+    cdef public object mode
+    cdef public object _start_stop_lock
+    cdef public object _new_info_callback
+    cdef public object _background_tasks
+    cdef public object scanner
+
+    cpdef void _async_detection_callback(
+        self,
+        object device,
+        object advertisement_data
+    )