DankMaterialShell/Services/BluetoothService.qml

pragma Singleton
pragma ComponentBehavior: Bound

import QtQuick
import Quickshell
import Quickshell.Bluetooth

Singleton {
    id: root

    readonly property BluetoothAdapter adapter: Bluetooth.defaultAdapter
    readonly property bool available: adapter !== null
    readonly property bool enabled: (adapter && adapter.enabled) ?? false
    readonly property bool discovering: (adapter && adapter.discovering) ?? false
    readonly property var devices: adapter ? adapter.devices : null
    readonly property var pairedDevices: {
        if (!adapter || !adapter.devices)
            return [];

        return adapter.devices.values.filter((dev) => {
            return dev && dev.paired && isValidDevice(dev);
        });
    }
    readonly property var availableDevices: {
        if (!adapter || !adapter.discovering || !Bluetooth.devices)
            return [];

        var filtered = Bluetooth.devices.values.filter((dev) => {
            return dev && !dev.paired && !dev.pairing && !dev.blocked && isValidDevice(dev) && (dev.signalStrength === undefined || dev.signalStrength > 0);
        });
        return sortBySignalStrength(filtered);
    }
    readonly property var allDevicesWithBattery: {
        if (!adapter || !adapter.devices)
            return [];

        return adapter.devices.values.filter((dev) => {
            return dev && dev.batteryAvailable && dev.battery > 0;
        });
    }

    function sortBySignalStrength(devices) {
        return devices.sort((a, b) => {
            var aSignal = (a.signalStrength !== undefined && a.signalStrength > 0) ? a.signalStrength : 0;
            var bSignal = (b.signalStrength !== undefined && b.signalStrength > 0) ? b.signalStrength : 0;
            return bSignal - aSignal;
        });
    }

    function isValidDevice(device) {
        if (!device)
            return false;

        var displayName = device.name || device.deviceName;
        if (!displayName || displayName.length < 2)
            return false;

        if (displayName.startsWith('/org/bluez') || displayName.includes('hci0'))
            return false;

        return displayName.length >= 3;
    }

    function getDeviceIcon(device) {
        if (!device)
            return "bluetooth";

        var name = (device.name || device.deviceName || "").toLowerCase();
        var icon = (device.icon || "").toLowerCase();
        if (icon.includes("headset") || icon.includes("audio") || name.includes("headphone") || name.includes("airpod") || name.includes("headset") || name.includes("arctis"))
            return "headset";

        if (icon.includes("mouse") || name.includes("mouse"))
            return "mouse";

        if (icon.includes("keyboard") || name.includes("keyboard"))
            return "keyboard";

        if (icon.includes("phone") || name.includes("phone") || name.includes("iphone") || name.includes("android") || name.includes("samsung"))
            return "smartphone";

        if (icon.includes("watch") || name.includes("watch"))
            return "watch";

        if (icon.includes("speaker") || name.includes("speaker"))
            return "speaker";

        if (icon.includes("display") || name.includes("tv"))
            return "tv";

        return "bluetooth";
    }

    function getDeviceType(device) {
        if (!device)
            return "bluetooth";

        var name = (device.name || device.deviceName || "").toLowerCase();
        var icon = (device.icon || "").toLowerCase();
        if (icon.includes("headset") || icon.includes("audio") || name.includes("headphone") || name.includes("airpod") || name.includes("headset") || name.includes("arctis"))
            return "headset";

        if (icon.includes("mouse") || name.includes("mouse"))
            return "mouse";

        if (icon.includes("keyboard") || name.includes("keyboard"))
            return "keyboard";

        if (icon.includes("phone") || name.includes("phone") || name.includes("iphone") || name.includes("android") || name.includes("samsung"))
            return "phone";

        if (icon.includes("watch") || name.includes("watch"))
            return "watch";

        if (icon.includes("speaker") || name.includes("speaker"))
            return "speaker";

        if (icon.includes("display") || name.includes("tv"))
            return "tv";

        return "bluetooth";
    }

    function canPair(device) {
        if (!device)
            return false;

        return !device.paired && !device.pairing && !device.blocked;
    }

    function debugDevice(device) {
        console.log("Device:", device.name, "paired:", device.paired, "connected:", device.connected, "signalStrength:", device.signalStrength);
    }

    function getPairingStatus(device) {
        if (!device)
            return "unknown";

        if (device.pairing)
            return "pairing";

        if (device.paired)
            return "paired";

        if (device.blocked)
            return "blocked";

        return "available";
    }

    function getSignalStrength(device) {
        if (!device || device.signalStrength === undefined || device.signalStrength <= 0)
            return "Unknown";

        var signal = device.signalStrength;
        if (signal >= 80)
            return "Excellent";

        if (signal >= 60)
            return "Good";

        if (signal >= 40)
            return "Fair";

        if (signal >= 20)
            return "Poor";

        return "Very Poor";
    }

    function getSignalIcon(device) {
        if (!device || device.signalStrength === undefined || device.signalStrength <= 0)
            return "signal_cellular_null";

        var signal = device.signalStrength;
        if (signal >= 80)
            return "signal_cellular_4_bar";

        if (signal >= 60)
            return "signal_cellular_3_bar";

        if (signal >= 40)
            return "signal_cellular_2_bar";

        if (signal >= 20)
            return "signal_cellular_1_bar";

        return "signal_cellular_0_bar";
    }

    function toggleAdapter() {
        if (adapter)
            adapter.enabled = !adapter.enabled;

    }

    function startScan() {
        if (adapter)
            adapter.discovering = true;

    }

    function stopScan() {
        if (adapter)
            adapter.discovering = false;

    }

    function connect(address) {
        var device = _findDevice(address);
        if (device)
            device.connect();

    }

    function disconnect(address) {
        var device = _findDevice(address);
        if (device)
            device.disconnect();

    }

    function pair(address) {
        var device = _findDevice(address);
        if (device && canPair(device))
            device.pair();

    }

    function forget(address) {
        var device = _findDevice(address);
        if (device)
            device.forget();

    }

    function toggle(address) {
        var device = _findDevice(address);
        if (device) {
            if (device.connected)
                device.disconnect();
            else
                device.connect();
        }
    }

    function _findDevice(address) {
        if (!adapter)
            return null;

        return adapter.devices.values.find((d) => {
            return d && d.address === address;
        }) || (Bluetooth.devices ? Bluetooth.devices.values.find((d) => {
            return d && d.address === address;
        }) : null);
    }

}