content.js

// content.js - Fixed version
let lastUrl = location.href;
let processedPrompts = new Set();
let isInitialized = false;

function sendMessageSafely(message, callback) {
  try {
    chrome.runtime.sendMessage(message, (response) => {
      if (chrome.runtime.lastError) {
        console.log('Message error:', chrome.runtime.lastError.message);
      } else if (callback) {
        callback(response);
      }
    });
  } catch (error) {
    console.log('Send message failed:', error);
  }
}

function detectAndSendPrompt() {
  // Try multiple selectors for ChatGPT user messages
  const selectors = [
    '[data-message-author-role="user"] .whitespace-pre-wrap',
    '[data-message-author-role="user"] div[class*="markdown"]',
    '.group\\/conversation-turn[data-author="user"] .whitespace-pre-wrap',
    'div[class*="user"] div[class*="markdown"]',
    'div[class*="from-user"] .whitespace-pre-wrap'
  ];

  for (const selector of selectors) {
    const elements = document.querySelectorAll(selector);
    elements.forEach(element => {
      const text = element.textContent?.trim();
      if (text && text.length > 5 && !processedPrompts.has(text)) {
        processedPrompts.add(text);
        console.log('Found prompt:', text.substring(0, 50) + '...');
        sendMessageSafely({ action: 'newPrompt', prompt: text });
      }
    });
  }
}

function startObserving() {
  const targetNode = document.body;
  const observer = new MutationObserver((mutations) => {
    let shouldCheck = false;
    
    mutations.forEach((mutation) => {
      if (mutation.addedNodes.length > 0) {
        mutation.addedNodes.forEach((node) => {
          if (node.nodeType === 1 && node.textContent?.length > 0) {
            shouldCheck = true;
          }
        });
      }
    });
    
    if (shouldCheck) {
      setTimeout(detectAndSendPrompt, 1000);
    }
  });

  observer.observe(targetNode, {
    childList: true,
    subtree: true
  });
}

// Handle scroll requests
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
  if (message.action === 'scrollToPrompt') {
    const prompt = message.prompt;
    const selectors = [
      '[data-message-author-role="user"] .whitespace-pre-wrap',
      '[data-message-author-role="user"] div[class*="markdown"]',
      '.group\\/conversation-turn[data-author="user"] .whitespace-pre-wrap'
    ];

    for (const selector of selectors) {
      const elements = document.querySelectorAll(selector);
      for (const element of elements) {
        if (element.textContent?.trim() === prompt) {
          element.scrollIntoView({ behavior: 'smooth', block: 'center' });
          sendResponse({ success: true });
          return;
        }
      }
    }
    sendResponse({ success: false });
  }
  return true;
});

// Monitor URL changes
function checkUrlChange() {
  const currentUrl = location.href;
  if (currentUrl !== lastUrl) {
    lastUrl = currentUrl;
    processedPrompts.clear();
    sendMessageSafely({ action: 'chatChanged', newUrl: currentUrl });
  }
}

// Initialize
function initialize() {
  if (isInitialized) return;
  isInitialized = true;
  
  console.log('ChatGPT Prompt Recorder initialized');
  detectAndSendPrompt();
  startObserving();
  setInterval(checkUrlChange, 2000);
}

// Start when ready
if (document.readyState === 'loading') {
  document.addEventListener('DOMContentLoaded', initialize);
} else {
  setTimeout(initialize, 1000);
}