From 2588b8dbb6a9fdd428cf47a9d6b7aa35e66f236e Mon Sep 17 00:00:00 2001
From: Eric Lippmann <eric.lippmann@netways.de>
Date: Thu, 10 Oct 2013 14:40:13 +0200
Subject: [PATCH] Add monitoring controller class

refs #4823
---
 .../library/Monitoring/Controller.php         | 93 +++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 modules/monitoring/library/Monitoring/Controller.php

diff --git a/modules/monitoring/library/Monitoring/Controller.php b/modules/monitoring/library/Monitoring/Controller.php
new file mode 100644
index 000000000..762cc2a58
--- /dev/null
+++ b/modules/monitoring/library/Monitoring/Controller.php
@@ -0,0 +1,93 @@
+<?php
+// {{{ICINGA_LICENSE_HEADER}}}
+// {{{ICINGA_LICENSE_HEADER}}}
+
+namespace Icinga\Module\Monitoring;
+
+use Icinga\Application\Config as IcingaConfig;
+use Icinga\Module\Monitoring\DataView\HostAndServiceStatus as HostAndServiceStatusView;
+use Icinga\Web\Controller\ActionController;
+
+/**
+ * Base class for all monitoring action controller
+ */
+class Controller extends ActionController
+{
+    /**
+     * Retrieve services from either given parameters or request
+     *
+     * @param   array $params
+     *
+     * @return  \Zend_Paginator
+     */
+    protected function fetchServices(array $params = null)
+    {
+        $columns = array(
+            'host_name',
+            'host_state',
+            'host_state_type',
+            'host_last_state_change',
+            'host_address',
+            'host_handled',
+            'service_description',
+            'service_display_name',
+            'service_state',
+            'service_in_downtime',
+            'service_acknowledged',
+            'service_handled',
+            'service_output',
+            'service_last_state_change',
+            'service_icon_image',
+            'service_long_output',
+            'service_is_flapping',
+            'service_state_type',
+            'service_handled',
+            'service_severity',
+            'service_last_check',
+            'service_notifications_enabled',
+            'service_action_url',
+            'service_notes_url',
+            'service_last_comment',
+            'service_active_checks_enabled',
+            'service_passive_checks_enabled'
+        );
+        if ($params === null) {
+            $query = HostAndServiceStatusView::fromRequest(
+                $this->_request,
+                $columns
+            )->getQuery();
+        } else {
+            $params['backend'] = $this->_request->getParam('backend');
+            $query = HostAndServiceStatusView::fromParams(
+                $params,
+                $columns
+            )->getQuery();
+        }
+        $this->handleFormatRequest($query);
+        return $query->paginate();
+    }
+
+    private function handleFormatRequest($query)
+    {
+        if ($this->_getParam('format') === 'sql'
+            && IcingaConfig::app()->global->get('environment', 'production') === 'development') {
+            echo '<pre>'
+                . htmlspecialchars(wordwrap($query->dump()))
+                . '</pre>';
+            exit;
+        }
+        if ($this->_getParam('format') === 'json'
+            || $this->_request->getHeader('Accept') === 'application/json')
+        {
+            header('Content-type: application/json');
+            echo json_encode($query->fetchAll());
+            exit;
+        }
+        if ($this->_getParam('format') === 'csv'
+            || $this->_request->getHeader('Accept') === 'text/csv') {
+            Csv::fromQuery($query)->dump();
+            exit;
+        }
+    }
+}
+// @codingStandardsIgnoreEnd