add amazon questions when the company tag is clicked

zubyj · zubyj · commit b1799f0e5491 · 2023-06-30T18:02:58.000-07:00
diff --git a/src/background/background.ts b/src/background/background.ts
@@ -59,6 +59,24 @@ chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
     }
 });
 
+chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
+    if (request.action === "openCompanyPage") {
+        chrome.tabs.create({
+            url: chrome.runtime.getURL("src/popup/company.html"),
+            active: true
+        }, function (tab) {
+            chrome.tabs.onUpdated.addListener(function listener(tabId, changedProps) {
+                // When the tab is done loading
+                if (tabId == tab.id && changedProps.status == "complete") {
+                    chrome.tabs.sendMessage(tabId, request);
+                    chrome.tabs.onUpdated.removeListener(listener);
+                }
+            });
+        });
+    }
+});
+
+
 chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) => {
     // If descriptions tab is opened or updated, update the description
     let urlPattern = /^https:\/\/leetcode\.com\/problems\/.*\/(description\/)?/;
diff --git a/src/content-script/update-description-tab.ts b/src/content-script/update-description-tab.ts
@@ -97,15 +97,21 @@ function loadCompanyTags(problemTitle: string, companyTagContainer: HTMLElement)
     chrome.storage.local.get(['leetcodeProblems'], (result) => {
         const problem = result.leetcodeProblems.questions.find((problem: problem) => problem.title === problemTitle);
         if (problem.companies && problem.companies.length > 0) {
-            // slice the array to get only the first five companies
             const topCompanies = problem.companies.slice(0, 5);
-
             // create a button for each company
             topCompanies.forEach((company: { name: string; score: any; }) => {
                 const button = document.createElement('button');
+                // opens the company page when the button is clicked
+                button.onclick = () => {
+                    chrome.runtime.sendMessage({
+                        // passes the company name and score to the background script
+                        action: 'openCompanyPage', company: company
+                    })
+                }
+
                 button.style.display = 'flex';
-                button.style.alignItems = 'center';  // align items vertically in the center
-                button.style.justifyContent = 'center';  // align items horizontally in the center
+                button.style.alignItems = 'center';
+                button.style.justifyContent = 'center';
 
                 const icon = document.createElement('img');
                 icon.src = `https://logo.clearbit.com/${company.name.toLowerCase().replace(/\s/g, '')}.com`; // replace spaces with nothing
diff --git a/src/popup/company.html b/src/popup/company.html
@@ -0,0 +1,20 @@
+<!DOCTYPE html>
+<html>
+
+<head>
+    <title>LeetCode Solutions</title>
+</head>
+
+<body>
+    <h1 id="test">Company tags</h1>
+    <table id="solutionTable">
+        <tr>
+            <th>Problem Number</th>
+            <th>Problem Title</th>
+            <th>Score</th>
+        </tr>
+    </table>
+    <script src="../../dist/popup/company.js"></script>
+</body>
+
+</html>
diff --git a/src/popup/company.ts b/src/popup/company.ts
@@ -0,0 +1,38 @@
+
+
+function main() {
+    chrome.storage.local.get("leetcodeProblems", function (data) {
+        const companyName = "Amazon";
+        let solutions = [];
+
+        console.log(data);
+
+        data.leetcodeProblems.questions.forEach(question => {
+            if (!question.companies) return;
+            question.companies.forEach(company => {
+                if (company.name === companyName) {
+                    solutions.push({
+                        id: question.frontend_id,
+                        title: question.title,
+                        score: company.score,
+                        url: `https://leetcode.com/problems/${question.title.replace(/\s/g, '-')}/`
+                    });
+                }
+            });
+        });
+
+        const table = document.getElementById("solutionTable");
+        solutions.forEach(solution => {
+            const row = table.insertRow(-1);
+            row.insertCell(0).innerText = solution.id;
+            const titleCell = row.insertCell(1);
+            titleCell.innerHTML = `<a href="${solution.url}" target="_blank">${solution.title}</a>`;
+            row.insertCell(2).innerText = solution.score;
+        });
+    });
+
+    document.getElementById("test")?.textContent = "Hello World!";
+}
+
+/* Run the script */
+main();
