|
| 1 | +<?php |
| 2 | + |
| 3 | +namespace CodingSocks\ChunkUploader\Driver; |
| 4 | + |
| 5 | +use Closure; |
| 6 | +use CodingSocks\ChunkUploader\Helper\ChunkHelpers; |
| 7 | +use CodingSocks\ChunkUploader\Identifier\Identifier; |
| 8 | +use CodingSocks\ChunkUploader\Range\NgFileUploadRange; |
| 9 | +use CodingSocks\ChunkUploader\Response\PercentageJsonResponse; |
| 10 | +use CodingSocks\ChunkUploader\StorageConfig; |
| 11 | +use Illuminate\Http\JsonResponse; |
| 12 | +use Illuminate\Http\Request; |
| 13 | +use Illuminate\Http\UploadedFile; |
| 14 | +use Illuminate\Support\Arr; |
| 15 | +use InvalidArgumentException; |
| 16 | +use Symfony\Component\HttpFoundation\Response; |
| 17 | +use Symfony\Component\HttpKernel\Exception\BadRequestHttpException; |
| 18 | +use Symfony\Component\HttpKernel\Exception\MethodNotAllowedHttpException; |
| 19 | + |
| 20 | +class NgFileUploadDriver extends UploadDriver |
| 21 | +{ |
| 22 | + use ChunkHelpers; |
| 23 | + |
| 24 | + /** |
| 25 | + * @var \CodingSocks\ChunkUploader\Identifier\Identifier |
| 26 | + */ |
| 27 | + private $identifier; |
| 28 | + |
| 29 | + /** |
| 30 | + * NgFileUploadDriver constructor. |
| 31 | + * |
| 32 | + * @param \CodingSocks\ChunkUploader\Identifier\Identifier $identifier |
| 33 | + */ |
| 34 | + public function __construct(Identifier $identifier) |
| 35 | + { |
| 36 | + $this->identifier = $identifier; |
| 37 | + } |
| 38 | + |
| 39 | + /** |
| 40 | + * @inheritDoc |
| 41 | + */ |
| 42 | + public function handle(Request $request, StorageConfig $config, Closure $fileUploaded = null): Response |
| 43 | + { |
| 44 | + if ($this->isRequestMethodIn($request, [Request::METHOD_GET])) { |
| 45 | + return $this->resume($request, $config); |
| 46 | + } |
| 47 | + |
| 48 | + if ($this->isRequestMethodIn($request, [Request::METHOD_POST])) { |
| 49 | + return $this->save($request, $config, $fileUploaded); |
| 50 | + } |
| 51 | + |
| 52 | + throw new MethodNotAllowedHttpException([ |
| 53 | + Request::METHOD_GET, |
| 54 | + Request::METHOD_POST, |
| 55 | + ]); |
| 56 | + } |
| 57 | + |
| 58 | + private function resume(Request $request, StorageConfig $config): Response |
| 59 | + { |
| 60 | + $request->validate([ |
| 61 | + 'file' => 'required', |
| 62 | + 'totalSize' => 'required', |
| 63 | + ]); |
| 64 | + |
| 65 | + $originalFilename = $request->get('file'); |
| 66 | + $totalSize = $request->get('totalSize'); |
| 67 | + $uuid = $this->identifier->generateFileIdentifier($totalSize, $originalFilename); |
| 68 | + |
| 69 | + if (!$this->chunkExists($config, $uuid)) { |
| 70 | + return new JsonResponse([ |
| 71 | + 'file' => $originalFilename, |
| 72 | + 'size' => 0, |
| 73 | + ]); |
| 74 | + } |
| 75 | + |
| 76 | + $chunk = Arr::last($this->chunks($config, $uuid)); |
| 77 | + $size = explode('-', basename($chunk))[1] + 1; |
| 78 | + |
| 79 | + return new JsonResponse([ |
| 80 | + 'file' => $originalFilename, |
| 81 | + 'size' => $size, |
| 82 | + ]); |
| 83 | + } |
| 84 | + |
| 85 | + private function save(Request $request, StorageConfig $config, Closure $fileUploaded = null): Response |
| 86 | + { |
| 87 | + $file = $request->file('file'); |
| 88 | + |
| 89 | + $this->validateUploadedFile($file); |
| 90 | + |
| 91 | + if ($this->isMonolithRequest($request)) { |
| 92 | + return $this->saveMonolith($file, $config, $fileUploaded); |
| 93 | + } |
| 94 | + |
| 95 | + $this->validateChunkRequest($request); |
| 96 | + |
| 97 | + return $this->saveChunk($file, $request, $config, $fileUploaded); |
| 98 | + } |
| 99 | + |
| 100 | + private function isMonolithRequest(Request $request) |
| 101 | + { |
| 102 | + return empty($request->post()); |
| 103 | + } |
| 104 | + |
| 105 | + /** |
| 106 | + * @param \Illuminate\Http\Request $request |
| 107 | + */ |
| 108 | + private function validateChunkRequest(Request $request): void |
| 109 | + { |
| 110 | + $request->validate([ |
| 111 | + '_chunkNumber' => 'required|numeric', |
| 112 | + '_chunkSize' => 'required|numeric', |
| 113 | + '_totalSize' => 'required|numeric', |
| 114 | + '_currentChunkSize' => 'required|numeric', |
| 115 | + ]); |
| 116 | + } |
| 117 | + |
| 118 | + /** |
| 119 | + * @param \Illuminate\Http\UploadedFile $file |
| 120 | + * @param \CodingSocks\ChunkUploader\StorageConfig $config |
| 121 | + * @param \Closure|null $fileUploaded |
| 122 | + * |
| 123 | + * @return \Symfony\Component\HttpFoundation\Response |
| 124 | + */ |
| 125 | + private function saveMonolith(UploadedFile $file, StorageConfig $config, Closure $fileUploaded = null): Response |
| 126 | + { |
| 127 | + $path = $file->store($config->getMergedDirectory(), [ |
| 128 | + 'disk' => $config->getDisk(), |
| 129 | + ]); |
| 130 | + |
| 131 | + $this->triggerFileUploadedEvent($config->getDisk(), $path, $fileUploaded); |
| 132 | + |
| 133 | + return new PercentageJsonResponse(100); |
| 134 | + } |
| 135 | + |
| 136 | + /** |
| 137 | + * @param \Illuminate\Http\UploadedFile $file |
| 138 | + * @param \Illuminate\Http\Request $request |
| 139 | + * @param \CodingSocks\ChunkUploader\StorageConfig $config |
| 140 | + * @param \Closure|null $fileUploaded |
| 141 | + * |
| 142 | + * @return \Symfony\Component\HttpFoundation\Response |
| 143 | + */ |
| 144 | + private function saveChunk(UploadedFile $file, Request $request, StorageConfig $config, Closure $fileUploaded = null): Response |
| 145 | + { |
| 146 | + try { |
| 147 | + $range = new NgFileUploadRange($request); |
| 148 | + } catch (InvalidArgumentException $e) { |
| 149 | + throw new BadRequestHttpException($e->getMessage(), $e); |
| 150 | + } |
| 151 | + |
| 152 | + $originalFilename = $file->getClientOriginalName(); |
| 153 | + $totalSize = $request->get('_totalSize'); |
| 154 | + $uuid = $this->identifier->generateFileIdentifier($totalSize, $originalFilename); |
| 155 | + |
| 156 | + $chunks = $this->storeChunk($config, $range, $file, $uuid); |
| 157 | + |
| 158 | + if (!$range->isLast()) { |
| 159 | + return new PercentageJsonResponse($range->getPercentage()); |
| 160 | + } |
| 161 | + |
| 162 | + $targetFilename = $file->hashName(); |
| 163 | + |
| 164 | + $path = $this->mergeChunks($config, $chunks, $targetFilename); |
| 165 | + |
| 166 | + if ($config->sweep()) { |
| 167 | + $this->deleteChunkDirectory($config, $uuid); |
| 168 | + } |
| 169 | + |
| 170 | + $this->triggerFileUploadedEvent($config->getDisk(), $path, $fileUploaded); |
| 171 | + |
| 172 | + return new PercentageJsonResponse(100); |
| 173 | + } |
| 174 | +} |
0 commit comments