<?php

/**
 * Observium
 *
 *   This file is part of Observium.
 *
 * @package        observium
 * @subpackage     graphs
 * @copyright  (C) Adam Armstrong
 *
 */

if (!isset($vars['stat'])) {
    $vars['stat'] = "bits";
}
if (is_numeric($vars['topn']) && $vars['topn'] > 0 && $vars['topn'] < 1024) {
    $topn = $vars['topn'];
} else {
    $topn = '10';
}

include_once($config['html_dir'] . "/includes/graphs/common.inc.php");

switch ($vars['stat']) {
    case "pkts":
        $units       = 'pps';
        $unit        = 'p';
        $multiplier  = '1';
        $colours_in  = 'purples';
        $colours_out = 'oranges';
        $prefix      = "P";
        if ($vars['sort'] == "in") {
            $sort = "pkts_input_rate";
        } elseif ($vars['sort'] == "out") {
            $sort = "pkts_output_rate";
        } else {
            $sort = "pps";
        }
        break;
    case "bits":
        $units       = 'bps';
        $unit        = 'B';
        $multiplier  = '8';
        $colours_in  = 'greens';
        $colours_out = 'blues';
        if ($vars['sort'] == "in") {
            $sort = "bytes_input_rate";
        } elseif ($vars['sort'] == "out") {
            $sort = "bytes_output_rate";
        } else {
            $sort = "bps";
        }
        break;
}

if (isset($sort)) {
    $query = "SELECT *, (bytes_input_rate + bytes_output_rate) AS bps,
                    (pkts_input_rate + pkts_output_rate) AS pps
                    FROM `mac_accounting`
                    WHERE `mac_accounting`.port_id = ?
                    ORDER BY " . $sort . " DESC LIMIT 0, " . $topn;

    $mas = dbFetchRows($query, [$port['port_id'], $sort, $topn]);
}

$pluses      = "";
$iter        = '0';
$rrd_options .= " COMMENT:'                                     In\: Current     Maximum      Total      Out\: Current     Maximum     Total\\n'";

foreach ($mas as $ma) {
    $this_rrd = get_rrd_path($device, "mac_acc-" . $port['ifIndex'] . "-" . $ma['vlan_id'] . "-" . $ma['mac'] . ".rrd");

    if (rrd_is_file($this_rrd)) {
        $mac  = format_mac($ma['mac']);
        $name = $mac;
        $addy = dbFetchRow("SELECT * FROM `ip_mac` WHERE `mac_address` = ? AND `port_id` = ?", [$ma['mac'], $ma['port_id']]);

        if ($addy) {
            $name = $addy['ipv4_address'] . " (" . $mac . ")";
            $peer = dbFetchRow("SELECT * FROM ipv4_addresses AS A, ports AS I, devices AS D
              WHERE A.ipv4_address = ? AND I.port_id = A.port_id AND D.device_id = I.device_id", [$addy['ipv4_address']]);
            if ($peer) {
                $name = $peer['hostname'] . " " . short_ifname($port['ifDescr'], NULL, FALSE) . " (" . $mac . ")";
            }

            //if (dbFetchCell("SELECT COUNT(*) FROM bgpPeers WHERE device_id = ? AND bgpPeerIdentifier = ?", array($ma['device_id'], $addy['ipv4_address'])))
            if (dbExist('bgpPeers', '`device_id` = ? AND `bgpPeerIdentifier` = ?', [$ma['device_id'], $addy['ipv4_address']])) {
                $peer_info = dbFetchRow("SELECT * FROM bgpPeers WHERE device_id = ? AND bgpPeerIdentifier = ?", [$ma['device_id'], $addy['ipv4_address']]);
                $name      .= " - AS" . $peer_info['bgpPeerRemoteAs'];
            }

            if ($peer_info) {
                $asn    = "AS" . $peer_info['bgpPeerRemoteAs'];
                $astext = $peer_info['astext'];
            } else {
                unset ($as);
                unset ($astext);
                unset($asn);
            }
        }

        $this_id = str_replace(".", "", $ma['mac']);
        if (!$config['graph_colours'][$colours_in][$iter] || !$config['graph_colours'][$colours_out][$iter]) {
            $iter = 0;
        }

        $colour_in   = $config['graph_colours'][$colours_in][$iter];
        $colour_out  = $config['graph_colours'][$colours_out][$iter];
        $descr       = rrdtool_escape($name, 36);
        $rrd_options .= " DEF:in" . $this_id . "=$this_rrd:" . $prefix . "IN:AVERAGE ";
        $rrd_options .= " DEF:out" . $this_id . "temp=$this_rrd:" . $prefix . "OUT:AVERAGE ";
        $rrd_options .= " CDEF:inB" . $this_id . "=in" . $this_id . ",$multiplier,* ";
        $rrd_options .= " CDEF:outB" . $this_id . "temp=out" . $this_id . "temp,$multiplier,*";
        $rrd_options .= " CDEF:outB" . $this_id . "=outB" . $this_id . "temp,-1,*";
        $rrd_options .= " CDEF:octets" . $this_id . "=inB" . $this_id . ",outB" . $this_id . "temp,+";
        $rrd_options .= " VDEF:totin" . $this_id . "=inB" . $this_id . ",TOTAL";
        $rrd_options .= " VDEF:totout" . $this_id . "=outB" . $this_id . "temp,TOTAL";
        $rrd_options .= " VDEF:tot" . $this_id . "=octets" . $this_id . ",TOTAL";
        $rrd_options .= " AREA:inB" . $this_id . "#" . $colour_in . ":'" . $descr . "':STACK";
        if ($rrd_optionsb) {
            $stack = ":STACK";
        }
        $rrd_optionsb .= " AREA:outB" . $this_id . "#" . $colour_out . ":$stack";
        $rrd_options  .= " GPRINT:inB" . $this_id . ":LAST:%6.2lf%s$units";
        $rrd_options  .= " GPRINT:inB" . $this_id . ":MAX:%6.2lf%s$units";
        $rrd_options  .= " GPRINT:totin" . $this_id . ":%6.2lf%s$unit";
        $rrd_options  .= " COMMENT:' '";
        $rrd_options  .= "  HRULE:999999999999999#" . $colour_out . ":' '";
        $rrd_options  .= " GPRINT:outB" . $this_id . "temp:LAST:%6.2lf%s$units";
        $rrd_options  .= " GPRINT:outB" . $this_id . "temp:MAX:%6.2lf%s$units";
        $rrd_options  .= " GPRINT:totout" . $this_id . ":%6.2lf%s$unit\\n";
        $iter++;
    }
}

$rrd_options .= $rrd_optionsb;
$rrd_options .= " HRULE:0#999999";

// EOF