content.js

const observer = new MutationObserver((mutations) => {
    mutations.forEach((mutation) => {
      if (mutation.type === 'childList') {
        mutation.addedNodes.forEach((node) => {
          if (node.tagName === 'YTD-PLAYER') {
            checkYouTubeShort(node);
          }
        });
      }
    });
  });
  
  observer.observe(document.body, { childList: true, subtree: true });
  
  function checkYouTubeShort(playerElement) {
    const videoId = getVideoIdFromUrl(window.location.href);
    
    if (videoId) {
      chrome.runtime.sendMessage({action: 'analyzeYouTubeShort', videoId: videoId}, (response) => {
        if (response && response.block) {
          blockYouTubeShort(playerElement, response.reason);
        }
      });
    } else {
      console.warn('No valid video ID found, skipping analysis');
    }
  }
  
  function getVideoIdFromUrl(url) {
    const patterns = [
      /(?:\/shorts\/|\/watch\?v=|\/embed\/|\/v\/|\/youtu.be\/)([a-zA-Z0-9_-]{11})/,
      /^([a-zA-Z0-9_-]{11})$/
    ];
  
    for (let pattern of patterns) {
      const match = url.match(pattern);
      if (match && match[1]) {
        return match[1];
      }
    }
  
    console.warn('Could not extract video ID from URL:', url);
    return null;
  }
  
  function blockYouTubeShort(playerElement, reason) {
    playerElement.style.display = 'none';
    const message = document.createElement('div');
    message.textContent = `This Short has been blocked: ${reason}`;
    message.style.cssText = 'padding: 20px; text-align: center; background: #f0f0f0;';
    playerElement.parentNode.insertBefore(message, playerElement);
    
    const skipButton = document.createElement('button');
    skipButton.textContent = 'Next Short';
    skipButton.onclick = () => {
      const event = new KeyboardEvent('keydown', {'keyCode': 39, 'which': 39});
      document.dispatchEvent(event);
    };
    message.appendChild(skipButton);
  
    const overrideButton = document.createElement('button');
    overrideButton.textContent = 'Watch Anyway';
    overrideButton.onclick = () => {
      message.remove();
      playerElement.style.display = '';
    };
    message.appendChild(overrideButton);
  }