Added additional query to block hosts in blocked file

This commit is contained in:
Jacob Windle 2019-02-13 16:26:47 -05:00
parent d27d048d7a
commit 7ab1ca6ae0

View File

@ -1,6 +1,7 @@
var sqlite3 = require('sqlite3').verbose(); var sqlite3 = require('sqlite3').verbose();
var db = new sqlite3.Database("./proxy.db"); var db = new sqlite3.Database("./proxy.db");
var winston = require("winston"); var winston = require("winston");
var url = require("url");
var fs = require('fs'); var fs = require('fs');
@ -97,6 +98,17 @@ function addHostToTable(hostname) {
*/ */
exports.visitHost = function (hostname, callbackFn) { exports.visitHost = function (hostname, callbackFn) {
return db.serialize(() => { 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) => { db.get("select * from hosts where hostname = ?", [hostname], (err, row) => {
if (row === undefined) { if (row === undefined) {
addHostToTable(hostname); addHostToTable(hostname);