function getSelectedCollection() {
    const selectedIcon = document.querySelector('.selected');
    if (selectedIcon) {
      return selectedIcon.dataset.collection;
    }
    return 'misc';
  }
  
  function addBookmark(e) {
    e.preventDefault();
    const collection = getSelectedCollection();
    const bookmark = {
      url: window.location.href
    };
    let bookmarks = JSON.parse(localStorage.getItem(`bookmarks_${collection}`)) || [];
    bookmarks.push(bookmark);
    localStorage.setItem(`bookmarks_${collection}`, JSON.stringify(bookmarks));
    renderBookmarkList();
  }
  
  function initialize() {
    renderBookmarkList();
  }
  
  initialize();
  
  const bookmarkList = document.getElementById('bookmark-list');
  
  function renderBookmarkList() {
    const collections = ['train', 'safehouse', 'bank', 'dust2', 'lake', 'italy', 'misc'];
    bookmarkList.innerHTML = '';
    collections.forEach(collection => {
      const bookmarks = JSON.parse(localStorage.getItem(`bookmarks_${collection}`)) || [];
      if (bookmarks.length > 0) {
        const collectionHeading = document.createElement('h3');
        collectionHeading.textContent = collection;
        bookmarkList.appendChild(collectionHeading);
      }
      bookmarks.forEach(bookmark => {
        const bookmarkItem = document.createElement('div');
        bookmarkItem.classList.add('bookmark-item');
        bookmarkItem.innerHTML = `
          <a href="${bookmark.url}" target="_blank">${bookmark.url}</a>
        `;
        bookmarkList.appendChild(bookmarkItem);
      });
    });
  }
  
  document.getElementById('add-bookmark-form').addEventListener('submit', addBookmark);
  
  document.getElementById('view-bookmarks').addEventListener('click', function() {
    chrome.tabs.create({url: 'bookmarks.html'});
  });
  document.getElementById("viewBookmarkBtn").addEventListener("click", function() {
    window.location.href = "bookmarks.html";
  });