Merge pull request #50 from Catmaniscatlord/main

PDF detection change
do-me · Apr 30, 2024 · 534356d · 534356d
2 parents c24ed0d + 1ffe9c0
commit 534356d
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/extension/public/manifest.json b/extension/public/manifest.json
@@ -5,6 +5,7 @@
   "version": "0.0.1",
   "host_permissions": ["http://*/*", "https://*/*"],
   "permissions": [
+    "scripting",
     "activeTab",
     "storage",
     "unlimitedStorage"

diff --git a/extension/src/content/content.js b/extension/src/content/content.js
@@ -53,8 +53,8 @@ chrome.runtime.onMessage.addListener(async function(request, sender) {
         if (request.type === "getText") {
             const numChars = await fetchNumChars();
             let texts = [];
-            // hacky support for pdf
-            if (currentURL.includes('pdf')) {
+
+      if (request.contentType == "application/pdf") {
                 let textContent = await fetchAndExtractPDFText(currentURL);
                 texts = splitReadableContent(textContent, numChars);
 

diff --git a/extension/src/popup/popup.vue b/extension/src/popup/popup.vue
@@ -108,7 +108,17 @@ export default {
         chrome.runtime.onMessage.addListener(this.handleMessage);
         const [tab] = await chrome.tabs.query({active: true, currentWindow: true});
 
-        chrome.tabs.sendMessage(tab.id, {type: "getText"});
+    const [res] = await chrome.scripting.executeScript({
+      target: { tabId: tab.id },
+      func: () => {
+        return document.contentType;
+      },
+    });
+
+    chrome.tabs.sendMessage(tab.id, {
+      type: "getText",
+      contentType: res.result,
+    });
         chrome.runtime.sendMessage({type: "load"});
 
         this.results = [];