From b0df56ecb367b8943ce7808d267497619aab6467 Mon Sep 17 00:00:00 2001
From: Tyler Hill <gotyhill@gmail.com>
Date: Wed, 3 Apr 2024 20:30:51 -0500
Subject: [PATCH] realBrowser

---
 package.json             |  2 +-
 src/background.ts        | 30 ++++++++++++++++--------------
 src/content.ts           |  7 +++++--
 src/pages/CoursePage.tsx |  1 +
 4 files changed, 23 insertions(+), 17 deletions(-)

diff --git a/package.json b/package.json
index 703f6f6..0b47a8a 100644
--- a/package.json
+++ b/package.json
@@ -6,7 +6,7 @@
   "author": "Nebula Labs",
   "packageManager": "npm@8.19.2",
   "scripts": {
-    "dev": "plasmo dev",
+    "dev": "plasmo dev --target=firefox-mv3",
     "build": "plasmo build",
     "package": "plasmo package",
     "format": "prettier --write .",
diff --git a/src/background.ts b/src/background.ts
index ed7028f..508a0b1 100644
--- a/src/background.ts
+++ b/src/background.ts
@@ -14,15 +14,17 @@ let scrapedCourseData: ShowCourseTabPayload = null;
 // for persistent state
 const storage = new Storage();
 
+const realBrowser = process.env.PLASMO_BROWSER === 'chrome' ? chrome : browser;
+
 /** Injects the content script if we hit a course page */
-chrome.webNavigation.onHistoryStateUpdated.addListener((details) => {
+realBrowser.webNavigation.onHistoryStateUpdated.addListener((details) => {
   if (
     /^.*:\/\/utdallas\.collegescheduler\.com\/terms\/.*\/courses\/.+$/.test(
       details.url,
     )
   ) {
     //Scrape data
-    chrome.scripting.executeScript(
+    realBrowser.scripting.executeScript(
       {
         target: {
           tabId: details.tabId,
@@ -40,28 +42,28 @@ chrome.webNavigation.onHistoryStateUpdated.addListener((details) => {
       },
     );
     //Listen for table change to rescrape data
-    chrome.tabs.sendMessage(details.tabId, 'disconnectObserver');
-    chrome.scripting.executeScript({
+    realBrowser.tabs.sendMessage(details.tabId, 'disconnectObserver');
+    realBrowser.scripting.executeScript({
       target: {
         tabId: details.tabId,
       },
       func: listenForTableChange,
     });
     //Store tab info
-    chrome.action.setBadgeText({ text: '!' });
-    chrome.action.setBadgeBackgroundColor({ color: 'green' });
+    realBrowser.action.setBadgeText({ text: '!' });
+    realBrowser.action.setBadgeBackgroundColor({ color: 'green' });
     courseTabId = details.tabId;
     storage.set('courseTabId', courseTabId);
     storage.set('courseTabUrl', details.url);
   } else {
-    chrome.action.setBadgeText({ text: '' });
+    realBrowser.action.setBadgeText({ text: '' });
   }
 });
 
 /** Rescrape data on table change */
-chrome.runtime.onMessage.addListener(function (message) {
+realBrowser.runtime.onMessage.addListener(function (message) {
   if (message === 'tableChange') {
-    chrome.scripting.executeScript(
+    realBrowser.scripting.executeScript(
       {
         target: {
           tabId: courseTabId,
@@ -80,14 +82,14 @@ chrome.runtime.onMessage.addListener(function (message) {
 });
 
 /** Sets the icon to be active if we're on a course tab */
-chrome.tabs.onActivated.addListener(async () => {
+realBrowser.tabs.onActivated.addListener(async () => {
   const cachedTabUrl: string = await storage.get('courseTabUrl');
   const currentTabUrl: string = (await getCurrentTab()).url;
   if (cachedTabUrl === currentTabUrl) {
-    chrome.action.setBadgeText({ text: '!' });
-    chrome.action.setBadgeBackgroundColor({ color: 'green' });
+    realBrowser.action.setBadgeText({ text: '!' });
+    realBrowser.action.setBadgeBackgroundColor({ color: 'green' });
   } else {
-    chrome.action.setBadgeText({ text: '' });
+    realBrowser.action.setBadgeText({ text: '' });
   }
 });
 
@@ -103,6 +105,6 @@ export async function getScrapedCourseData() {
 async function getCurrentTab() {
   const queryOptions = { active: true, lastFocusedWindow: true };
   // `tab` will either be a `tabs.Tab` instance or `undefined`.
-  const [tab] = await chrome.tabs.query(queryOptions);
+  const [tab] = await realBrowser.tabs.query(queryOptions);
   return tab;
 }
diff --git a/src/content.ts b/src/content.ts
index 77fb366..6a64426 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -44,6 +44,7 @@ export async function scrapeCourseData() {
   async function getCourseInfo(): Promise<CourseHeader> {
     const course = await waitForElement('h1');
     const courseData = course.innerText.split(' ');
+    console.log({ subjectPrefix: courseData[0], courseNumber: courseData[1] });
     return { subjectPrefix: courseData[0], courseNumber: courseData[1] };
   }
 
@@ -93,10 +94,12 @@ export async function scrapeCourseData() {
       // collapse section details
       sectionDetailsButton.click();
     });
+    console.log([...new Set(professors)]);
     return [...new Set(professors)];
   }
 }
 
+const realBrowser = process.env.PLASMO_BROWSER === 'chrome' ? chrome : browser;
 /** This listens for clicks on the buttons that switch between the enabled and disabled professor tabs and reports back to background.ts */
 export function listenForTableChange() {
   const observer = new MutationObserver((mutationsList) => {
@@ -107,7 +110,7 @@ export function listenForTableChange() {
       ) {
         //button corresponding to shown table is given an active class
         if (mutation.target.classList.contains('active')) {
-          chrome.runtime.sendMessage('tableChange');
+          realBrowser.runtime.sendMessage('tableChange');
         }
       }
     }
@@ -117,7 +120,7 @@ export function listenForTableChange() {
     subtree: true,
   });
   //remove observer when ordered by backgroud.ts to avoid duplicates
-  chrome.runtime.onMessage.addListener(function (message) {
+  realBrowser.runtime.onMessage.addListener(function (message) {
     if (message === 'disconnectObserver') {
       observer.disconnect();
     }
diff --git a/src/pages/CoursePage.tsx b/src/pages/CoursePage.tsx
index eac9321..705007a 100644
--- a/src/pages/CoursePage.tsx
+++ b/src/pages/CoursePage.tsx
@@ -28,6 +28,7 @@ export const CoursePage = () => {
     if (!state) {
       setOnCoursebook(true);
       getCourseData().then((payload) => {
+        console.log(payload);
         if (payload === null) {
           setOnCoursebook(false);
         } else {