1 files changed, 138 insertions, 0 deletions
diff --git a/application/forms/SimulationForm.php b/application/forms/SimulationForm.php
new file mode 100644
index 0000000..04a0f56
--- /dev/null
+++ b/application/forms/SimulationForm.php
@@ -0,0 +1,138 @@
+<?php
+
+namespace Icinga\Module\Businessprocess\Forms;
+
+use Icinga\Module\Businessprocess\MonitoredNode;
+use Icinga\Module\Businessprocess\Simulation;
+use Icinga\Module\Businessprocess\Web\Form\BpConfigBaseForm;
+use Icinga\Web\View;
+
+class SimulationForm extends BpConfigBaseForm
+{
+    /** @var MonitoredNode */
+    protected $node;
+
+    /** @var ?MonitoredNode */
+    protected $simulatedNode;
+
+    /** @var Simulation */
+    protected $simulation;
+
+    public function setup()
+    {
+        $states = $this->enumStateNames();
+
+        // TODO: Fetch state from object
+        if ($this->simulatedNode) {
+            $simulatedState = $this->simulatedNode->getState();
+            $states[$simulatedState] = sprintf(
+                '%s (%s)',
+                $this->node->getStateName($simulatedState),
+                $this->translate('Current simulation')
+            );
+            $node = $this->simulatedNode;
+            $hasSimulation = true;
+        } else {
+            $hasSimulation = false;
+            $node = $this->node;
+        }
+
+        /** @var View $view */
+        $view = $this->getView();
+        if ($hasSimulation) {
+            $title = $this->translate('Modify simulation for %s');
+        } else {
+            $title = $this->translate('Add simulation for %s');
+        }
+        $this->addHtml(
+            '<h2>'
+            . $view->escape(sprintf($title, $node->getAlias() ?? $node->getName()))
+            . '</h2>'
+        );
+
+        $this->addElement('select', 'state', array(
+            'label'        => $this->translate('State'),
+            'multiOptions' => $states,
+            'class'        => 'autosubmit',
+            'value'        => $this->simulatedNode ? $node->getState() : null,
+        ));
+
+        $sentState = $this->getSentValue('state');
+        if (in_array($sentState, array('0', '99'))) {
+            return;
+        }
+
+        if ($hasSimulation || ($sentState !== null && ctype_digit($sentState))) {
+            $this->addElement('checkbox', 'acknowledged', array(
+                'label' => $this->translate('Acknowledged'),
+                'value' => $node->isAcknowledged(),
+            ));
+
+            $this->addElement('checkbox', 'in_downtime', array(
+                'label' => $this->translate('In downtime'),
+                'value' => $node->isInDowntime(),
+            ));
+        }
+
+        $this->setSubmitLabel($this->translate('Apply'));
+    }
+
+    public function setNode($node)
+    {
+        $this->node = $node;
+        return $this;
+    }
+
+    public function setSimulation(Simulation $simulation)
+    {
+        $this->simulation = $simulation;
+
+        $name = $this->node->getName();
+        if ($simulation->hasNode($name)) {
+            $this->simulatedNode = clone($this->node);
+            $s = $simulation->getNode($name);
+            $this->simulatedNode->setState($s->state)
+                ->setAck($s->acknowledged)
+                ->setDowntime($s->in_downtime)
+                ->setMissing(false);
+        }
+
+        return $this;
+    }
+
+    public function onSuccess()
+    {
+        $nodeName = $this->node->getName();
+        $state = $this->getValue('state');
+
+        if ($state !== null && ctype_digit($state)) {
+            $this->notifySuccess($this->translate('Simulation has been set'));
+            $this->simulation->set($nodeName, (object) array(
+                'state'        => $this->getValue('state'),
+                'acknowledged' => $this->getValue('acknowledged'),
+                'in_downtime'  => $this->getValue('in_downtime'),
+            ));
+        } else {
+            if ($this->simulation->remove($nodeName)) {
+                $this->notifySuccess($this->translate('Simulation has been removed'));
+            }
+        }
+
+        parent::onSuccess();
+    }
+
+    /**
+     * @return array
+     */
+    protected function enumStateNames()
+    {
+        $states = array(
+            null => sprintf(
+                $this->translate('Use current state (%s)'),
+                $this->translate($this->node->getStateName())
+            )
+        ) + $this->node->enumStateNames();
+
+        return $states;
+    }
+}