icingaweb2/modules/monitoring/application/forms/Config/InstanceConfigForm.php

<?php
// {{{ICINGA_LICENSE_HEADER}}}
// {{{ICINGA_LICENSE_HEADER}}}

namespace Icinga\Module\Monitoring\Form\Config;

use InvalidArgumentException;
use Icinga\Exception\ConfigurationError;
use Icinga\Form\ConfigForm;
use Icinga\Module\Monitoring\Command\Transport\LocalCommandFile;
use Icinga\Module\Monitoring\Command\Transport\RemoteCommandFile;
use Icinga\Module\Monitoring\Form\Config\Instance\LocalInstanceForm;
use Icinga\Module\Monitoring\Form\Config\Instance\RemoteInstanceForm;
use Icinga\Web\Notification;
use Icinga\Web\Request;

/**
 * Form for modifying/creating monitoring instances
 */
class InstanceConfigForm extends ConfigForm
{
    /**
     * (non-PHPDoc)
     * @see Form::init() For the method documentation.
     */
    public function init()
    {
        $this->setName('form_config_monitoring_instance');
        $this->setSubmitLabel(mt('monitoring', 'Save Changes'));
    }

    /**
     * Get a form object for the given instance type
     *
     * @param   string $type                The instance type for which to return a form
     *
     * @return  LocalInstanceForm|RemoteInstanceForm
     *
     * @throws  InvalidArgumentException    In case the given instance type is invalid
     */
    public function getInstanceForm($type)
    {
        switch (strtolower($type)) {
            case LocalCommandFile::TRANSPORT:
                $form = new LocalInstanceForm();
                break;
            case RemoteCommandFile::TRANSPORT;
                $form = new RemoteInstanceForm();
                break;
            default:
                throw new InvalidArgumentException(
                    sprintf(mt('monitoring', 'Invalid instance type "%s" given'), $type)
                );
        }
        return $form;
    }

    /**
     * Add a new instance
     *
     * The resource to add is identified by the array-key `name'.
     *
     * @param   array   $values             The values to extend the configuration with
     *
     * @return  self
     *
     * @throws  InvalidArgumentException    In case the resource already exists
     */
    public function add(array $values)
    {
        $name = isset($values['name']) ? $values['name'] : '';
        if (! $name) {
            throw new InvalidArgumentException(mt('monitoring', 'Instance name missing'));
        }
        if (isset($this->config->{$name})) {
            throw new InvalidArgumentException(mt('monitoring', 'Instance already exists'));
        }

        unset($values['name']);
        $this->config->{$name} = $values;
        return $this;
    }

    /**
     * Edit an existing instance
     *
     * @param   string      $name           The name of the resource to edit
     * @param   array       $values         The values to edit the configuration with
     *
     * @return  array                       The edited resource configuration
     *
     * @throws  InvalidArgumentException    In case the resource name is missing or invalid
     */
    public function edit($name, array $values)
    {
        if (! $name) {
            throw new InvalidArgumentException(mt('monitoring', 'Old instance name missing'));
        } elseif (! ($newName = isset($values['name']) ? $values['name'] : '')) {
            throw new InvalidArgumentException(mt('monitoring', 'New instance name missing'));
        } elseif (! ($instanceConfig = $this->config->get($name))) {
            throw new InvalidArgumentException(mt('monitoring', 'Unknown instance name provided'));
        }

        unset($values['name']);
        unset($this->config->{$name});
        $this->config->{$newName} = $values;
        return $this->config->{$newName};
    }

    /**
     * Remove a instance
     *
     * @param   string      $name           The name of the resource to remove
     *
     * @return  array                       The removed resource configuration
     *
     * @throws  InvalidArgumentException    In case the resource name is missing or invalid
     */
    public function remove($name)
    {
        if (! $name) {
            throw new InvalidArgumentException(mt('monitoring', 'Instance name missing'));
        } elseif (! ($instanceConfig = $this->config->get($name))) {
            throw new InvalidArgumentException(mt('monitoring', 'Unknown instance name provided'));
        }

        unset($this->config->{$name});
        return $instanceConfig;
    }

    /**
     * @see     Form::onRequest()   For the method documentation.
     * @throws  ConfigurationError  In case the instance name is missing or invalid
     */
    public function onRequest(Request $request)
    {
        $instanceName = $request->getQuery('instance');
        if ($instanceName !== null) {
            if (! $instanceName) {
                throw new ConfigurationError(mt('monitoring', 'Instance name missing'));
            }
            if (! isset($this->config->{$instanceName})) {
                throw new ConfigurationError(mt('monitoring', 'Unknown instance name given'));
            }

            $instanceConfig = $this->config->{$instanceName}->toArray();
            $instanceConfig['name'] = $instanceName;
            $this->populate($instanceConfig);
        }
    }

    /**
     * (non-PHPDoc)
     * @see Form::onSuccess() For the method documentation.
     */
    public function onSuccess(Request $request)
    {
        $instanceName = $request->getQuery('instance');
        try {
            if ($instanceName === null) { // create new instance
                $this->add($this->getValues());
                $message = mt('monitoring', 'Instance "%s" created successfully.');
            } else { // edit existing instance
                $this->edit($instanceName, $this->getValues());
                $message = mt('monitoring', 'Instance "%s" edited successfully.');
            }
        } catch (InvalidArgumentException $e) {
            Notification::error($e->getMessage());
            return;
        }

        if ($this->save()) {
            Notification::success(sprintf($message, $this->getElement('name')->getValue()));
        } else {
            return false;
        }
    }

    /**
     * (non-PHPDoc)
     * @see Form::createElements() For the method documentation.
     */
    public function createElements(array $formData = array())
    {
        $instanceType = isset($formData['transport']) ? $formData['transport'] : LocalCommandFile::TRANSPORT;

        $this->addElements(array(
            array(
                'text',
                'name',
                array(
                    'required'  => true,
                    'label'     => mt('monitoring', 'Instance Name')
                )
            ),
            array(
                'select',
                'transport',
                array(
                    'required'      => true,
                    'autosubmit'    => true,
                    'label'         => mt('monitoring', 'Instance Type'),
                    'multiOptions'  => array(
                        LocalCommandFile::TRANSPORT     => mt('monitoring', 'Local Command File'),
                        RemoteCommandFile::TRANSPORT    => mt('monitoring', 'Remote Command File')
                    ),
                    'value' => $instanceType
                )
            )
        ));

        $this->addElements($this->getInstanceForm($instanceType)->createElements($formData)->getElements());
    }
}