<?php

/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2013 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Alchemy\Phrasea\Command\Task;

use Alchemy\Phrasea\Command\Command;
use Alchemy\Phrasea\Exception\RuntimeException;
use Alchemy\Phrasea\TaskManager\Event\FinishedJobRemoverSubscriber;
use Alchemy\Phrasea\TaskManager\Job\JobData;
use Alchemy\TaskManager\Event\JobSubscriber\DurationLimitSubscriber;
use Alchemy\TaskManager\Event\JobSubscriber\LockFileSubscriber;
use Alchemy\TaskManager\Event\JobSubscriber\MemoryLimitSubscriber;
use Alchemy\TaskManager\Event\JobSubscriber\SignalControlledSubscriber;
use Alchemy\TaskManager\Event\JobSubscriber\StopSignalSubscriber;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputOption;

class TaskRun extends Command
{
    public function __construct()
    {
        parent::__construct('task-manager:task:run');

        $this->addArgument('task_id', InputArgument::REQUIRED, 'The id of the task to run', null)
            ->addOption('max-memory', null, InputOption::VALUE_REQUIRED, '')
            ->addOption('max-duration', null, InputOption::VALUE_REQUIRED, '')
            ->addOption('listen-signal', null, InputOption::VALUE_NONE, '')
        ;
    }

    protected function doExecute(InputInterface $input, OutputInterface $output)
    {
        declare(ticks=1);

        if (null === $task = $this->container['manipulator.task']->getRepository()->find($input->getArgument('task_id'))) {
            throw new RuntimeException('Invalid task_id');
        }

        $job = $factory = $this->container['task-manager.job-factory']->create($task->getJobId());
        $logger = $this->container['task-manager.logger'];

        $job->addSubscriber(new LockFileSubscriber('task-'.$task->getId(), $logger, $this->container['root.path'].'/tmp/locks'));
        $job->addSubscriber(new StopSignalSubscriber($this->container['signal-handler'], $logger));

        if ($input->getOption('listen-signal')) {
            $job->addSubscriber(new SignalControlledSubscriber($this->container['signal-handler'], 2, $logger));
        }
        if (null !== $maxDuration = $input->getOption('max-duration')) {
            $job->addSubscriber(new DurationLimitSubscriber($maxDuration, $logger));
        }
        if (null !== $maxMemory = $input->getOption('max-memory')) {
            $job->addSubscriber(new MemoryLimitSubscriber($maxMemory, $logger));
        }
        if ($task->isSingleRun()) {
            $job->addSubscriber(new FinishedJobRemoverSubscriber($this->container['EM']));
        }

        $job->run(new JobData($this->container, $task));
    }
}