arno/pi-hole
1
0
Fork 0
mirror of https://github.com/pi-hole/pi-hole synced 2024-12-22 14:58:08 +00:00
Code Issues Releases Wiki Activity

Store regex in gravity.db as well

Browse Source 
Signed-off-by: DL6ER <dl6er@dl6er.de>
This commit is contained in:
DL6ER 2019-02-03 15:13:18 +01:00
parent dcf0a605cf
commit 93f1859bab
No known key found for this signature in database
GPG Key ID: FB60471F0575164A
2 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
advanced/Templates/gravity.db.schema
View File

@ -1,5 +1,6 @@
CREATE TABLE whitelist (domain TEXT UNIQUE NOT NULL, disabled BOOLEAN DEFAULT 0, comment TEXT, DateAdded DATETIME);
CREATE TABLE blacklist (domain TEXT UNIQUE NOT NULL, disabled BOOLEAN DEFAULT 0, comment TEXT, DateAdded DATETIME);
CREATE TABLE regex (domain TEXT UNIQUE NOT NULL, disabled BOOLEAN DEFAULT 0, comment TEXT, DateAdded DATETIME);
CREATE TABLE gravity (domain TEXT UNIQUE NOT NULL);
CREATE VIEW vw_gravity AS SELECT DISTINCT a.domain
@ -14,3 +15,7 @@ WHERE a.disabled != 1 AND
CREATE VIEW vw_whitelist AS SELECT DISTINCT a.domain
FROM whitelist a
WHERE a.disabled != 1;
CREATE VIEW vw_regex AS SELECT DISTINCT a.domain
FROM regex a
WHERE a.disabled != 1;

20
gravity.sh
View File

@ -124,7 +124,7 @@ gravity_store_in_database() {
gravity_Cleanup "error"
fi
if [ "$table" == "whitelist" ] || [ "$table" == "blacklist" ]; then
if [ "$table" == "whitelist" ] || [ "$table" == "blacklist" ] || [ "$table" == "regex" ]; then
# Set disabled to false
output=$( { sqlite3 "${gravityDBfile}" <<< "UPDATE ${table} SET disabled = 0 WHERE disabled IS NULL;"; } 2>&1 )
status="$?"
@ -135,13 +135,16 @@ gravity_store_in_database() {
fi
fi
# Empty $template if it already exists, otherwise, create it
output=$( { : > "${template}"; } 2>&1 )
status="$?"
# Only create template file if asked to
if [ "${template}" != "-" ]; then
# Empty $template if it already exists, otherwise, create it
output=$( { : > "${template}"; } 2>&1 )
status="$?"
if [[ "${status}" -ne 0 ]]; then
echo -e "\\n ${CROSS} Unable to create empty ${template}\\n ${output}"
gravity_Cleanup "error"
if [[ "${status}" -ne 0 ]]; then
echo -e "\\n ${CROSS} Unable to create empty ${template}\\n ${output}"
gravity_Cleanup "error"
fi
fi
}
@ -586,6 +589,9 @@ gravity_ShowBlockCount() {
num=$(grep -cv "^#" "${regexFile}")
echo -e " ${INFO} Number of regex filters: ${num}"
fi
# Store regex files in gravity database
gravity_store_in_database "regex" "${regexFile}" "-"
}
# Parse list of domains into hosts format