From 5a24cd0160ba1ed8e366bf34e3c994162b55aeab Mon Sep 17 00:00:00 2001
From: atavism <paul@getlantern.org>
Date: Sat, 22 Jun 2024 12:08:54 -0700
Subject: [PATCH] show VPNBandwidth widget on desktop

---
 desktop/app/app.go   | 3 +++
 lib/vpn/vpn_tab.dart | 4 ++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/desktop/app/app.go b/desktop/app/app.go
index bec9b3ea4..6e5affb9d 100644
--- a/desktop/app/app.go
+++ b/desktop/app/app.go
@@ -120,6 +120,9 @@ func NewApp(flags flashlight.Flags, configDir string, proClient proclient.ProCli
 	onProStatusChange(func(isPro bool) {
 		app.statsTracker.SetIsPro(isPro)
 	})
+	datacap.AddDataCapListener(func(hitDataCap bool) {
+		app.statsTracker.SetHitDataCap(hitDataCap)
+	})
 
 	log.Debugf("Using configdir: %v", configDir)
 
diff --git a/lib/vpn/vpn_tab.dart b/lib/vpn/vpn_tab.dart
index 90fd79551..5e40a4ba8 100644
--- a/lib/vpn/vpn_tab.dart
+++ b/lib/vpn/vpn_tab.dart
@@ -87,8 +87,8 @@ class VPNTab extends StatelessWidget {
                         if (Platform.isAndroid) ...{
                           const CDivider(height: 32.0),
                           SplitTunnelingWidget(),
-                          if (!proUser) const VPNBandwidth(),
-                        }
+                        },
+                        if (!proUser) const VPNBandwidth(),
                       ],
                     ),
                   ),