1 files changed, 42 insertions, 22 deletions

diff --git a/apps/files/lib/Activity/Helper.php b/apps/files/lib/Activity/Helper.php
index b9a5ae887ec..7bbaf44ab4c 100644
--- a/apps/files/lib/Activity/Helper.php
+++ b/apps/files/lib/Activity/Helper.php
@@ -4,6 +4,7 @@
  *
  * @author Christoph Wurst <christoph@winzerhof-wurst.at>
  * @author Joas Schilling <coding@schilljs.com>
+ * @author Ferdinand Thiessen <opensource@fthiessen.de>
  *
  * @license AGPL-3.0
  *
@@ -23,30 +24,30 @@
 namespace OCA\Files\Activity;
 
 use OCP\Files\Folder;
+use OCP\Files\IRootFolder;
+use OCP\Files\Node;
 use OCP\ITagManager;
 
 class Helper {
 	/** If a user has a lot of favorites the query might get too slow and long */
 	public const FAVORITE_LIMIT = 50;
 
-	/** @var ITagManager */
-	protected $tagManager;
-
-	/**
-	 * @param ITagManager $tagManager
-	 */
-	public function __construct(ITagManager $tagManager) {
-		$this->tagManager = $tagManager;
+	public function __construct(
+		protected ITagManager $tagManager,
+		protected IRootFolder $rootFolder,
+	) {
 	}
 
 	/**
-	 * Returns an array with the favorites
+	 * Return an array with nodes marked as favorites
 	 *
-	 * @param string $user
-	 * @return array
+	 * @param string $user User ID
+	 * @param bool $foldersOnly Only return folders (default false)
+	 * @return Node[]
+	 * @psalm-return ($foldersOnly is true ? Folder[] : Node[])
 	 * @throws \RuntimeException when too many or no favorites where found
 	 */
-	public function getFavoriteFilePaths($user) {
+	public function getFavoriteNodes(string $user, bool $foldersOnly = false): array {
 		$tags = $this->tagManager->load('files', [], false, $user);
 		$favorites = $tags->getFavorites();
 
@@ -57,26 +58,45 @@ class Helper {
 		}
 
 		// Can not DI because the user is not known on instantiation
-		$rootFolder = \OC::$server->getUserFolder($user);
-		$folders = $items = [];
+		$userFolder = $this->rootFolder->getUserFolder($user);
+		$favoriteNodes = [];
 		foreach ($favorites as $favorite) {
-			$nodes = $rootFolder->getById($favorite);
+			$nodes = $userFolder->getById($favorite);
 			if (!empty($nodes)) {
-				/** @var \OCP\Files\Node $node */
 				$node = array_shift($nodes);
-				$path = substr($node->getPath(), strlen($user . '/files/'));
-
-				$items[] = $path;
-				if ($node instanceof Folder) {
-					$folders[] = $path;
+				if (!$foldersOnly || $node instanceof Folder) {
+					$favoriteNodes[] = $node;
 				}
 			}
 		}
 
-		if (empty($items)) {
+		if (empty($favoriteNodes)) {
 			throw new \RuntimeException('No favorites', 1);
 		}
 
+		return $favoriteNodes;
+	}
+
+	/**
+	 * Returns an array with the favorites
+	 *
+	 * @param string $user
+	 * @return array
+	 * @throws \RuntimeException when too many or no favorites where found
+	 */
+	public function getFavoriteFilePaths(string $user): array {
+		$userFolder = $this->rootFolder->getUserFolder($user);
+		$nodes = $this->getFavoriteNodes($user);
+		$folders = $items = [];
+		foreach ($nodes as $node) {
+			$path = $userFolder->getRelativePath($node->getPath());
+
+			$items[] = $path;
+			if ($node instanceof Folder) {
+				$folders[] = $path;
+			}
+		}
+
 		return [
 			'items' => $items,
 			'folders' => $folders,