background.js

chrome.runtime.onInstalled.addListener(() => {
  chrome.storage.local.set({
    enabled: true,
    threshold: 0.5,
    categories: ['Porn', 'Hentai', 'Sexy'],
    disabledDomains: []
  });
});

chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
  if (request.action === 'getSettings') {
    chrome.storage.local.get(['enabled', 'threshold', 'categories', 'disabledDomains'], (result) => {
      sendResponse(result);
    });
    return true;
  }
  
  if (request.action === 'updateSettings') {
    chrome.storage.local.set(request.settings, () => {
      sendResponse({ success: true });
    });
    return true;
  }
  
  if (request.action === 'fetchImage') {
    fetch(request.url)
      .then(response => response.blob())
      .then(blob => {
        if (!blob.type || !blob.type.startsWith('image/')) {
          sendResponse({ error: 'Non-image content' });
          return;
        }
        const reader = new FileReader();
        reader.onloadend = () => {
          sendResponse({ dataUri: reader.result });
        };
        reader.onerror = () => {
          sendResponse({ error: 'Failed to read blob' });
        };
        reader.readAsDataURL(blob);
      })
      .catch(error => {
        sendResponse({ error: error.message });
      });
    return true;
  }
});

chrome.action.onClicked.addListener((tab) => {
  chrome.action.openPopup();
});