1 files changed, 135 insertions, 0 deletions
diff --git a/library/Director/Objects/IcingaScheduledDowntime.php b/library/Director/Objects/IcingaScheduledDowntime.php
new file mode 100644
index 0000000..7fc3f78
--- /dev/null
+++ b/library/Director/Objects/IcingaScheduledDowntime.php
@@ -0,0 +1,135 @@
+<?php
+
+namespace Icinga\Module\Director\Objects;
+
+use Icinga\Module\Director\IcingaConfig\IcingaConfigHelper as c;
+use RuntimeException;
+
+class IcingaScheduledDowntime extends IcingaObject
+{
+    protected $table = 'icinga_scheduled_downtime';
+
+    protected $defaultProperties = [
+        'id'                => null,
+        'uuid'              => null,
+        'zone_id'           => null,
+        'object_name'       => null,
+        'object_type'       => null,
+        'disabled'          => 'n',
+        'author'            => null,
+        'comment'           => null,
+        'fixed'             => null,
+        'duration'          => null,
+        'apply_to'          => null,
+        'assign_filter'     => null,
+        'with_services'     => null,
+    ];
+
+    protected $uuidColumn = 'uuid';
+
+    protected $supportsImports = true;
+
+    protected $supportsRanges = true;
+
+    protected $supportsApplyRules = true;
+
+    protected $relations = [
+        'zone' => 'IcingaZone',
+    ];
+
+    protected $booleans = [
+        'fixed' => 'fixed',
+    ];
+
+    protected $intervalProperties = [
+        'duration' => 'duration',
+    ];
+
+    protected $propertiesNotForRendering = [
+        'id',
+        'apply_to',
+        'object_name',
+        'object_type',
+        'with_services',
+    ];
+
+    /**
+     * @return string
+     */
+    protected function renderObjectHeader()
+    {
+        if ($this->isApplyRule()) {
+            if (($to = $this->get('apply_to')) === null) {
+                throw new RuntimeException(sprintf(
+                    'Applied notification "%s" has no valid object type',
+                    $this->getObjectName()
+                ));
+            }
+
+            return sprintf(
+                "%s %s %s to %s {\n",
+                $this->getObjectTypeName(),
+                $this->getType(),
+                c::renderString($this->getObjectName()),
+                ucfirst($to)
+            );
+        } else {
+            return parent::renderObjectHeader();
+        }
+    }
+
+    public function getOnDeleteUrl()
+    {
+        if ($this->isApplyRule()) {
+            return 'director/scheduled-downtimes/applyrules';
+        } elseif ($this->isTemplate()) {
+            return 'director/scheduled-downtimes/templates';
+        } else {
+            return 'director/scheduled-downtimes';
+        }
+    }
+
+    public function isActive($now = null)
+    {
+        if ($now === null) {
+            $now = time();
+        }
+
+        foreach ($this->ranges()->getRanges() as $range) {
+            if ($range->isActive($now)) {
+                return true;
+            }
+        }
+
+        // TODO: no range currently means (and renders) "never", Icinga behaves
+        //       different. Figure out whether and how we should support this
+        return false;
+    }
+
+    /**
+     * @return string
+     */
+    protected function renderSuffix()
+    {
+        if ($this->get('with_services') === 'y' && $this->get('apply_to') === 'host') {
+            return parent::renderSuffix() . $this->renderCloneForServices();
+        } else {
+            return parent::renderSuffix();
+        }
+    }
+
+    protected function prefersGlobalZone()
+    {
+        return false;
+    }
+
+    protected function renderCloneForServices()
+    {
+        $services = clone($this);
+        $services
+            ->set('with_services', 'n')
+            ->set('apply_to', 'service');
+
+        return $services->toConfigString();
+    }
+}