1 files changed, 110 insertions, 0 deletions
diff --git a/application/forms/IcingaScheduledDowntimeRangeForm.php b/application/forms/IcingaScheduledDowntimeRangeForm.php
new file mode 100644
index 0000000..b5f95d0
--- /dev/null
+++ b/application/forms/IcingaScheduledDowntimeRangeForm.php
@@ -0,0 +1,110 @@
+<?php
+
+namespace Icinga\Module\Director\Forms;
+
+use Icinga\Module\Director\Objects\IcingaObject;
+use Icinga\Module\Director\Objects\IcingaScheduledDowntime;
+use Icinga\Module\Director\Objects\IcingaScheduledDowntimeRange;
+use Icinga\Module\Director\Web\Form\DirectorObjectForm;
+
+class IcingaScheduledDowntimeRangeForm extends DirectorObjectForm
+{
+    /** @var IcingaScheduledDowntime */
+    private $downtime;
+
+    /**
+     * @throws \Zend_Form_Exception
+     */
+    public function setup()
+    {
+        $this->addHidden('scheduled_downtime_id', $this->downtime->get('id'));
+        $this->addElement('text', 'range_key', [
+            'label'       => $this->translate('Day(s)'),
+            'description' => $this->translate(
+                'Might be monday, tuesday or 2016-01-28 - have a look at the documentation for more examples'
+            ),
+        ]);
+
+        $this->addElement('text', 'range_value', [
+            'label'       => $this->translate('Timeperiods'),
+            'description' => $this->translate(
+                'One or more time periods, e.g. 00:00-24:00 or 00:00-09:00,17:00-24:00'
+            ),
+        ]);
+
+        $this->setButtons();
+    }
+
+    public function setScheduledDowntime(IcingaScheduledDowntime $downtime)
+    {
+        $this->downtime = $downtime;
+        $this->setDb($downtime->getConnection());
+        return $this;
+    }
+
+    /**
+     * @param IcingaScheduledDowntimeRange $object
+     * @throws \Icinga\Module\Director\Exception\DuplicateKeyException
+     */
+    protected function deleteObject($object)
+    {
+        $key = $object->get('range_key');
+        $downtime = $this->downtime;
+        $downtime->ranges()->remove($key);
+        $downtime->store();
+        $msg = sprintf(
+            $this->translate('Time range "%s" has been removed from %s'),
+            $key,
+            $downtime->getObjectName()
+        );
+
+        $url = $this->getSuccessUrl()->without(
+            ['range', 'range_type']
+        );
+
+        $this->setSuccessUrl($url);
+        $this->redirectOnSuccess($msg);
+    }
+
+    /**
+     * @throws \Icinga\Module\Director\Exception\DuplicateKeyException
+     */
+    public function onSuccess()
+    {
+        $object = $this->object();
+        if ($object->hasBeenModified()) {
+            $this->downtime->ranges()->setRange(
+                $this->getValue('range_key'),
+                $this->getValue('range_value')
+            );
+        }
+
+        if ($this->downtime->hasBeenModified()) {
+            if (! $object->hasBeenLoadedFromDb()) {
+                $this->setHttpResponseCode(201);
+            }
+
+            $msg = sprintf(
+                $object->hasBeenLoadedFromDb()
+                ? $this->translate('The %s has successfully been stored')
+                : $this->translate('A new %s has successfully been created'),
+                $this->translate($this->getObjectShortClassName())
+            );
+
+            $this->downtime->store($this->db);
+        } else {
+            if ($this->isApiRequest()) {
+                $this->setHttpResponseCode(304);
+            }
+            $msg = $this->translate('No action taken, object has not been modified');
+        }
+        if ($object instanceof IcingaObject) {
+            $this->setSuccessUrl(
+                'director/' . strtolower($this->getObjectShortClassName()),
+                $object->getUrlParams()
+            );
+        }
+
+        $this->redirectOnSuccess($msg);
+    }
+}