from fabric.widgets.box import Box
from fabric.widgets.label import Label
from fabric.widgets.image import Image
from fabric.widgets.button import Button
from fabric.widgets.centerbox import CenterBox
from fabric.widgets.scrolledwindow import ScrolledWindow
from fabric.bluetooth import BluetoothClient, BluetoothDevice
import utils.icons as icons


class BluetoothDeviceSlot(CenterBox):
    def __init__(self, device: BluetoothDevice, **kwargs):
        super().__init__(name="bluetooth-device", **kwargs)
        self.device = device
        self.device.connect("changed", self.on_changed)
        self.device.connect(
            "notify::closed", lambda *_: self.device.closed and self.destroy()
        )

        self.connection_label = Label(
            name="bluetooth-connection", markup=icons.bluetooth_disconnected
        )
        self.connect_button = Button(
            name="bluetooth-connect",
            label="Connect",
            on_clicked=lambda *_: self.device.set_connecting(not self.device.connected),
        )

        self.start_children = [
            Box(
                spacing=8,
                children=[
                    Image(icon_name=f"{device.icon_name}-symbolic", size=32),
                    Label(label=device.name),
                    self.connection_label,
                ],
            )
        ]
        self.end_children = self.connect_button

        self.device.emit("changed")  # to update display status

    def on_changed(self, *_):
        self.connection_label.set_markup(
            icons.bluetooth_connected
            if self.device.connected
            else icons.bluetooth_disconnected
        )
        if self.device.connecting:
            self.connect_button.set_label(
                "Connecting..." if not self.device.connecting else "Disconnecting..."
            )
        else:
            self.connect_button.set_label(
                "Connect" if not self.device.connected else "Disconnect"
            )
        return


class BluetoothConnections(Box):
    def __init__(self, **kwargs):
        super().__init__(
            name="bluetooth",
            spacing=8,
            orientation="vertical",
            **kwargs,
        )

        self.launcher = kwargs["launcher"]

        self.buttons = self.launcher.dashboard.bluetooth_button
        self.bt_status_text = self.buttons.bluetooth_status_text
        self.bt_status_button = self.buttons.bluetooth_status_button
        self.bt_icon = self.buttons.bluetooth_icon
        self.bt_label = self.buttons.bluetooth_label
        self.bt_menu_button = self.buttons.bluetooth_menu_button
        self.bt_menu_label = self.buttons.bluetooth_menu_label

        self.client = BluetoothClient(on_device_added=self.on_device_added)
        self.scan_button = Button(
            name="bluetooth-scan",
            child=Label(name="bluetooth-scan-icon", markup=icons.scan_stop),
            on_clicked=lambda *_: self.client.toggle_scan(),
        )
        self.toggle_button = Button(
            name="bluetooth-toggle",
            child=Label(name="bluetooth-toggle-icon", markup=icons.bluetooth_off),
            on_clicked=lambda *_: self.client.toggle_power(),
        )

        self.client.connect(
            "notify::enabled",
            lambda *_: self.status_label(),
        )
        self.client.connect(
            "notify::scanning",
            lambda *_: self.scan_button.get_child().set_markup(
                icons.scan_stop if self.client.scanning else icons.scan
            ),
        )

        self.paired_box = Box(spacing=2, orientation="vertical")
        self.available_box = Box(spacing=2, orientation="vertical")

        self.children = [
            CenterBox(
                name="bluetooth-header",
                start_children=self.scan_button,
                center_children=Label(name="bluetooth-text", label="Bluetooth Devices"),
                end_children=self.toggle_button,
            ),
            Label(name="bluetooth-section", label="Paired"),
            ScrolledWindow(
                name="bluetooth-paired",
                min_content_size=(-1, -1),
                child=self.paired_box,
                v_expand=True,
            ),
            Label(name="bluetooth-section", label="Available"),
            ScrolledWindow(
                name="bluetooth-available",
                min_content_size=(-1, -1),
                child=self.available_box,
                v_expand=True,
            ),
        ]

        # to run notify closures thus display the status
        # without having to wait until an actual change
        self.client.notify("scanning")
        self.client.notify("enabled")

    def status_label(self):
        print(self.client.enabled)
        if self.client.enabled:
            self.toggle_button.get_child().set_markup(icons.bluetooth)
            self.bt_status_text.set_label("Enabled")
            for i in [
                self.bt_status_button,
                self.bt_status_text,
                self.bt_icon,
                self.bt_label,
                self.bt_menu_button,
                self.bt_menu_label,
            ]:
                i.remove_style_class("disabled")
            self.bt_icon.set_markup(icons.bluetooth)
        else:
            self.toggle_button.get_child().set_markup(icons.bluetooth_off)
            self.bt_status_text.set_label("Disabled")
            for i in [
                self.bt_status_button,
                self.bt_status_text,
                self.bt_icon,
                self.bt_label,
                self.bt_menu_button,
                self.bt_menu_label,
            ]:
                i.add_style_class("disabled")
            self.bt_icon.set_markup(icons.bluetooth_off)

    def on_device_added(self, client: BluetoothClient, address: str):
        if not (device := client.get_device(address)):
            return
        slot = BluetoothDeviceSlot(device)

        if device.paired:
            self.paired_box.add(slot)
        else:
            self.available_box.add(slot)