icingaweb2/application/controllers/NavigationController.php

<?php
/* Icinga Web 2 | (c) 2013-2015 Icinga Development Team | GPLv2+ */

namespace Icinga\Controllers;

use Exception;
use Icinga\Application\Config;
use Icinga\Application\Icinga;
use Icinga\Exception\NotFoundError;
use Icinga\Data\DataArray\ArrayDatasource;
use Icinga\Forms\ConfirmRemovalForm;
use Icinga\Forms\Navigation\NavigationConfigForm;
use Icinga\Web\Controller;
use Icinga\Web\Form;
use Icinga\Web\Notification;
use Icinga\Web\Url;

/**
 * Navigation configuration
 */
class NavigationController extends Controller
{
    /**
     * The default item types provided by Icinga Web 2
     *
     * @var array
     */
    protected $defaultItemTypes;

    /**
     * {@inheritdoc}
     */
    public function init()
    {
        parent::init();

        $this->defaultItemTypes = array(
            'menu-item' => $this->translate('Menu Entry'),
            'dashlet'   => 'Dashlet'
        );
    }

    /**
     * Return a list of available navigation item types
     *
     * @return  array
     */
    protected function listItemTypes()
    {
        $types = $this->defaultItemTypes;
        foreach (Icinga::app()->getModuleManager()->getLoadedModules() as $module) {
            $moduleTypes = $module->getNavigationItems();
            if (! empty($moduleTypes)) {
                $types = array_merge($types, $moduleTypes);
            }
        }

        return $types;
    }

    /**
     * Show the current user a list of his/her navigation items
     */
    public function indexAction()
    {
        $user = $this->Auth()->getUser();

        $ds = new ArrayDatasource(array_merge(
            Config::app('navigation')->select()->where('owner', $user->getUsername())->fetchAll(),
            iterator_to_array($user->loadNavigationConfig())
        ));
        $ds->setKeyColumn('name');
        $query = $ds->select();

        $this->view->types = $this->listItemTypes();
        $this->view->items = $query;

        $this->getTabs()->add(
            'navigation',
            array(
                'title'     => $this->translate('List and configure your own navigation items'),
                'label'     => $this->translate('Navigation'),
                'url'       => 'navigation'
            )
        )->activate('navigation');
        $this->setupSortControl(
            array(
                'type'  => $this->translate('Type'),
                'name'  => $this->translate('Shared Navigation')
            ),
            $query
        );
    }

    /**
     * List all shared navigation items
     */
    public function sharedAction()
    {
        $this->assertPermission('config/application/navigation');
        $config = Config::app('navigation');
        $config->getConfigObject()->setKeyColumn('name');
        $query = $config->select();

        $removeForm = new Form();
        $removeForm->setUidDisabled();
        $removeForm->setAction(Url::fromPath('navigation/unshare'));
        $removeForm->addElement('hidden', 'name', array(
            'decorators'    => array('ViewHelper')
        ));
        $removeForm->addElement('hidden', 'redirect', array(
            'value'         => Url::fromPath('navigation/shared'),
            'decorators'    => array('ViewHelper')
        ));
        $removeForm->addElement('button', 'btn_submit', array(
            'escape'        => false,
            'type'          => 'submit',
            'class'         => 'link-like spinner',
            'value'         => 'btn_submit',
            'decorators'    => array('ViewHelper'),
            'label'         => $this->view->icon('trash'),
            'title'         => $this->translate('Unshare this navigation item')
        ));

        $this->view->removeForm = $removeForm;
        $this->view->types = $this->listItemTypes();
        $this->view->items = $query;

        $this->getTabs()->add(
            'navigation/shared',
            array(
                'title'     => $this->translate('List and configure shared navigation items'),
                'label'     => $this->translate('Shared Navigation'),
                'url'       => 'navigation/shared'
            )
        )->activate('navigation/shared');
        $this->setupSortControl(
            array(
                'type'  => $this->translate('Type'),
                'owner' => $this->translate('Owner'),
                'name'  => $this->translate('Shared Navigation')
            ),
            $query
        );
    }

    /**
     * Add a navigation item
     */
    public function addAction()
    {
        $form = new NavigationConfigForm();
        $form->setRedirectUrl('navigation');
        $form->setItemTypes($this->listItemTypes());
        $form->setTitle($this->translate('Create New Navigation Item'));
        $form->addDescription($this->translate('Create a new navigation item, such as a menu entry or dashlet.'));
        $form->setUser($this->Auth()->getUser());
        $form->setShareConfig(Config::app('navigation'));
        $form->setOnSuccess(function (NavigationConfigForm $form) {
            try {
                $form->add(array_filter($form->getValues()));
            } catch (Exception $e) {
                $form->error($e->getMessage());
                return false;
            }

            if ($form->save()) {
                Notification::success(t('Navigation item successfully created'));
                return true;
            }

            return false;
        });
        $form->handleRequest();

        $this->view->form = $form;
        $this->render('form');
    }

    /**
     * Edit a navigation item
     */
    public function editAction()
    {
        $itemName = $this->params->getRequired('name');

        $form = new NavigationConfigForm();
        $form->setRedirectUrl('navigation');
        $form->setItemTypes($this->listItemTypes());
        $form->setTitle(sprintf($this->translate('Edit Navigation Item %s'), $itemName));
        $form->setUser($this->Auth()->getUser());
        $form->setShareConfig(Config::app('navigation'));
        $form->setOnSuccess(function (NavigationConfigForm $form) use ($itemName) {
            try {
                $form->edit($itemName, array_map(
                    function ($v) {
                        return $v !== '' ? $v : null;
                    },
                    $form->getValues()
                ));
            } catch (NotFoundError $e) {
                throw $e;
            } catch (Exception $e) {
                $form->error($e->getMessage());
                return false;
            }

            if ($form->save()) {
                Notification::success(sprintf(t('Navigation item "%s" successfully updated'), $itemName));
                return true;
            }

            return false;
        });

        try {
            $form->load($itemName);
            $form->handleRequest();
        } catch (NotFoundError $_) {
            $this->httpNotFound(sprintf($this->translate('Navigation item "%s" not found'), $itemName));
        }

        $this->view->form = $form;
        $this->render('form');
    }

    /**
     * Remove a navigation item
     */
    public function removeAction()
    {
        $itemName = $this->params->getRequired('name');

        $navigationConfigForm = new NavigationConfigForm();
        $navigationConfigForm->setUser($this->Auth()->getUser());
        $navigationConfigForm->setShareConfig(Config::app('navigation'));
        $form = new ConfirmRemovalForm();
        $form->setRedirectUrl('navigation');
        $form->setTitle(sprintf($this->translate('Remove Navigation Item %s'), $itemName));
        $form->setOnSuccess(function (ConfirmRemovalForm $form) use ($itemName, $navigationConfigForm) {
            try {
                $navigationConfigForm->delete($itemName);
            } catch (NotFoundError $e) {
                Notification::success(sprintf(t('Navigation Item "%s" not found. No action required'), $itemName));
                return true;
            } catch (Exception $e) {
                $form->error($e->getMessage());
                return false;
            }

            if ($navigationConfigForm->save()) {
                Notification::success(sprintf(t('Navigation Item "%s" successfully removed'), $itemName));
                return true;
            }

            return false;
        });
        $form->handleRequest();

        $this->view->form = $form;
        $this->render('form');
    }

    /**
     * Unshare a navigation item
     */
    public function unshareAction()
    {
        $this->assertPermission('config/application/navigation');
        $this->assertHttpMethod('POST');

        $navigationConfigForm = new NavigationConfigForm();
        $navigationConfigForm->setUser($this->Auth()->getUser());
        $navigationConfigForm->setShareConfig(Config::app('navigation'));

        $form = new Form(array(
            'onSuccess' => function ($form) use ($navigationConfigForm) {
                try {
                    $navigationConfigForm->unshare($form->getValue('name'));
                } catch (NotFoundError $e) {
                    throw $e;
                } catch (Exception $e) {
                    Notification::error($e->getMessage());
                }

                if ($navigationConfigForm->save()) {
                    Notification::success(sprintf(
                        t('Navigation item "%s" has been unshared'),
                        $form->getValue('name')
                    ));
                } else {
                    // TODO: It failed obviously to write one of the configs, so we're leaving the user in
                    //       a inconsistent state. Luckily, it's nothing lost but possibly duplicated...
                    Notification::error(sprintf(
                        t('Failed to unshare navigation item "%s"'),
                        $form->getValue('name')
                    ));
                }

                $redirect = $form->getValue('redirect');
                if (! empty($redirect)) {
                    $form->setRedirectUrl(htmlspecialchars_decode($redirect));
                }

                return true;
            }
        ));
        $form->setUidDisabled();
        $form->setSubmitLabel('btn_submit'); // Required to ensure that isSubmitted() is called
        $form->addElement('hidden', 'name', array('required' => true));
        $form->addElement('hidden', 'redirect');

        try {
            $form->handleRequest();
        } catch (NotFoundError $_) {
            $this->httpNotFound(sprintf($this->translate('Navigation item "%s" not found'), $form->getValue('name')));
        }
    }
}