icingaweb2-module-businessprocess/application/forms/BpConfigForm.php

<?php

namespace Icinga\Module\Businessprocess\Forms;

use Icinga\Application\Config;
use Icinga\Module\Businessprocess\BusinessProcess;
use Icinga\Module\Businessprocess\Form;
use Icinga\Web\Notification;
use Icinga\Web\Request;
use Icinga\Web\Url;

class BpConfigForm extends Form
{
    protected $storage;

    protected $backend;

    protected $config;

    protected $node;

    protected $objectList = array();

    protected $processList = array();

    public function setup()
    {
        $this->addElement('text', 'name', array(
            'label'       => $this->translate('Name'),
            'required'    => true,
            'description' => $this->translate(
                'This is the unique identifier of this process'
            ),
        ));

        $this->addElement('text', 'title', array(
            'label'       => $this->translate('Title'),
            'description' => $this->translate(
                'Usually this title will be shown for this process. Equals name'
              . ' if not given'
            ),
        ));

        $this->addElement('select', 'backend_name', array(
            'label'       => $this->translate('Backend'),
            'description' => $this->translate(
                'Icinga Web Monitoring Backend where current object states for'
              . ' this process should be retrieved from'
            ),
            'multiOptions' => array(
                null => $this->translate('Use the configured default backend'),
            ) + $this->listAvailableBackends()
        ));

        $this->addElement('select', 'state_type', array(
            'label'       => $this->translate('State Type'),
            'required'    => true,
            'description' => $this->translate(
                'Whether this process should be based on Icinga hard or soft states'
            ),
            'multiOptions' => array(
                'hard' => $this->translate('Use HARD states'),
                'soft' => $this->translate('Use SOFT states'),
            )
        ));
        $this->addElement('submit', $this->translate('Store'));
    }

    protected function listAvailableBackends()
    {
        $keys = array_keys(Config::module('monitoring', 'backends')->toArray());
        return array_combine($keys, $keys);
    }

    public function setStorage($storage)
    {
        $this->storage = $storage;
        return $this;
    }

    public function setProcessConfig($config)
    {
        $this->config = $config;
        $this->getElement('name')->setValue($config->getName());
        $this->getElement('name')->setAttrib('readonly', true);

        if ($config->hasTitle()) {
            $this->getElement('title')->setValue($config->getTitle());
        }

        if ($config->hasBackend()) {
            $this->getElement('backend_name')->setValue(
                $config->getBackend()->getName()
            );
        }

        if ($config->usesSoftStates()) {
            $this->getElement('state_type')->setValue('soft');
        } else {
            $this->getElement('state_type')->setValue('hard');
        }

        return $this;
    }

    public function onSuccess()
    {
        $name    = $this->getValue('name');
        $title   = $this->getValue('title');
        $backend = $this->getValue('backend');

        if ($this->config === null) {
            // New config
            $config = new BusinessProcess();
            $config->setName($name);
            if ($title) {
                $config->setTitle($title);
            }
            if ($backend) {
                $config->setBackendName($backend);
            }
            if ($this->getValue('state_type') === 'soft') {
                $config->useSoftStates();
            } else {
                $config->useHardStates();
            }
            $this->storage->storeProcess($config);
            $config->clearAppliedChanges();
            $this->setRedirectUrl(
                $this->getRedirectUrl()->setParams(
                    array('config' => $name, 'unlocked' => true)
                )
            );

            Notification::success(sprintf('Process %s has been created', $name));
        } else {
            // Existing config
            $config = $this->config;

            if ($title) {
                $config->setTitle($title);
            }
            if ($backend) {
                $config->setBackendName($backend);
            }
            if ($this->getValue('state_type') === 'soft') {
                $config->useSoftStates();
            } else {
                $config->useHardStates();
            }

            $this->storage->storeProcess($config);
            $config->clearAppliedChanges();
            $this->getRedirectUrl()->setParam('config', $name);
            Notification::success(sprintf('Process %s has been stored', $name));
        }
    }
}