contentScript.js

function applyFocusIfEnabled() {
  // Only apply on video pages
  if (window.location.pathname === '/watch') {
    chrome.storage.sync.get('focusEnabled', (data) => {
      toggleFocusMode(!!data.focusEnabled);
    });
  }
}

function toggleFocusMode(enabled) {
  const toggle = (selector, hide) => {
    document.querySelectorAll(selector).forEach(el => {
      el.style.display = hide ? 'none' : '';
    });
  };
  toggle('#related, #comments, ytd-watch-next-secondary-results-renderer, #masthead-container, #secondary', enabled);
}

// Initial application
applyFocusIfEnabled();

// Ensure focus mode is applied after DOM is ready (for first navigation after browser start)
document.addEventListener('DOMContentLoaded', applyFocusIfEnabled);

// Listen for YouTube navigation events (SPA navigation)
window.addEventListener('yt-navigate-finish', applyFocusIfEnabled);

// Observe DOM changes to re-apply focus mode if YouTube re-renders parts of the page
const observer = new MutationObserver(() => {
  applyFocusIfEnabled();
});
observer.observe(document.body, { childList: true, subtree: true });