diff options
Diffstat (limited to 'extension.js')
| -rw-r--r-- | extension.js | 240 |
1 files changed, 240 insertions, 0 deletions
diff --git a/extension.js b/extension.js new file mode 100644 index 0000000..718c92d --- /dev/null +++ b/extension.js @@ -0,0 +1,240 @@ +// Copyright 2018 Bartosz Jaroszewski +// SPDX-License-Identifier: GPL-2.0-or-later +// +// 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 2 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/>. + +const Main = imports.ui.main; +const GnomeBluetooth = imports.gi.GnomeBluetooth; +const PopupMenu = imports.ui.popupMenu; +const Util = imports.misc.util; +const GLib = imports.gi.GLib; + +const ExtensionUtils = imports.misc.extensionUtils; +const Me = ExtensionUtils.getCurrentExtension(); +const Convenience = Me.imports.convenience; + +class BluetoothDevice { + constructor(model, device) { + this._name = model.get_value(device, GnomeBluetooth.Column.NAME); + this._isConnected = model.get_value(device, GnomeBluetooth.Column.CONNECTED); + this._isPaired = model.get_value(device, GnomeBluetooth.Column.PAIRED); + this._mac = model.get_value(device, GnomeBluetooth.Column.ADDRESS); + } + + get name() { + return this._name; + } + + get isConnected() { + return this._isConnected; + } + + get isPaired() { + return this._isPaired; + } + + get mac() { + return this._mac; + } + + get item() { + if (!this._item) + this._buildMenuItem(); + + return this._item; + } + + _buildMenuItem() { + this._item = new PopupMenu.PopupSwitchMenuItem(this.name, this.isConnected); + this._item.isDeviceSwitcher = true; + this._item.connect('toggled', (item, state) => { + if (state) + this._connect(); + else + this._disconnect(); + }); + } + + _disconnect() { + this._call_bluetoothctl(`disconnect ${this.mac}`) + } + + _connect() { + this._call_bluetoothctl(`connect ${this.mac}`) + } + + _call_bluetoothctl(command) { + let btctl_command = `echo -e "${command}\\n" | bluetoothctl`; + Util.spawn(['/bin/bash', '-c', btctl_command]); + } +} + +class BluetoothQuickConnect { + constructor(bluetooth, settings) { + this._menu = bluetooth._item.menu; + this._proxy = bluetooth._proxy; + this._settings = settings; + + this._signals = []; + } + + enable() { + this._loadBluetoothModel(); + this._connectSignal(this._menu, 'open-state-changed', (menu, isOpen) => { + if (isOpen && this._autoPowerOnEnabled()) + this._proxy.BluetoothAirplaneMode = false; + }); + + this._connectSignal(this._model, 'row-changed', () => this._sync()); + this._connectSignal(this._model, 'row-deleted', () => this._sync()); + this._connectSignal(this._model, 'row-inserted', () => this._sync()); + + this._proxy.BluetoothAirplaneMode = false; + this._idleMonitor(); + this._sync(); + } + + disable() { + this._destroy(); + } + + test() { + try { + GLib.spawn_command_line_sync("bluetoothctl --version"); + } catch(error) { + Main.notifyError(`Bluetooth quick connect: error trying to execute "bluetoothctl"`); + } + } + + _idleMonitor() { + this._idleMonitorId = GLib.timeout_add(GLib.PRIORITY_DEFAULT, this._autoPowerOffCheckingInterval() * 1000, () => { + if (this._autoPowerOffEnabled() && this._getConnectedDevices().length === 0) + this._proxy.BluetoothAirplaneMode = true; + + return true; + }); + } + + _connectSignal(subject, signal_name, method) { + let signal_id = subject.connect(signal_name, method); + this._signals.push({ + subject: subject, + signal_id: signal_id + }); + } + + _loadBluetoothModel() { + this._client = new GnomeBluetooth.Client(); + this._model = this._client.get_model(); + } + + _getDefaultAdapter() { + let [ret, iter] = this._model.get_iter_first(); + while (ret) { + let isDefault = this._model.get_value(iter, GnomeBluetooth.Column.DEFAULT); + let isPowered = this._model.get_value(iter, GnomeBluetooth.Column.POWERED); + if (isDefault && isPowered) + return iter; + ret = this._model.iter_next(iter); + } + return null; + } + + _getDevices() { + let adapter = this._getDefaultAdapter(); + if (!adapter) + return []; + + let devices = []; + + let [ret, iter] = this._model.iter_children(adapter); + while (ret) { + devices.push(new BluetoothDevice(this._model, iter)); + ret = this._model.iter_next(iter); + } + + return devices; + } + + _getPairedDevices() { + return this._getDevices().filter((device) => { + return device.isPaired || device.isConnected; + }); + } + + _getConnectedDevices() { + return this._getDevices().filter((device) => { + return device.isConnected; + }); + } + + _sync() { + this._removeDevicesFromMenu(); + this._addDevicesToMenu(); + } + + _addDevicesToMenu() { + this._getPairedDevices().forEach((device) => { + this._menu.addMenuItem(device.item, 1); + }); + } + + _removeDevicesFromMenu() { + this._menu._getMenuItems().forEach((item) => { + if (item.isDeviceSwitcher) { + item.destroy(); + } + }); + } + + _destroy() { + this._signals.forEach((signal) => { + signal.subject.disconnect(signal.signal_id); + }); + this._signals = []; + this._removeDevicesFromMenu(); + + if (this._idleMonitorId) + GLib.Source.remove(this._idleMonitorId); + } + + _autoPowerOnEnabled() { + return this._settings.get_boolean('bluetooth-auto-power-on'); + } + + _autoPowerOffEnabled() { + return this._settings.get_boolean('bluetooth-auto-power-off'); + } + + _autoPowerOffCheckingInterval() { + return this._settings.get_int('bluetooth-auto-power-off-interval'); + } +} + +let bluetoothQuickConnect = null; + +function init() { + let bluetooth = Main.panel.statusArea.aggregateMenu._bluetooth; + let settings = Convenience.getSettings(); + bluetoothQuickConnect = new BluetoothQuickConnect(bluetooth, settings); +} + +function enable() { + bluetoothQuickConnect.test(); + bluetoothQuickConnect.enable(); +} + +function disable() { + bluetoothQuickConnect.disable(); +} |