<?php
/**
 * Observium
 *
 *   This file is part of Observium.
 *
 * @package    observium
 * @subpackage web
 * @copyright  (C) 2006-2013 Adam Armstrong, (C) 2013-2022 Observium Limited
 *
 */

register_html_title("Poller/Discovery Timing");

// Generate cache of Pollers
$pollers = [ 0 => [ 'poller_name' => "Default", 'poller_id' => 0 ] ];
foreach(dbFetchRows("SELECT * FROM `pollers`") as $entry) {
  $pollers[$entry['poller_id']] = $entry;
}

$navbar = [ 'brand' => "Poller", 'class' => "navbar-narrow" ];

if ($_SESSION['userlevel'] >= 7) {
  $navbar['options']['wrapper']['text']        = 'Wrapper';
}
$navbar['options']['devices']['text']        = 'Per-Device';
$navbar['options']['modules']['text']        = 'Per-Module';

//if (dbFetchCell("SELECT COUNT(*) FROM `pollers`"))
if (OBS_DISTRIBUTED) {

  $navbar['options']['pollers']['text']        = 'Partitions';

  $navbar['options_right']['poller_id']['text']        = 'Poller Partition (All)';

  if (!safe_empty($vars['poller_id'])) {
      $navbar['options_right']['poller_id']['suboptions']['all']['text'] = 'All Partitions';
      $navbar['options_right']['poller_id']['suboptions']['all']['url']  = generate_url($vars, [ 'poller_id' => NULL ] );
  }

  foreach($pollers as $poller) {
    $navbar['options_right']['poller_id']['suboptions'][$poller['poller_id']]['text'] = escape_html($poller['poller_name']);
    $navbar['options_right']['poller_id']['suboptions'][$poller['poller_id']]['url']  = generate_url($vars, [ 'poller_id' => $poller['poller_id'] ] );
    if (!safe_empty($vars['poller_id']) && $vars['poller_id'] == $poller['poller_id']) {
      $navbar['options_right']['poller_id']['suboptions'][$poller['poller_id']]['class'] = "active";
      $navbar['options_right']['poller_id']['text'] = 'Poller Partition ('.$poller['poller_name'].')';
    }
  }
}

foreach ($navbar['options'] as $option => $array) {
  if (!isset($vars['view'])) { $vars['view'] = $option; }
  if ($vars['view'] == $option) { $navbar['options'][$option]['class'] .= " active"; }
  $navbar['options'][$option]['url'] = generate_url($vars, array('view' => $option));
}

print_navbar($navbar);
unset($navbar);

// Generate statistics

$totals['pollery']  = 0;
$totals['disovery'] = 0;
$totals['count']    = 0;

$proc['avg2']['poller']    = 0;
$proc['avg2']['discovery'] = 0;
$proc['max']['poller']    = 0;
$proc['max']['discovery'] = 0;

$mod_total = 0;
$mods = [];

// Make poller table
$devices = [];
foreach ($cache['devices']['hostname'] as $hostname => $id) {
  // Reference the cache.
  $device = &$cache['devices']['id'][$id];

  if ($device['disabled'] == 1 && !$config['web_show_disabled']) { continue; }
  if (!safe_empty($vars['poller_id']) && $device['poller_id'] != $vars['poller_id']) {
      // Restricting devices list to matching poller domain.
      //unset($devices[$device['device_id']]);
      continue;
  }

  // Convert empty times to numeric
  $device['last_polled_timetaken'] = (float) $device['last_polled_timetaken'];
  $device['last_discovered_timetaken'] = (float) $device['last_discovered_timetaken'];

  $devices[$device['device_id']] = $device;

  // Find max poller/discovery times
  if ($device['status']) {
    if ($device['last_polled_timetaken'] > $proc['max']['poller']) {
      $proc['max']['poller'] = $device['last_polled_timetaken'];
    }
    if ($device['last_discovered_timetaken'] > $proc['max']['discovery']) {
      $proc['max']['discovery'] = $device['last_discovered_timetaken'];
    }
  }
  $proc['avg2']['poller']    += $device['last_polled_timetaken'] ** 2;
  $proc['avg2']['discovery'] += $device['last_discovered_timetaken'] ** 2;

  $totals['count']++;
  $totals['poller'] += $device['last_polled_timetaken'];
  $totals['discovery'] += $device['last_discovered_timetaken'];

  $devices[$id] = array_merge($devices[$id], [
    'html_row_class'            => $device['html_row_class'],
    'device_hostname'           => $device['hostname'],
    'device_link'               => generate_device_link($device),
    'device_status'             => $device['status'],
    'device_disabled'           => $device['disabled'],
    'last_polled_timetaken'     => $device['last_polled_timetaken'],
    //'last_polled'               => $device['last_polled'],
    'last_discovered_timetaken' => $device['last_discovered_timetaken'],
    //'last_discovered'           => $device['last_discovered']
    ]
  );

  foreach($device['state']['poller_mod_perf'] as $mod => $time) {
    $mods[$mod]['time'] += $time;
    $mods[$mod]['count']++;
    $mod_total += $time;
  }
}

$proc['avg']['poller']    = round(float_div($totals['polling'], $totals['count']), 2);
$proc['avg']['discovery'] = round(float_div($totals['discovery'], $totals['count']), 2);

// End generate statistics

if ($vars['view'] === "modules") {

  if ($_SESSION['userlevel'] >= 7) {
    echo generate_box_open([ 'header-border' => TRUE, 'title' => 'Poller Modules' ]);

    $graph_array = [
      'type'   => 'global_pollermods',
      'from'   => get_time('week'),
      'to'     => get_time(),
      'legend' => 'no'
    ];

    if (!safe_empty($vars['poller_id']) && is_numeric($vars['poller_id'])) {
        $graph_array['poller_id'] = $vars['poller_id'];
    }

    print_graph_row($graph_array);

    echo generate_box_close();
  }


  echo generate_box_open();
  if ($_SESSION['userlevel'] >= 7) {
    echo('<table class="'.OBS_CLASS_TABLE_STRIPED_TWO.'">' . PHP_EOL);
  } else {
    echo('<table class="'.OBS_CLASS_TABLE_STRIPED.'">' . PHP_EOL);
  }

  $mods = array_sort_by($mods, 'time', SORT_DESC, SORT_NUMERIC);

  foreach($mods as $mod => $data) {

    $perc = round(float_div($data['time'], $mod_total) * 100);
    $bg   = get_percentage_colours($perc);

    echo '<tr>';
    echo '  <td><h3>'.$mod.'</h3></td>';
    echo '  <td width="200">'.print_percentage_bar ('100%', '20', $perc, $perc.'%', "ffffff", $bg['left'], '', "ffffff", $bg['right']).'</td>';
    echo '  <td width="60">'.$data['count'].'</td>';
    echo '  <td width="60">'.round($data['time'], 3).'s</td>';
    echo '</tr>';
    if ($_SESSION['userlevel'] >= 7) {
      echo '<tr>';
      echo '  <td colspan=6>';

      $graph_array = [
        'type'   => 'global_pollermod',
        'module' => $mod,
        'legend' => 'no'
      ];

      if (!safe_empty($vars['poller_id']) && is_numeric($vars['poller_id'])) {
        $graph_array['poller_id'] = $vars['poller_id'];
      }

      print_graph_row($graph_array);

      echo '  </td>';
      echo '</tr>';
    }

  }

?>
  </tbody>
</table>

<?php

  echo generate_box_close();

} elseif ($vars['view'] === "wrapper" && $_SESSION['userlevel'] >= 7) {

  $rrd_file = $config['rrd_dir'].'/poller-wrapper.rrd';
  if (rrd_is_file($rrd_file, TRUE)) {
    echo generate_box_open([ 'header-border' => TRUE, 'title' => 'Poller Wrapper History' ]);

    $graph_array = [
      'type'   => 'poller_wrapper_threads',
      //'operation' => 'poll',
      //'width'  => 1158,
      'height' => 100,
      'from'   => get_time('week'),
      'to'     => get_time()
    ];
    //echo(generate_graph_tag($graph_array));
    print_graph_row($graph_array);

    //$graph_array = array('type'   => 'poller_wrapper_count',
    //                     //'operation' => 'poll',
    //                     'width'  => 1158,
    //                     'height' => 100,
    //                     'from'   => $config['time']['week'],
    //                     'to'     => $config['time']['now'],
    //                     );
    //echo(generate_graph_tag($graph_array));
    //echo "<h3>Poller wrapper Total time</h3>";
    $graph_array = [
      'type'   => 'poller_wrapper_times',
      //'operation' => 'poll',
      //'width'  => 1158,
      'height' => 100,
      'from'   => get_time('week'),
      'to'     => get_time()
    ];
    //echo(generate_graph_tag($graph_array));
    print_graph_row($graph_array);

    echo generate_box_close([ 'footer_content' => '<b>Please note:</b> The total time for the poller wrapper is not the same as the timings below. Total poller wrapper time is real polling time for all devices and all threads.' ]);
  }

} elseif ($vars['view'] === "devices") {

  if ($_SESSION['userlevel'] >= 7) {
    echo generate_box_open([ 'header-border' => TRUE, 'title' => 'All Devices Poller Performance' ]);

    $graph_array = [
      'type'   => 'global_poller',
      'from'   => get_time('week'),
      'to'     => get_time(),
      'legend' => 'no'
    ];

    if (!safe_empty($vars['poller_id']) && is_numeric($vars['poller_id'])) {
      $graph_array['poller_id'] = $vars['poller_id'];
    }

    print_graph_row($graph_array);

    echo generate_box_close();
  }

  echo generate_box_open([ 'header-border' => TRUE, 'title' => 'Poller/Discovery Timing' ]);
  echo('<table class="'.OBS_CLASS_TABLE_STRIPED_MORE.'">' . PHP_EOL);

// FIXME -- table header generator / sorting

?>

  <thead>
    <tr>
      <th class="state-marker"></th>
      <th>Device</th>
      <th colspan="3">Last Polled</th>
      <th colspan="3">Last Discovered</th>
      <?php if(safe_empty($vars['poller_id'])) { echo '<th>Poller</th>'; } ?>
    </tr>
  </thead>
  <tbody>
<?php


// Sort poller table
// sort order: $polled > $discovered > $hostname
  $devices = array_sort_by($devices, 'device_status',             SORT_DESC, SORT_NUMERIC,
                                     'last_polled_timetaken',     SORT_DESC, SORT_NUMERIC,
                                     'last_discovered_timetaken', SORT_DESC, SORT_NUMERIC,
                                     'device_hostname',           SORT_ASC,  SORT_STRING);

// Print poller table
  foreach ($devices as $row) {
    $proc['time']['poller']     = round(float_div($row['last_polled_timetaken'], $proc['max']['poller']) * 100);
    $proc['color']['poller']    = "success";
    if     ($row['last_polled_timetaken'] >  ($proc['max']['poller'] * 0.75)) { $proc['color']['poller'] = "danger"; }
    elseif ($row['last_polled_timetaken'] >  ($proc['max']['poller'] * 0.5))  { $proc['color']['poller'] = "warning"; }
    elseif ($row['last_polled_timetaken'] >= ($proc['max']['poller'] * 0.25)) { $proc['color']['poller'] = "info"; }
    $proc['time']['discovery']  = round(float_div($row['last_discovered_timetaken'], $proc['max']['discovery']) * 100);
    $proc['color']['discovery'] = "success";
    if     ($row['last_discovered_timetaken'] >  ($proc['max']['discovery'] * 0.75)) { $proc['color']['discovery'] = "danger"; }
    elseif ($row['last_discovered_timetaken'] >  ($proc['max']['discovery'] * 0.5))  { $proc['color']['discovery'] = "warning"; }
    elseif ($row['last_discovered_timetaken'] >= ($proc['max']['discovery'] * 0.25)) { $proc['color']['discovery'] = "info"; }

    $poll_bg     = get_percentage_colours($proc['time']['poller']);
    $discover_bg = get_percentage_colours($proc['time']['discovery']);

    // Poller times
    echo('    <tr class="'.$row['html_row_class'].'">
      <td class="state-marker"></td>
      <td class="entity">'.$row['device_link'].'</td>
      <td style="width: 12%;">'.
        print_percentage_bar ('100%', '20', $proc['time']['poller'], $proc['time']['poller'].'%', "ffffff", $poll_bg['left'], '', "ffffff", $poll_bg['right'])
      .'</td>
      <td style="width: 7%">
        '.$row['last_polled_timetaken'].'s
      </td>
      <!-- <td>'.format_timestamp($row['last_polled']).' </td> -->
      <td>'.format_uptime($config['time']['now'] - strtotime($row['last_polled']), 'shorter').' ago</td>');

    // Discovery times
    echo('
      <td style="width: 12%;">'.
        print_percentage_bar ('100%', '20', $proc['time']['discovery'], $proc['time']['discovery'].'%', "ffffff", $discover_bg['left'], '', "ffffff", $discover_bg['right'])
      .'</td>
      <td style="width: 7%">
        '.$row['last_discovered_timetaken'].'s
      </td>
      <!-- <td>'.format_timestamp($row['last_discovered']).'</td> -->
      <td>'.format_uptime($config['time']['now'] - strtotime($row['last_discovered']), 'shorter').' ago</td> ');

    if (safe_empty($vars['poller_id'])) {
      echo '
   <td>' . get_type_class_label($pollers[$row['poller_id']]['poller_name'], 'poller') . '</td>';
    }

    echo '
    </tr>
';
  }

  // Calculate root mean square
  $proc['avg2']['poller']    = sqrt(float_div($proc['avg2']['poller'], $totals['count']));
  $proc['avg2']['poller']    = round($proc['avg2']['poller'], 2);
  $proc['avg2']['discovery'] = sqrt(float_div($proc['avg2']['discovery'], $totals['count']));
  $proc['avg2']['discovery'] = round($proc['avg2']['discovery'], 2);

  echo('    <tr>
      <th></th>
      <th style="text-align: right;">Total time for all devices (average per device):</th>
      <th></th>
      <th colspan="3">'.$totals['poller'].'s ('.$proc['avg2']['poller'].'s)</th>
      <th></th>
      <th colspan="3">'.$totals['discovery'].'s ('.$proc['avg2']['discovery'].'s)</th>
    </tr>
');

  unset($row);

?>
  </tbody>
</table>

<?php

echo generate_box_close();

} elseif ($view = 'pollers') {

  //$pollers = dbFetchRows("SELECT * FROM `pollers`");
  unset($pollers[0]); // remove default poller here

  echo generate_box_open();

  echo '<table class="'.OBS_CLASS_TABLE_STRIPED.'">' . PHP_EOL;

  echo get_table_header([
    [ 'ID',              'style="width: 20px;"' ],
    [ 'Poller Name',     'style="width: 10%;"' ],
    [ 'Host ID / Uname', 'style="width: 30%;"' ],
    'Assigned Devices'
    ]);
  //echo '<tr><td>Poller Name</td><td>Assigned Devices</td></tr>';

  foreach($pollers as $poller) {
    $device_list = [];
    echo '<tr>';
    echo '<td class="entity-name">'.$poller['poller_id'].'</td>';
    echo '<td class="entity-name">'.$poller['poller_name'].'</td>';
    echo '<td><small>'.$poller['host_id'].'<br /><i>'.$poller['host_uname'].'</i></small></td>';
    echo '<td>';

    foreach(dbFetchRows("SELECT * FROM `devices` WHERE `poller_id` = ?", [ $poller['poller_id'] ]) as $device) {
      $device_list[] = generate_device_link($device);
    }

    echo implode(', ', $device_list);

    echo '</td>';
    echo '</tr>';
  }

  echo '</table>';

  echo generate_box_close();

  foreach ($pollers as $poller) {
    echo generate_box_open([ 'header-border' => TRUE, 'title' => 'Poller Wrapper ('.$poller['poller_name'].') History' ]);

    //r($poller);
    $graph_array = [
      'type'   => 'poller_partitioned_wrapper_threads',
      //'operation' => 'poll',
      'id' => $poller['poller_id'],
      // 'width'  => 1158,
      'height' => 100,
      'from'   => get_time('week'),
      'to'     => get_time(),
    ];
    //echo(generate_graph_tag($graph_array));
    print_graph_row($graph_array);

    $graph_array = [
      'type'   => 'poller_partitioned_wrapper_times',
      //'operation' => 'poll',
      'id' => $poller['poller_id'],
      // 'width'  => 1158,
      'height' => 100,
      'from'   => get_time('week'),
      'to'     => get_time(),
    ];
    //echo(generate_graph_tag($graph_array));
    print_graph_row($graph_array);

    echo generate_box_close();

    if ($actions = dbFetchRows('SELECT * FROM `observium_actions` WHERE `poller_id` = ?', [ $poller['poller_id'] ])) {
      echo generate_box_open(array('header-border' => TRUE, 'title' => 'Poller Actions ('.$poller['poller_name'].')'));
      $options = [
        'columns' => [
          'ID', 'Poller', 'Action', 'Identifier', 'Variables', 'Added'
        ],
        'vars' => 'json',
        'added' => 'unixtime'
      ];
      echo build_table($actions, $options);
      echo generate_box_close();
    }

  }
}

unset($devices, $proc, $pollers);

// EOF