<?php
    $domains = Array();
    $log = Array();
    $ipv6 = file_exists("/etc/pihole/.useIPv6");

    /*******   Public Members ********/
    function getSummaryData() {
        global $ipv6;
        $domains = readInBlockList();
        $log = readInLog();
        $domains_being_blocked = count($domains) / ($ipv6 ? 2 : 1);

        $dns_queries_today = count(getDnsQueries($log));

        $ads_blocked_today = count(getBlockedQueries($log));

        $ads_percentage_today = $dns_queries_today > 0 ? ($ads_blocked_today / $dns_queries_today * 100) : 0;

        return array(
            'domains_being_blocked' => $domains_being_blocked,
            'dns_queries_today' => $dns_queries_today,
            'ads_blocked_today' => $ads_blocked_today,
            'ads_percentage_today' => $ads_percentage_today,
        );
    }

    function getOverTimeData() {
        $domains = readInBlockList();
        $log = readInLog();
        $dns_queries = getDnsQueries($log);
        $ads_blocked = getBlockedQueries($log);

        $domains_over_time = overTime($dns_queries);
        $ads_over_time = overTime($ads_blocked);
        alignTimeArrays($ads_over_time, $domains_over_time);
        return Array(
            'domains_over_time' => $domains_over_time,
            'ads_over_time' => $ads_over_time,
        );
    }

    function getTopItems() {
        $domains = readInBlockList();
        $log = readInLog();
        $dns_queries = getDnsQueries($log);
        $ads_blocked = getBlockedQueries($log);

        $topAds = topItems($ads_blocked);
        $topQueries = topItems($dns_queries, $topAds);

        return Array(
            'top_queries' => $topQueries,
            'top_ads' => $topAds,
        );
    }

    function getRecentItems($qty) {
        $log = readInLog();
        $dns_queries = getDnsQueries($log);
        return Array(
            'recent_queries' => getRecent($dns_queries, $qty)
        );
    }

    function getIpvType() {
        $log = readInLog();
        $dns_queries = getDnsQueries($log);
        $queryTypes = array();

        foreach($dns_queries as $query) {
            $info = trim(explode(": ", $query)[1]);
            $queryType = explode(" ", $info)[0];
            if (isset($queryTypes[$queryType])) {
                $queryTypes[$queryType]++;
            }
            else {
                $queryTypes[$queryType] = 1;
            }
        }

        return $queryTypes;
    }

    function getForwardDestinations() {
        $log = readInLog();
        $forwards = getForwards($log);
        $destinations = array();
        foreach ($forwards as $forward) {
            $exploded = explode(" ", trim($forward));
            $dest = $exploded[count($exploded) - 1];
            if (isset($destinations[$dest])) {
                $destinations[$dest]++;
            }
            else {
                $destinations[$dest] = 0;
            }
        }

        return $destinations;

    }

    function getQuerySources() {
        $log = readInLog();
        $dns_queries = getDnsQueries($log);
        $sources = array();
        foreach($dns_queries as $query) {
            $exploded = explode(" ", $query);
            $ip = trim($exploded[count($exploded)-1]);
            if (isset($sources[$ip])) {
                $sources[$ip]++;
            }
            else {
                $sources[$ip] = 1;
            }
        }
        return $sources;
    }

    function getAllQueries() {
        $allQueries = array("data" => array());
        $log = readInLog();
        $dns_queries = getDnsQueries($log);
        
        $fileName = '/etc/pihole/gravity.list';
        //Turn gravity.list into an array
        $lines = explode("\n", file_get_contents($fileName));
        
        //Create a new array and set domain name as index instead of value, with value as 1
        foreach(array_values($lines) as $v){
            $new_lines[trim(strstr($v, ' '))] = 1;
        }
        
        foreach ($dns_queries as $query) {
            $time = date_create(substr($query, 0, 16));
            
            $exploded = explode(" ", trim($query));
            
            //Is index of the domain name set?
            if (isset($new_lines[$exploded[6]])){
            	$extra = "Pi-holed";
            }
            else
            {
            	$extra = "OK";
            }            
            array_push($allQueries['data'], array(
                $time->format('Y-m-d\TH:i:s'),
                substr($exploded[count($exploded)-4], 6, -1),
                $exploded[count($exploded)-3],
                $exploded[count($exploded)-1],
                $extra,
            ));
        }
    
        return $allQueries;
    }

    /******** Private Members ********/
    function readInBlockList() {
        global $domains;
        return count($domains) > 1 ? $domains :
            file("/etc/pihole/gravity.list");
    }
    function readInLog() {
        global $log;
        return count($log) > 1 ? $log :
            file("/var/log/pihole.log");
    }
    function getDnsQueries($log) {
        return array_filter($log, "findQueries");
    }
    function getBlockedQueries($log) {
        return array_filter($log, "findAds");
    }
    function getForwards($log) {
        return array_filter($log, "findForwards");
    }


    function topItems($queries, $exclude = array(), $qty=10) {
        $splitQueries = array();
        foreach ($queries as $query) {
            $exploded = explode(" ", $query);
            $domain = trim($exploded[count($exploded) - 3]);
            if (!isset($exclude[$domain])) {
                if (isset($splitQueries[$domain])) {
                    $splitQueries[$domain]++;
                }
                else {
                    $splitQueries[$domain] = 1;
                }
            }
        }
        arsort($splitQueries);
        return array_slice($splitQueries, 0, $qty);
    }

    function overTime($entries) {
        $byTime = array();
        foreach ($entries as $entry) {
            $time = date_create(substr($entry, 0, 16));
            $hour = $time->format('G');

            if (isset($byTime[$hour])) {
                $byTime[$hour]++;
            }
            else {
                $byTime[$hour] = 1;
            }
        }
        return $byTime;
    }

    function alignTimeArrays(&$times1, &$times2) {
        $max = max(array(max(array_keys($times1)), max(array_keys($times2))));
        $min = min(array(min(array_keys($times1)), min(array_keys($times2))));
        
        for ($i = $min; $i <= $max; $i++) {
            if (!isset($times2[$i])) {
                $times2[$i] = 0;
            }
            if (!isset($times1[$i])) {
                $times1[$i] = 0;
            }
        }

        ksort($times1);
        ksort($times2);
    }

    function getRecent($queries, $qty){
        $recent = array();
        foreach (array_slice($queries, -$qty) as $query) {
            $queryArray = array();
            $exploded = explode(" ", $query);
            $time = date_create(substr($query, 0, 16));
            $queryArray['time'] = $time->format('h:i:s a');
            $queryArray['domain'] = trim($exploded[count($exploded) - 3]);
            $queryArray['ip'] = trim($exploded[count($exploded)-1]);
            array_push($recent, $queryArray);

        }
        return array_reverse($recent);
    }

    function findQueries($var) {
        return strpos($var, ": query[") !== false;
    }

    function findAds($var) {
        return strpos($var, "gravity.list") !== false;
    }

    function findForwards($var) {
        return strpos($var, ": forwarded") !== false;
    }

/*
    $data = array(
        'domains_being_blocked' => $domains_being_blocked,
        'dns_queries_today' => $dns_queries_today,
        'ads_blocked_today' => $ads_blocked_today,
        'ads_percentage_today' => $ads_percentage_today,
        'top_queries' => $topQueries,
        'top_ads' => $topAds,
        'domains_over_time' => $domains_over_time,
        'ads_over_time' => $ads_over_time,
        'recent_queries' => getRecent($dns_queries, 20),
    );

 */
?>