From 8a91e3cfc1eb588756bc053763f3b1a751076381 Mon Sep 17 00:00:00 2001
From: Tyler Hill <gotyhill@gmail.com>
Date: Thu, 4 Jan 2024 02:09:59 -0600
Subject: [PATCH] Initial

---
 package.json      |  3 ++-
 src/background.ts | 33 ++++++++++++++++++++++++++++++++-
 src/content.ts    | 25 +++++++++++++++++++++++++
 3 files changed, 59 insertions(+), 2 deletions(-)

diff --git a/package.json b/package.json
index fd357c1..703f6f6 100644
--- a/package.json
+++ b/package.json
@@ -100,7 +100,8 @@
       "storage",
       "scripting",
       "webNavigation",
-      "tabs"
+      "tabs",
+      "nativeMessaging"
     ],
     "host_permissions": [
       "https://utdallas.collegescheduler.com/terms/*/courses/*",
diff --git a/src/background.ts b/src/background.ts
index 49b537c..51e89c9 100644
--- a/src/background.ts
+++ b/src/background.ts
@@ -1,6 +1,6 @@
 import { Storage } from '@plasmohq/storage';
 
-import { CourseHeader, scrapeCourseData } from '~content';
+import { CourseHeader, listenForTableChange, scrapeCourseData } from '~content';
 
 export interface ShowCourseTabPayload {
   header: CourseHeader;
@@ -16,6 +16,7 @@ const storage = new Storage();
 
 /** Injects the content script if we hit a course page */
 chrome.webNavigation.onHistoryStateUpdated.addListener((details) => {
+  console.log('hello');
   if (
     /^.*:\/\/utdallas\.collegescheduler\.com\/terms\/.*\/courses\/.+$/.test(
       details.url,
@@ -38,6 +39,13 @@ chrome.webNavigation.onHistoryStateUpdated.addListener((details) => {
         }
       },
     );
+    chrome.tabs.sendMessage(details.tabId, 'disconnect');
+    chrome.scripting.executeScript({
+      target: {
+        tabId: details.tabId,
+      },
+      func: listenForTableChange,
+    });
     chrome.action.setBadgeText({ text: '!' });
     chrome.action.setBadgeBackgroundColor({ color: 'green' });
     courseTabId = details.tabId;
@@ -48,6 +56,29 @@ chrome.webNavigation.onHistoryStateUpdated.addListener((details) => {
   }
 });
 
+chrome.runtime.onMessage.addListener(function (message) {
+  if (message === 'tableChange') {
+    console.log('hi');
+    chrome.scripting.executeScript(
+      {
+        target: {
+          tabId: courseTabId,
+        },
+        // content script injection only works reliably on the prod packaged extension
+        // b/c of the plasmo dev server connections
+        func: scrapeCourseData,
+      },
+      async function (resolve) {
+        if (resolve && resolve[0] && resolve[0].result) {
+          const result: ShowCourseTabPayload = resolve[0].result;
+          scrapedCourseData = result;
+          await storage.set('scrapedCourseData', scrapedCourseData);
+        }
+      },
+    );
+  }
+});
+
 /** Sets the icon to be active if we're on a course tab */
 chrome.tabs.onActivated.addListener(async () => {
   const cachedTabUrl: string = await storage.get('courseTabUrl');
diff --git a/src/content.ts b/src/content.ts
index 25e0e78..2fb3b45 100644
--- a/src/content.ts
+++ b/src/content.ts
@@ -91,3 +91,28 @@ export async function scrapeCourseData() {
     return [...new Set(professors)];
   }
 }
+
+export async function listenForTableChange() {
+  const observer = new MutationObserver((mutationsList) => {
+    for (const mutation of mutationsList) {
+      if (
+        mutation.type === 'attributes' &&
+        mutation.attributeName === 'class'
+      ) {
+        if (mutation.target.classList.contains('active')) {
+          console.log(mutation.target.innerText.split(' ')[0]);
+          chrome.runtime.sendMessage('tableChange');
+        }
+      }
+    }
+  });
+  observer.observe(document.body, {
+    attributes: true,
+    subtree: true,
+  });
+  chrome.runtime.onMessage.addListener(function (message) {
+    if (message === 'disconnect') {
+      observer.disconnect();
+    }
+  });
+}