1 files changed, 184 insertions, 0 deletions
diff --git a/library/Eventdb/Eventdb.php b/library/Eventdb/Eventdb.php
new file mode 100644
index 0000000..b885258
--- /dev/null
+++ b/library/Eventdb/Eventdb.php
@@ -0,0 +1,184 @@
+<?php
+/* Icinga Web 2 | (c) 2016 Icinga Development Team | GPLv2+ */
+
+namespace Icinga\Module\Eventdb;
+
+use Icinga\Application\Config;
+use Icinga\Data\ConfigObject;
+use Icinga\Data\Filter\Filter;
+use Icinga\Data\Filter\FilterExpression;
+use Icinga\Data\ResourceFactory;
+use Icinga\Exception\ConfigurationError;
+use Icinga\Repository\DbRepository;
+use Icinga\Repository\RepositoryQuery;
+
+class Eventdb extends DbRepository
+{
+    /**
+     * {@inheritdoc}
+     */
+    const DATETIME_FORMAT = 'Y-m-d G:i:s';
+
+    /**
+     * {@inheritdoc}
+     */
+    protected $tableAliases = array(
+        'comment' => 'c',
+        'event'   => 'e',
+    );
+
+    /**
+     * Default query columns
+     *
+     * @var array
+     */
+    protected static $defaultQueryColumns = array(
+        'event'   => array(
+            'id',
+            'host_name',
+            'host_address',
+            'type',
+            'facility',
+            'priority',
+            'program',
+            'message',
+            'alternative_message',
+            'ack',
+            'created',
+            'modified',
+        ),
+        'comment' => array(
+            'id',
+            'event_id',
+            'type',
+            'message',
+            'created',
+            'modified',
+            'user'
+        )
+    );
+
+    protected static $edbcQueryColumns = array(
+        'event' => array(
+            'group_active',
+            'group_id',
+            'group_count',
+            'group_leader',
+            'group_autoclear',
+            'flags',
+            'alternative_message'
+        )
+    );
+
+    /** @var bool */
+    protected $hasCorrelatorExtensions = null;
+
+    /**
+     * Checks if Event repository has EDBC columns
+     *
+     * @return bool
+     */
+    public function hasCorrelatorExtensions()
+    {
+        if ($this->hasCorrelatorExtensions === null) {
+            $dba = $this->getDataSource()->getDbAdapter();
+            $result = $dba->fetchRow("SHOW COLUMNS FROM `event` LIKE 'group_leader'");
+
+            $this->hasCorrelatorExtensions = ! ! $result;
+        }
+        return $this->hasCorrelatorExtensions;
+    }
+
+    public function filterGroups(RepositoryQuery $query)
+    {
+        if ($this->hasCorrelatorExtensions()) {
+            $query->addFilter(Filter::matchAny(
+                Filter::expression('group_leader', '=', -1),
+                new FilterExpression('group_leader', 'IS', new \Zend_Db_Expr('NULL'))
+            ));
+        }
+        return $this;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function initializeQueryColumns()
+    {
+        $additionalColumns = Config::module('eventdb', 'columns')->keys();
+        $queryColumns = static::$defaultQueryColumns;
+        if ($this->hasCorrelatorExtensions()) {
+            foreach (static::$edbcQueryColumns as $table => $fields) {
+                if (array_key_exists($table, $queryColumns)) {
+                    $queryColumns[$table] = array_merge($queryColumns[$table], $fields);
+                } else {
+                    $queryColumns[$table] = $fields;
+                }
+            }
+        }
+        if ($additionalColumns !== null) {
+            $eventColumns = $queryColumns['event'];
+            $queryColumns['event'] = array_merge($eventColumns, array_diff($additionalColumns, $eventColumns));
+        }
+        return $queryColumns;
+    }
+
+    /**
+     * Create and return a new instance of the Eventdb
+     *
+     * @param   ConfigObject    $config     The configuration to use, otherwise the module's configuration
+     *
+     * @return  static
+     *
+     * @throws  ConfigurationError          In case no resource has been configured in the module's configuration
+     */
+    public static function fromConfig(ConfigObject $config = null)
+    {
+        if ($config === null) {
+            $moduleConfig = Config::module('eventdb');
+            if (($resourceName = $moduleConfig->get('backend', 'resource')) === null) {
+                throw new ConfigurationError(
+                    mt('eventdb', 'You need to configure a resource to access the EventDB database first')
+                );
+            }
+
+            $resource = ResourceFactory::create($resourceName);
+        } else {
+            $resource = ResourceFactory::createResource($config);
+        }
+
+        return new static($resource);
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function initializeConversionRules()
+    {
+        return array('event' => array('host_address' => 'ip_address'));
+    }
+
+    /**
+     * Convert an IP address into its human-readable form
+     *
+     * @param   string  $rawAddress
+     *
+     * @return  string
+     */
+    protected function retrieveIpAddress($rawAddress)
+    {
+        return $rawAddress === null ? null : inet_ntop($rawAddress);
+    }
+
+    /**
+     * Convert an IP address into its binary form
+     *
+     * @param   string  $address
+     *
+     * @return  string
+     */
+    protected function persistIpAddress($address)
+    {
+        return $address === null ? null : inet_pton($address);
+    }
+}