Icinga\Module\Trapdirector\TrapsController

Complexity

Total Complexity

46

Size/Duplication

Total Lines	297
Duplicated Lines	0 %

Importance

Changes	24
Bugs	1	Features	0

21 Methods

Rating	Name	Duplication	Size	Complexity
A	getHandlerListTable()	0	8	2
A	getTrapHostListTable()	0	8	2
A	getModuleConfig()	0	12	3
A	getTrapListTable()	0	6	2
A	getUIDatabase()	0	8	2
A	getHandlersCategory()	0	12	2
A	isDirectorInstalled()	0	12	3
A	getIdoConn()	0	25	4
A	checkReadPermission()	0	4	2
A	getMIB()	0	14	3
A	displayExitError()	0	3	1
A	applyPaginationLimits()	0	8	2
A	delHandlersCategory()	0	6	1
A	setHandlerCategory()	0	9	3
A	itemListDisplay()	0	3	1
A	setitemListDisplay()	0	3	1
A	addHandlersCategory()	0	8	4
A	checkConfigPermission()	0	4	2
A	getTrapClass()	0	11	2
A	renameHandlersCategory()	0	5	1
A	checkModuleConfigPermission()	0	10	3

<?php

namespace Icinga\Module\Trapdirector;

use Icinga\Web\Controller;

use Icinga\Data\Paginatable;

//use Exception;

use Icinga\Module\Trapdirector\Config\TrapModuleConfig;
use Icinga\Module\Trapdirector\Tables\TrapTableList;
use Icinga\Module\Trapdirector\Tables\TrapTableHostList;
use Icinga\Module\Trapdirector\Tables\HandlerTableList;
use Icinga\Module\Trapdirector\Config\MIBLoader;
use Icinga\Module\Trapdirector\TrapsActions\UIDatabase;

use Trapdirector\Trap;

use Icinga\Data\ConfigObject;
use Icinga\Exception\ProgrammingError;

class TrapsController extends Controller
{
	/** @var TrapModuleConfig $moduleConfig TrapModuleConfig instance */
	protected $moduleConfig;
	/** @var TrapTableList $trapTableList (by date)*/
	protected $trapTableList;
	/** @var TrapTableHostList $trapTableHostList TrapTableList (by hosts)*/
	protected $trapTableHostList;
	/** @var HandlerTableList $handlerTableList HandlerTableList instance*/
	protected $handlerTableList;
	/** @var ConfigObject $trapDB Trap database */
	protected $trapDB;
	/** @var ConfigObject $icingaDB Icinga IDO database */
	protected $icingaDB;
	/** @var MIBLoader $MIBData MIBLoader class */
	protected $MIBData;
	/** @var Trap $trapClass Trap class for bin/trap_class.php */
	protected $trapClass;
	/** @var UIDatabase $UIDatabase */
	protected $UIDatabase = NULL;
	/** @var Icinga2API $IcingaAPI */
	protected $icingaAPI = NULL;
	/** @var bool $apiMode connection to icinngaDB is by api (true) od ido DB (false) */
	protected $apiMode = FALSE;
	
	
	
	
	/** Get instance of TrapModuleConfig class
	*	@return TrapModuleConfig
	*/
	public function getModuleConfig() 
	{
		if ($this->moduleConfig == Null) 
		{
			$db_prefix=$this->Config()->get('config', 'database_prefix');
			if ($db_prefix === null) 
			{
				$this->redirectNow('trapdirector/settings?message=No database prefix');
			}
			$this->moduleConfig = new TrapModuleConfig($db_prefix);
		}
		return $this->moduleConfig;
	}
	
	/**
	 * Get instance of TrapTableList
	 * @return \Icinga\Module\Trapdirector\Tables\TrapTableList
	 */
	public function getTrapListTable() {
		if ($this->trapTableList == Null) {
			$this->trapTableList = new TrapTableList();
			$this->trapTableList->setConfig($this->getModuleConfig());
		}
		return $this->trapTableList;
	}
	
	/**
	 * @return \Icinga\Module\Trapdirector\Tables\TrapTableHostList
	 */
	public function getTrapHostListTable()
	{
	    if ($this->trapTableHostList == Null) 
		{
	        $this->trapTableHostList = new TrapTableHostList();
	        $this->trapTableHostList->setConfig($this->getModuleConfig());
	    }
	    return $this->trapTableHostList;
	}
	
	/**
	 * @return \Icinga\Module\Trapdirector\Tables\HandlerTableList
	 */
	public function getHandlerListTable() 
	{
		if ($this->handlerTableList == Null) 
		{
			$this->handlerTableList = new HandlerTableList();
			$this->handlerTableList->setConfig($this->getModuleConfig());
		}
		return $this->handlerTableList;
	}	

	/**
	 * @return UIDatabase
	 */
	public function getUIDatabase()
	{
	    if ($this->UIDatabase == Null)
	    {
	        $this->UIDatabase = new UIDatabase($this);
	       
	    }
	    return $this->UIDatabase;
	}

	/**
	 * Get Ido connection on API (first) or Database as fallback
	 * @return \Icinga\Module\Trapdirector\TrapsActions\UIDatabase|\Icinga\Module\Trapdirector\Icinga2API
	 */
	public function getIdoConn()
	{
	    if ($this->icingaAPI === NULL)
	    {
    	    $host = $this->Config()->get('config', 'icingaAPI_host');
    	    if ($host == '')
    	    {
    	        $this->apiMode = FALSE;
    	        return $this->getUIDatabase();
    	    }
    	    $port = $this->Config()->get('config', 'icingaAPI_port');
    	    $user = $this->Config()->get('config', 'icingaAPI_user');
    	    $pass = $this->Config()->get('config', 'icingaAPI_password');
    	    $this->icingaAPI = new Icinga2API($host,$port);
    	    $this->icingaAPI->setCredentials($user, $pass);
    	    list($ret,$message) = $this->icingaAPI->test($this->getModuleConfig()->getapiUserPermissions());
    	    if ($ret === TRUE) // On error, switch to ido DB
    	    {
    	        $this->apiMode = FALSE;
    	        return $this->getUIDatabase();
    	    }
    	    $this->apiMode = TRUE;
    	    
	    }
	    return $this->icingaAPI;
	    
	}
	
    protected function applyPaginationLimits(Paginatable $paginatable, $limit = 25, $offset = null)
    {
        $limit = $this->params->get('limit', $limit);
        $page = $this->params->get('page', $offset);

        $paginatable->limit($limit, $page > 0 ? ($page - 1) * $limit : 0);

        return $paginatable;
    }	
	
	public function displayExitError($source,$message)
	{	// TODO : check better ways to transmit data (with POST ?)
		$this->redirectNow('trapdirector/error?source='.$source.'&message='.$message);
	}
	
	protected function checkReadPermission()
	{
        if (! $this->Auth()->hasPermission('trapdirector/view')) {
            $this->displayExitError('Permissions','No permission fo view content');
        }		
	}

	protected function checkConfigPermission()
	{
        if (! $this->Auth()->hasPermission('trapdirector/config')) {
            $this->displayExitError('Permissions','No permission fo configure');
        }		
	}
	
    /**
     * Check if user has write permission
     * @param number $check optional : if set to 1, return true (user has permission) or false instead of displaying error page
     * @return boolean : user has permission
     */
	protected function checkModuleConfigPermission($check=0)
	{
        if (! $this->Auth()->hasPermission('trapdirector/module_config')) {
            if ($check == 0)
            {
                $this->displayExitError('Permissions','No permission fo configure module');
            }
            return false;
        }
        return true;
	}

	/*************************  Trap class get **********************/
	public function getTrapClass()
	{ // TODO : try/catch here ? or within caller
		if ($this->trapClass == null)
		{
			require_once($this->Module()->getBaseDir() .'/bin/trap_class.php');
			$icingaweb2_etc=$this->Config()->get('config', 'icingaweb2_etc');
			//$debug_level=4;
			$this->trapClass = new Trap($icingaweb2_etc);
			//$Trap->setLogging($debug_level,'syslog');
		}
		return $this->trapClass;
	}
	
	/************************** MIB related **************************/
	
	/** Get MIBLoader class
	*	@return MIBLoader class
	*/
	protected function getMIB()
	{
		if ($this->MIBData == null)
		{
		    $dbConn = $this->getUIDatabase()->getDbConn();
		    if ($dbConn === null) throw new \ErrorException('uncatched db error');
			$this->MIBData=new MIBLoader(
				$this->Config()->get('config', 'snmptranslate'),
				$this->Config()->get('config', 'snmptranslate_dirs'),
			    $dbConn,
				$this->getModuleConfig()
			);
		}
		return $this->MIBData;
	}	
	
	/**************************  Database queries *******************/		
	
	/** Check if director is installed
	*	@return bool true/false
	*/
	protected function isDirectorInstalled()
	{
	    $output=array();
	    exec('icingacli module list',$output);
	    foreach ($output as $line)
		{
			if (preg_match('/^director .*enabled/',$line))
			{
				return true;
			}
		}
		return false;
	}


	/************************ UI elements **************************/
	
    /**
     * get max rows to display before paging.
     * @return number
     */
	public function itemListDisplay()
	{
	    return $this->getUIDatabase()->getDBConfigValue('max_rows_in_list');
	}

	public function setitemListDisplay(int $maxRows)
	{
	    return $this->getUIDatabase()->setDBConfigValue('max_rows_in_list',$maxRows);
	}
	
    /**
     * get Handlers categories list (index => textvalue).
     * @return array
     */	
	public function getHandlersCategory()
	{
	    //<index>:<name>!<index>:<name>
	    $catList = $this->getUIDatabase()->getDBConfigValue('handler_categories');
	    $catListArray=explode('!',$catList);
	    $retArray=array();
	    foreach ($catListArray as $category)
	    {
	        $catArray=explode(':',$category);
	        $retArray[$catArray[0]] = $catArray[1];
	    }
	    return $retArray; 
	}

	public function setHandlerCategory(array $catArray)
	{
	    $catString='';
	    foreach ($catArray as $index => $value)
	    {
	        if ($catString != '' ) $catString .= '!';
	        $catString .= $index . ':' . $value;
	    }
	    $this->getUIDatabase()->setDBConfigValue('handler_categories', $catString);
	}
	
	public function addHandlersCategory(string $catName)
	{
	    $catArray = $this->getHandlersCategory();
	    $i=1;
	    while (isset($catArray[$i]) && $i < 100) $i++;
	    if ($i == 100) throw new ProgrammingError('Category array error');
	    $catArray[$i] = $catName;
	    $this->setHandlerCategory($catArray);
	}
	
	public function delHandlersCategory(int $catIndex)
	{
	    $catArray = $this->getHandlersCategory();
	    unset($catArray[$catIndex]);
	    $this->setHandlerCategory($catArray);
	    $this->getUIDatabase()->updateHandlersOnCategoryDelete($catIndex);
	}
	
	public function renameHandlersCategory(int $catIndex, string $catName)
	{
	    $catArray = $this->getHandlersCategory();
	    $catArray[$catIndex] = $catName;
	    $this->setHandlerCategory($catArray);
	}
	
}