icingaweb2-module-director/library/Director/Db.php

<?php

namespace Icinga\Module\Director;

use Icinga\Data\Db\DbConnection;

class Db extends DbConnection
{
    protected $modules = array();

    protected static $zoneCache;

    protected static $commandCache;

    protected function db()
    {
        return $this->getDbAdapter();
    }

    public function fetchActivityLogEntryById($id)
    {
        $sql = 'SELECT * FROM director_activity_log WHERE id = ' . (int) $id;

        return $this->db()->fetchRow($sql);
    }

    public function fetchActivityLogEntry($checksum)
    {
        if ($this->getDbType() === 'pgsql') {
            $checksum = new \Zend_Db_Expr("\\x" . bin2hex($checksum));
        }

        $sql = 'SELECT * FROM director_activity_log WHERE checksum = ?';
        $ret = $this->db()->fetchRow($sql, $checksum);

        if (is_resource($ret->checksum)) {
            $ret->checksum = stream_get_contents($ret->checksum);
        }

        if (is_resource($ret->parent_checksum)) {
            $ret->checksum = stream_get_contents($ret->parent_checksum);
        }

        return $ret;
    }

    public function getLastActivityChecksum()
    {
        if ($this->getDbType() === 'pgsql') {
            $select = "SELECT checksum FROM (SELECT * FROM (SELECT 1 AS pos, LOWER(ENCODE(checksum, 'hex')) AS checksum FROM director_activity_log ORDER BY change_time DESC LIMIT 1) a UNION SELECT 2 AS pos, '' AS checksum) u ORDER BY pos LIMIT 1";
        } else {
            $select = "SELECT checksum FROM (SELECT * FROM (SELECT 1 AS pos, LOWER(HEX(checksum)) AS checksum FROM director_activity_log ORDER BY change_time DESC LIMIT 1) a UNION SELECT 2 AS pos, '' AS checksum) u ORDER BY pos LIMIT 1";
        }

        return $this->db()->fetchOne($select);
    }

    public function fetchLatestImportedRows($source, $columns = null)
    {
        $db = $this->db();
        $lastRun = $db->select()->from('import_run', array('rowset_checksum'));

        if (is_int($source) || ctype_digit($source)) {
            $lastRun->where('source_id = ?', $source);
        } else {
            $lastRun->where('source_name = ?', $source);
        }

        $lastRun->order('start_time DESC')->limit(1);
        $checksum = $db->fetchOne($lastRun);

        return $this->fetchImportedRowsetRows($checksum, $columns);
    }

    public function listImportedRowsetColumnNames($checksum)
    {
        $db = $this->db();

        $query = $db->select()->distinct()->from(
            array('p' => 'imported_property'),
            'property_name'
        )->join(
            array('rp' => 'imported_row_property'),
            'rp.property_checksum = p.checksum',
            array()
        )->join(
            array('rsr' => 'imported_rowset_row'),
            'rsr.row_checksum = rp.row_checksum',
            array()
        )->where('rsr.rowset_checksum = ?', $checksum);

        return $db->fetchCol($query);
    }

    public function fetchImportedRowsetRows($checksum, $columns = null)
    {
        $db = $this->db();

        $query = $db->select()->from(
            array('r' => 'imported_row'),
            array()
        )->join(
            array('rsr' => 'imported_rowset_row'),
            'rsr.row_checksum = r.checksum',
            array()
        )->where('rsr.rowset_checksum = ?', $checksum);

        $propertyQuery = $db->select()->from(
            array('rp' => 'imported_row_property'),
            array(
                'property_value' => 'p.property_value',
                'row_checksum'   => 'rp.row_checksum'
            )
        )->join(
            array('p' => 'imported_property'),
            'rp.property_checksum = p.checksum',
            array()
        );

        $fetchColumns = array('object_name' => 'r.object_name');
        if ($columns === null) {
            $columns = $this->listImportedRowsetColumnNames($checksum);
        }

        foreach ($columns as $c) {
            $fetchColumns[$c] = sprintf('p_%s.property_value', $c);
            $p = clone($propertyQuery);
            $query->joinLeft(
                array(sprintf('p_' . $c) => $p->where('p.property_name = ?', $c)),
                sprintf('p_%s.row_checksum = r.checksum', $c),
                array()
            );
        }

        $query->columns($fetchColumns);

        return $db->fetchAll($query);
    }

    public function enumCheckcommands()
    {
        $select = $this->db()->select()->from('icinga_command', array(
            'id',
            'object_name',
        ))->where('object_type IN (?)', array('object', 'external_object'))
          ->where('methods_execute IN (?)', array('PluginCheck', 'IcingaCheck'))
          ->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function enumCommands()
    {
        if (self::$commandCache === null) {
            $select = $this->db()->select()->from('icinga_command', array(
                'id',
                'object_name',
            ))
              ->order('object_name ASC');

            self::$commandCache = $this->db()->fetchPairs($select);
        }
        return self::$commandCache;
    }

    public function enumZones()
    {
        if (self::$zoneCache === null) {
            $select = $this->db()->select()->from('icinga_zone', array(
                'id',
                'object_name',
            ))->where('object_type = ?', 'object')->order('object_name ASC');

            self::$zoneCache = $this->db()->fetchPairs($select);
        }

        return self::$zoneCache;
    }

    public function getZoneName($id)
    {
        $objects = $this->enumZones();
        return $objects[$id];
    }

    public function getCommandName($id)
    {
        $objects = $this->enumCommands();
        return $objects[$id];
    }

    public function enumHosts()
    {
        $select = $this->db()->select()->from('icinga_host', array(
            'id',
            'object_name',
        ))->where('object_type = ?', 'object')->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function enumHostgroups()
    {
        $select = $this->db()->select()->from('icinga_hostgroup', array(
            'id',
            'object_name',
        ))->where('object_type = ?', 'object')->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function enumServices()
    {
        $select = $this->db()->select()->from('icinga_service', array(
            'id',
            'object_name',
        ))->where('object_type = ?', 'object')->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function enumServicegroups()
    {
        $select = $this->db()->select()->from('icinga_servicegroup', array(
            'id',
            'object_name',
        ))->where('object_type = ?', 'object')->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function enumUsers()
    {
        $select = $this->db()->select()->from('icinga_user', array(
            'id',
            'object_name',
        ))->where('object_type = ?', 'object')->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function enumUsergroups()
    {
        $select = $this->db()->select()->from('icinga_usergroup', array(
            'id',
            'object_name',
        ))->where('object_type = ?', 'object')->order('object_name ASC');
        return $this->db()->fetchPairs($select);
    }

    public function clearZoneCache()
    {
        // TODO: wipe cache on update/insert/delete
        self::$zoneCache = null;
    }

    public function clearCommandCache()
    {
        // TODO: wipe cache on update/insert/delete
        self::$commandCache = null;
    }
}