update chronometer

This commit is contained in:
Tommy Huff 2016-10-24 17:41:14 -04:00
parent 44a36368a2
commit 99c3c57968

View file

@ -12,137 +12,153 @@
#Functions############################################################################################################## #Functions##############################################################################################################
piLog="/var/log/pihole.log"
gravity="/etc/pihole/gravity.list"
today=$(date "+%b %e") #move to pihole
statsUpdateJSON() {
CalcBlockedDomains() { if [[ -z "${AdminLink}" ]] ; then
CheckIPv6 AdminLink="http://127.0.0.1/admin"
if [ -e "$gravity" ]; then fi
#Are we IPV6 or IPV4? local x=$(curl -s ${AdminLink}/api.php?summaryRaw)
if [[ -n ${piholeIPv6} ]];then #check if json is valid
#We are IPV6 if echo "${x}" | python -m json.tool > /dev/null ; then
blockedDomainsTotal=$(wc -l /etc/pihole/gravity.list | awk '{print $1/2}') echo "${x}"
else else
#We are IPV4 echo "Error"
blockedDomainsTotal=$(wc -l /etc/pihole/gravity.list | awk '{print $1}') fi
fi
else
blockedDomainsTotal="Err."
fi
} }
CalcQueriesToday() { #move to pihole
if [ -e "$piLog" ];then statsBlockedDomains() {
queriesToday=$(cat "$piLog" | grep "$today" | awk '/query/ {print $6}' | wc -l) if [[ -z "${json}" ]] ; then
else json=$(statsUpdateJSON)
queriesToday="Err." fi
fi if [[ "${json}" != "Error" ]] ; then
local x=$(echo "${json}" | python -c "import sys, json; print json.load(sys.stdin)['domains_being_blocked']")
echo ${x}
else
echo "Error"
fi
} }
CalcblockedToday() { #move to pihole
if [ -e "$piLog" ] && [ -e "$gravity" ];then statsQueriesToday() {
blockedToday=$(cat ${piLog} | awk '/\/etc\/pihole\/gravity.list/ && !/address/ {print $6}' | wc -l) if [[ -z "${json}" ]] ; then
else json=$(statsUpdateJSON)
blockedToday="Err." fi
fi if [[ "${json}" != "Error" ]] ; then
local x=$(echo "${json}" | python -c "import sys, json; print json.load(sys.stdin)['dns_queries_today']")
echo ${x}
else
echo "Error"
fi
} }
CalcPercentBlockedToday() { #move to pihole
if [ "$queriesToday" != "Err." ] && [ "$blockedToday" != "Err." ]; then statsBlockedToday() {
if [ "$queriesToday" != 0 ]; then #Fixes divide by zero error :) if [[ -z "${json}" ]] ; then
#scale 2 rounds the number down, so we'll do scale 4 and then trim the last 2 zeros json=$(statsUpdateJSON)
percentBlockedToday=$(echo "scale=4; $blockedToday/$queriesToday*100" | bc) fi
percentBlockedToday=$(sed 's/.\{2\}$//' <<< "$percentBlockedToday") if [[ "${json}" != "Error" ]] ; then
else local x=$(echo "${json}" | python -c "import sys, json; print json.load(sys.stdin)['ads_blocked_today']")
percentBlockedToday=0 echo ${x}
fi else
fi echo "Error"
fi
} }
CheckIPv6() { #move to pihole
piholeIPv6file="/etc/pihole/.useIPv6" statsPercentBlockedToday() {
if [[ -f ${piholeIPv6file} ]];then if [[ -z "${json}" ]] ; then
# If the file exists, then the user previously chose to use IPv6 in the automated installer json=$(statsUpdateJSON)
piholeIPv6=$(ip -6 route get 2001:4860:4860::8888 | awk -F " " '{ for(i=1;i<=NF;i++) if ($i == "src") print $(i+1) }') fi
fi if [[ "${json}" != "Error" ]] ; then
local x=$(echo "${json}" | python -c "import sys, json; print round(float(json.load(sys.stdin)['ads_percentage_today']), 2)")
echo ${x}
else
echo "Error"
fi
} }
outputJSON() {
CalcQueriesToday
CalcblockedToday
CalcPercentBlockedToday
CalcBlockedDomains #start script
printf '{"domains_being_blocked":"%s","dns_queries_today":"%s","ads_blocked_today":"%s","ads_percentage_today":"%s"}\n' "$blockedDomainsTotal" "$queriesToday" "$blockedToday" "$percentBlockedToday" setupVars="/etc/pihole/setupVars.conf"
if [[ -f "${setupVars}" ]] ; then
. "${setupVars}"
else
echo "::: WARNING: /etc/pihole/setupVars.conf missing. Possible installation failure."
echo "::: Please run 'pihole -r', and choose the 'reconfigure' option to reconfigure."
exit 1
fi
#remove CIDR from IPv4
IPv4_address=${IPv4_address%/*}
center(){
cols=$(tput cols)
length=${#1}
center=$(expr $cols / 2)
halfstring=$(expr $length / 2 )
pad=$(expr $center + $halfstring)
printf "%${pad}s\n" "$1"
} }
normalChrono() { normalChrono() {
for (( ; ; )) for (( ; ; ))
do do
clear ## prepare all lines before clear to remove flashing
# Displays a colorful Pi-hole logo json=$(statsUpdateJSON)
echo " ___ _ _ _" load=$(uptime | cut -d' ' -f11-)
echo "| _ (_)___| |_ ___| |___" uptime=$(uptime | awk -F'( |,|:)+' '{if ($7=="min") m=$6; else {if ($7~/^day/) {d=$6;h=$8;m=$9} else {h=$6;m=$7}}} {print d+0,"days,",h+0,"hours,",m+0,"minutes."}')
echo "| _/ |___| ' \/ _ \ / -_)" list=$(statsBlockedDomains)
echo "|_| |_| |_||_\___/_\___|" hits=$(statsQueriesToday)
echo "" blocked=$(statsBlockedToday)
echo " $(ifconfig eth0 | awk '/inet addr/ {print $2}' | cut -d':' -f2)" percent=$(statsPercentBlockedToday)
echo "" #for moving functions to pihole
uptime | cut -d' ' -f11- # list=$(pihole stats list)
#uptime -p #Doesn't work on all versions of uptime # hits=$(pihole stats hits)
uptime | awk -F'( |,|:)+' '{if ($7=="min") m=$6; else {if ($7~/^day/) {d=$6;h=$8;m=$9} else {h=$6;m=$7}}} {print d+0,"days,",h+0,"hours,",m+0,"minutes."}' # blocked=$(pihole stats blocked)
echo "-------------------------------" # percent=$(pihole stats percent)
# Uncomment to continually read the log file and display the current domain being blocked clear
#tail -f /var/log/pihole.log | awk '/\/etc\/pihole\/gravity.list/ {if ($7 != "address" && $7 != "name" && $7 != "/etc/pihole/gravity.list") print $7; else;}' # Displays a colorful Pi-hole logo
echo " ___ _ _ _"
#uncomment next 4 lines to use original query count calculation echo "| _ (_)___| |_ ___| |___"
#today=$(date "+%b %e") echo "| _/ |___| ' \/ _ \ / -_)"
#todaysQueryCount=$(cat /var/log/pihole.log | grep "$today" | awk '/query/ {print $7}' | wc -l) echo "|_| |_| |_||_\___/_\___|"
#todaysQueryCountV4=$(cat /var/log/pihole.log | grep "$today" | awk '/query/ && /\[A\]/ {print $7}' | wc -l) echo ""
#todaysQueryCountV6=$(cat /var/log/pihole.log | grep "$today" | awk '/query/ && /\[AAAA\]/ {print $7}' | wc -l) center ${IPv4_address}
center ${IPv6_address}
echo "${load}"
CalcQueriesToday echo "${uptime}"
CalcblockedToday echo "-------------------------------"
CalcPercentBlockedToday echo "Blocking: ${list}"
echo "Queries: ${hits}"
CalcBlockedDomains echo "Pi-holed: ${blocked} (${percent})%"
sleep 5
echo "Blocking: $blockedDomainsTotal" done
#below commented line does not add up to todaysQueryCount
#echo "Queries: $todaysQueryCountV4 / $todaysQueryCountV6"
echo "Queries: $queriesToday" #same total calculation as dashboard
echo "Pi-holed: $blockedToday ($percentBlockedToday%)"
sleep 5
done
} }
displayHelp() {
cat << EOM function displayHelp(){
::: Displays stats about your piHole! echo "::: Displays stats about your piHole!"
::: echo ":::"
::: Usage: sudo pihole -c [optional:-j] echo "::: Usage: sudo pihole -c [optional:-j]"
::: Note: If no option is passed, then stats are displayed on screen, updated every 5 seconds echo "::: Note: If no option is passed, then stats are displayed on screen, updated every 5 seconds"
::: echo ":::"
::: Options: echo "::: Options:"
::: -j, --json output stats as JSON formatted string echo "::: -j, --json output stats as JSON formatted string"
::: -h, --help display this help text echo "::: -h, --help display this help text"
EOM exit 1
exit 1
} }
if [[ $# = 0 ]]; then if [[ $# = 0 ]]; then
normalChrono normalChrono
fi fi
for var in "$@" for var in "$@"
do do
case "$var" in case "$var" in
"-j" | "--json" ) outputJSON;; "-j" | "--json" ) statsUpdateJSON;;
"-h" | "--help" ) displayHelp;; "-h" | "--help" ) displayHelp;;
* ) exit 1;; * ) exit 1;;
esac esac