From 7ab1ca6ae063356ef10fb163923a7b19c9f66431 Mon Sep 17 00:00:00 2001
From: jaketothepast <windlejacob12@gmail.com>
Date: Wed, 13 Feb 2019 16:26:47 -0500
Subject: [PATCH] Added additional query to block hosts in blocked file

---
 lib/db.js | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/lib/db.js b/lib/db.js
index e89928e..9dcd3d3 100644
--- a/lib/db.js
+++ b/lib/db.js
@@ -1,6 +1,7 @@
 var sqlite3 = require('sqlite3').verbose();
 var db = new sqlite3.Database("./proxy.db");
 var winston = require("winston");
+var url = require("url");
 
 var fs = require('fs');
 
@@ -97,6 +98,17 @@ function addHostToTable(hostname) {
  */
 exports.visitHost = function (hostname, callbackFn) {
     return db.serialize(() => {
+
+        // let host = new url.URL(hostname).hostname
+        // console.log(hostname)
+
+        db.get("select * from blocked_hosts where hostname = ?", [hostname], (err, row) => {
+            if (row !== undefined) {
+                logger.info("Blocked " + hostname + " due to blocked.json")
+                callbackFn(false)
+            }
+        })
+
         db.get("select * from hosts where hostname = ?", [hostname], (err, row) => {
             if (row === undefined) {
                 addHostToTable(hostname);