/* Pi-hole: A black hole for Internet advertisements
*  (c) 2017 Pi-hole, LLC (https://pi-hole.net)
*  Network-wide ad blocking via your own hardware.
*
*  This file is copyright under the latest version of the EUPL.
*  Please see LICENSE file for your rights under this license. */

/*global
    moment
*/

var start__ = moment().subtract(6, "days");
var from = moment(start__).utc().valueOf()/1000;
var end__ = moment();
var until = moment(end__).utc().valueOf()/1000;

var timeoutWarning = $("#timeoutWarning");

$(function () {
    $("#querytime").daterangepicker(
    {
      timePicker: true, timePickerIncrement: 15,
      locale: { format: "MMMM Do YYYY, HH:mm" },
      ranges: {
        "Today": [moment().startOf("day"), moment()],
        "Yesterday": [moment().subtract(1, "days").startOf("day"), moment().subtract(1, "days").endOf("day")],
        "Last 7 Days": [moment().subtract(6, "days"), moment()],
        "Last 30 Days": [moment().subtract(29, "days"), moment()],
        "This Month": [moment().startOf("month"), moment()],
        "Last Month": [moment().subtract(1, "month").startOf("month"), moment().subtract(1, "month").endOf("month")],
        "This Year": [moment().startOf("year"), moment()],
        "All Time": [moment(0), moment()]
      },
      "opens": "center", "showDropdowns": true
    },
    function (startt, endt) {
      from = moment(startt).utc().valueOf()/1000;
      until = moment(endt).utc().valueOf()/1000;
    });


});

// Credit: http://stackoverflow.com/questions/1787322/htmlspecialchars-equivalent-in-javascript/4835406#4835406
function escapeHtml(text) {
  var map = {
    "&": "&",
    "<": "&lt;",
    ">": "&gt;",
    "\"": "&quot;",
    "\'": "&#039;"
  };

  return text.replace(/[&<>"']/g, function(m) { return map[m]; });
}

function padNumber(num) {
    return ("00" + num).substr(-2,2);
}

// Helper function needed for converting the Objects to Arrays

function objectToArray(p){
    var keys = Object.keys(p);
    keys.sort(function(a, b) {
        return a - b;
    });

    var arr = [], idx = [];
    for (var i = 0; i < keys.length; i++) {
        arr.push(p[keys[i]]);
        idx.push(keys[i]);
    }
    return [idx,arr];
}

var timeLineChart;

function compareNumbers(a, b) {
  return a - b;
}

function updateQueriesOverTime() {
    $("#queries-over-time .overlay").show();
    timeoutWarning.show();
    $.getJSON("api_db.php?getGraphData&from="+from+"&until="+until, function(data) {

        // convert received objects to arrays
        data.domains_over_time = objectToArray(data.domains_over_time);
        data.ads_over_time = objectToArray(data.ads_over_time);
        // Remove possibly already existing data
        timeLineChart.data.labels = [];
        timeLineChart.data.datasets[0].data = [];
        timeLineChart.data.datasets[1].data = [];

        var dates = [], hour;

        for (hour in data.domains_over_time[0]) {
            if ({}.hasOwnProperty.call(data.domains_over_time[0], hour)) {
                dates.push(parseInt(data.domains_over_time[0][hour]));
            }
        }

        for (hour in data.ads_over_time[0]) {
            if ({}.hasOwnProperty.call(data.ads_over_time[0], hour)) {
                if(!dates.includes(parseInt(data.ads_over_time[0][hour])))
                {
                    dates.push(parseInt(data.ads_over_time[0][hour]));
                }
            }
        }

        dates.sort(compareNumbers);

        // Add data for each hour that is available
        for (hour in dates) {
            if ({}.hasOwnProperty.call(dates, hour)) {
                var d, dom = 0, ads = 0;
                d = new Date(1000*dates[hour]);

                var idx = data.domains_over_time[0].indexOf(dates[hour].toString());
                if (idx > -1)
                {
                    dom = data.domains_over_time[1][idx];
                }

                idx = data.ads_over_time[0].indexOf(dates[hour].toString());
                if (idx > -1)
                {
                    ads = data.ads_over_time[1][idx];
                }

                timeLineChart.data.labels.push(d);
                timeLineChart.data.datasets[0].data.push(dom);
                timeLineChart.data.datasets[1].data.push(ads);
            }
        }

        timeLineChart.options.scales.xAxes[0].display=true;
        $("#queries-over-time .overlay").hide();
        timeoutWarning.hide();
        timeLineChart.update();
    });
}

/* global Chart */
$(document).ready(function() {
    var ctx = document.getElementById("queryOverTimeChart").getContext("2d");
    timeLineChart = new Chart(ctx, {
        type: "line",
        data: {
            labels: [ 0 ],
            datasets: [
                {
                    label: "Total DNS Queries",
                    fill: true,
                    backgroundColor: "rgba(220,220,220,0.5)",
                    borderColor: "rgba(0, 166, 90,.8)",
                    pointBorderColor: "rgba(0, 166, 90,.8)",
                    pointRadius: 1,
                    pointHoverRadius: 5,
                    data: [],
                    pointHitRadius: 5,
                    cubicInterpolationMode: "monotone"
                },
                {
                    label: "Blocked DNS Queries",
                    fill: true,
                    backgroundColor: "rgba(0,192,239,0.5)",
                    borderColor: "rgba(0,192,239,1)",
                    pointBorderColor: "rgba(0,192,239,1)",
                    pointRadius: 1,
                    pointHoverRadius: 5,
                    data: [],
                    pointHitRadius: 5,
                    cubicInterpolationMode: "monotone"
                }
            ]
        },
        options: {
            tooltips: {
                enabled: true,
                responsive: true,
                mode: "x-axis",
                callbacks: {
                    title: function(tooltipItem, data) {
                        var label = tooltipItem[0].xLabel;
                        var time = new Date(label);
                        var date = time.getFullYear()+"-"+padNumber(time.getMonth()+1)+"-"+padNumber(time.getDate());
                        var h = time.getHours();
                        var m = time.getMinutes();
                        var from = padNumber(h)+":"+padNumber(m)+":00";
                        var to = padNumber(h)+":"+padNumber(m+9)+":59";
                        return "Queries from "+from+" to "+to+" on "+date;
                    },
                    label: function(tooltipItems, data) {
                        if(tooltipItems.datasetIndex === 1)
                        {
                            var percentage = 0.0;
                            var total = parseInt(data.datasets[0].data[tooltipItems.index]);
                            var blocked = parseInt(data.datasets[1].data[tooltipItems.index]);
                            if(total > 0)
                            {
                                percentage = 100.0*blocked/total;
                            }
                            return data.datasets[tooltipItems.datasetIndex].label + ": " + tooltipItems.yLabel + " (" + percentage.toFixed(1) + "%)";
                        }
                        else
                        {
                            return data.datasets[tooltipItems.datasetIndex].label + ": " + tooltipItems.yLabel;
                        }
                    }
                }
            },
            legend: {
                display: false
            },
            scales: {
                xAxes: [{
                    type: "time",
                    display: false,
                    time: {
                        displayFormats: {
                           "minute": "HH:mm",
                           "hour": "HH:mm",
                           "day": "HH:mm",
                           "week": "MMM DD HH:mm",
                           "month": "MMM DD",
                           "quarter": "MMM DD",
                           "year": "MMM DD"
                        }
                    }
                }],
                yAxes: [{
                    ticks: {
                        beginAtZero: true
                    }
                }]
            },
            maintainAspectRatio: false
        }
    });
});

$("#querytime").on("apply.daterangepicker", function(ev, picker) {
    $("#queries-over-time").show();
    updateQueriesOverTime();
});

$("#queryOverTimeChart").click(function(evt){
    var activePoints = timeLineChart.getElementAtEvent(evt);
    if(activePoints.length > 0)
    {
        //get the internal index in the chart
        var clickedElementindex = activePoints[0]["_index"];

        //get specific label by index
        var label = timeLineChart.data.labels[clickedElementindex];

        //get value by index
        var from = label/1000;
        var until = label/1000 + 600;
        window.location.href = "db_queries.php?from="+from+"&until="+until;
    }
    return false;
});