From 53c4c7bfc9f0c0fbf9089dd03a40cb60f707bcb3 Mon Sep 17 00:00:00 2001
From: Reinout Meliesie <zedfrigg@kernelmaft.com>
Date: Sat, 30 Mar 2024 01:40:04 +0100
Subject: [PATCH] feat: add support for VPN primary connection

---
 src/modules/networkmanager.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/modules/networkmanager.rs b/src/modules/networkmanager.rs
index 4786c82..d503176 100644
--- a/src/modules/networkmanager.rs
+++ b/src/modules/networkmanager.rs
@@ -32,6 +32,7 @@ pub enum NetworkmanagerState {
     Cellular,
     Offline,
     Unknown,
+    Vpn,
     Wired,
     Wireless,
     WirelessDisconnected,
@@ -111,6 +112,7 @@ impl Module<gtk::Box> for NetworkmanagerModule {
                 NetworkmanagerState::Cellular => "network-cellular-symbolic",
                 NetworkmanagerState::Offline => "network-wireless-disabled-symbolic",
                 NetworkmanagerState::Unknown => "dialog-question-symbolic",
+                NetworkmanagerState::Vpn => "network-vpn-symbolic",
                 NetworkmanagerState::Wired => "network-wired-symbolic",
                 NetworkmanagerState::Wireless => "network-wireless-symbolic",
                 NetworkmanagerState::WirelessDisconnected => "network-wireless-acquiring-symbolic",
@@ -147,8 +149,10 @@ async fn get_network_state(
             "cdma" => Ok(NetworkmanagerState::Cellular),
             "gsm" => Ok(NetworkmanagerState::Cellular),
             "pppoe" => Ok(NetworkmanagerState::Wired),
+            "vpn" => Ok(NetworkmanagerState::Vpn),
             "wifi-p2p" => Ok(NetworkmanagerState::Wireless),
             "wimax" => Ok(NetworkmanagerState::Cellular),
+            "wireguard" => Ok(NetworkmanagerState::Vpn),
             "wpan" => Ok(NetworkmanagerState::Wireless),
             _ => Ok(NetworkmanagerState::Unknown),
         }