sysPass/app/modules/web/Controllers/AccountFileController.php

<?php
/**
 * sysPass
 *
 * @author    nuxsmin
 * @link      https://syspass.org
 * @copyright 2012-2018, Rubén Domínguez nuxsmin@$syspass.org
 *
 * This file is part of sysPass.
 *
 * sysPass is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * sysPass is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 *  along with sysPass.  If not, see <http://www.gnu.org/licenses/>.
 */

namespace SP\Modules\Web\Controllers;

use SP\Core\Acl\Acl;
use SP\Core\Events\Event;
use SP\Core\Events\EventMessage;
use SP\Core\Exceptions\SPException;
use SP\DataModel\FileData;
use SP\Html\Html;
use SP\Http\JsonResponse;
use SP\Modules\Web\Controllers\Helpers\ItemsGridHelper;
use SP\Modules\Web\Controllers\Traits\ItemTrait;
use SP\Modules\Web\Controllers\Traits\JsonTrait;
use SP\Mvc\Controller\CrudControllerInterface;
use SP\Services\Account\AccountFileService;
use SP\Services\Account\AccountService;
use SP\Util\ErrorUtil;
use SP\Util\FileUtil;
use SP\Util\Util;

/**
 * Class AccountFileController
 *
 * @package SP\Modules\Web\Controllers
 */
final class AccountFileController extends ControllerBase implements CrudControllerInterface
{
    use JsonTrait, ItemTrait;

    /**
     * @var AccountFileService
     */
    protected $accountFileService;

    /**
     * View action
     *
     * @param $id
     */
    public function viewAction($id)
    {
        try {
            if (null === ($fileData = $this->accountFileService->getById($id))) {
                throw new SPException(__u('El archivo no existe'), SPException::INFO);
            }

            $this->view->addTemplate('file', 'itemshow');

            if (FileUtil::isImage($fileData)) {
                $this->view->assign('data', chunk_split(base64_encode($fileData->getContent())));
                $this->view->assign('fileData', $fileData);
                $this->view->assign('isImage', 1);

                $this->eventDispatcher->notifyEvent('show.accountFile',
                    new Event($this,
                        EventMessage::factory()
                            ->addDescription(__u('Archivo visualizado'))
                            ->addDetail(__u('Archivo'), $fileData->getName()))
                );

                $this->returnJsonResponseData(['html' => $this->render()]);
            }

            if (mb_strtoupper($fileData->getExtension()) === 'TXT') {
                $this->view->assign('data', htmlentities($fileData->getContent()));

                $this->eventDispatcher->notifyEvent('show.accountFile',
                    new Event($this,
                        EventMessage::factory()
                            ->addDescription(__u('Archivo visualizado'))
                            ->addDetail(__u('Archivo'), $fileData->getName()))
                );

                $this->returnJsonResponseData(['html' => $this->render()]);
            }
        } catch (\Exception $e) {
            processException($e);

            $this->returnJsonResponseException($e);
        }

        $this->returnJsonResponse(JsonResponse::JSON_WARNING, __u('Archivo no soportado para visualizar'));
    }

    /**
     * Download action
     *
     * @param $id
     */
    public function downloadAction($id)
    {
        try {
            if (null === ($fileData = $this->accountFileService->getById($id))) {
                throw new SPException(__u('El archivo no existe'), SPException::INFO);
            }

            // Enviamos el archivo al navegador
            header('Set-Cookie: fileDownload=true; path=/');
            header('Cache-Control: max-age=60, must-revalidate');
            header('Content-length: ' . $fileData->getSize());
            header('Content-type: ' . $fileData->getType());
            header('Content-Disposition: attachment; filename="' . $fileData->getName() . '"');
            header('Content-Description: PHP Generated Data');
            header('Content-transfer-encoding: binary');

            $this->eventDispatcher->notifyEvent('download.accountFile',
                new Event($this, EventMessage::factory()
                    ->addDescription(__u('Archivo descargado'))
                    ->addDetail(__u('Archivo'), $fileData->getName()))
            );

            exit($fileData->getContent());
        } catch (\Exception $e) {
            processException($e);
        }
    }

    /**
     * Upload action
     *
     * @param int $accountId
     */
    public function uploadAction($accountId)
    {
        try {
            $file = $this->router->request()->files()->get('inFile');

            if ($accountId === 0 || null === $file) {
                throw new SPException(__u('CONSULTA INVÁLIDA'), SPException::ERROR);
            }

            $allowedExts = $this->configData->getFilesAllowedExts();

            if (empty($allowedExts)) {
                throw new SPException(__u('No hay extensiones permitidas'), SPException::ERROR);
            }

            $fileData = new FileData();
            $fileData->setAccountId($accountId);
            $fileData->setName(Html::sanitize($file['name']));
            $fileData->setSize($file['size']);
            $fileData->setType($file['type']);

            if ($fileData->getName() !== '') {
                // Comprobamos la extensión del archivo
                $fileData->setExtension(mb_strtoupper(pathinfo($fileData->getName(), PATHINFO_EXTENSION)));

                if (!in_array($fileData->getExtension(), $allowedExts, true)) {
                    throw new SPException(
                        __u('Tipo de archivo no soportado'),
                        SPException::ERROR,
                        sprintf(__('Extensión: %s'), $fileData->getExtension())
                    );
                }
            } else {
                throw new SPException(
                    __u('Archivo inválido'),
                    SPException::ERROR,
                    sprintf(__u('Archivo: %s'), $fileData->getName())
                );
            }

            if (!file_exists($file['tmp_name'])) {
                throw new SPException(
                    __u('Error interno al leer el archivo'),
                    SPException::ERROR,
                    sprintf(__u('Máximo tamaño: %s'), Util::getMaxUpload())
                );
            }

            $allowedSize = $this->configData->getFilesAllowedSize();

            if ($fileData->getSize() > ($allowedSize * 1000)) {
                throw new SPException(
                    __u('Tamaño de archivo superado'),
                    SPException::ERROR,
                    sprintf(__u('Máximo tamaño: %d KB'),
                        $fileData->getRoundSize())
                );
            }

            // Leemos el archivo a una variable
            $fileData->setContent(file_get_contents($file['tmp_name']));

            if ($fileData->getContent() === false) {
                throw new SPException(__u('Error interno al leer el archivo'));
            }

            $this->accountFileService->create($fileData);

            $account = $this->dic->get(AccountService::class)
                ->getById($accountId)
                ->getAccountVData();

            $this->eventDispatcher->notifyEvent('upload.accountFile',
                new Event($this,
                    EventMessage::factory()
                        ->addDescription(__u('Archivo guardado'))
                        ->addDetail(__u('Archivo'), $fileData->getName())
                        ->addDetail(__u('Cuenta'), $account->getName())
                        ->addDetail(__u('Cliente'), $account->getClientName())
                        ->addDetail(__u('Tipo'), $fileData->getType())
                        ->addDetail(__u('Tamaño'), $fileData->getRoundSize() . 'KB')
                )
            );

            $this->returnJsonResponse(0, __u('Archivo guardado'));
        } catch (SPException $e) {
            processException($e);

            $this->returnJsonResponse(1, $e->getMessage(), [$e->getHint()]);
        } catch (\Exception $e) {
            processException($e);

            $this->returnJsonResponseException($e);
        }
    }

    /**
     * Search action
     *
     * @throws \SP\Core\Exceptions\ConstraintException
     * @throws \SP\Core\Exceptions\QueryException
     */
    public function searchAction()
    {
        if (!$this->acl->checkUserAccess(Acl::ACCOUNT_FILE_SEARCH)) {
            return;
        }

        $this->view->addTemplate('datagrid-table', 'grid');
        $this->view->assign('index', $this->request->analyzeInt('activetab', 0));
        $this->view->assign('data', $this->getSearchGrid());

        $this->returnJsonResponseData(['html' => $this->render()]);
    }

    /**
     * getSearchGrid
     *
     * @return $this
     * @throws \SP\Core\Exceptions\ConstraintException
     * @throws \SP\Core\Exceptions\QueryException
     */
    protected function getSearchGrid()
    {
        $itemsGridHelper = $this->dic->get(ItemsGridHelper::class);
        $itemSearchData = $this->getSearchData($this->configData->getAccountCount(), $this->request);

        return $itemsGridHelper->updatePager($itemsGridHelper->getFilesGrid($this->accountFileService->search($itemSearchData)), $itemSearchData);
    }

    /**
     * Create action
     */
    public function createAction()
    {
        throw new \RuntimeException('Not implemented');
    }

    /**
     * Edit action
     *
     * @param $id
     */
    public function editAction($id)
    {
        throw new \RuntimeException('Not implemented');
    }

    /**
     * Delete action
     *
     * @param $id
     */
    public function deleteAction($id = null)
    {
        try {
            if ($id === null) {
                $this->accountFileService->deleteByIdBatch($this->getItemsIdFromRequest($this->request));

                $this->eventDispatcher->notifyEvent('delete.accountFile.selection',
                    new Event($this, EventMessage::factory()
                        ->addDescription(__u('Archivos eliminados')))
                );

                $this->returnJsonResponse(0, __u('Archivos eliminados'));
            } else {
                $this->eventDispatcher->notifyEvent('delete.accountFile',
                    new Event($this, EventMessage::factory()
                        ->addDescription(__u('Archivo eliminado'))
                        ->addDetail(__u('Archivo'), $id))
                );

                $this->accountFileService->delete($id);

                $this->returnJsonResponse(0, __u('Archivo Eliminado'));
            }
        } catch (\Exception $e) {
            processException($e);

            $this->returnJsonResponseException($e);
        }
    }

    /**
     * Saves create action
     */
    public function saveCreateAction()
    {
        throw new \RuntimeException('Not implemented');
    }

    /**
     * Saves edit action
     *
     * @param $id
     */
    public function saveEditAction($id)
    {
        throw new \RuntimeException('Not implemented');
    }

    /**
     * Obtener los datos para la vista de archivos de una cuenta
     *
     * @param int $accountId Account's ID
     *
     * @throws \Psr\Container\ContainerExceptionInterface
     */
    public function listAction($accountId)
    {
        if (!$this->configData->isFilesEnabled()) {
            die(__('Gestión de archivos deshabilitada'));
        }

        try {
            $this->view->addTemplate('files-list', 'account');

            $this->view->assign('deleteEnabled', $this->request->analyzeInt('del', false));
            $this->view->assign('files', $this->dic->get(AccountFileService::class)->getByAccountId($accountId));
            $this->view->assign('sk', $this->session->getSecurityKey());
            $this->view->assign('fileViewRoute', Acl::getActionRoute(Acl::ACCOUNT_FILE_VIEW));
            $this->view->assign('fileDownloadRoute', Acl::getActionRoute(Acl::ACCOUNT_FILE_DOWNLOAD));
            $this->view->assign('fileDeleteRoute', Acl::getActionRoute(Acl::ACCOUNT_FILE_DELETE));

            if (!is_array($this->view->files) || count($this->view->files) === 0) {
                return;
            }

            $this->eventDispatcher->notifyEvent('list.accountFile', new Event($this));
        } catch (\Exception $e) {
            processException($e);

            ErrorUtil::showErrorInView($this->view, ErrorUtil::ERR_EXCEPTION);
        }

        $this->view();
    }

    /**
     * Initialize class
     *
     * @throws \Psr\Container\ContainerExceptionInterface
     * @throws \Psr\Container\NotFoundExceptionInterface
     * @throws \SP\Services\Auth\AuthException
     */
    protected function initialize()
    {
        $this->checkLoggedIn();

        $this->accountFileService = $this->dic->get(AccountFileService::class);
    }
}