From bf5c5575f563a6661e4aef4d9b091fe292ebae35 Mon Sep 17 00:00:00 2001
From: skidoodle <contact@albert.lol>
Date: Mon, 24 Mar 2025 20:01:31 +0100
Subject: [PATCH] refactor icon handling to use webextension-polyfill for
 browser compatibility

---
 utils/icon.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/utils/icon.ts b/utils/icon.ts
index ec143e5..8e2d70c 100644
--- a/utils/icon.ts
+++ b/utils/icon.ts
@@ -1,3 +1,5 @@
+import { browser } from 'webextension-polyfill-ts'
+
 export async function updateIcon(countryCode: string | null) {
   let validCode
   if (countryCode === 'cloudflare') {
@@ -57,12 +59,12 @@ export async function updateIcon(countryCode: string | null) {
 
   try {
     const bitmap = await loadImageBitmap(validCode)
-    chrome.action.setIcon({ imageData: await processImage(bitmap) })
+    browser.action.setIcon({ imageData: await processImage(bitmap) })
   } catch (error) {
     console.error('Primary flag failed, trying unknown:', error)
     try {
       const unknownBitmap = await loadImageBitmap('unknown')
-      chrome.action.setIcon({ imageData: await processImage(unknownBitmap) })
+      browser.action.setIcon({ imageData: await processImage(unknownBitmap) })
     } catch (fallbackError) {
       console.error('Both flag assets failed:', fallbackError)
     }