first commit!

aln/AirPods/Pro2.py

@@ -0,0 +1,14 @@
+from ..Capabilites import Capabilites
+class Pro2:
+    def __init__(self):
+        self.name = 'AirPods Pro 2'
+        self.capabilites = {
+            Capabilites.NOISE_CANCELLATION: [
+                Capabilites.NoiseCancellation.Off,
+                Capabilites.NoiseCancellation.On,
+                Capabilites.NoiseCancellation.Transparency,
+                Capabilites.NoiseCancellation.Adaptive,
+            ],
+            Capabilites.CONVERSATION_AWARENESS: True,
+            Capabilites.CUSTOMIZABLE_ADAPTIVE_TRANSPARENCY: True
+        }

aln/AirPods/__init__.py

@@ -0,0 +1,3 @@
+from . import Pro2
+
+Pro2 = Pro2.Pro2

aln/Capabilites/__init__.py

@@ -0,0 +1,17 @@
+class NoiseCancellation:
+    Off = b"\x01"
+    On = b"\x02"
+    Transparency = b"\x03"
+    Adaptive = b"\x04"
+
+class ConversationAwareness:
+    Off = b"\x02"
+    On = b"\x01"
+
+class Capabilites:
+    NOISE_CANCELLATION = b"\x0d"
+    CONVERSATION_AWARENESS = b"\x28"
+    CUSTOMIZABLE_ADAPTIVE_TRANSPARENCY = b"\x01\x02"
+    EAR_DETECTION = b"\x06"
+    NoiseCancellation = NoiseCancellation
+    ConversationAwareness = ConversationAwareness

aln/Notifications/Battery.py

@@ -0,0 +1,64 @@
+class BatteryComponent:
+    LEFT = 4
+    RIGHT = 2
+    CASE = 8
+    pass
+class BatteryStatus:
+    CHARGING = 1
+    NOT_CHARGING = 2
+    DISCONNECTED = 4
+    pass
+
+class Battery:
+
+    def get_name(self, cls, value):
+        for key, val in cls.__dict__.items():
+            if val == value:
+                return key
+        return None
+    
+    def get_component(self):
+        return self.get_name(BatteryComponent, self.component)
+    
+    def get_level(self):
+        return self.level
+    
+    def get_status(self):
+        return self.get_name(BatteryStatus, self.status)
+    
+    def __init__(self, component: int, level: int, status: int):
+        self.component = component
+        self.level = level
+        self.status = status
+        pass
+    
+class BatteryNotification:
+    def __init__(self):
+        self.first = Battery(BatteryComponent.LEFT, 0, BatteryStatus.DISCONNECTED)
+        self.second = Battery(BatteryComponent.RIGHT, 0, BatteryStatus.DISCONNECTED)
+        self.case = Battery(BatteryComponent.CASE, 0, BatteryStatus.DISCONNECTED)
+        pass
+    
+    def isBatteryData(self, data):
+        if len(data) != 22:
+            return False
+        if data[0] == 0x04 and data[1] == 0x00 and data[2] == 0x04 and data[3] == 0x00 and data[4] == 0x04 and data[5] == 0x00:
+            return True
+        else:
+            return False
+    def setBattery(self, data):
+        self.count = data[6]
+        self.first = Battery(data[7], data[9], data[10])
+        self.second = Battery(data[12], data[14], data[15])
+        self.case = Battery(data[17], data[19], data[20])
+        pass
+    
+    def getBattery(self):
+        if self.first.component == BatteryComponent.LEFT:
+            self.left = self.first
+            self.right = self.second
+        else:
+            self.left = self.second
+            self.right = self.first
+        self.case = self.case
+        return [self.left, self.right, self.case]

aln/Notifications/EarDetection.py

@@ -0,0 +1,27 @@
+from ..Capabilites import Capabilites
+from ..enums import enums
+import logging
+from typing import Literal
+
+class EarDetectionNotification:
+    NOTIFICATION_BIT = Capabilites.EAR_DETECTION
+    NOTIFICATION_PREFIX = enums.SEND_PREFIX + NOTIFICATION_BIT
+    IN_EAR = 0x00
+    OUT_OF_EAR = 0x01
+    def __init__(self):
+        pass
+    
+    def isEarDetectionData(self, data: bytes):
+        if len(data) != 8:
+            return False
+        if data.hex().startswith(self.NOTIFICATION_PREFIX.hex()):
+            return True
+    
+    def setEarDetection(self, data: bytes):
+        self.first = data[6]
+        self.second = data[7]
+
+    def getEarDetection(self):
+        return [self.first, self.second]
+        pass
+        

aln/Notifications/Listener.py

@@ -0,0 +1,43 @@
+from bluetooth import BluetoothSocket
+import threading
+from .Battery import BatteryNotification
+from .EarDetection import EarDetectionNotification
+import logging
+
+logging = logging.getLogger(__name__)
+
+class NotificationListener:
+    BATTERY_UPDATED = 0x01
+    ANC_UPDATED = 0x02
+    EAR_DETECTION_UPDATED = 0x03
+
+    def __init__(self, socket: BluetoothSocket, callback: callable):
+        self.socket = socket
+        self.BatteryNotification = BatteryNotification()
+        self.EarDetectionNotification = EarDetectionNotification()
+        self.callback = callback
+        pass
+    
+    def __start(self):
+        while True:
+            data = self.socket.recv(1024)
+            if len(data) == 0:
+                break
+            if self.BatteryNotification.isBatteryData(data):
+                self.BatteryNotification.setBattery(data)
+                self.callback(self.BATTERY_UPDATED)
+                pass
+            if self.EarDetectionNotification.isEarDetectionData(data):
+                self.EarDetectionNotification.setEarDetection(data)
+                self.callback(self.EAR_DETECTION_UPDATED)
+            pass
+        pass
+    
+    def start(self):
+        threading.Thread(target=self.__start).start()
+        pass
+    
+    def stop(self):
+        self.socket.close()
+        pass
+        

aln/Notifications/__init__.py

@@ -0,0 +1,36 @@
+from .Listener import NotificationListener
+from ..enums import enums
+import bluetooth
+import logging
+
+logging = logging.getLogger(__name__)
+
+enums = enums()
+
+class Notifications:
+    BATTERY_UPDATED = NotificationListener.BATTERY_UPDATED
+    ANC_UPDATED = NotificationListener.ANC_UPDATED
+    EAR_DETECTION_UPDATED = NotificationListener.EAR_DETECTION_UPDATED
+    
+    def __init__(self, socket: bluetooth.BluetoothSocket, callback: callable):
+        self.socket = socket
+        self.notificationListener = NotificationListener(self.socket, callback)
+        self.BatteryNotification = self.notificationListener.BatteryNotification
+        self.EarDetectionNotification = self.notificationListener.EarDetectionNotification
+        pass
+
+    def initialize(self):
+        try:
+            self.socket.send(enums.REQUEST_NOTIFICATIONS)
+            self.notificationListener.start()
+            
+        except bluetooth.btcommon.BluetoothError as e:
+            logging.error(f'Failed to send data to {self.mac_address}: {e}')
+            return False
+        return True
+
+    def __del__(self):
+        self.notificationListener.stop()
+        self.socket.close()
+        pass
+    pass

aln/__init__.py

@@ -0,0 +1,56 @@
+from .Notifications import Notifications
+from .Capabilites import Capabilites
+from .enums import enums
+
+import bluetooth
+import logging
+
+logging = logging.getLogger("Connection Handler")
+
+class Connection:
+    def __init__(self, mac_address: str):
+        self.mac_address = mac_address
+        self.socket = bluetooth.BluetoothSocket(bluetooth.L2CAP)
+    def connect(self):
+        try:
+            self.socket.connect((self.mac_address, 0x1001))
+        except bluetooth.btcommon.BluetoothError as e:
+            logging.error(f'Failed to connect to {self.mac_address}: {e}')
+            return False
+        return True
+    
+    def initialize_notifications(self):
+        self.notifications = Notifications(self.socket, self.notification_callback)
+        self.notificationListener = self.notifications.notificationListener
+        self.BatteryNotification = self.notifications.BatteryNotification
+        self.notifications.initialize()
+
+    def send(self, data: bytes):
+        try:
+            logging.debug(f'Sending data to {self.mac_address}: {data.hex()}')
+            self.socket.send(data)
+            logging.debug(f'Sent data to {self.mac_address}')
+        except bluetooth.btcommon.BluetoothError as e:
+            logging.error(f'Failed to send data to {self.mac_address}: {e}')
+            return False
+        return True
+    
+    def notification_callback(self, notification_type: int):
+        import logging
+        logging = logging.getLogger("Notification Callback")
+        if notification_type == Notifications.BATTERY_UPDATED:
+            for i in self.notificationListener.BatteryNotification.getBattery():
+                logging.debug(f'{i.get_component()} - {i.get_status()}: {i.get_level()}')
+            pass
+        elif notification_type == Notifications.EAR_DETECTION_UPDATED:
+            logging.debug(f'{self.notificationListener.EarDetectionNotification.getEarDetection()}')
+            pass
+        pass
+
+    def disconnect(self):
+        self.socket.close()
+        pass
+
+    def __del__(self):
+        self.socket.close()
+        pass

aln/enums.py

@@ -0,0 +1,19 @@
+from .Capabilites import Capabilites
+
+class enums:
+    NOISE_CANCELLATION = Capabilites.NOISE_CANCELLATION
+    CONVERSATION_AWARENESS = Capabilites.CONVERSATION_AWARENESS
+    CUSTOMIZABLE_ADAPTIVE_TRANSPARENCY = Capabilites.CUSTOMIZABLE_ADAPTIVE_TRANSPARENCY
+    SEND_PREFIX = b'\x04\x00\x04\x00'
+    SETTINGS = b"\x09\x00"
+    SETTINGS_SEND_SUFFIX = b'\x00\x00\x00'
+    NOTIFICATION_FILTER = b'\x0f'
+    REQUEST_NOTIFICATIONS = SEND_PREFIX + NOTIFICATION_FILTER + b"\x00\xff\xff\xff\xff"
+    HANDSHAKE = b"\x00\x00\x04\x00\x01\x00\x02\x00\x00\x00\x00\x00\x00\x00\x00\x00"
+
+    SET_NOISE_CANCELLATION_OFF = SEND_PREFIX + SETTINGS + NOISE_CANCELLATION + Capabilites.NoiseCancellation.Off + SETTINGS_SEND_SUFFIX
+    SET_NOISE_CANCELLATION_ON = SEND_PREFIX + SETTINGS + NOISE_CANCELLATION + Capabilites.NoiseCancellation.On + SETTINGS_SEND_SUFFIX
+    SET_NOISE_CANCELLATION_TRANSPARENCY = SEND_PREFIX + SETTINGS + NOISE_CANCELLATION + Capabilites.NoiseCancellation.Transparency + SETTINGS_SEND_SUFFIX
+    SET_NOISE_CANCELLATION_ADAPTIVE = SEND_PREFIX + SETTINGS + NOISE_CANCELLATION + Capabilites.NoiseCancellation.Adaptive + SETTINGS_SEND_SUFFIX
+    SET_CONVERSATION_AWARENESS_OFF = SEND_PREFIX + SETTINGS + CONVERSATION_AWARENESS + Capabilites.ConversationAwareness.Off + SETTINGS_SEND_SUFFIX
+    SET_CONVERSATION_AWARENESS_ON = SEND_PREFIX + SETTINGS + CONVERSATION_AWARENESS + Capabilites.ConversationAwareness.On + SETTINGS_SEND_SUFFIX