alltube/controllers/FrontController.php

<?php
/**
 * FrontController class.
 */
namespace Alltube\Controller;

use Alltube\Config;
use Alltube\PasswordException;
use Alltube\VideoDownload;
use Interop\Container\ContainerInterface;
use Slim\Container;
use Slim\Http\Request;
use Slim\Http\Response;
use Slim\Http\Stream;

/**
 * Main controller.
 */
class FrontController
{
    /**
     * Config instance.
     *
     * @var Config
     */
    private $config;

    /**
     * VideoDownload instance.
     *
     * @var VideoDownload
     */
    private $download;

    /**
     * Slim dependency container.
     *
     * @var ContainerInterface
     */
    private $container;

    /**
     * Session segment used to store session variables.
     *
     * @var \Aura\Session\Segment
     */
    private $sessionSegment;

    /**
     * FrontController constructor.
     *
     * @param Container $container Slim dependency container
     */
    public function __construct(ContainerInterface $container)
    {
        $this->config = Config::getInstance();
        $this->download = new VideoDownload();
        $this->container = $container;
        $session_factory = new \Aura\Session\SessionFactory();
        $session = $session_factory->newInstance($_COOKIE);
        $this->sessionSegment = $session->getSegment('Alltube\Controller\FrontController');
    }

    /**
     * Display index page.
     *
     * @param Request  $request  PSR-7 request
     * @param Response $response PSR-7 response
     *
     * @return void
     */
    public function index(Request $request, Response $response)
    {
        if ($this->container instanceof Container) {
            $this->container->view->render(
                $response,
                'index.tpl',
                [
                    'convert'     => $this->config->convert,
                    'class'       => 'index',
                    'description' => 'Easily download videos from Youtube, Dailymotion, Vimeo and other websites.',
                ]
            );
        }
    }

    /**
     * Display a list of extractors.
     *
     * @param Request  $request  PSR-7 request
     * @param Response $response PSR-7 response
     *
     * @return void
     */
    public function extractors(Request $request, Response $response)
    {
        if ($this->container instanceof Container) {
            $this->container->view->render(
                $response,
                'extractors.tpl',
                [
                    'extractors'  => $this->download->listExtractors(),
                    'class'       => 'extractors',
                    'title'       => 'Supported websites',
                    'description' => 'List of all supported websites from which Alltube Download '.
                        'can extract video or audio files',
                ]
            );
        }
    }

    /**
     * Display a password prompt.
     *
     * @param Request  $request  PSR-7 request
     * @param Response $response PSR-7 response
     *
     * @return Response HTTP response
     */
    public function password(Request $request, Response $response)
    {
        if ($this->container instanceof Container) {
            $this->container->view->render(
                $response,
                'password.tpl',
                [
                    'class'       => 'password',
                    'title'       => 'Password prompt',
                    'description' => 'You need a password in order to download this video with Alltube Download',
                ]
            );
        }
    }

    /**
     * Dislay information about the video.
     *
     * @param Request  $request  PSR-7 request
     * @param Response $response PSR-7 response
     *
     * @return Response HTTP response
     */
    public function video(Request $request, Response $response)
    {
        $params = $request->getQueryParams();
        if (isset($params['url'])) {
            $password = $request->getParam('password');
            if (isset($password)) {
                $this->sessionSegment->setFlash($params['url'], $password);
            }
            if (isset($params['audio'])) {
                try {
                    $url = $this->download->getURL($params['url'], 'mp3[protocol^=http]', $password);

                    return $response->withRedirect($url);
                } catch (PasswordException $e) {
                    return $this->password($request, $response);
                } catch (\Exception $e) {
                    $response = $response->withHeader(
                        'Content-Disposition',
                        'attachment; filename="'.
                        $this->download->getAudioFilename($params['url'], 'bestaudio/best', $password).'"'
                    );
                    $response = $response->withHeader('Content-Type', 'audio/mpeg');

                    if ($request->isGet() || $request->isPost()) {
                        $process = $this->download->getAudioStream($params['url'], 'bestaudio/best', $password);
                        $response = $response->withBody(new Stream($process));
                    }

                    return $response;
                }
            } else {
                try {
                    $video = $this->download->getJSON($params['url'], null, $password);
                } catch (PasswordException $e) {
                    return $this->password($request, $response);
                }
                if ($this->container instanceof Container) {
                    $this->container->view->render(
                        $response,
                        'video.tpl',
                        [
                            'video'       => $video,
                            'class'       => 'video',
                            'title'       => $video->title,
                            'description' => 'Download "'.$video->title.'" from '.$video->extractor_key,
                        ]
                    );
                }
            }
        } else {
            return $response->withRedirect($this->container->get('router')->pathFor('index'));
        }
    }

    /**
     * Display an error page.
     *
     * @param Request    $request   PSR-7 request
     * @param Response   $response  PSR-7 response
     * @param \Exception $exception Error to display
     *
     * @return Response HTTP response
     */
    public function error(Request $request, Response $response, \Exception $exception)
    {
        if ($this->container instanceof Container) {
            $this->container->view->render(
                $response,
                'error.tpl',
                [
                    'errors' => $exception->getMessage(),
                    'class'  => 'video',
                    'title'  => 'Error',
                ]
            );
        }

        return $response->withStatus(500);
    }

    /**
     * Redirect to video file.
     *
     * @param Request  $request  PSR-7 request
     * @param Response $response PSR-7 response
     *
     * @return Response HTTP response
     */
    public function redirect(Request $request, Response $response)
    {
        $params = $request->getQueryParams();
        if (isset($params['url'])) {
            try {
                $url = $this->download->getURL($params['url'], $request->getParam('format'), $this->sessionSegment->getFlash($params['url']));

                return $response->withRedirect($url);
            } catch (PasswordException $e) {
                return $response->withRedirect($this->container->get('router')->pathFor('video').'?url='.urlencode($params['url']));
            } catch (\Exception $e) {
                $response->getBody()->write($e->getMessage());

                return $response->withHeader('Content-Type', 'text/plain');
            }
        }
    }

    /**
     * Output JSON info about the video.
     *
     * @param Request  $request  PSR-7 request
     * @param Response $response PSR-7 response
     *
     * @return Response HTTP response
     */
    public function json(Request $request, Response $response)
    {
        $params = $request->getQueryParams();
        if (isset($params['url'])) {
            try {
                $video = $this->download->getJSON($params['url']);

                return $response->withJson($video);
            } catch (\Exception $e) {
                return $response->withJson(
                    ['success' => false, 'error' => $e->getMessage()]
                );
            }
        }
    }
}