#!/usr/bin/env bash
# Pi-hole: A black hole for Internet advertisements
# (c) 2015, 2016 by Jacob Salmela
# Network-wide ad blocking via your Raspberry Pi
# http://pi-hole.net
# Calculates stats and displays to an LCD
#
# Pi-hole is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.


#Functions##############################################################################################################

#move to pihole
statsUpdateJSON() {
  if [[ -z "${AdminLink}" ]] ; then
    AdminLink="http://127.0.0.1/admin"
  fi
  local x=$(curl -s ${AdminLink}/api.php?summaryRaw)
  #check if json is valid
  if python -c "import sys, json;json.loads('${x}')" ; then
    echo "${x}"
  else
    echo "Error"
  fi
}

#move to pihole
statsBlockedDomains() {
  if [[ -z "${json}" ]] ; then
    json=$(statsUpdateJSON)
  fi
  if [[ "${json}" != "Error" ]] ; then
    local x=$(python -c "import sys, json;print(json.loads('${json}')['domains_being_blocked'])")
    echo ${x}
  else
    echo "Error"
  fi
}

#move to pihole
statsQueriesToday() {
  if [[ -z "${json}" ]] ; then
   json=$(statsUpdateJSON)
  fi
  if [[ "${json}" != "Error" ]] ; then
    local x=$(python -c "import sys, json;print(json.loads('${json}')['dns_queries_today'])")
    echo ${x}
  else
    echo "Error"
  fi
}

#move to pihole
statsBlockedToday() {
  if [[ -z "${json}" ]] ; then
    json=$(statsUpdateJSON)
  fi
  if [[ "${json}" != "Error" ]] ; then
    local x=$(python -c "import sys, json;print(json.loads('${json}')['ads_blocked_today'])")
    echo ${x}
  else
    echo "Error"
  fi
}

#move to pihole
statsPercentBlockedToday() {
  if [[ -z "${json}" ]] ; then
    json=$(statsUpdateJSON)
  fi
  if [[ "${json}" != "Error" ]] ; then
    local x=$(python -c "import sys, json;print(round(float(json.loads('${json}')['ads_percentage_today']), 2))")
    echo ${x}
  else
    echo "Error"
  fi
}


#start script

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

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() {
  for (( ; ; ))
  do
    ## prepare all lines before clear to remove flashing
    json=$(statsUpdateJSON)
    load=$(uptime | cut -d' ' -f11-)
    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."}')
    list=$(statsBlockedDomains)
    hits=$(statsQueriesToday)
    blocked=$(statsBlockedToday)
    percent=$(statsPercentBlockedToday)
#for moving functions to pihole
#    list=$(pihole stats list)
#    hits=$(pihole stats hits)
#    blocked=$(pihole stats blocked)
#    percent=$(pihole stats percent)
    clear
    # Displays a colorful Pi-hole logo
    echo " ___ _     _        _"
    echo "| _ (_)___| |_  ___| |___"
    echo "|  _/ |___| ' \/ _ \ / -_)"
    echo "|_| |_|   |_||_\___/_\___|"
    echo ""
    center ${IPv4_address}
    center ${IPv6_address}
    echo "${load}"
    echo "${uptime}"
    echo "-------------------------------"
    echo "Blocking:      ${list}"
    echo "Queries:       ${hits}"
    echo "Pi-holed:      ${blocked} (${percent})%"
    sleep 5
  done
}


function displayHelp(){
  echo "::: Displays stats about your piHole!"
  echo ":::"
  echo "::: Usage: sudo pihole -c [optional:-j]"
  echo "::: Note: If no option is passed, then stats are displayed on screen, updated every 5 seconds"
  echo ":::"
  echo "::: Options:"
  echo ":::  -j, --json		output stats as JSON formatted string"
  echo ":::  -h, --help		display this help text"
  exit 1
}

if [[ $# = 0 ]]; then
  normalChrono
fi

for var in "$@"
do
  case "$var" in
    "-j" | "--json"  ) statsUpdateJSON;;
    "-h" | "--help"  ) displayHelp;;
    *                ) exit 1;;
  esac
done