1 files changed, 124 insertions, 73 deletions

diff --git a/lib/Controller/ChecksumController.php b/lib/Controller/ChecksumController.php
index da31f59..8953ba1 100644
--- a/lib/Controller/ChecksumController.php
+++ b/lib/Controller/ChecksumController.php
@@ -1,82 +1,133 @@
 <?php
+
+declare(strict_types=1);
+
 namespace OCA\Checksum\Controller;
 
 use OCP\AppFramework\Controller;
-use OCP\IRequest;
-use OC\Files\Filesystem;
 use OCP\AppFramework\Http\JSONResponse;
-
+use OCP\Files\Mount\IMountManager;
+use OCP\IL10N;
+use OCP\IRequest;
+use OCP\IUserSession;
+use OCP\L10N\IFactory;
 
 class ChecksumController extends Controller {
 
-		protected $language;
-
-		public function __construct($appName, IRequest $request) {
-
-				parent::__construct($appName, $request);
-
-				// get i10n
-				$this->language = \OC::$server->getL10N('checksum');
-
-		}
-
-		/**
-		 * callback function to get md5 hash of a file
-		 * @NoAdminRequired
-		 * @param (string) $source - filename
-		 * @param (string) $type - hash algorithm type
-		 */
-	  public function check($source, $type) {
-	  		if(!$this->checkAlgorithmType($type)) {
-	  			return new JSONResponse(
-							array(
-									'response' => 'error',
-									'msg' => $this->language->t('The algorithm type "%s" is not a valid or supported algorithm type.', array($type))
-							)
-					);
-	  		}
-
-				if($hash = $this->getHash($source, $type)){
-						return new JSONResponse(
-								array(
-										'response' => 'success',
-										'msg' => $hash
-								)
-						);
-				} else {
-						return new JSONResponse(
-								array(
-										'response' => 'error',
-										'msg' => $this->language->t('File not found.')
-								)
-						);
-				};
-
-	  }
-
-	  protected function getHash($source, $type) {
-
-	  	if($info = Filesystem::getLocalFile($source)) {
-	  			return hash_file($type, $info);
-	  	}
-
-	  	return false;
-	  }
-
-	  protected function checkAlgorithmType($type) {
-	  	$list_algos = hash_algos();
-	  	return in_array($type, $this->getAllowedAlgorithmTypes()) && in_array($type, $list_algos);
-	  }
-
-	  protected function getAllowedAlgorithmTypes() {
-	  	return array(
-				'md5',
-				'sha1',
-				'sha256',
-				'sha384',
-				'sha512',
-				'crc32'
-			);
-		}
+    /**
+     * @var IL10N
+     */
+    private IL10N $language;
+
+    /**
+     * @var IMountManager
+     */
+    private IMountManager $mountManager;
+
+    /**
+     * @var IUserSession
+     */
+    private IUserSession $userSession;
+
+    /**
+     * ChecksumController constructor.
+     *
+     * @param string $appName
+     * @param IRequest $request
+     * @param IFactory $languageFactory
+     * @param IMountManager $mountManager
+     * @param IUserSession $userSession
+     */
+    public function __construct(
+        string $appName,
+        IRequest $request,
+        IFactory $languageFactory,
+        IMountManager $mountManager,
+        IUserSession $userSession
+    ) {
+        parent::__construct($appName, $request);
+
+        $this->language = $languageFactory->get('checksum');
+        $this->mountManager = $mountManager;
+        $this->userSession = $userSession;
+    }
+
+    /**
+     * Compute the hash of a file.
+     *
+     * @NoAdminRequired
+     * @param string $source file path relative to user home
+     * @param string $type hash algorithm
+     * @return JSONResponse
+     */
+    public function check(string $source, string $type): JSONResponse {
+        if (!$this->checkAlgorithmType($type)) {
+            return new JSONResponse(
+                [
+                    'response' => 'error',
+                    'msg' => $this->language->t(
+                        'The algorithm type "%s" is not a valid or supported algorithm type.',
+                        [$type]
+                    )
+                ]
+            );
+        }
+
+        $hash = $this->getHash($source, $type);
+        if ($hash) {
+            return new JSONResponse(
+                [
+                    'response' => 'success',
+                    'msg' => $hash
+                ]
+            );
+        }
+
+        return new JSONResponse(
+            [
+                'response' => 'error',
+                'msg' => $this->language->t('File not found.')
+            ]
+        );
+    }
+
+    private function getHash(string $source, string $type): ?string {
+        $user = $this->userSession->getUser();
+        if (!$user) {
+            return null;
+        }
+
+        $absPath = $user->getUID() . '/files/' . $source;
+        $mount = $this->mountManager->find($absPath);
+        if (!$mount) {
+            return null;
+        }
+
+        $internalPath = $mount->getInternalPath($absPath);
+        $file = $mount->getStorage()->fopen($internalPath, 'rb');
+        if (!$file) {
+            return null;
+        }
+
+        $hash = hash_init($type);
+        hash_update_stream($hash, $file);
+        fclose($file);
+
+        return hash_final($hash);
+    }
+
+    private function checkAlgorithmType(string $type): bool {
+        return in_array($type, $this->getAllowedAlgorithmTypes()) && in_array($type, hash_algos());
+    }
+
+    private function getAllowedAlgorithmTypes(): array {
+        return [
+            'md5',
+            'sha1',
+            'sha256',
+            'sha384',
+            'sha512',
+            'crc32'
+        ];
+    }
 }
-