add cloudflare

entrypoints/background.ts

@@ -42,7 +42,14 @@ async function handleTabUpdate(url: string) {
     const apiResponse = await fetch(`https://ip.albert.lol/${ip}`)
     const apiData = await apiResponse.json()
 
-    await updateIcon(apiData.country || null)
+    const country = apiData.country || null
+    const asn = apiData.org?.split(' ')[0]
+    let iconCode = country
+    if (!iconCode && asn === 'AS13335') {
+      iconCode = 'cloudflare'
+    }
+
+    await updateIcon(iconCode)
   } catch (error) {
     console.error('Failed to handle tab update:', error)
     await updateIcon(null)
@@ -60,7 +67,8 @@ browser.tabs.onUpdated.addListener(async (_tabId, changeInfo) => {
 
 export default defineBackground({
   main() {
-    browser.runtime.onMessage.addListener((request: any, _sender, sendResponse) => {
+    browser.runtime.onMessage.addListener(
+      (request: any, _sender, sendResponse) => {
         if (request.type === 'FETCH_SERVER_INFO') {
           ;(async () => {
             try {
@@ -76,7 +84,13 @@ export default defineBackground({
               const parsed = psl.parse(request.hostname)
               const origin = 'domain' in parsed ? parsed.domain : null
 
-            await updateIcon(apiData.country)
+              const country = apiData.country || null
+              const asn = apiData.org?.split(' ')[0]
+              let iconCode = country
+              if (!iconCode && asn === 'AS13335') {
+                iconCode = 'cloudflare'
+              }
+              await updateIcon(iconCode)
 
               sendResponse({
                 error: null,
@@ -102,6 +116,7 @@ export default defineBackground({
 
         sendResponse({ error: 'Unknown request type', data: null })
         return true
-    })
+      }
+    )
   },
 })

utils/icon.ts

@@ -1,9 +1,14 @@
 export async function updateIcon(countryCode: string | null) {
-  const validCode =
+  let validCode
+  if (countryCode === 'cloudflare') {
+    validCode = 'cloudflare'
+  } else {
+    validCode =
       countryCode?.match(/^[A-Z]{2}$/i)?.[0]?.toLowerCase() || 'unknown'
+  }
 
   const loadImageBitmap = async (code: string): Promise<ImageBitmap> => {
-    const url = browser.runtime.getURL("/")
+    const url = browser.runtime.getURL('/')
     try {
       const response = await fetch(url + `${code}.webp`)
       if (!response.ok) throw new Error('Flag not found')