import subprocess import time from fabric.bluetooth import BluetoothClient from fabric.utils import get_relative_path from fabric.widgets.box import Box from fabric.widgets.button import Button from fabric.widgets.circularprogressbar import CircularProgressBar from fabric.widgets.label import Label from fabric.widgets.overlay import Overlay from gi.repository import GLib import config.data as data import utils.icons as icons from .notifications import Notifications from services.network import NetworkClient class WifiIndicator(Button): def __init__(self, **kwargs): super().__init__(name="button-bar-network", **kwargs) self.network = NetworkClient() self.progress_bar = CircularProgressBar( name="button-network", size=28, line_width=2, start_angle=150, end_angle=390, ) self.wifi_label = Label(name="network-label", markup=icons.loader) self.add(Overlay(child=self.progress_bar, overlays=self.wifi_label)) self.network.connect("ready", self.on_network_ready) if self.network.is_ready: self.setup_signals() def on_network_ready(self, *args): self.setup_signals() def setup_signals(self): self.network.connect("changed", self.on_network_changed) self.network.connect("wifi-device-added", self.on_wifi_device_added) self.network.connect("wifi-device-removed", self.on_wifi_device_removed) self.network.connect("ethernet-device-added", self.on_ethernet_device_added) self.network.connect("ethernet-device-removed", self.on_ethernet_device_removed) if self.network.wifi_device: self.network.wifi_device.connect("changed", self.on_wifi_changed) self.network.wifi_device.connect("ap-added", self.on_wifi_changed) self.network.wifi_device.connect("ap-removed", self.on_wifi_changed) if self.network.ethernet_device: self.network.ethernet_device.connect("changed", self.on_ethernet_changed) self.update_state() def update_state(self): if ( self.network.ethernet_device and self.network.ethernet_device.state == "activated" ): self.wifi_label.set_markup(icons.world) self.progress_bar.value = 1.0 # Full circle for ethernet tooltip = f"Connected to Ethernet ({self.network.ethernet_device.speed})" self.set_tooltip_text(tooltip) return if not self.network.wifi_device: self.wifi_label.set_markup(icons.cloud_off) self.progress_bar.value = 0 tooltip = "No WiFi device found" elif not self.network.wifi_device.wireless_enabled: self.wifi_label.set_markup(icons.cloud_off) self.progress_bar.value = 0 tooltip = "WiFi disabled" else: active_ap = self.network.wifi_device.active_access_point if active_ap: strength = active_ap.strength self.progress_bar.value = strength / 100 if strength >= 75: self.wifi_label.set_markup(icons.wifi_3) elif strength >= 50: self.wifi_label.set_markup(icons.wifi_2) elif strength >= 25: self.wifi_label.set_markup(icons.wifi_1) else: self.wifi_label.set_markup(icons.wifi_0) tooltip = f"Connected to {active_ap.ssid} ({strength}%)" else: self.wifi_label.set_markup(icons.world_off) self.progress_bar.value = 0 tooltip = "Not connected" self.set_tooltip_text(tooltip) def on_wifi_device_added(self, *args): self.network.wifi_device.connect("changed", self.on_wifi_changed) self.network.wifi_device.connect("ap-added", self.on_wifi_changed) self.network.wifi_device.connect("ap-removed", self.on_wifi_changed) self.update_state() def on_wifi_device_removed(self, *args): self.update_state() def on_network_changed(self, *args): self.update_state() def on_wifi_changed(self, *args): self.update_state() def on_ethernet_device_added(self, *args): if self.network.ethernet_device: self.network.ethernet_device.connect("changed", self.on_ethernet_changed) self.update_state() def on_ethernet_device_removed(self, *args): self.update_state() def on_ethernet_changed(self, *args): self.update_state() class BluetoothIndicator(Button): def __init__(self, **kwargs): super().__init__(name="button-bar-bluetooth", **kwargs) self.bluetooth = BluetoothClient() self.bt_label = Label(name="bluetooth-label", markup=icons.loader) self.add(self.bt_label) self.bluetooth.connect("changed", self.on_bluetooth_changed) self.bluetooth.connect("device-added", self.on_device_added) self.bluetooth.connect("device-removed", self.on_device_removed) self.update_state() def update_state(self): if not self.bluetooth.enabled: self.bt_label.set_markup(icons.bluetooth_off) tooltip = "Bluetooth disabled" else: connected_devices = self.bluetooth.connected_devices if connected_devices: self.bt_label.set_markup(icons.bluetooth_connected) if len(connected_devices) == 1: device = connected_devices[0] tooltip = f"Connected to {device.alias}" if device.battery_percentage > 0: tooltip += f" ({device.battery_percentage:.0f}%)" else: tooltip = f"Connected to {len(connected_devices)} devices" else: if self.bluetooth.scanning: self.bt_label.set_markup(icons.loader) tooltip = "Scanning for devices..." else: self.bt_label.set_markup(icons.bluetooth) tooltip = "No devices connected" self.set_tooltip_text(tooltip) def on_bluetooth_changed(self, *args): self.update_state() def on_device_added(self, _, address): self.update_state() def on_device_removed(self, _, address): self.update_state() class RecordingIndicator(Button): def __init__(self, **kwargs): super().__init__(name="button-bar-recording", **kwargs) self.script_path = get_relative_path("../../../scripts/screen-capture.sh") self.recording_start_time = None self.recording_label = Label(name="recording-label", markup=icons.screenrecord) self.time_label = Label(name="recording-time-label", markup="00:00") # Adapt layout based on dock position if data.VERTICAL: # Vertical dock: show only icon self.recording_box = Box( orientation="v", spacing=2, children=[self.recording_label] ) else: # Horizontal dock: place icon and time side by side self.recording_box = Box( orientation="h", spacing=4, children=[self.recording_label, self.time_label] ) self.add(self.recording_box) self.connect("clicked", self.on_stop_recording) self.hide() # Delay initial check to prevent showing during dock initialization GLib.timeout_add(1000, self._delayed_init) def check_recording_status(self): try: result = subprocess.run( [self.script_path, "status"], capture_output=True, text=True, timeout=2 ) is_recording = result.stdout.strip() == "true" if is_recording: if not self.get_visible(): self.show() # Get the recording start time if we don't have it if self.recording_start_time is None: self.recording_start_time = self.get_recording_start_time() # Update the recording time display if self.recording_start_time: elapsed_seconds = int(time.time() - self.recording_start_time) minutes = elapsed_seconds // 60 seconds = elapsed_seconds % 60 # Only update time label in horizontal mode if not data.VERTICAL: time_text = f"{minutes:02d}:{seconds:02d}" self.time_label.set_markup(time_text) self.set_tooltip_text(f"Recording in progress ({minutes:02d}:{seconds:02d}) - Click to stop") else: self.set_tooltip_text("Recording in progress - Click to stop") else: if self.get_visible(): self.hide() self.recording_start_time = None except Exception: # If we can't check status, hide the indicator if self.get_visible(): self.hide() self.recording_start_time = None return True # Continue the timeout def get_recording_start_time(self): """Get the recording start time from the file""" try: with open("/tmp/recording_start_time.txt", "r") as f: return float(f.read().strip()) except Exception: return None def on_stop_recording(self, *args): try: subprocess.Popen([self.script_path, "record", "stop"]) except Exception as e: print(f"Error stopping recording: {e}") def _delayed_init(self): """Initialize recording status check after a delay to prevent showing during dock startup""" try: self.check_recording_status() self.timeout_id = GLib.timeout_add(1000, self.check_recording_status) except Exception as e: print(f"[DEBUG] Error in delayed recording indicator init: {e}") return False # Don't repeat this timeout def cleanup(self): if hasattr(self, "timeout_id"): GLib.source_remove(self.timeout_id) class Indicators(Box): def __init__(self, **kwargs): self.recording_indicator = RecordingIndicator() self.notifications = Notifications() super().__init__( name="indicator", orientation="h" if not data.VERTICAL else "v", spacing=4, children=[ WifiIndicator(), BluetoothIndicator(), self.notifications, self.recording_indicator, ], **kwargs, ) self.show_all() def cleanup(self): if hasattr(self, "recording_indicator"): self.recording_indicator.cleanup()