/* mirror_status.js
 * Homepage: https://projects.archlinux.org/archweb.git/
 * Copyright: 2012-2013 The Archweb Team (Dan McGee)
 * License: GPLv2
 *
 * This file is part of Archweb.
 *
 * Archweb is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * Archweb is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Archweb.  If not, see <http://www.gnu.org/licenses/>.
 */

function draw_graphs(location_url, log_url, container_id) {
    jQuery.when(jQuery.getJSON(location_url), jQuery.getJSON(log_url))
        .then(function(loc_data, log_data) {
            /* use the same color selection for a given URL in every graph */
            var color = d3.scale.category10();
            jQuery.each(loc_data[0].locations, function(i, val) {
                mirror_status(container_id, val, log_data[0], color);
            });
        });
}

function mirror_status(container_id, check_loc, log_data, color) {

    var draw_graph = function(chart_id, data) {
        var jq_div = jQuery(chart_id);
        var margin = {top: 20, right: 20, bottom: 30, left: 40},
            width = jq_div.width() - margin.left - margin.right,
            height = jq_div.height() - margin.top - margin.bottom;

        var x = d3.time.scale.utc().range([0, width]),
            y = d3.scale.linear().range([height, 0]),
            x_axis = d3.svg.axis().scale(x).orient("bottom"),
            y_axis = d3.svg.axis().scale(y).orient("left");

        /* remove any existing graph first if we are redrawing after resize */
        d3.select(chart_id).select("svg").remove();
        var svg = d3.select(chart_id).append("svg")
            .attr("width", width + margin.left + margin.right)
            .attr("height", height + margin.top + margin.bottom)
            .append("g")
            .attr("transform", "translate(" + margin.left + "," + margin.top + ")");

        x.domain([
                d3.min(data, function(c) { return d3.min(c.logs, function(v) { return v.check_time; }); }),
                d3.max(data, function(c) { return d3.max(c.logs, function(v) { return v.check_time; }); })
        ]).nice(d3.time.hour);
        y.domain([
                0,
                d3.max(data, function(c) { return d3.max(c.logs, function(v) { return v.duration; }); })
        ]).nice();

        /* build the axis lines... */
        svg.append("g")
            .attr("class", "x axis")
            .attr("transform", "translate(0," + height + ")")
            .call(x_axis)
            .append("text")
            .attr("class", "label")
            .attr("x", width)
            .attr("y", -6)
            .style("text-anchor", "end")
            .text("Check Time (UTC)");

        svg.append("g")
            .attr("class", "y axis")
            .call(y_axis)
            .append("text")
            .attr("class", "label")
            .attr("transform", "rotate(-90)")
            .attr("y", 6)
            .attr("dy", ".71em")
            .style("text-anchor", "end")
            .text("Duration (seconds)");

        var line = d3.svg.line()
            .interpolate("basis")
            .x(function(d) { return x(d.check_time); })
            .y(function(d) { return y(d.duration); });

        /* ...then the points and lines between them. */
        var urls = svg.selectAll(".url")
            .data(data)
            .enter()
            .append("g")
            .attr("class", "url");

        urls.append("path")
            .attr("class", "url-line")
            .attr("d", function(d) { return line(d.logs); })
            .style("stroke", function(d) { return color(d.url); });

        urls.selectAll("circle")
            .data(function(u) {
                return jQuery.map(u.logs, function(l, i) {
                    return {url: u.url, check_time: l.check_time, duration: l.duration};
                });
            })
            .enter()
            .append("circle")
            .attr("class", "url-dot")
            .attr("r", 3.5)
            .attr("cx", function(d) { return x(d.check_time); })
            .attr("cy", function(d) { return y(d.duration); })
            .style("fill", function(d) { return color(d.url); })
            .append("title")
            .text(function(d) { return d.url + "\n" + d.duration.toFixed(3) + " secs\n" + d.check_time.toUTCString(); });

        /* add a legend for good measure */
        var active = jQuery.map(data, function(item, i) { return item.url; });
        var legend = svg.selectAll(".legend")
            .data(active)
            .enter().append("g")
            .attr("class", "legend")
            .attr("transform", function(d, i) { return "translate(0," + i * 20 + ")"; });

        legend.append("rect")
            .attr("x", width - 18)
            .attr("width", 18)
            .attr("height", 18)
            .style("fill", color);

        legend.append("text")
            .attr("x", width - 24)
            .attr("y", 9)
            .attr("dy", ".35em")
            .style("text-anchor", "end")
            .text(function(d) { return d; });
    };

    var filter_data = function(json, location_id) {
        return jQuery.map(json.urls, function(url, i) {
            var logs = jQuery.map(url.logs, function(log, j) {
                if (!log.is_success) {
                    return null;
                }
                /* screen by location ID if we were given one */
                if (location_id && log.location_id !== location_id) {
                    return null;
                }
                return {
                    duration: log.duration,
                    check_time: new Date(log.check_time)
                };
            });
            /* don't return URLs without any log info */
            if (logs.length === 0) {
                return null;
            }
            return {
                url: url.url,
                logs: logs
            };
        });
    };

    var cached_data = filter_data(log_data, check_loc.id);
    /* we had a check location with no log data handed to us, skip graphing */
    if (cached_data.length === 0) {
        return;
    }

    /* create the containers, defer the actual graph drawing */
    var chart_id = 'status-chart-' + check_loc.id;
    jQuery(container_id).append('<h3><span class="fam-flag fam-flag-' + check_loc.country_code.toLowerCase() + '" title="' + check_loc.country + '"></span> ' + check_loc.country + ' (' + check_loc.source_ip + '), IPv' + check_loc.ip_version + '</h3>');
    jQuery(container_id).append('<div id="' + chart_id + '" class="visualize-mirror visualize-chart"></div>');
    jQuery(container_id).append('<br/>');
    setTimeout(function() {
        draw_graph('#' + chart_id, cached_data);
    }, 0);

    /* then hook up a resize handler to redraw if necessary */
    var resize_timeout = null;
    var real_resize = function() {
        resize_timeout = null;
        draw_graph('#' + chart_id, cached_data);
    };
    jQuery(window).resize(function() {
        if (resize_timeout) {
            clearTimeout(resize_timeout);
        }
        resize_timeout = setTimeout(real_resize, 200);
    });
}