dashboard.js

document.addEventListener('DOMContentLoaded', () => {
  const listViewBtn = document.getElementById('list-view-btn');
  const calendarViewBtn = document.getElementById('calendar-view-btn');
  const listView = document.getElementById('list-view');
  const calendarView = document.getElementById('calendar-view');
  const timelineList = document.getElementById('timeline-list');
  const searchBar = document.getElementById('search-bar');
  const calendarEl = document.getElementById('calendar');
  const addTaskBtn = document.getElementById('add-task-btn');
  const taskTitleInput = document.getElementById('task-title');
  const taskContentInput = document.getElementById('task-content');

  let allNotes = [];
  let calendar;

  // 1. Load notes from local storage
  chrome.storage.local.get({ notes: [] }, (result) => {
    allNotes = result.notes.sort((a, b) => b.timestamp - a.timestamp); // Newest first
    displayNotesAsList(allNotes);
    initializeCalendar(allNotes);
  });

  // 2. Display notes in the list view
  function displayNotesAsList(notes) {
    timelineList.innerHTML = '';
    if (notes.length === 0) {
      timelineList.innerHTML = '<li>No notes saved yet. Right-click on selected text to save a note.</li>';
      return;
    }

    notes.forEach((note, index) => {
      const listItem = document.createElement('li');
      listItem.className = 'timeline-item';
      listItem.innerHTML = `
        <div class="timeline-date">${new Date(note.timestamp).toLocaleDateString()}</div>
        <div class="timeline-content">
          <h3>${note.title}</h3>
          <p>${note.content.substring(0, 200)}...</p>
          <a href="${note.url}" target="_blank" class="source-link">Go to source</a>
          <button class="remove-task-btn" data-index="${index}">Remove</button>
        </div>
      `;
      timelineList.appendChild(listItem);
    });

    // Add event listeners to remove buttons
    document.querySelectorAll('.remove-task-btn').forEach(button => {
      button.addEventListener('click', (e) => {
        const indexToRemove = parseInt(e.target.dataset.index, 10);
        removeNote(indexToRemove);
      });
    });
  }

  // 3. Initialize FullCalendar
  function initializeCalendar(notes) {
    const events = notes.map(note => ({
      title: note.title,
      start: new Date(note.timestamp),
      url: note.url
    }));

    calendar = new FullCalendar.Calendar(calendarEl, {
      initialView: 'dayGridMonth',
      events: events,
      eventClick: function(info) {
        info.jsEvent.preventDefault(); // prevent browser navigation
        if (info.event.url) {
          window.open(info.event.url, '_blank');
        }
      }
    });

    calendar.render();
  }

  // 4. Search functionality
  searchBar.addEventListener('input', (e) => {
    const query = e.target.value.toLowerCase();
    const filteredNotes = allNotes.filter(note =>
      note.title.toLowerCase().includes(query) ||
      note.content.toLowerCase().includes(query)
    );
    displayNotesAsList(filteredNotes);
  });

  // 5. View toggling
  listViewBtn.addEventListener('click', () => {
    listView.style.display = 'block';
    calendarView.style.display = 'none';
    listViewBtn.classList.add('active');
    calendarViewBtn.classList.remove('active');
  });

  calendarViewBtn.addEventListener('click', () => {
    listView.style.display = 'none';
    calendarView.style.display = 'block';
    listViewBtn.classList.remove('active');
    calendarViewBtn.classList.add('active');
    calendar.render(); // Re-render the calendar when switching to it
  });

  // 6. Add a new task
  addTaskBtn.addEventListener('click', () => {
    const title = taskTitleInput.value.trim();
    const content = taskContentInput.value.trim();

    if (title && content) {
      const newNote = {
        title: title,
        content: content,
        timestamp: new Date().getTime(),
        url: '' // No URL for manually added tasks
      };

      allNotes.unshift(newNote); // Add to the beginning of the array
      chrome.storage.local.set({ notes: allNotes }, () => {
        displayNotesAsList(allNotes);
        initializeCalendar(allNotes);
        taskTitleInput.value = '';
        taskContentInput.value = '';
      });
    }
  });

  // 7. Remove a note
  function removeNote(index) {
    allNotes.splice(index, 1);
    chrome.storage.local.set({ notes: allNotes }, () => {
      displayNotesAsList(allNotes);
      initializeCalendar(allNotes);
    });
  }
});