mirror of https://github.com/MISP/MISP
122 lines
4.4 KiB
PHP
122 lines
4.4 KiB
PHP
<?php
|
|
/**
|
|
* Org Events widget which reportes the number of events created monthly by each local organizations
|
|
*
|
|
*/
|
|
class OrgEventsWidget
|
|
{
|
|
public $title = 'Org Events';
|
|
public $render = 'MultiLineChart';
|
|
public $width = 8;
|
|
public $height = 6;
|
|
public $description = 'A graph to show the monthly number of events per organisation';
|
|
public $cacheLifetime = 10;
|
|
public $autoRefreshDelay = false;
|
|
public $params = array (
|
|
'blocklist_orgs' => 'A list of organisation names to filter out',
|
|
'months' => 'Number of past months to consider for the graph',
|
|
'logarithmic' => 'Visualize data on logarithmic scale'
|
|
);
|
|
|
|
public $placeholder =
|
|
'{
|
|
"blocklist_orgs": ["Orgs to filter"],
|
|
"months": "6",
|
|
"logarithmic": "true"
|
|
}';
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
* Target_month must be from 1 to 12
|
|
* Target year must be 4 digits
|
|
*/
|
|
private function org_events_count($user, $org, $target_month, $target_year) {
|
|
$events_count = 0;
|
|
|
|
$start_date = $target_year.'-'.$target_month.'-01';
|
|
if($target_month == 12) {
|
|
$end_date = ($target_year+1).'-01-01';
|
|
} else {
|
|
$end_date = $target_year.'-'.($target_month+1).'-01';
|
|
}
|
|
$conditions = array('Event.orgc_id' => $org['Organisation']['id'], 'Event.date >=' => $start_date, 'Event.date <' => $end_date);
|
|
|
|
//This is required to enforce the ACL (not pull directly from the DB)
|
|
$eventIds = $this->Event->fetchSimpleEventIds($user, array('conditions' => $conditions));
|
|
|
|
if(!empty($eventIds)) {
|
|
$params = array('Event.id' => $eventIds);
|
|
$events = $this->Event->find('all', array('conditions' => array('AND' => $params)));
|
|
foreach($events as $event) {
|
|
$events_count+= 1;
|
|
}
|
|
}
|
|
return $events_count;
|
|
}
|
|
|
|
private function filter_ghost_orgs(&$data, $orgs){
|
|
foreach ($data['data'] as &$item) {
|
|
foreach(array_keys($orgs) as $org_name) {
|
|
unset($item[$org_name]);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function handler($user, $options = array())
|
|
{
|
|
$this->Log = ClassRegistry::init('Log');
|
|
$this->Org = ClassRegistry::init('Organisation');
|
|
$this->Event = ClassRegistry::init('Event');
|
|
$orgs = $this->Org->find('all', array( 'conditions' => array('Organisation.local' => 1)));
|
|
$current_month = date('n');
|
|
$current_year = date('Y');
|
|
$limit = 6; // months
|
|
if(!empty($options['months'])) {
|
|
$limit = (int) ($options['months']);
|
|
}
|
|
$offset = 0;
|
|
$ghost_orgs = array(); // track orgs without any contribution
|
|
// We start by putting all orgs_id in there:
|
|
foreach($orgs as $org) {
|
|
// We check for blocklisted orgs
|
|
if(!empty($options['blocklist_orgs']) && in_array($org['Organisation']['name'], $options['blocklist_orgs'])) {
|
|
unset($orgs[$offset]);
|
|
} else {
|
|
$ghost_orgs[$org['Organisation']['name']] = true;
|
|
}
|
|
$offset++;
|
|
}
|
|
$data = array();
|
|
$data['data'] = array();
|
|
for ($i=0; $i < $limit; $i++) {
|
|
$target_month = $current_month - $i;
|
|
$target_year = $current_year;
|
|
if ($target_month < 1) {
|
|
$target_month += 12;
|
|
$target_year -= 1;
|
|
}
|
|
$item = array();
|
|
$item ['date'] = $target_year.'-'.$target_month.'-01';
|
|
foreach($orgs as $org) {
|
|
$count = $this->org_events_count($user, $org, $target_month, $target_year);
|
|
if($options['logarithmic'] === "true" || $options['logarithmic'] === "1") {
|
|
$item[$org['Organisation']['name']] = (int) round(log($count, 1.1)); // taking the logarithmic view
|
|
} else if(empty($options['logarithmic']) || $options['logarithmic'] === "true" || $options['logarithmic'] === "1"){
|
|
$item[$org['Organisation']['name']] = $count;
|
|
}
|
|
// if a positive score is detected at least once it's enough to be
|
|
// considered for the graph
|
|
if($count > 0) {
|
|
unset($ghost_orgs[$org['Organisation']['name']]);
|
|
}
|
|
}
|
|
$data['data'][] = $item;
|
|
}
|
|
$this->filter_ghost_orgs($data, $ghost_orgs);
|
|
return $data;
|
|
}
|
|
}
|