<?php

/*

 * This file is part of the FOSUserBundle package.

 *

 * (c) FriendsOfSymfony <http://friendsofsymfony.github.com/>

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

namespace FOS\UserBundle\Command;

use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;

use Symfony\Component\Console\Input\InputArgument;

use Symfony\Component\Console\Input\InputOption;

use Symfony\Component\Console\Input\InputInterface;

use Symfony\Component\Console\Output\OutputInterface;

use Symfony\Component\Security\Core\Authentication\Token\UsernamePasswordToken;

use FOS\UserBundle\Model\User;

use FOS\UserBundle\Util\UserManipulator;

/**

 * @author Lenar Lõhmus <lenar@city.ee>

 */

abstract class RoleCommand extends ContainerAwareCommand

{

    /**

     * @see Command

     */

    protected function configure()

    {

        $this

            ->setDefinition(array(

                new InputArgument('username', InputArgument::REQUIRED, 'The username'),

                new InputArgument('role', InputArgument::OPTIONAL, 'The role'),

                new InputOption('super', null, InputOption::VALUE_NONE, 'Instead specifying role, use this to quickly add the super administrator role'),

            ));

    }

    /**

     * @see Command

     */

    protected function execute(InputInterface $input, OutputInterface $output)

    {

        $username = $input->getArgument('username');

        $role = $input->getArgument('role');

        $super = (true === $input->getOption('super'));

        if (null !== $role && $super) {

            throw new \InvalidArgumentException('You can pass either the role or the --super option (but not both simultaneously).');

        }

        if (null === $role && !$super) {

            throw new \RuntimeException('Not enough arguments.');

        }

        $manipulator = $this->getContainer()->get('fos_user.util.user_manipulator');

        $this->executeRoleCommand($manipulator, $output, $username, $super, $role);

    }

    /**

     * @see Command

     */

    abstract protected function executeRoleCommand(UserManipulator $manipulator, OutputInterface $output, $username, $super, $role);

    /**

     * @see Command

     */

    protected function interact(InputInterface $input, OutputInterface $output)

    {

        if (!$input->getArgument('username')) {

            $username = $this->getHelper('dialog')->askAndValidate(

                $output,

                'Please choose a username:',

                function($username)

                {

                    if (empty($username)) {

                        throw new \Exception('Username can not be empty');

                    }

                    return $username;

                }

            );

            $input->setArgument('username', $username);

        }

        if ((true !== $input->getOption('super')) && !$input->getArgument('role')) {

            $role = $this->getHelper('dialog')->askAndValidate(

                $output,

                'Please choose a role:',

                function($role)

                {

                    if (empty($role)) {

                        throw new \Exception('Role can not be empty');

                    }

                    return $role;

                }

            );

            $input->setArgument('role', $role);

        }

    }

}