move everything to linux folder

linux/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
+        }

linux/aln/AirPods/__init__.py

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

linux/aln/Capabilites/__init__.py

@@ -0,0 +1,18 @@
+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

linux/aln/Notifications/ANC.py

@@ -0,0 +1,25 @@
+from ..enums import enums
+from ..Capabilites import Capabilites
+class ANCNotification:
+    NOTIFICATION_PREFIX = enums.NOISE_CANCELLATION_PREFIX
+    OFF = Capabilites.NoiseCancellation.OFF
+    ON = Capabilites.NoiseCancellation.ON
+    TRANSPARENCY = Capabilites.NoiseCancellation.TRANSPARENCY
+    ADAPTIVE = Capabilites.NoiseCancellation.ADAPTIVE
+    
+    def __init__(self):
+        pass
+    
+    def isANCData(self, data: bytes):
+        # 04 00 04 00 09 00 0D 01 00 00 00
+        if len(data) != 11:
+            return False
+        
+        if data.hex().startswith(self.NOTIFICATION_PREFIX.hex()):
+            return True
+        else:
+            return False
+        
+    def setData(self, data: bytes):
+        self.status = data[7]
+        pass

linux/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]

linux/aln/Notifications/ConversationalAwareness.py

@@ -0,0 +1,19 @@
+# 04 00 04 00 4b 00 02 00 01 [level]
+
+from ..enums import enums
+
+class ConversationalAwarenessNotification:
+    NOTIFICATION_PREFIX = enums.CONVERSATION_AWARENESS_RECEIVE_PREFIX
+    
+    def __init__(self):
+        pass
+    
+    def isConversationalAwarenessData(self, data: bytes):
+        if len(data) != 10:
+            return False
+        if data.hex().startswith(self.NOTIFICATION_PREFIX.hex()):
+            return True
+    
+    def setData(self, data: bytes):
+        self.status = data[9]
+        pass

linux/aln/Notifications/EarDetection.py

@@ -0,0 +1,26 @@
+from ..Capabilites import Capabilites
+from ..enums import enums
+from typing import Literal
+
+class EarDetectionNotification:
+    NOTIFICATION_BIT = Capabilites.EAR_DETECTION
+    NOTIFICATION_PREFIX = enums.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
+        

linux/aln/Notifications/Listener.py

@@ -0,0 +1,58 @@
+from bluetooth import BluetoothSocket
+import threading
+from .Battery import BatteryNotification
+from .EarDetection import EarDetectionNotification
+from .ConversationalAwareness import ConversationalAwarenessNotification
+from .ANC import ANCNotification
+import logging
+
+logging = logging.getLogger(__name__)
+
+class NotificationListener:
+    BATTERY_UPDATED = 0x01
+    ANC_UPDATED = 0x02
+    EAR_DETECTION_UPDATED = 0x03
+    CA_UPDATED = 0x04
+    UNKNOWN = 0x00
+
+    def __init__(self, socket: BluetoothSocket, callback: callable):
+        self.socket = socket
+        self.BatteryNotification = BatteryNotification()
+        self.EarDetectionNotification = EarDetectionNotification()
+        self.ANCNotification = ANCNotification()
+        self.ConversationalAwarenessNotification = ConversationalAwarenessNotification()
+        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, data)
+                pass
+            if self.EarDetectionNotification.isEarDetectionData(data):
+                self.EarDetectionNotification.setEarDetection(data)
+                self.callback(self.EAR_DETECTION_UPDATED, data)
+            if self.ANCNotification.isANCData(data):
+                self.ANCNotification.setData(data)
+                self.callback(self.ANC_UPDATED, data)
+            if self.ConversationalAwarenessNotification.isConversationalAwarenessData(data):
+                self.ConversationalAwarenessNotification.setData(data)
+                self.callback(self.CA_UPDATED, data)
+            else:
+                self.callback(self.UNKNOWN, data)
+                pass
+            pass
+        pass
+    
+    def start(self):
+        threading.Thread(target=self.__start).start()
+        pass
+    
+    def stop(self):
+        self.socket.close()
+        pass
+        

linux/aln/Notifications/__init__.py

@@ -0,0 +1,40 @@
+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
+    CA_UPDATED = NotificationListener.CA_UPDATED
+    EAR_DETECTION_UPDATED = NotificationListener.EAR_DETECTION_UPDATED
+    UNKNOWN = NotificationListener.UNKNOWN
+    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
+        self.ANCNotification = self.notificationListener.ANCNotification
+        self.ConversationalAwarenessNotification = self.notificationListener.ConversationalAwarenessNotification
+        pass
+
+    def initialize(self):
+        try:
+            self.socket.send(enums.SET_SPECIFIC_FEATURES)
+            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

linux/aln/__init__.py

@@ -0,0 +1,70 @@
+from .Notifications import Notifications
+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, callback: callable = None):
+        if callback is None:
+            callback = self.notification_callback
+        self.notifications = Notifications(self.socket, callback)
+        self.notificationListener = self.notifications.notificationListener
+        self.BatteryNotification = self.notifications.BatteryNotification
+        self.ANCNotification = self.notifications.ANCNotification
+        self.EarDetectionNotification = self.notifications.EarDetectionNotification
+        self.ConversationalAwarenessNotification = self.notifications.ConversationalAwarenessNotification
+        self.notifications.initialize()
+
+    def send(self, data: bytes):
+        try:
+            self.socket.send(data)
+        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, data: bytes):
+        import logging
+        if notification_type == Notifications.BATTERY_UPDATED:
+            logging = logging.getLogger("Battery Status")
+            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 = logging.getLogger("In-Ear Status")
+            logging.debug(f'{self.notificationListener.EarDetectionNotification.getEarDetection()}')
+            pass
+        elif notification_type == Notifications.ANC_UPDATED:
+            logging = logging.getLogger("ANC Status")
+            logging.debug(f'{self.notificationListener.ANCNotification.status}')
+            pass
+        elif notification_type == Notifications.CA_UPDATED:
+            logging = logging.getLogger("Conversational Awareness Status")
+            logging.debug(f'{self.notificationListener.ConversationalAwarenessNotification.status}')
+            pass
+        elif notification_type == Notifications.UNKNOWN:
+            logging = logging.getLogger("Unknown Notification")
+            hex_data = ' '.join(f'{byte:02x}' for byte in data)
+            logging.debug(f'{hex_data}')
+            pass
+        pass
+
+    def disconnect(self):
+        self.socket.close()
+        pass
+
+    def __del__(self):
+        self.socket.close()
+        pass

linux/aln/enums.py

@@ -0,0 +1,24 @@
+from .Capabilites import Capabilites
+
+class enums:
+    NOISE_CANCELLATION = Capabilites.NOISE_CANCELLATION
+    CONVERSATION_AWARENESS = Capabilites.CONVERSATION_AWARENESS
+    CUSTOMIZABLE_ADAPTIVE_TRANSPARENCY = Capabilites.CUSTOMIZABLE_ADAPTIVE_TRANSPARENCY
+    PREFIX = b'\x04\x00\x04\x00'
+    SETTINGS = b"\x09\x00"
+    SUFFIX = b'\x00\x00\x00'
+    NOTIFICATION_FILTER = b'\x0f'
+    SPECIFIC_FEATURES = b'\x4d'
+    SET_SPECIFIC_FEATURES = PREFIX + SPECIFIC_FEATURES + b"\x00\xff\x00\x00\x00\x00\x00\x00\x00"
+    REQUEST_NOTIFICATIONS = 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"
+    NOISE_CANCELLATION_PREFIX = PREFIX + SETTINGS + NOISE_CANCELLATION
+    NOISE_CANCELLATION_OFF = NOISE_CANCELLATION_PREFIX + Capabilites.NoiseCancellation.OFF + SUFFIX
+    NOISE_CANCELLATION_ON = NOISE_CANCELLATION_PREFIX + Capabilites.NoiseCancellation.ON + SUFFIX
+    NOISE_CANCELLATION_TRANSPARENCY = NOISE_CANCELLATION_PREFIX + Capabilites.NoiseCancellation.TRANSPARENCY + SUFFIX
+    NOISE_CANCELLATION_ADAPTIVE = NOISE_CANCELLATION_PREFIX + Capabilites.NoiseCancellation.ADAPTIVE + SUFFIX
+    
+    SET_CONVERSATION_AWARENESS_OFF = PREFIX + SETTINGS + CONVERSATION_AWARENESS + Capabilites.ConversationAwareness.OFF + SUFFIX
+    SET_CONVERSATION_AWARENESS_ON = PREFIX + SETTINGS + CONVERSATION_AWARENESS + Capabilites.ConversationAwareness.ON + SUFFIX
+    
+    CONVERSATION_AWARENESS_RECEIVE_PREFIX = PREFIX + b"\x4b\x00\x02\00"