mirror of
https://github.com/kavishdevar/librepods.git
synced 2026-01-28 22:01:50 +00:00
90 lines
3.5 KiB
Python
90 lines
3.5 KiB
Python
# AirPods like Normal (ALN) - Bringing Apple-only features to Linux and Android for seamless AirPods functionality!
|
|
# Copyright (C) 2024 Kavish Devar
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import socket
|
|
import argparse
|
|
from aln import enums
|
|
import logging
|
|
|
|
class CustomFormatter(logging.Formatter):
|
|
# Define color codes for different log levels
|
|
COLORS = {
|
|
logging.DEBUG: "\033[48;5;240;38;5;15m%s\033[1;0m", # Grey background, white bold text
|
|
logging.INFO: "\033[48;5;34;38;5;15m%s\033[1;0m", # Green background, white bold text
|
|
logging.WARNING: "\033[1;48;5;214;38;5;0m%s\033[1;0m", # Orange background, black bold text
|
|
logging.ERROR: "\033[1;48;5;202;38;5;15m%s\033[1;0m", # Orange-red background, white bold text
|
|
logging.CRITICAL: "\033[1;48;5;196;38;5;15m%s\033[1;0m", # Pure red background, white bold text
|
|
}
|
|
|
|
def format(self, record):
|
|
# Apply color to the level name
|
|
levelname = self.COLORS.get(record.levelno, "%s") % record.levelname.ljust(8)
|
|
record.levelname = levelname
|
|
|
|
# Format the message
|
|
formatted_message = super().format(record)
|
|
|
|
return formatted_message
|
|
|
|
# Custom formatter with fixed width for level name
|
|
formatter = CustomFormatter('\033[2;37m%(asctime)s\033[1;0m - %(levelname)s - %(message)s')
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
# Set the custom formatter for the root logger
|
|
logging.getLogger().handlers[0].setFormatter(formatter)
|
|
|
|
|
|
SOCKET_PATH = "/tmp/airpods_daemon.sock"
|
|
|
|
def send_command(command):
|
|
"""Send a command to the daemon via UNIX domain socket."""
|
|
try:
|
|
# Create a socket connection to the daemon
|
|
logging.info("Connecting to daemon...")
|
|
client_socket = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
client_socket.connect(SOCKET_PATH)
|
|
logging.info("Connected to daemon")
|
|
|
|
# Send the command
|
|
client_socket.sendall(command)
|
|
logging.info(f"Sent command: {command}")
|
|
|
|
# Close the connection
|
|
client_socket.close()
|
|
logging.info("Socket closed")
|
|
except Exception as e:
|
|
logging.error(f"Error communicating with daemon: {e}")
|
|
|
|
def parse_arguments():
|
|
parser = argparse.ArgumentParser(description="Set AirPods ANC mode.")
|
|
parser.add_argument("mode", choices=["off", "on", "transparency", "adaptive", "1", "2", "3", "4"], help="ANC mode to set")
|
|
return parser.parse_args()
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_arguments()
|
|
|
|
if args.mode == "off" or args.mode == "1":
|
|
command = enums.NOISE_CANCELLATION_OFF
|
|
elif args.mode == "on" or args.mode == "2":
|
|
command = enums.NOISE_CANCELLATION_ON
|
|
elif args.mode == "transparency" or args.mode == "3":
|
|
command = enums.NOISE_CANCELLATION_TRANSPARENCY
|
|
elif args.mode == "adaptive" or args.mode == "4":
|
|
command = enums.NOISE_CANCELLATION_ADAPTIVE
|
|
|
|
send_command(command)
|