Ano-sys
121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
from ignis.widgets import Widget
|
|
from ignis.utils import Utils
|
|
from .qs_button import QSButton
|
|
from typing import List
|
|
from ignis.services.network import NetworkService, VpnConnection
|
|
|
|
|
|
network = NetworkService.get_default()
|
|
|
|
|
|
class VpnNetworkItem(Widget.Button):
|
|
def __init__(self, conn: VpnConnection):
|
|
|
|
super().__init__(
|
|
css_classes=["vpn-connection-item", "unset"],
|
|
on_click=lambda x: conn.toggle_connection(),
|
|
child=Widget.Box(
|
|
child=[
|
|
Widget.Label(
|
|
label=conn.name,
|
|
ellipsize="end",
|
|
max_width_chars=20,
|
|
halign="start",
|
|
css_classes=["vpn-connection-label"],
|
|
),
|
|
Widget.Button(
|
|
child=Widget.Label(
|
|
label=conn.bind(
|
|
"is_connected",
|
|
lambda value: "Disconnect" if value else "Connect",
|
|
)
|
|
),
|
|
css_classes=["vpn-connection-item-connect-label", "unset"],
|
|
halign="end",
|
|
hexpand=True,
|
|
),
|
|
]
|
|
),
|
|
)
|
|
|
|
|
|
def vpn_qsbutton() -> QSButton:
|
|
networks_list = Widget.Revealer(
|
|
transition_duration=300,
|
|
transition_type="slide_down",
|
|
child=Widget.Box(
|
|
vertical=True,
|
|
css_classes=["control-center-menu"],
|
|
child=[
|
|
Widget.Box(
|
|
css_classes=["vpn-header-box"],
|
|
child=[
|
|
Widget.Icon(
|
|
icon_name="network-vpn-symbolic", pixel_size=28),
|
|
Widget.Label(
|
|
label="VPN connections",
|
|
css_classes=["vpn-header-label"],
|
|
),
|
|
],
|
|
),
|
|
Widget.Box(
|
|
vertical=True,
|
|
child=network.vpn.bind(
|
|
"connections",
|
|
transform=lambda value: [
|
|
VpnNetworkItem(i) for i in value],
|
|
),
|
|
),
|
|
Widget.Separator(
|
|
css_classes=["vpn-connection-list-separator"]),
|
|
Widget.Button(
|
|
css_classes=["vpn-connection-item", "unset"],
|
|
on_click=lambda x: Utils.exec_sh_async(
|
|
"nm-connection-editor"),
|
|
style="margin-bottom: 0;",
|
|
child=Widget.Box(
|
|
child=[
|
|
Widget.Icon(image="preferences-system-symbolic"),
|
|
Widget.Label(
|
|
label="Network Manager",
|
|
halign="start",
|
|
css_classes=["vpn-connection-label"],
|
|
),
|
|
]
|
|
),
|
|
),
|
|
],
|
|
),
|
|
)
|
|
|
|
def get_label(id: str) -> str:
|
|
if id:
|
|
return id
|
|
else:
|
|
return "VPN"
|
|
|
|
def get_icon(icon_name: str) -> str:
|
|
if network.vpn.is_connected:
|
|
return icon_name
|
|
else:
|
|
return "network-vpn-symbolic"
|
|
|
|
def toggle_list(x) -> None:
|
|
networks_list.toggle()
|
|
|
|
return QSButton(
|
|
label=network.vpn.bind("active_vpn_id", get_label),
|
|
icon_name=network.vpn.bind("icon-name", get_icon),
|
|
on_activate=toggle_list,
|
|
on_deactivate=toggle_list,
|
|
active=network.vpn.bind("is-connected"),
|
|
content=networks_list,
|
|
)
|
|
|
|
|
|
def vpn_control() -> List[QSButton]:
|
|
if len(network.vpn.connections) > 0:
|
|
return [vpn_qsbutton()]
|
|
else:
|
|
return []
|