From 293d2e7d21c4b73c51aca34e79d98d9735be1a3d Mon Sep 17 00:00:00 2001
From: Brice Johnson <bricej13@gmail.com>
Date: Thu, 14 Jan 2016 23:26:58 -0600
Subject: [PATCH] Added top domains for queries and ads

---
 api.php | 41 +++++++++++++++++++++++++++++++++++++----
 1 file changed, 37 insertions(+), 4 deletions(-)

diff --git a/api.php b/api.php
index 0611d835..dac2a580 100644
--- a/api.php
+++ b/api.php
@@ -1,13 +1,46 @@
 <?php
-    $domains_being_blocked = exec("wc -l /etc/pihole/gravity.list | awk '{print $1}'");
-    $dns_queries_today = exec("cat /var/log/pihole.log | awk '/query/ {print $6}' | wc -l");
-    $ads_blocked_today = exec("cat /var/log/pihole.log | awk '/\/etc\/pihole\/gravity.list/ && !/address/ {print $6}' | wc -l");
+    function findQueries($var) {
+        return strpos($var, "query") != false;
+    }
+    function findAds($var) {
+        return strpos($var, "gravity.list") != false;
+    }
+
+    $domains = file("/etc/pihole/gravity.list");
+    $log = file("/var/log/pihole.log");
+    $domains_being_blocked = count($domains);
+
+    $dns_queries = array_filter($log, "findQueries");
+    $dns_queries_today = count($dns_queries);
+
+    $ads_blocked= array_filter($log, "findAds");
+    $ads_blocked_today = count($ads_blocked);
+
     $ads_percentage_today = $ads_blocked_today / $dns_queries_today * 100;
+
+    function topItems($queries) {
+        $splitQueries = array();
+        foreach ($queries as $query) {
+            $exploded = explode(" ", $query);
+            $domain = trim($exploded[5]);
+            if (isset($splitQueries[$domain])) {
+                $splitQueries[$domain] = $splitQueries[$domain] + 1;
+            }
+            else {
+                $splitQueries[$domain] = 1;
+            }
+        }
+        arsort($splitQueries);
+        return array_slice($splitQueries, 0, 10);
+    }
+
     $arr = 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
+        'ads_percentage_today' => $ads_percentage_today,
+        'top_queries' => topItems($dns_queries),
+        'top_ads' => topItems($ads_blocked),
     );
 
     header('Content-type: application/json');