Patch invalidate method

ScreamZ · Dec 22, 2015 · 6d540bb · 6d540bb
1 parent 1b8b4b3
commit 6d540bb
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/Services/SecureDownloader.php b/Services/SecureDownloader.php
@@ -83,15 +83,22 @@ public function generateHash($filePath, $accessKey, $documentHashTTL = null)
     /**
      * Mark the download request as stale, another call using this hash will make the request miss.
      *
-     * @param DownloadRequest $downloadRequest
+     * @param string $documentHash
+     * @param string $accessKey
      *
      * @return bool
+     * @throws DownloadRequestException
      */
-    public function invalidate(DownloadRequest $downloadRequest)
+    public function invalidate($documentHash, $accessKey)
     {
-        $downloadRequest = $this->stash->getItem($this->stashPrefixKey.'/'.$downloadRequest->getHash());
+        $downloadRequest = $this->initiateDownloadRequest($documentHash, $accessKey);
+        if ($downloadRequest->isProcessable()) {
+            $downloadRequest = $this->stash->getItem($this->stashPrefixKey.'/'.$downloadRequest->getHash());
 
-        return $downloadRequest->clear();
+            return $downloadRequest->clear();
+        } else {
+            throw new DownloadRequestException($downloadRequest);
+        }
     }
 
     /**