Extension_WebBlocker/background.js at master · krish1000/Extension_WebBlocker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// //connect to html button
// const url1 = document.getElementById("submit");

// //event listeners
// url1.addEventListener("click", function () {
//   getUrl();
// });

// //fetches url
// function getUrl() {
//   console.log("fetched url sucessfully");
//   var urlString = document.getElementById("wlist").value;
//   console.log(urlString);
//   //document.getElementById("desc").innerHTML = "asdf";
//   whitelist(urlString);
// }

// function whitelist(urlString) {
//   //   chrome.runtime.sendMessage({ greeting: "hello" }, function (response) {
//   //     console.log(response.farewell);
//   //   });

//   chrome.webRequest.onBeforeRequest.addListener(
//     function (details) {
//       return {
//         cancel: details.url.indexOf("://docs.google.com/" != -1),
//       };
//     },
//     { urls: ["<all_urls>"] },
//     ["blocking"]
//   );
//   console.log("whitelisted.");
// }

// whitelist();
////////////////////////BLACK LIST
function blockRequest(details) {
  return { cancel: true }; //enables cancel:true, when used in blacklistB, makes it always blacklist
}

//stores urls that get blacklisted
var arr = [];

function blacklistB(urls) {
  //refresh listeners (apending to the array in the next instruction)
  if (chrome.webRequest.onBeforeRequest.hasListener(blockRequest)) {
    chrome.webRequest.onBeforeRequest.removeListener(blockRequest);
  }
  chrome.webRequest.onBeforeRequest.addListener(blockRequest, { urls: arr }, [
    "blocking",
  ]);

  //This is not catching error when user inputs malformed url.
  // if (chrome.runtime.lastError) {
  //   arr.splice(arr.length - 1, 1); //1 means remove, and this removes last index that gave the error.
  //   console.log("caught error");
  //   alert("Please input valid urls");
  // } else {
  //   console.log("no errors");
  // }

  console.log("blacklistB called.");
}

chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
  sendResponse({ reply: "sucess call to background.js" });
  arr.push(request.link1);
  blacklistB(); //calls method to block the url
});

//Unfortunately this listener isn't catching the errors for when user inputs a malformed url.
// chrome.webRequest.onErrorOccurred.addListener(
//   function (callback) {
//     arr.splice(arr.length - 1, 1); //1 means remove, and this removes last index that gave the error.
//     console.log("caught error");
//   },
//   { urls: ["<all_urls>"] }
// );

//////////////////////////////BLACK LIST^

/** White listing doesn't properly work, as it disables script links as well*/
// function whitelistB(urls) {
//   chrome.webRequest.onBeforeRequest.addListener(
//     function (details) {
//       return {
//         cancel:
//           details.url.indexOf(".facebook.com") == -1 &&
//           details.url.indexOf(".youtube.com") == -1,
//       };
//     },
//     { urls: ["<all_urls>"] },
//     ["blocking"]
//   );
// }
// chrome.runtime.onMessage.addListener(function (request, sender, sendResponse) {
//   //arr.push(request.link1);
//   //blacklistAll();
//   whitelistB();
// });

console.log("background.js running");

// console.log(
//   sender.tab
//     ? "from a content script:" + sender.tab.url
//     : "from the extension"
// );