From 0c5185f8ba0b658a8de6c2d6c9705fc54e4b0f65 Mon Sep 17 00:00:00 2001 From: DL6ER Date: Wed, 4 Dec 2019 21:02:46 +0000 Subject: [PATCH] Also display how many unique domains we have caught in the event horizon. Signed-off-by: DL6ER --- gravity.sh | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/gravity.sh b/gravity.sh index 3225b28d..72f39ad4 100755 --- a/gravity.sh +++ b/gravity.sh @@ -529,19 +529,24 @@ gravity_ParseFileIntoDomains() { gravity_Table_Count() { local table="${1}" local str="${2}" - local extra="${3}" local num - num="$(sqlite3 "${gravityDBfile}" "SELECT COUNT(*) FROM ${table} ${extra};")" - echo -e " ${INFO} Number of ${str}: ${num}" + num="$(sqlite3 "${gravityDBfile}" "SELECT COUNT(*) FROM ${table};")" + if [[ "${table}" == "vw_gravity" ]]; then + local unique + unique="$(sqlite3 "${gravityDBfile}" "SELECT COUNT(DISTINCT domain) FROM ${table};")" + echo -e " ${INFO} Number of ${str}: ${num} (${unique} unique domains)" + else + echo -e " ${INFO} Number of ${str}: ${num}" + fi } # Output count of blacklisted domains and regex filters gravity_ShowCount() { - gravity_Table_Count "gravity" "gravity domains" "" - gravity_Table_Count "blacklist" "exact blacklisted domains" "WHERE enabled = 1" - gravity_Table_Count "regex_blacklist" "regex blacklist filters" "WHERE enabled = 1" - gravity_Table_Count "whitelist" "exact whitelisted domains" "WHERE enabled = 1" - gravity_Table_Count "regex_whitelist" "regex whitelist filters" "WHERE enabled = 1" + gravity_Table_Count "vw_gravity" "gravity domains" "" + gravity_Table_Count "vw_blacklist" "exact blacklisted domains" + gravity_Table_Count "vw_regex_blacklist" "regex blacklist filters" + gravity_Table_Count "vw_whitelist" "exact whitelisted domains" + gravity_Table_Count "vw_regex_whitelist" "regex whitelist filters" } # Parse list of domains into hosts format