option to blacklist sites (#69)

gautamkrishnar · Oct 30, 2021 · 727def7 · 727def7
1 parent 37ad89a
commit 727def7
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 0 deletions.
diff --git a/app/_locales/en/messages.json b/app/_locales/en/messages.json
@@ -42,5 +42,13 @@
   "keyInputPlaceholder": {
     "message": "Motrix API Key",
     "description": "Motrix API Key"
+  },
+  "blacklist": {
+    "message": "Blacklist",
+    "description": "Blacklist"
+  },
+  "saveBlacklist": {
+    "message": "Save blacklist",
+    "description": "Save blacklist"
   }
 }
diff --git a/app/scripts/background.js b/app/scripts/background.js
@@ -229,20 +229,31 @@ function downloadAgent() {
       'enableNotifications',
       'enableDownloadPrompt',
       'minFileSize',
+      'blacklist',
     ]);
 
     getResult.then(async (result) => {
       if (!result.extensionStatus) {
         // Extension is disabled
         return;
       }
+
       if (
         downloadItem.fileSize > 0 &&
         downloadItem.fileSize < result.minFileSize * 1024 * 1024
       ) {
         // File size is known and it is smaller than the minimum file size (in mb)
         return;
       }
+
+      // If url is on the blacklist then skip
+      if (
+        result.blacklist
+          .map((x) => downloadItem.url.includes(x))
+          .reduce((prev, curr) => prev || curr, false)
+      ) {
+        return;
+      }
       if (!result.motrixAPIkey) {
         // API KEY is not set, triggers an alert to the user
         alert(

diff --git a/app/scripts/config.js b/app/scripts/config.js
@@ -17,6 +17,7 @@ function ConfigView() {
   const [enableNotifications, setEnableNotifications] = useState(false);
   const [enableDownloadPrompt, setEnableDownloadPrompt] = useState(false);
   const [minFileSize, setMinFileSize] = useState('');
+  const [blacklist, setBlacklist] = useState([]);
 
   useEffect(() => {
     browser.storage.sync
@@ -26,6 +27,7 @@ function ConfigView() {
         'enableNotifications',
         'enableDownloadPrompt',
         'minFileSize',
+        'blacklist',
       ])
       .then(
         (result) => {
@@ -65,6 +67,13 @@ function ConfigView() {
           } else {
             setEnableDownloadPrompt(result.enableDownloadPrompt);
           }
+
+          if (typeof result.blacklist === 'undefined') {
+            browser.storage.sync.set({ blacklist: [] });
+            setBlacklist([]);
+          } else {
+            setBlacklist(result.blacklist);
+          }
         },
         (error) => {
           console.error(`Error: ${error}`);
@@ -170,6 +179,33 @@ function ConfigView() {
             />
           </Box>
         </Grid>
+
+        {/* Blacklist */}
+        <Grid item xs={8}>
+          <TextField
+            label="__MSG_blacklist__"
+            multiline
+            fullWidth
+            rows={4}
+            value={blacklist.join('\n')}
+            onChange={(e) => setBlacklist(e.target.value.split('\n'))}
+          />
+        </Grid>
+        {/* Save blacklist button */}
+        <Grid item xs={6} />
+        <Grid item xs={2}>
+          <Button
+            variant="outlined"
+            style={{ width: '100%', height: '56px' }}
+            onClick={() =>
+              browser.storage.sync.set({
+                blacklist: blacklist.filter((x) => x !== ''),
+              })
+            }
+          >
+            __MSG_saveBlacklist__
+          </Button>
+        </Grid>
       </Grid>
     </Container>
   );