content.js

chrome.runtime.onMessage.addListener(function(request, sender, sendResponse) {
  if (request.action === 'exportFollowers') {
    exportData('followers', sendResponse);
    return true; // 非同期処理のためにtrueを返す
  } else if (request.action === 'exportFollowing') {
    exportData('following', sendResponse);
    return true; // 非同期処理のためにtrueを返す
  }
  // もしどっちのアクションでもなかったら…
  console.log('あれ？どっちのアクションでもないみたい…🤔', request.action);
  sendResponse({ status: 'error', message: 'Unknown action.' }); // 一応レスポンス返しとく
  return false; // 非同期処理じゃない場合はfalse
});

async function exportData(type, sendResponse) {
  let data = [];
  let listElement;

  if (type === 'followers') {
    // フォロワーリストのモーダルを開くトリガーをクリック
    const followersLink = document.querySelector('a[href$="/followers/"]');
    if (followersLink) {
      followersLink.click();
      await sleep(2000); // モーダルが開くのを待つ
    } else {
      console.error('フォロワーリンクが見つからないんだけど…😭');
      sendResponse({ status: 'error', message: 'Could not find followers link. Please navigate to your profile page.' });
      return;
    }

    // モーダル内のフォロワーリストの親要素を探す
    // クラス名が変更される可能性が高いので、開発者ツールで要確認
    const userListWrapper = document.querySelector('div[role="dialog"] div[style*="overflow: hidden auto"]');

    if (userListWrapper && userListWrapper.parentElement) {
      listElement = userListWrapper.parentElement;
      const styles = window.getComputedStyle(listElement);
    } else {
      console.error('ユーザーリストのラッパーか、その親要素が見つかんないんだけど…😱 セレクタ再確認プリーズ！');
      sendResponse({ status: 'error', message: 'Could not find the scrollable list element. Check selectors.' });
      return;
    }

  } else if (type === 'following') {
    // フォローリストのモーダルを開くトリガーをクリック
    const followingLink = document.querySelector('a[href$="/following/"]');
    if (followingLink) {
      followingLink.click();
      await sleep(2000); // モーダルが開くのを待つ
    } else {
      console.error('フォロー中リンクが見つからないんだけど…😭'); // ★デバッグログ
      sendResponse({ status: 'error', message: 'Could not find following link. Please navigate to your profile page.' });
      return;
    }

    // モーダル内のフォロー中リストの親要素を探す
    // クラス名が変更される可能性が高いので、開発者ツールで要確認
    const userListWrapper = document.querySelector('div[role="dialog"] div[style*="overflow: hidden auto"]');

    if (userListWrapper && userListWrapper.parentElement) {
      listElement = userListWrapper.parentElement;
      const styles = window.getComputedStyle(listElement);
    } else {
      console.error('ユーザーリストのラッパーか、その親要素が見つかんないんだけど…😱 セレクタ再確認プリーズ！');
      sendResponse({ status: 'error', message: 'Could not find the scrollable list element. Check selectors.' });
      return;
    }
  }

  // listElementがちゃんと取れてるかチェック！
  if (!listElement) {
    console.error('結局listElementが取れてないじゃん！😭 exportDataを中断するね…');
    return;
  }

  // 無限スクロール処理
  let previousHeight = 0;
  let currentHeight = listElement.scrollHeight;
  let scrollAttempts = 0; // スクロール試行回数をカウントする変数
  const MAX_SCROLL_ATTEMPTS = 3; // スクロールしても高さが変わらない場合の最大試行回数

  while (true) { // 基本的に無限ループにして、脱出条件を明確にする
    previousHeight = listElement.scrollHeight; // スクロール前の高さを記憶
    listElement.scrollTop = listElement.scrollHeight; // リストの一番下までスクロール
    await sleep(2000); // 新しいコンテンツが読み込まれるのをしっかり待つ

    currentHeight = listElement.scrollHeight; // スクロール後の高さを取得

    if (currentHeight === previousHeight) {
      scrollAttempts++; // 高さが変わらなかったら試行回数を増やす
      console.log(`高さ変わんない…試行回数: ${scrollAttempts}`); // ★デバッグログ
      if (scrollAttempts >= MAX_SCROLL_ATTEMPTS) {
        console.log('最大試行回数に達したから、スクロール終わりにするね！🏁'); // ★デバッグログ
        break; // もう読み込むものがなさそうなのでループを抜ける
      }
      // 念のため、さらに少し待ってから再度高さを確認する
      await sleep(2000);
      currentHeight = listElement.scrollHeight;
      if (currentHeight === previousHeight) {
      } else {
        scrollAttempts = 0; // 高さが変わったらリセット
      }
    } else {
      scrollAttempts = 0; // 高さが変わったらリセット
    }
  }

  const userElements = listElement.querySelectorAll('div.x1dm5mii.x16mil14.xiojian.x1yutycm.x1lliihq.x193iq5w.xh8yej3');

  userElements.forEach(element => {
    let displayName = '';
    let userId = '';

    // ユーザーID (プロフィールID) を取得する
    const userIdSpanElement = element.querySelector('a span._ap3a._aaco._aacw._aacx._aad7._aade');
    if (userIdSpanElement) {
      userId = userIdSpanElement.textContent.trim();
      console.log('ユーザーID (プロフィールID) を見つけたよ！:', userId, userIdSpanElement);
    } else {
      console.warn('あれれ、ユーザーID (プロフィールID) が見つからなかった…😢', element);
    }

    const displayNameSpan = element.querySelector('span.x1lliihq.x193iq5w.x6ikm8r.x10wlt62.xlyipyv.xuxw1ft');
    if (displayNameSpan && displayNameSpan.textContent.trim()) {
      displayName = displayNameSpan.textContent.trim();
      console.log('表示名を見つけたよ！:', displayName, displayNameSpan);
    } else {
      console.warn('うーん、指定したクラス名のspanが見つからないか、中身が空っぽだった…🤔', element);
    }

    if (displayName || userId) {
      data.push({ username: displayName, userId: userId });
    } else {
      console.warn('結局、表示名もIDも見つからなかった…しょぼん(´・ω・｀)', element);
    }
  });

  if (data.length > 0) {
    console.log('データあった！CSVダウンロードするね！😉', data); // ★デバッグログ
    downloadCSV(data, `${type}_list.csv`);
    sendResponse({ status: 'success' });
  } else {
    console.warn('あれ、データが空っぽだ…🤔'); // ★デバッグログ
    sendResponse({ status: 'error', message: 'No data found.' });
  }
}

// ヘルパー関数: 指定ミリ秒待機
function sleep(ms) {
  return new Promise(resolve => setTimeout(resolve, ms));
}

// CSVダウンロード関数
function downloadCSV(data, filename) {
  const csvRows = [];
  const headers = Object.keys(data[0]); // ヘッダーとしてオブジェクトのキーを使用
  csvRows.push(headers.join(',')); // ヘッダー行を追加

  for (const row of data) {
    const values = headers.map(header => {
      const escaped = ('' + row[header]).replace(/"/g, '""'); // ダブルクォートをエスケープ
      return `"${escaped}"`; // ダブルクォートで囲む
    });
    csvRows.push(values.join(','));
  }

  const csvString = csvRows.join('\n');
  const blob = new Blob([csvString], { type: 'text/csv;charset=utf-8;' });
  const url = URL.createObjectURL(blob);

  const link = document.createElement('a');
  link.setAttribute('href', url);
  link.setAttribute('download', filename);
  link.style.visibility = 'hidden';
  document.body.appendChild(link);
  link.click();
  document.body.removeChild(link);
}