Refresh the extension badge when toggling proxy

mullvad · Apr 18, 2024 · 8b90c28 · 8b90c28
1 parent 9089b8d
commit 8b90c28
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/composables/useSocksProxy.ts b/src/composables/useSocksProxy.ts
@@ -6,6 +6,7 @@ import {
   ProxyInfoType,
   ProxyOperationArgs,
 } from '@/helpers/socksProxy.types';
+import { updateTabsProxyBadges } from '@/helpers/browserAction';
 
 import useActiveTab from '@/composables/useActiveTab';
 import useConnection from '@/composables/useConnection';
@@ -36,9 +37,11 @@ const currentHostProxyDNSEnabled = computed(() => currentHostProxyDetails.value?
 
 const toggleGlobalProxy = () => {
   globalProxyDetails.value.socksEnabled = !globalProxyDetails.value.socksEnabled;
+  updateTabsProxyBadges();
 };
 const toggleCurrentHostProxy = () => {
   hostProxiesDetails.value[activeTabHost.value].socksEnabled = !currentHostProxyEnabled.value;
+  updateTabsProxyBadges();
 };
 
 const toggleGlobalProxyDNS = () => {

diff --git a/src/helpers/browserAction.ts b/src/helpers/browserAction.ts
@@ -33,7 +33,7 @@ const updateTabProxyBadge = async (tab: browser.tabs.Tab, activeProxyDetails: Pr
 };
 
 // Update state of the proxy badge & title, for all tabs
-const updateTabsProxyBadges = async () => {
+export const updateTabsProxyBadges = async () => {
   const tabs = await browser.tabs.query({});
   const activeProxyDetails = await getActiveProxyDetails();