1 files changed, 137 insertions, 0 deletions
diff --git a/application/forms/Command/Object/ScheduleCheckForm.php b/application/forms/Command/Object/ScheduleCheckForm.php
new file mode 100644
index 0000000..9b32ea1
--- /dev/null
+++ b/application/forms/Command/Object/ScheduleCheckForm.php
@@ -0,0 +1,137 @@
+<?php
+
+/* Icinga DB Web | (c) 2021 Icinga GmbH | GPLv2 */
+
+namespace Icinga\Module\Icingadb\Forms\Command\Object;
+
+use DateInterval;
+use DateTime;
+use Generator;
+use Icinga\Module\Icingadb\Command\Object\ScheduleCheckCommand;
+use Icinga\Module\Icingadb\Forms\Command\CommandForm;
+use Icinga\Module\Icingadb\Model\Host;
+use Icinga\Web\Notification;
+use ipl\Html\Attributes;
+use ipl\Html\HtmlElement;
+use ipl\Html\Text;
+use ipl\Web\FormDecorator\IcingaFormDecorator;
+use ipl\Web\Widget\Icon;
+use Traversable;
+
+use function ipl\Stdlib\iterable_value_first;
+
+class ScheduleCheckForm extends CommandForm
+{
+    public function __construct()
+    {
+        $this->on(self::ON_SUCCESS, function () {
+            if ($this->errorOccurred) {
+                return;
+            }
+
+            $countObjects = count($this->getObjects());
+            if (iterable_value_first($this->getObjects()) instanceof Host) {
+                $message = sprintf(
+                    tp('Scheduled check successfully', 'Scheduled check for %d hosts successfully', $countObjects),
+                    $countObjects
+                );
+            } else {
+                $message = sprintf(
+                    tp('Scheduled check successfully', 'Scheduled check for %d services successfully', $countObjects),
+                    $countObjects
+                );
+            }
+
+            Notification::success($message);
+        });
+    }
+
+    protected function assembleElements()
+    {
+        $this->addHtml(new HtmlElement(
+            'div',
+            Attributes::create(['class' => 'form-description']),
+            new Icon('info-circle', ['class' => 'form-description-icon']),
+            new HtmlElement(
+                'ul',
+                null,
+                new HtmlElement(
+                    'li',
+                    null,
+                    Text::create(t(
+                        'This command is used to schedule the next check of hosts or services. Icinga'
+                        . ' will re-queue the hosts or services to be checked at the time you specify.'
+                    ))
+                )
+            )
+        ));
+
+        $decorator = new IcingaFormDecorator();
+
+        $this->addElement(
+            'localDateTime',
+            'check_time',
+            [
+                'data-use-datetime-picker'  => true,
+                'required'                  => true,
+                'label'                     => t('Check Time'),
+                'description'               => t('Set the date and time when the check should be scheduled.'),
+                'value'                     => (new DateTime())->add(new DateInterval('PT1H'))
+            ]
+        );
+        $decorator->decorate($this->getElement('check_time'));
+
+        $this->addElement(
+            'checkbox',
+            'force_check',
+            [
+                'label'         => t('Force Check'),
+                'description'   => t(
+                    'If you select this option, Icinga will force a check regardless of both what time the'
+                    . ' scheduled check occurs and whether or not checks are enabled.'
+                )
+            ]
+        );
+        $decorator->decorate($this->getElement('force_check'));
+    }
+
+    protected function assembleSubmitButton()
+    {
+        $this->addElement(
+            'submit',
+            'btn_submit',
+            [
+                'required'  => true,
+                'label'     => tp('Schedule check', 'Schedule checks', count($this->getObjects()))
+            ]
+        );
+
+        (new IcingaFormDecorator())->decorate($this->getElement('btn_submit'));
+    }
+
+    protected function getCommands(Traversable $objects): Traversable
+    {
+        $granted = (function () use ($objects): Generator {
+            foreach ($objects as $object) {
+                if (
+                    $this->isGrantedOn('icingadb/command/schedule-check', $object)
+                    || (
+                        $object->active_checks_enabled
+                        && $this->isGrantedOn('icingadb/command/schedule-check/active-only', $object)
+                    )
+                ) {
+                    yield $object;
+                }
+            }
+        })();
+
+        if ($granted->valid()) {
+            $command = new ScheduleCheckCommand();
+            $command->setObjects($granted);
+            $command->setForced($this->getElement('force_check')->isChecked());
+            $command->setCheckTime($this->getValue('check_time')->getTimestamp());
+
+            yield $command;
+        }
+    }
+}