1 files changed, 136 insertions, 0 deletions
diff --git a/application/forms/Command/CommandForm.php b/application/forms/Command/CommandForm.php
new file mode 100644
index 0000000..7fb7966
--- /dev/null
+++ b/application/forms/Command/CommandForm.php
@@ -0,0 +1,136 @@
+<?php
+
+/* Icinga DB Web | (c) 2021 Icinga GmbH | GPLv2 */
+
+namespace Icinga\Module\Icingadb\Forms\Command;
+
+use Exception;
+use Icinga\Application\Logger;
+use Icinga\Module\Icingadb\Command\IcingaCommand;
+use Icinga\Module\Icingadb\Command\Transport\CommandTransport;
+use Icinga\Web\Notification;
+use Icinga\Web\Session;
+use ipl\Html\Form;
+use ipl\Orm\Model;
+use ipl\Web\Common\CsrfCounterMeasure;
+
+abstract class CommandForm extends Form
+{
+    use CsrfCounterMeasure;
+
+    protected $defaultAttributes = ['class' => 'icinga-form icinga-controls'];
+
+    /** @var mixed */
+    protected $objects;
+
+    /**
+     * Whether an error occurred while sending the command
+     *
+     * Prevents the success message from being rendered simultaneously
+     *
+     * @var bool
+     */
+    protected $errorOccurred = false;
+
+    /**
+     * Set the objects to issue the command for
+     *
+     * @param mixed $objects A traversable that is also countable
+     *
+     * @return $this
+     */
+    public function setObjects($objects): self
+    {
+        $this->objects = $objects;
+
+        return $this;
+    }
+
+    /**
+     * Get the objects to issue the command for
+     *
+     * @return mixed
+     */
+    public function getObjects()
+    {
+        return $this->objects;
+    }
+
+    /**
+     * Create and add form elements representing the command's options
+     *
+     * @return void
+     */
+    abstract protected function assembleElements();
+
+    /**
+     * Create and add a submit button to the form
+     *
+     * @return void
+     */
+    abstract protected function assembleSubmitButton();
+
+    /**
+     * Get the command to issue for the given object
+     *
+     * @param Model $object
+     *
+     * @return IcingaCommand|IcingaCommand[]|null NULL in case no command should be issued for the object
+     */
+    abstract protected function getCommand(Model $object);
+
+    protected function assemble()
+    {
+        $this->assembleElements();
+        $this->assembleSubmitButton();
+        $this->addElement($this->createCsrfCounterMeasure(Session::getSession()->getId()));
+    }
+
+    protected function onSuccess()
+    {
+        $errors = [];
+        foreach ($this->getObjects() as $object) {
+            $commands = $this->getCommand($object);
+            if ($commands === null) {
+                continue;
+            }
+
+            if ($commands instanceof IcingaCommand) {
+                $commands = [$commands];
+            }
+
+            foreach ($commands as $command) {
+                try {
+                    $this->sendCommand($command);
+                } catch (Exception $e) {
+                    Logger::error($e->getMessage());
+                    $errors[] = $e->getMessage();
+                }
+            }
+        }
+
+        if (! empty($errors)) {
+            if (count($errors) > 1) {
+                Notification::warning(
+                    t('Some commands were not transmitted. Please check the log. The first error follows.')
+                );
+            }
+
+            $this->errorOccurred = true;
+
+            Notification::error($errors[0]);
+        }
+    }
+
+    /**
+     * Transmit the given command
+     *
+     * @param IcingaCommand $command
+     *
+     * @return void
+     */
+    protected function sendCommand(IcingaCommand $command)
+    {
+        (new CommandTransport())->send($command);
+    }
+}