<?php

/*
 * Copyright (C) 2015 IRSTEA
 * All rights reserved.
 */

namespace Irstea\FileUploadBundle\Controller;

use Irstea\FileUploadBundle\Entity\UploadedFile;
use Irstea\FileUploadBundle\Exception\RejectedFileException;
use Irstea\FileUploadBundle\Http\UploadedFileResponse;
use Irstea\FileUploadBundle\Model\FileManagerInterface;
use Irstea\FileUploadBundle\Model\UploadedFileInterface;
use Sensio\Bundle\FrameworkExtraBundle\Configuration\Method;
use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
use Symfony\Bundle\FrameworkBundle\Controller\Controller;
use Symfony\Component\Form\Extension\Csrf\CsrfProvider\CsrfProviderInterface;
use Symfony\Component\HttpFoundation\JsonResponse;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\HttpFoundation\ResponseHeaderBag;
use Symfony\Component\HttpKernel\Exception\BadRequestHttpException;
use Symfony\Component\HttpKernel\Exception\HttpException;
use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
use Symfony\Component\Routing\RouterInterface;
use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;

/**
 * @Route("/files", service="irstea_file_upload.upload_controller")
 */
class UploadController extends Controller
{
    const CSRF_INTENTION = "uploaded_file";

    /**
     *
     * @var FileManagerInterface
     */
    protected $fileManager;

    /**
     *
     * @var RouterInterface
     */
    protected $router;

    /**
     * @var CsrfProviderInterface
     */
    protected $csrfProvider;

    /**
     * @var TokenStorageInterface
     */
    protected $tokenStorage;

    /**
     *
     */
    public function __construct(FileManagerInterface $fileManager, RouterInterface $router, CsrfProviderInterface $csrfProvider, TokenStorageInterface $tokenStorage)
    {
        $this->fileManager = $fileManager;
        $this->router = $router;
        $this->csrfProvider = $csrfProvider;
        $this->tokenStorage = $tokenStorage;
    }

    /**
     * @Route("", name="file_upload_create")
     * @Method("POST")
     * @param Request $request
     */
    public function createAction(Request $request)
    {
        try {
            $data = $request->request->get('file');

            $token = $this->tokenStorage->getToken();

            $file = $this->fileManager->create(
                $data['name'],
                $data['size'],
                $data['type'],
                isset($data['lastModified']) ? $data['lastModified'] : null,
                null !== $token ? $token->getUsername() : null,
                $request->getClientIp()
            );

            $parameters = ['id' => $file->getId()];

            $deleteUrl = $this->router->generate('file_upload_delete', $parameters);

            return $this->createResponse(
                Response::HTTP_CREATED,
                'New file created',
                array_merge(
                    $file->toArray(),
                    [
                        'put_url'     => $this->router->generate('file_upload_put_content', $parameters),
                        'delete_type' => 'DELETE',
                        'delete_url'  => $deleteUrl
                    ]
                ),
                // On a pas de get pour l'instant, le DELETE et ce qui y ressemble le plus
                [ 'Location' => $deleteUrl ]
            );
        } catch(\Exception $ex) {
            return $this->createExceptionResponse($ex);
        }
    }

    /**
     * @Route("/{id}/content", name="file_upload_put_content")
     * @Method("PUT")
     * @param Request $request
     * @param UploadedFile $file
     */
    public function putContentAction(Request $request, UploadedFile $file)
    {
        try {
            $this->validateCsrfToken($request);

            list($offset, $maxlen, $complete) = $this->handleRangeHeader($request);

            // Demande un filehandle plutôt que charger le contenu en mémoire
            $input = $request->getContent(true);
            $file->copyFrom($input, $maxlen, $offset);
            fclose($input);

            if($complete) {
                return $this->completeUpload($file);
            }

            return $this->createResponse(Response::HTTP_OK, 'Chunk received');

        } catch(\Exception $ex) {
            return $this->createExceptionResponse($ex);
        }
    }

    /**
     *
     * @param Request $request
     * @return array
     */
    protected function handleRangeHeader(Request $request)
    {

        if(null === $range = $request->headers->get('Content-Range', null)) {
            return [0, PHP_INT_MAX, true];
        }

        $matches = [];
        if(!preg_match('@^bytes (\d+)-(\d+)/(\d+)$@', $range, $matches)) {
            throw new BadRequestHttpException("Invalid Content-Range");
        }

        $start = intval($matches[1]);
        $end   = intval($matches[2]);
        $total = intval($matches[3]);

        if($start < 0 || $start >= $end || $end >= $total) {
            throw new HttpException(Response::HTTP_REQUESTED_RANGE_NOT_SATISFIABLE);
        }

        return [$start, 1 + ($end - $start), $end === ($total-1)];
    }

    /**
     *
     * @param UploadedFileInterface $file
     * @return Response
     */
    protected function completeUpload(UploadedFileInterface $file)
    {
        $status = Response::HTTP_OK;
        $message = 'File uploaded';
        try {
            $this->fileManager->completed($file);
        } catch(RejectedFileException $ex) {
            $status = Response::HTTP_FORBIDDEN;
            $message = 'File rejected: '.$ex->getMessage();
        }

        $parameters = ['id' => $file->getId()];

        return $this->createResponse(
            $status,
            $message,
            [
                'files' => [
                    array_merge(
                        $file->toArray(),
                        [
                            'url'         => $this->router->generate('file_upload_get_content', $parameters),
                            'delete_type' => 'DELETE',
                            'delete_url'  => $this->router->generate('file_upload_delete', $parameters),
                        ]
                    )
                ]
            ]
        );
    }

    /**
     * @Route("/{id}/content", name="file_upload_get_content")
     * @Method("GET")
     * @param Request $request
     * @param UploadedFile $file
     */
    public function getContentAction(Request $request, UploadedFile $file)
    {
        $this->validateCsrfToken($request);

        if(!$file->isValid()) {
            throw new NotFoundHttpException();
        }

        $response = UploadedFileResponse::create($file, 200, [], false, ResponseHeaderBag::DISPOSITION_ATTACHMENT);
        $response->isNotModified($request);
        return $response;
    }

    /**
     * @Route("/{id}", name="file_upload_delete")
     * @Method("DELETE")
     * @param Request $request
     * @param UploadedFile $file
     */
    public function deleteAction(Request $request, UploadedFile $file)
    {
        try {
            $this->validateCsrfToken($request);

            $this->fileManager->delete($file);

            return $this->createResponse();
        } catch(\Exception $ex) {
            return $this->createExceptionResponse($ex);
        }
    }

    /**
     *
     * @param Request $request
     * @throws HttpException
     */
    protected function validateCsrfToken(Request $request)
    {
        if(!$this->csrfProvider->isCsrfTokenValid(self::CSRF_INTENTION, $request->query->get('token', null))) {
            throw new HttpException(Response::HTTP_FORBIDDEN, 'Invalid CSRF token');
        }
    }

    /**
     *
     * @param int $status
     * @param string $message
     * @param array $data
     * @param array $headers
     * @return JsonResponse
     */
    protected function createResponse($status = Response::HTTP_OK, $message = 'OK', array $data = [], array $headers = [])
    {
        $data['status'] = $status;
        $data['message'] = $message;
        $response = new JsonResponse($data, $status, $headers);
        $response->setStatusCode($status, $message);
        return $response;
    }

    /**
     *
     * @param \Exception $ex
     * @return JsonResponse
     */
    protected function createExceptionResponse(\Exception $ex)
    {
        return $this->createResponse(
            $ex instanceof HttpException ? $ex->getStatusCode() : Response::HTTP_INTERNAL_SERVER_ERROR,
            $ex->getMessage(),
            [
                'exception' => [
                    'class' => get_class($ex),
                    'file' => $ex->getFile(),
                    'line' => $ex->getLine(),
                    'code' => $ex->getCode(),
                    'trace' => $ex->getTraceAsString(),
                ]
            ]
        );
    }
}