Repect .nomedia and .noimage files

Signed-off-by: Arne Hamann <git@arne.email>

1 files changed, 96 insertions, 51 deletions

diff --git a/lib/Service/GeophotoService.php b/lib/Service/GeophotoService.php
index 9fc19aad..f1cd0504 100644
--- a/lib/Service/GeophotoService.php
+++ b/lib/Service/GeophotoService.php
@@ -12,7 +12,12 @@
 
 namespace OCA\Maps\Service;
 
+use OC\Files\Search\SearchBinaryOperator;
+use OC\Files\Search\SearchComparison;
+use OC\Files\Search\SearchQuery;
 use OCP\Files\FileInfo;
+use OCP\Files\Search\ISearchBinaryOperator;
+use OCP\Files\Search\ISearchComparison;
 use OCP\IL10N;
 use OCP\Files\IRootFolder;
 use OCP\Files\Storage\IStorage;
@@ -60,9 +65,11 @@ class GeophotoService {
 
     /**
      * @param string $userId
+	 * @param bool $respectNomediaAndNoimage=true
      * @return array with geodatas of all photos
      */
-     public function getAllFromDB($userId) {
+     public function getAllFromDB(string $userId, bool $respectNomediaAndNoimage=true) {
+		$ignoredPaths = $respectNomediaAndNoimage ? $this->getIgnoredPaths($userId) : [];
         $photoEntities = $this->photoMapper->findAll($userId);
         $userFolder = $this->getFolderForUser($userId);
         $filesById = [];
@@ -70,7 +77,7 @@ class GeophotoService {
         $previewEnableMimetypes = $this->getPreviewEnabledMimetypes();
         foreach ($photoEntities as $photoEntity) {
             $cacheEntry = $cache->get($photoEntity->getFileId());
-            if ($cacheEntry) {
+			if ($cacheEntry) {
                 // this path is relative to owner's storage
                 //$path = $cacheEntry->getPath();
                 // but we want it relative to current user's storage
@@ -83,26 +90,35 @@ class GeophotoService {
                     continue;
                 }
 				$path = $userFolder->getRelativePath( $file->getPath());
-				$isRoot = $file === $userFolder;
+				$isIgnored = false;
+				foreach ($ignoredPaths as $ignoredPath) {
+					if (str_starts_with($path, $ignoredPath)) {
+						$isIgnored = true;
+						break;
+					}
+				}
+				if (!$isIgnored) {
+					$isRoot = $file === $userFolder;
 
-				$file_object = new \stdClass();
-                $file_object->fileId = $photoEntity->getFileId();
-				$file_object->fileid = $file_object->fileId;
-                $file_object->lat = $photoEntity->getLat();
-                $file_object->lng = $photoEntity->getLng();
-                $file_object->dateTaken = $photoEntity->getDateTaken() ?? \time();
-                $file_object->basename = $isRoot ? '' : $file->getName();
-                $file_object->filename = $this->normalizePath($path);
-                $file_object->etag = $cacheEntry->getEtag();
+					$file_object = new \stdClass();
+					$file_object->fileId = $photoEntity->getFileId();
+					$file_object->fileid = $file_object->fileId;
+					$file_object->lat = $photoEntity->getLat();
+					$file_object->lng = $photoEntity->getLng();
+					$file_object->dateTaken = $photoEntity->getDateTaken() ?? \time();
+					$file_object->basename = $isRoot ? '' : $file->getName();
+					$file_object->filename = $this->normalizePath($path);
+					$file_object->etag = $cacheEntry->getEtag();
 //Not working for NC21 as Viewer requires String representation of permissions
 //                $file_object->permissions = $file->getPermissions();
-                $file_object->type = $file->getType();
-                $file_object->mime = $file->getMimetype();
-                $file_object->lastmod = $file->getMTime();
-                $file_object->size = $file->getSize();
-                $file_object->path = $path;
-                $file_object->hasPreview = in_array($cacheEntry->getMimeType(), $previewEnableMimetypes);
-                $filesById[] = $file_object;
+					$file_object->type = $file->getType();
+					$file_object->mime = $file->getMimetype();
+					$file_object->lastmod = $file->getMTime();
+					$file_object->size = $file->getSize();
+					$file_object->path = $path;
+					$file_object->hasPreview = in_array($cacheEntry->getMimeType(), $previewEnableMimetypes);
+					$filesById[] = $file_object;
+				}
             }
         }
         shuffle($filesById);
@@ -111,9 +127,11 @@ class GeophotoService {
 
     /**
      * @param string $userId
+	 * @param bool $respectNomediaAndNoimage
      * @return array with geodatas of all nonLocalizedPhotos
      */
-    public function getNonLocalizedFromDB ($userId) {
+    public function getNonLocalizedFromDB (string $userId, bool $respectNomediaAndNoimage=true) {
+		$ignoredPaths = $respectNomediaAndNoimage ? $this->getIgnoredPaths($userId) : [];
         $foo = $this->loadTimeorderedPointSets($userId);
         $photoEntities = $this->photoMapper->findAllNonLocalized($userId);
         $userFolder = $this->getFolderForUser($userId);
@@ -136,32 +154,39 @@ class GeophotoService {
 				}
 				$path = $userFolder->getRelativePath( $file->getPath());
 				$isRoot = $file === $userFolder;
-
-                $date = $photoEntity->getDateTaken() ?? \time();
-                $locations = $this->getLocationGuesses($date);
-                foreach ($locations as $location) {
-                    $file_object = new \stdClass();
-                    $file_object->fileId = $photoEntity->getFileId();
-					$file_object->fileid = $file_object->fileId;
-                    $file_object->path = $this->normalizePath($path);
-                    $file_object->hasPreview = in_array($cacheEntry->getMimeType(), $previewEnableMimetypes);
-                    $file_object->lat = $location[0];
-                    $file_object->lng = $location[1];
-                    $file_object->dateTaken = $date;
-					$file_object->basename = $isRoot ? '' : $file->getName();
-					$file_object->filename = $this->normalizePath($path);
-					$file_object->etag = $cacheEntry->getEtag();
+				$isIgnored = false;
+				foreach ($ignoredPaths as $ignoredPath) {
+					if (str_starts_with($path, $ignoredPath)) {
+						$isIgnored = true;
+						break;
+					}
+				}
+				if (!$isIgnored) {
+					$date = $photoEntity->getDateTaken() ?? \time();
+					$locations = $this->getLocationGuesses($date);
+					foreach ($locations as $location) {
+						$file_object = new \stdClass();
+						$file_object->fileId = $photoEntity->getFileId();
+						$file_object->fileid = $file_object->fileId;
+						$file_object->path = $this->normalizePath($path);
+						$file_object->hasPreview = in_array($cacheEntry->getMimeType(), $previewEnableMimetypes);
+						$file_object->lat = $location[0];
+						$file_object->lng = $location[1];
+						$file_object->dateTaken = $date;
+						$file_object->basename = $isRoot ? '' : $file->getName();
+						$file_object->filename = $this->normalizePath($path);
+						$file_object->etag = $cacheEntry->getEtag();
 //Not working for NC21 as Viewer requires String representation of permissions
 //                $file_object->permissions = $file->getPermissions();
-					$file_object->type = $file->getType();
-					$file_object->mime = $file->getMimetype();
-					$file_object->lastmod = $file->getMTime();
-					$file_object->size = $file->getSize();
-					$file_object->path = $path;
-					$file_object->hasPreview = in_array($cacheEntry->getMimeType(), $previewEnableMimetypes);
-                    $filesById[] = $file_object;
-                }
-
+						$file_object->type = $file->getType();
+						$file_object->mime = $file->getMimetype();
+						$file_object->lastmod = $file->getMTime();
+						$file_object->size = $file->getSize();
+						$file_object->path = $path;
+						$file_object->hasPreview = in_array($cacheEntry->getMimeType(), $previewEnableMimetypes);
+						$filesById[] = $file_object;
+					}
+				}
             }
 
         }
@@ -169,6 +194,32 @@ class GeophotoService {
         return $filesById;
     }
 
+	/**
+	 * @return array
+	 */
+	private function getIgnoredPaths($userId){
+		$ignoredPaths = [];
+		$userFolder = $this->root->getUserFolder($userId);
+		$excludedNodes = $userFolder->search(new SearchQuery(
+			new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
+				new SearchBinaryOperator( ISearchBinaryOperator::OPERATOR_NOT, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'mimetype', FileInfo::TYPE_FOLDER)
+				]),
+				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_OR, [
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'name', '.nomedia'),
+					new SearchComparison(ISearchComparison::COMPARE_EQUAL, 'name', '.noimage'),
+				]),
+			]),
+			0,
+			0,
+			[]
+		));
+		foreach($excludedNodes as $node) {
+			$ignoredPaths[] = $userFolder->getRelativePath($node->getParent()->getPath());
+		}
+		return $ignoredPaths;
+	}
+
     /**
      * returns a array of locations for a given date
      * @param $dateTaken
@@ -301,13 +352,7 @@ class GeophotoService {
      * @return Folder
      */
     private function getFolderForUser ($userId) {
-        $path = '/' . $userId . '/files';
-        if ($this->root->nodeExists($path)) {
-            $folder = $this->root->get($path);
-        } else {
-            $folder = $this->root->newFolder($path);
-        }
-        return $folder;
+		return $this->root->getUserFolder($userId);
     }
 
 }