From 7f7604a6af97d581144f429ed03353d682c75242 Mon Sep 17 00:00:00 2001
From: DL6ER <derigs@ph1.uni-koeln.de>
Date: Tue, 22 Nov 2016 15:23:30 +0100
Subject: [PATCH] Add "Query adlists" feature

---
 header.php            |  6 ++++++
 js/pihole/queryads.js | 36 ++++++++++++++++++++++++++++++++++++
 php/queryads.php      | 35 +++++++++++++++++++++++++++++++++++
 queryads.php          | 23 +++++++++++++++++++++++
 4 files changed, 100 insertions(+)
 create mode 100644 js/pihole/queryads.js
 create mode 100644 php/queryads.php
 create mode 100644 queryads.php

diff --git a/header.php b/header.php
index 54b4da1e..76284ef6 100644
--- a/header.php
+++ b/header.php
@@ -242,6 +242,12 @@
                         <i class="fa fa-arrow-circle-down"></i> <span>Update lists</span>
                     </a>
                 </li>
+                <!-- Query adlists -->
+                <li>
+                    <a href="queryads.php">
+                        <i class="fa fa-search"></i> <span>Query adlists</span>
+                    </a>
+                </li>
                 <!-- Toggle -->
                 <?php
                 if ($pistatus == "1") {
diff --git a/js/pihole/queryads.js b/js/pihole/queryads.js
new file mode 100644
index 00000000..d0da2412
--- /dev/null
+++ b/js/pihole/queryads.js
@@ -0,0 +1,36 @@
+function eventsource() {
+    var ta = $("#output");
+    var domain = $("#domain");
+    if(domain.val().length === 0)
+        return;
+    var source = new EventSource("php/queryads.php?domain="+domain.val());
+
+    // Reset and show field
+    ta.empty();
+    ta.show();
+
+    source.addEventListener("message", function(e) {
+        ta.append(e.data);
+    }, false);
+
+    // Will be called when script has finished
+    source.addEventListener("error", function(e) {
+        source.close();
+    }, false);
+}
+
+// $(function(){
+//    eventsourcetest();
+// });
+
+// Handle enter button for adding domains
+$(document).keypress(function(e) {
+    if(e.which === 13 && $("#domain").is(":focus")) {
+        // Enter was pressed, and the input has focus
+        eventsource();
+    }
+});
+// Handle button
+$("#btnSearch").on("click", function() {
+    eventsource();
+});
diff --git a/php/queryads.php b/php/queryads.php
new file mode 100644
index 00000000..5c56a273
--- /dev/null
+++ b/php/queryads.php
@@ -0,0 +1,35 @@
+<?php
+// Test if domain is set
+if(isset($_GET["domain"]))
+{
+    // Remove illegal characters
+    $url = filter_var($_GET["domain"], FILTER_SANITIZE_URL);
+    if(!filter_var("http://".$url, FILTER_VALIDATE_URL ) === true)
+    {
+       die("Invalid domain!");
+    }
+}
+else
+{
+    die("No domain provided");
+}
+
+ob_end_flush();
+ini_set("output_buffering", "0");
+ob_implicit_flush(true);
+header('Content-Type: text/event-stream');
+header('Cache-Control: no-cache');
+
+function echoEvent($datatext) {
+    echo "data: ".implode("\ndata: ", explode("\n", $datatext))."\n\n";
+}
+
+// echoEvent("***START***");
+
+$proc = popen("sudo pihole -q ".$url, 'r');
+while (!feof($proc)) {
+    echoEvent(fread($proc, 4096));
+}
+
+// echoEvent("***END***");
+?>
diff --git a/queryads.php b/queryads.php
new file mode 100644
index 00000000..82d22603
--- /dev/null
+++ b/queryads.php
@@ -0,0 +1,23 @@
+<?php
+    require "header.php";
+?>
+<!-- Title -->
+<div class="page-header">
+    <h1>Query list of ad-serving domains</h1>
+</div>
+<!-- Domain Input -->
+<div class="form-group input-group">
+    <input id="domain" type="text" class="form-control" placeholder="Domain to look for (example.com or sub.example.com)">
+    <span class="input-group-btn">
+        <button id="btnSearch" class="btn btn-default" type="button">Search</button>
+    </span>
+</div>
+
+<pre id="output" style="width: 100%; height: 100%;" hidden="true"></pre>
+
+<?php
+    require "footer.php";
+?>
+
+
+<script src="js/pihole/queryads.js"></script>