Phraseanet/lib/Alchemy/Phrasea/Controller/Prod/DoDownload.php

<?php

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

namespace Alchemy\Phrasea\Controller\Prod;

use Silex\Application;
use Silex\ControllerProviderInterface;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\HttpKernel\KernelEvents;

class DoDownload implements ControllerProviderInterface
{

    /**
     * {@inheritDoc}
     */
    public function connect(Application $app)
    {
        $controllers = $app['controllers_factory'];

        $controllers->before(function(Request $request) use ($app) {

                /**
                 * @todo do not authenticate user
                 */
                if (null === $token = $request->attributes->get('token')) {
                    $app->abort(403);
                }

                try {
                    $auth = new \Session_Authentication_Token($app, $token);
                    $app->openAccount($auth);
                } catch (\Exception $e) {
                    $app->abort(403);
                }
            });

        /**
         * Prepare a set of documents for download
         *
         * name         : prepare_download
         *
         * description  : Prepare a set of documents for download
         *
         * method       : POST
         *
         * parameters   : none
         *
         * return       : HTML Response
         */
        $controllers->post('/{token}/prepare/', $this->call('prepareDownload'))
            ->bind('prepare_download')
            ->assert('token', '^[a-zA-Z0-9]{8,16}$');

        /**
         * Download a set of documents
         *
         * name         : document_download
         *
         * description  : Download a set of documents
         *
         * method       : POST
         *
         * parameters   : none
         *
         * return       : HTML Response
         */
        $controllers->post('/{token}/get/', $this->call('downloadDocuments'))
            ->bind('document_download')
            ->assert('token', '^[a-zA-Z0-9]{8,16}$');

        /**
         * Build a zip with all downloaded documents
         *
         * name         : execute_download
         *
         * description  :  Build a zip with all downloaded documents
         *
         * method       : POST
         *
         * parameters   : none
         *
         * return       : JSON Response
         */
        $controllers->post('/{token}/execute/', $this->call('downloadExecute'))
            ->bind('execute_download')
            ->assert('token', '^[a-zA-Z0-9]{8,16}$');


        return $controllers;
    }

    /**
     * Prepare a set of documents for download
     *
     * @param   Application $app
     * @param   Request     $request
     * @return  Response
     */
    public function prepareDownload(Application $app, Request $request, $token)
    {
        try {
            $datas = \random::helloToken($app, $token);
        } catch (\Exception_NotFound $e) {
            $app->abort(404);
        }

        if (!is_string($datas['datas'])) {
            $app->abort(404);
        }

        if (false === $list = @unserialize($datas['datas'])) {
            $app->abort(500);
        }

        $records = array();
        foreach($list['files'] as $file) {
            $sbasId = \phrasea::sbasFromBas($app, $file['base_id']);
            $records[sprintf('%s_%s', $sbasId, $file['record_id'])] = new \record_adapter($app, $sbasId, $file['record_id']);
        }

        return new Reponse($app['twig']->render(
            '/prod/actions/Download/prepare.html.twig', array(
            'datas'   => $list,
            'records' => $records,
            'token'   => $token
        )));
    }

    /**
     * Download a set of documents
     *
     * @param   Application $app
     * @param   Request     $request
     * @return  Response
     */
    public function downloadDocuments(Application $app, Request $request, $token)
    {
        try {
            $datas = \random::helloToken($app, $token);
        } catch (\Exception_NotFound $e) {
            $app->abort(404);
        }

        if (false === $list = @unserialize((string) $datas['datas'])) {
            $app->abort(500);
        }

        $exportName = $list['export_name'];

        if($list['count'] === 1) {
            $file = array_pop($list['files']);
            $subdef = array_pop($file['subdefs']);
            $exportName = sprintf('%s%s.%s', $file['export_name'], $subdef['ajout'], $subdef['exportExt']);
            $exportFile = \p4string::addEndSlash($subdef['path']) . $subdef['file'];
            $mime = $subdef['mime'];
        } else {
            $exportFile = $app['phraseanet.registry']->get('GV_RootPath') . 'tmp/download/' . $datas['value'] . '.zip';
            $mime = 'application/zip';
        }

        if(!$app['file-system']->exists($exportFile)) {
            $app->abort(404);
        }

        $app['dispatcher']->addListener(KernelEvents::TERMINATE, function () use ($list, $request) {
            \set_export::log_download($list, $request->request->get('type'));
        });

        $response = \set_export::stream_file(
            $app['phraseanet.registry'],
            $exportFile,
            $exportName,
            $mime,
            'attachment'
        );

        return $response;
    }


    /**
     * Build a zip of downloaded documents
     *
     * @param   Application $app
     * @param   Request     $request
     * @return  Response
     */
    public function downloadExecute(Application $app, Request $request, $token)
    {
        try {
            $datas = \random::helloToken($app, $token);
        } catch (\Exception_NotFound $e) {
            return $app->json(array(
                'success' => false,
                'message' => 'Token not found'
            ));
        }

        if (false === $list = @unserialize((string) $datas['datas'])) {
            return $app->json(array(
                'success' => false,
                'message' => 'Invalid datas'
            ));
        }

        set_time_limit(0);
        // Force the session to be saved and closed.
        $app['session']->save();
        ignore_user_abort(true);

        \set_export::build_zip(
            new Filesystem(),
            $token,
            $list,
            $app['phraseanet.registry']->get('GV_RootPath') . 'tmp/download/' . $datas['value'] . '.zip' // Dest file
        );

        return $app->json(array(
            'success' => true,
            'message' => ''
        ));
    }

    /**
     * Prefix the method to call with the controller class name
     *
     * @param  string $method The method to call
     * @return string
     */
    private function call($method)
    {
        return sprintf('%s::%s', __CLASS__, $method);
    }
}