<?php
/*
* Copyright 2005-2015 Centreon
* Centreon is developped by : Julien Mathis and Romain Le Merlus under
* GPL Licence 2.0.
*
* This program is free software; you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free Software
* Foundation ; either version 2 of the License.
*
* This program 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
* this program; if not, see <http://www.gnu.org/licenses>.
*
* Linking this program statically or dynamically with other modules is making a
* combined work based on this program. Thus, the terms and conditions of the GNU
* General Public License cover the whole combination.
*
* As a special exception, the copyright holders of this program give Centreon
* permission to link this program with independent modules to produce an executable,
* regardless of the license terms of these independent modules, and to copy and
* distribute the resulting executable under terms of Centreon choice, provided that
* Centreon also meet, for each linked independent module, the terms and conditions
* of the license of that module. An independent module is a module which is not
* derived from this program. If you modify this program, you may extend this
* exception to your version of the program, but you are not obliged to do so. If you
* do not wish to do so, delete this exception statement from your version.
*
* For more information : contact@centreon.com
*
*/
/**
* Hostgroups objects
*
* @author jmathis
*/
class CentreonHostgroups
{
/**
*
* @var type
*/
private $DB;
/**
*
* @var type
*/
private $relationCache;
/**
*
* @var type
*/
private $dataTree;
/**
* Constructor
*
* @param $pearDB
*/
public function __construct($pearDB)
{
$this->DB = $pearDB;
}
/**
* Enter description here ...
*
* @param unknown_type $hg_id
* @param unknown_type $searchHost
* @param unknown_type $level
*/
public function getHostGroupHosts($hg_id = null)
{
if (!$hg_id) {
return;
}
if (!count($this->relationCache)) {
$this->setHgHgCache();
}
$hosts = array();
$DBRESULT = $this->DB->query(
"SELECT hgr.host_host_id " .
"FROM hostgroup_relation hgr, host h " .
"WHERE hgr.hostgroup_hg_id = '".$this->DB->escape($hg_id)."' " .
"AND h.host_id = hgr.host_host_id " .
"ORDER by h.host_name"
);
while ($elem = $DBRESULT->fetchRow()) {
$ref[$elem["host_host_id"]] = $elem["host_host_id"];
$hosts[] = $elem["host_host_id"];
}
$DBRESULT->free();
unset($elem);
if (isset($hostgroups) && count($hostgroups)) {
foreach ($hostgroups as $hg_id2) {
$ref[$hg_id2] = array();
$tmp = $this->getHostGroupHosts($hg_id2, "", 1);
foreach ($tmp as $id) {
print " host: $id<br>";
}
unset($tmp);
}
}
return $hosts;
}
/**
* Get Hostgroup Name
*
* @param int $hg_id
* @return string
*/
public function getHostgroupName($hg_id)
{
static $names = array();
if (!isset($names[$hg_id])) {
$query = "SELECT hg_name FROM hostgroup WHERE hg_id = " . $this->DB->escape($hg_id);
$res = $this->DB->query($query);
if ($res->numRows()) {
$row = $res->fetchRow();
$names[$hg_id] = $row['hg_name'];
}
}
if (isset($names[$hg_id])) {
return $names[$hg_id];
}
return "";
}
/**
* Get Hostgroup Id/Name
*
* @param int $hg_id
* @return string
*/
public function getHostsgroups($hg_id = array())
{
$arrayReturn = array();
if (!empty($hg_id)) {
$query = "SELECT hg_id, hg_name FROM hostgroup
WHERE hg_id IN (" . $this->DB->escape(implode(",", $hg_id)).")";
$res = $this->DB->query($query);
$arrayReturn = array();
while ($row = $res->fetchRow()) {
$arrayReturn[] = array("id" => $row['hg_id'], "name" => $row['hg_name']);
}
}
return $arrayReturn;
}
/**
* Get Hostgroup Id
*
* @param string $hg_name
* @return int
*/
public function getHostgroupId($hg_name)
{
static $ids = array();
if (!isset($ids[$hg_name])) {
$query = "SELECT hg_id FROM hostgroup WHERE hg_name = '" . $this->DB->escape($hg_name) . "'";
$res = $this->DB->query($query);
if ($res->numRows()) {
$row = $res->fetchRow();
$ids[$hg_name] = $row['hg_id'];
}
}
if (isset($ids[$hg_name])) {
return $ids[$hg_name];
}
return 0;
}
/**
* @param null $hg_id
* @return array|void
*/
public function getHostGroupHostGroups($hg_id = null)
{
if (!$hg_id) {
return;
}
$hosts = array();
$DBRESULT = $this->DB->query(
"SELECT hg_child_id " .
"FROM hostgroup_hg_relation, hostgroup " .
"WHERE hostgroup_hg_relation.hg_parent_id = '".$this->DB->escape($hg_id)."' " .
"AND hostgroup.hg_id = hostgroup_hg_relation.hg_child_id " .
"ORDER BY hostgroup.hg_name"
);
while ($elem = $DBRESULT->fetchRow()) {
$hosts[$elem["hg_child_id"]] = $elem["hg_child_id"];
}
$DBRESULT->free();
unset($elem);
return $hosts;
}
/**
*
*/
private function setHgHgCache()
{
$this->relationCache = array();
$DBRESULT = $this->DB->query("SELECT /* SQL_CACHE */ hg_parent_id, hg_child_id FROM hostgroup_hg_relation");
while ($data = $DBRESULT->fetchRow()) {
if (!isset($this->relationCache[$data["hg_parent_id"]])) {
$this->relationCache[$data["hg_parent_id"]] = array();
}
$this->relationCache[$data["hg_parent_id"]][$data["hg_child_id"]] = 1;
}
$DBRESULT->free();
unset($data);
}
public function getAllHostgroupsInCache($DB)
{
$hostgroups = array();
$this->unsetCache();
$DBRESULT = $DB->query(
"SELECT * FROM hostgroup WHERE hg_id NOT IN (SELECT hg_child_id FROM hostgroup_hg_relation)"
);
while ($data = $DBRESULT->fetchRow()) {
$this->dataTree[$data['hg_id']] = $this->getHostGroupHosts($data['hg_id'], $this->dataTree);
}
$DBRESULT->free();
return $hostgroups;
}
/**
*
*/
private function unsetCache()
{
$this->dataTree = array();
}
/**
*
* @param integer $field
* @return array
*/
public static function getDefaultValuesParameters($field)
{
$parameters = array();
$parameters['currentObject']['table'] = 'hostgroup';
$parameters['currentObject']['id'] = 'hg_id';
$parameters['currentObject']['name'] = 'hg_name';
$parameters['currentObject']['comparator'] = 'hg_id';
switch ($field) {
case 'hg_hosts':
$parameters['type'] = 'relation';
$parameters['externalObject']['object'] = 'centreonHost';
$parameters['externalObject']['table'] = 'host';
$parameters['externalObject']['id'] = 'host_id';
$parameters['externalObject']['name'] = 'host_name';
$parameters['externalObject']['comparator'] = 'host_id';
$parameters['relationObject']['table'] = 'hostgroup_relation';
$parameters['relationObject']['field'] = 'host_host_id';
$parameters['relationObject']['comparator'] = 'hostgroup_hg_id';
break;
}
return $parameters;
}
/**
*
* @param array $values
* @return array
*/
public function getObjectForSelect2($values = array(), $options = array())
{
global $centreon;
$items = array();
// get list of authorized hostgroups
if (!$centreon->user->access->admin) {
$hgAcl = $centreon->user->access->getHostGroupAclConf(
null,
'broker',
array(
'distinct' => true,
'fields' => array('hostgroup.hg_id'),
'get_row' => 'hg_id',
'keys' => array('hg_id'),
'conditions' => array(
'hostgroup.hg_id' => array(
'IN',
$values
)
)
),
true
);
}
$explodedValues = '';
$queryValues = array();
if (!empty($values)) {
foreach ($values as $k => $v) {
$explodedValues .= '?,';
$queryValues[] = (int)$v;
}
$explodedValues = rtrim($explodedValues, ',');
} else {
$explodedValues .= '""';
}
// get list of selected hostgroups
$query = "SELECT hg_id, hg_name "
. "FROM hostgroup "
. "WHERE hg_id IN (" . $explodedValues . ") "
. "ORDER BY hg_name ";
$stmt = $this->DB->prepare($query);
$resRetrieval = $this->DB->execute($stmt, $queryValues);
if (PEAR::isError($resRetrieval)) {
throw new Exception('Bad host groups query params');
}
while ($row = $resRetrieval->fetchRow()) {
// hide unauthorized hostgroups
$hide = false;
if (!$centreon->user->access->admin && !in_array($row['hg_id'], $hgAcl)) {
$hide = true;
}
$items[] = array(
'id' => $row['hg_id'],
'text' => $row['hg_name'],
'hide' => $hide
);
}
return $items;
}
/**
* @param $hgName
* @return array
*/
public function getHostsByHostgroupName($hgName)
{
$hostList = array();
$query = "SELECT host_name, host_id " .
"FROM hostgroup_relation hgr, host h, hostgroup hg " .
"WHERE hgr.host_host_id = h.host_id " .
"AND hgr.hostgroup_hg_id = hg.hg_id " .
"AND h.host_activate = '1' " .
"AND hg.hg_name = '" . $this->DB->escape($hgName) . "'";
$result = $this->DB->query($query);
while ($elem = $result->fetchrow()) {
$hostList[] = array(
'host' => $elem['host_name'],
'host_id' => $elem['host_id'],
'hg_name' => $hgName
);
}
return $hostList;
}
}