hostinfo/entrypoints/background.ts

import { browser } from 'wxt/browser';
import { Tab } from '@/services/tab';
import { StorageService } from '@/utils/storage';

export default defineBackground({
  main() {
    // Listen for Network Responses (Source of Truth for IPs)
    browser.webRequest.onResponseStarted.addListener(
      async (details) => {
        if (details.tabId === -1 || details.type !== 'main_frame' || !details.ip) return;
        await Tab.process(details.tabId, details.url, details.ip);
      },
      { urls: ['<all_urls>'] }
    );

    // Listen for Navigation
    browser.tabs.onUpdated.addListener(async (tabId, changeInfo, tab) => {
      if (changeInfo.status === 'complete' && tab.url && tab.url.startsWith('http')) {
        // We might not have the IP yet if it was cached, so we trigger a process
        // If IP is missing, Tab waits or we can force a HEAD request here
        await Tab.process(tabId, tab.url);

        // Force connection to ensure webRequest fires if cached
        try {
          await fetch(tab.url, { method: 'HEAD', cache: 'no-store', mode: 'no-cors' });
        } catch { /* ignore */ }
      }
    });

    // 3. Cleanup
    browser.tabs.onRemoved.addListener(async (tabId) => {
      await StorageService.remove(tabId);
    });
  },
});