Phraseanet/lib/Alchemy/Phrasea/Command/Setup/H264MappingGenerator.php

<?php

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

namespace Alchemy\Phrasea\Command\Setup;

use Alchemy\Phrasea\Command\Command;
use Alchemy\Phrasea\Databox\DataboxPathExtractor;
use Alchemy\Phrasea\Http\H264PseudoStreaming\H264Factory;
use Alchemy\Phrasea\Model\Manipulator\TokenManipulator;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Yaml\Yaml;

class H264MappingGenerator extends Command
{
    public function __construct($name = null)
    {
        parent::__construct('h264-pseudo-streaming:generate-mapping');

        $this->addOption('write', 'w', null, 'Writes the configuration')
             ->addOption('enabled', 'e', null, 'Set the enable toggle to `true`')
             ->addArgument('type', InputArgument::REQUIRED, 'The configuration type, either `nginx` or `apache`')
             ->setDescription('Generates Phraseanet H264 pseudo streaming mapping configuration depending on databoxes configuration');
    }

    /**
     * {@inheritdoc}
     */
    protected function doExecute(InputInterface $input, OutputInterface $output)
    {
        $extractor = new DataboxPathExtractor($this->container->getApplicationBox());
        $paths = $extractor->extractPaths();
        foreach ($paths as $path) {
            $this->container['filesystem']->mkdir($path);
        }

        $type = strtolower($input->getArgument('type'));
        $enabled = $input->getOption('enabled');

        $factory = new H264Factory($this->container['monolog'], true, $type, $this->computeMapping($paths));
        $mode = $factory->createMode(true);

        $currentConf = isset($this->container['phraseanet.configuration']['h264-pseudo-streaming']) ? $this->container['phraseanet.configuration']['h264-pseudo-streaming'] : [];
        $currentMapping = (isset($currentConf['mapping']) && is_array($currentConf['mapping'])) ? $currentConf['mapping'] : [];

        $conf = [
            'enabled' => $enabled,
            'type' => $type,
            'mapping' => $mode->getMapping(),
        ];

        if ($input->getOption('write')) {
            $output->write("Writing configuration ...");
            $this->container['phraseanet.configuration']['h264-pseudo-streaming'] = $conf;
            $output->writeln(" <info>OK</info>");
            $output->writeln("");
            $output->write("It is now strongly recommended to use <info>h264-pseudo-streaming:dump-configuration</info> command to upgrade your virtual-host");
        } else {
            $output->writeln("Configuration will <info>not</info> be written, use <info>--write</info> option to write it");
            $output->writeln("");
            $output->writeln(Yaml::dump(['h264-pseudo-streaming' => $conf], 4));
        }

        return 0;
    }

    private function computeMapping($paths)
    {
        $paths = array_unique($paths);

        $ret = [];

        foreach ($paths as $path) {
            $sanitizedPath = rtrim($path, '/');
            if (array_key_exists($sanitizedPath, $ret)) {
                continue;
            }
            $ret[$sanitizedPath] = $this->pathsToConf($sanitizedPath);
        }

        return $ret;
    }

    private function pathsToConf($path)
    {
        static $n = 0;
        $n++;

        return ['mount-point' => 'mp4-videos-'.$n, 'directory' => $path, 'passphrase' => $this->container['random.low']->generateString(32, TokenManipulator::LETTERS_AND_NUMBERS)];
    }
}