SonataAdminBundle/Admin/Extension/LockExtension.php

<?php

/*
 * This file is part of the Sonata Project package.
 *
 * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Sonata\AdminBundle\Admin\Extension;

use Sonata\AdminBundle\Admin\AdminExtension;
use Sonata\AdminBundle\Admin\AdminInterface;
use Sonata\AdminBundle\Form\FormMapper;
use Sonata\AdminBundle\Model\LockInterface;
use Symfony\Component\Form\FormEvent;
use Symfony\Component\Form\FormEvents;

/**
 * @author Emmanuel Vella <vella.emmanuel@gmail.com>
 */
class LockExtension extends AdminExtension
{
    protected $fieldName = '_lock_version';

    /**
     * {@inheritdoc}
     */
    public function configureFormFields(FormMapper $form)
    {
        $admin = $form->getAdmin();
        $formBuilder = $form->getFormBuilder();

        // PHP 5.3 BC
        $fieldName = $this->fieldName;

        $formBuilder->addEventListener(FormEvents::PRE_SET_DATA, function (FormEvent $event) use ($admin, $fieldName) {
            $data = $event->getData();
            $form = $event->getForm();

            if (null === $data || $form->getParent()) {
                return;
            }

            $modelManager = $admin->getModelManager();

            if (!$modelManager instanceof LockInterface) {
                return;
            }

            if (null === $lockVersion = $modelManager->getLockVersion($data)) {
                return;
            }

            $form->add($fieldName, 'hidden', array(
                'mapped' => false,
                'data'   => $lockVersion,
            ));
        });
    }

    /**
     * {@inheritdoc}
     */
    public function preUpdate(AdminInterface $admin, $object)
    {
        if (!$admin->hasRequest() || !$data = $admin->getRequest()->get($admin->getUniqid())) {
            return;
        }

        if (!isset($data[$this->fieldName])) {
            return;
        }

        $modelManager = $admin->getModelManager();

        if (!$modelManager instanceof LockInterface) {
            return;
        }

        $modelManager->lock($object, $data[$this->fieldName]);
    }
}