Adding blocked_hosts table, ensure only new hosts are inserted.
This commit is contained in:
parent
b10d007b4d
commit
5f577071ee
24
lib/db.js
24
lib/db.js
@ -39,7 +39,29 @@ exports.initializeDatabase = function() {
|
|||||||
if (err !== null)
|
if (err !== null)
|
||||||
throw err
|
throw err
|
||||||
|
|
||||||
console.log(JSON.parse(data))
|
// console.log(JSON.parse(data))
|
||||||
|
let blockedData = JSON.parse(data)
|
||||||
|
db.run("create table if not exists blocked_hosts (hostname TEXT)", (err, row) => {
|
||||||
|
if (err !== null) {
|
||||||
|
logErrorAndExit("Error creating table blocked_hosts: " + err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
db.get("select count(*) from blocked_hosts", (err, row) => {
|
||||||
|
if (err !== null) {
|
||||||
|
logErrorAndExit("Error getting count of blocked_hosts")
|
||||||
|
}
|
||||||
|
// if (blockedData.hosts.length row['count(*)'])
|
||||||
|
for (let host of blockedData.hosts) {
|
||||||
|
db.get("select hostname from blocked_hosts where hostname = ?", [host], (err, row) => {
|
||||||
|
if (err !== null) {
|
||||||
|
db.run("insert into blocked_hosts (hostname) values (?)", [hostname]);
|
||||||
|
}
|
||||||
|
// TODO -- run a diff for whatever has been deleted from the config file.
|
||||||
|
})
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
})
|
})
|
||||||
} catch(err) {
|
} catch(err) {
|
||||||
console.log(err)
|
console.log(err)
|
||||||
|
Loading…
Reference in New Issue
Block a user