background.js

const CONFIG = {
  API_URLS: {
    ip: 'https://www.virustotal.com/api/v3/ip_addresses/',
    domain: 'https://www.virustotal.com/api/v3/domains/',
    hash: 'https://www.virustotal.com/api/v3/files/'
  }
};

chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
  if (request.action === 'fetchIndicatorData') {
    fetchIndicatorData(request.value, request.type)
      .then(data => sendResponse({ success: true, data }))
      .catch(error => sendResponse({ success: false, error: error.message }));
    
    return true;
  }
  
  if (request.action === 'updateBadge') {
    if (sender.tab && sender.tab.id) {
      updateBadge(sender.tab.id, request.count);
    }
  }
});

async function fetchIndicatorData(value, type) {
  const result = await chrome.storage.sync.get('vtApiKey');
  const apiKey = result.vtApiKey;
  
  if (!apiKey) {
    throw new Error('No API key configured. Please add your VirusTotal API key in extension settings.');
  }
  
  const baseUrl = CONFIG.API_URLS[type];
  if (!baseUrl) {
    throw new Error(`Unsupported indicator type: ${type}`);
  }
  
  const response = await fetch(`${baseUrl}${value}`, {
    headers: {
      'x-apikey': apiKey
    }
  });
  
  if (!response.ok) {
    if (response.status === 429) {
      throw new Error('API rate limit exceeded');
    }
    if (response.status === 401) {
      throw new Error('Invalid API key. Please check your settings.');
    }
    if (response.status === 404) {
      throw new Error('Not found in VirusTotal database');
    }
    throw new Error(`API error: ${response.status}`);
  }
  
  const data = await response.json();
  return data;
}
function updateBadge(tabId, count) {
  if (count > 0) {
    chrome.action.setBadgeText({
      tabId: tabId,
      text: count.toString()
    });
    
    chrome.action.setBadgeBackgroundColor({
      tabId: tabId,
      color: '#667eea'
    });
  } else {
    chrome.action.setBadgeText({
      tabId: tabId,
      text: ''
    });
  }
}

chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) => {
  if (changeInfo.status === 'loading') {
    chrome.action.setBadgeText({
      tabId: tabId,
      text: ''
    });
  }
});