BetterSEQTA-Plus/src/plugins/built-in/notificationCollector/index.ts

import type { Plugin } from '../../core/types';

interface NotificationCollectorStorage {
  lastNotificationCount: number;
  lastCheckedTime: string;
}

const notificationCollectorPlugin: Plugin<{}, NotificationCollectorStorage> = {
  id: 'notificationCollector',
  name: 'Notification Collector',
  description: 'Collects and displays SEQTA notifications',
  version: '1.0.0',
  settings: {},
  disableToggle: true,

  run: async (api) => {
    let pollInterval: number | null = null;

    // Store last notification count in storage
    if (!api.storage.lastNotificationCount) {
      api.storage.lastNotificationCount = 0;
    }

    const checkNotifications = async () => {
      try {
        const alertDiv = document.querySelector("[class*='notifications__bubble___']") as HTMLElement;

        if (api.storage.lastNotificationCount !== 0) {
          alertDiv.textContent = api.storage.lastNotificationCount.toString();
        }
        
        const response = await fetch(`${location.origin}/seqta/student/heartbeat?`, {
          method: 'POST',
          headers: {
            'Content-Type': 'application/json; charset=utf-8'
          },
          body: JSON.stringify({
            timestamp: "1970-01-01 00:00:00.0",
            hash: "#?page=/home",
          })
        });

        const data = await response.json();
        
        // Store notification count for history
        const notificationCount = data.payload.notifications.length;
        api.storage.lastNotificationCount = notificationCount;
        api.storage.lastCheckedTime = new Date().toISOString();
        
        if (alertDiv) {
          alertDiv.textContent = notificationCount.toString();
        } else {
          console.info("[BetterSEQTA+] No notifications currently");
        }
      } catch (error) {
        console.error("[BetterSEQTA+] Error fetching notifications:", error);
      }
    };

    const startPolling = () => {
      if (pollInterval) return; // Already polling
      checkNotifications();
      pollInterval = window.setInterval(checkNotifications, 30000);
    };

    const stopPolling = () => {
      if (pollInterval) {
        window.clearInterval(pollInterval);
        pollInterval = null;
        const alertDiv = document.querySelector("[class*='notifications__bubble___']") as HTMLElement;
        if (alertDiv) {
          if (api.storage.lastNotificationCount > 9) {
            alertDiv.textContent = "9+";
          } else {
            alertDiv.textContent = api.storage.lastNotificationCount.toString();
          }
        }
      }
    };

    api.seqta.onMount("[class*='notifications__bubble___']", (_) => {
      startPolling();
    });

    return () => {
      stopPolling();
    };
  }
};

export default notificationCollectorPlugin;