show dropdown list ofall companies

zubyj · zubyj · commit 9758406960a3 · 2023-08-05T00:07:19.000-07:00
diff --git a/src/problems-by-company/company.html b/src/problems-by-company/company.html
@@ -9,6 +9,7 @@
 <body>
     <nav id="navbar">
     </nav>
+    <select id="companySelect"></select>
     <h1 id="title">Amazon</h1>
     <p>
         <span id="score">Score</span> = How likely the company is to ask this question in an interview on a scale of
diff --git a/src/problems-by-company/company.ts b/src/problems-by-company/company.ts
@@ -5,7 +5,7 @@ const companies = [
     'Adobe', 'Apple', 'Bloomberg', 'Cisco', 'Facebook', 'Google', 'Microsoft', 'Spotify'
 ];
 
-function main() {
+async function main() {
     chrome.storage.local.get('clickedCompany', function (data: { [key: string]: any; }) {
         companyName = data.clickedCompany;
         const title: HTMLElement | null = document.getElementById('title');
@@ -19,6 +19,7 @@ function main() {
     document.getElementById('Score')?.addEventListener('click', () => sortBy('Score'));
 
     addNavbarLinks();
+    await addCompaniesToSelect();
 }
 
 function addNavbarLinks() {
@@ -61,11 +62,10 @@ function addNavbarLinks() {
         button.appendChild(companyName);
 
         navbar?.appendChild(button);
+
     });
 }
 
-
-
 interface Company {
     name: string;
     score: number;
@@ -127,6 +127,39 @@ function addCompanyProblems(sortMethod: string) {
     });
 }
 
+async function addCompaniesToSelect() {
+    const companySelect = document.getElementById('companySelect') as HTMLSelectElement;
+
+    let uniqueCompanies = new Set<string>();
+
+    const data = await new Promise<{ leetcodeProblems: LeetcodeProblems }>(resolve => {
+        chrome.storage.local.get('leetcodeProblems', function (items: { [key: string]: any; }) {
+            resolve(items as { leetcodeProblems: LeetcodeProblems });
+        });
+    });
+
+    data.leetcodeProblems.questions.forEach((question: Question) => {
+        if (question.companies) {
+            question.companies.forEach((company: Company) => {
+                uniqueCompanies.add(company.name);
+            });
+        }
+    });
+
+    uniqueCompanies.forEach((company) => {
+        const option = document.createElement('option');
+        option.value = company;
+        option.text = company;
+        companySelect.appendChild(option);
+    });
+
+    companySelect.addEventListener('change', () => {
+        chrome.storage.local.set({ clickedCompany: companySelect.value }, () => {
+            location.reload();
+        });
+    });
+}
+
 function sortBy(column: string) {
     if (column === 'Score') {
         solutions.sort((a, b) => b.score - a.score);
