Allow apps to register SystemTags plugins

2 files changed, 21 insertions, 2 deletions

diff --git a/apps/dav/lib/RootCollection.php b/apps/dav/lib/RootCollection.php
index 356aad4e03a..f83b0495e1c 100644
--- a/apps/dav/lib/RootCollection.php
+++ b/apps/dav/lib/RootCollection.php
@@ -71,7 +71,8 @@ class RootCollection extends SimpleCollection {
 			\OC::$server->getSystemTagManager(),
 			\OC::$server->getSystemTagObjectMapper(),
 			\OC::$server->getUserSession(),
-			\OC::$server->getGroupManager()
+			\OC::$server->getGroupManager(),
+			\OC::$server->getEventDispatcher()
 		);
 		$commentsCollection = new Comments\RootCollection(
 			\OC::$server->getCommentsManager(),
diff --git a/apps/dav/lib/SystemTag/SystemTagsRelationsCollection.php b/apps/dav/lib/SystemTag/SystemTagsRelationsCollection.php
index b09f95ccd65..7c4c613dc47 100644
--- a/apps/dav/lib/SystemTag/SystemTagsRelationsCollection.php
+++ b/apps/dav/lib/SystemTag/SystemTagsRelationsCollection.php
@@ -25,10 +25,12 @@ namespace OCA\DAV\SystemTag;
 
 use OCP\SystemTag\ISystemTagManager;
 use OCP\SystemTag\ISystemTagObjectMapper;
+use OCP\SystemTag\SystemTagsEntityEvent;
 use Sabre\DAV\Exception\Forbidden;
 use Sabre\DAV\SimpleCollection;
 use OCP\IUserSession;
 use OCP\IGroupManager;
+use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 
 class SystemTagsRelationsCollection extends SimpleCollection {
 
@@ -39,12 +41,14 @@ class SystemTagsRelationsCollection extends SimpleCollection {
 	 * @param ISystemTagObjectMapper $tagMapper
 	 * @param IUserSession $userSession
 	 * @param IGroupManager $groupManager
+	 * @param EventDispatcherInterface $dispatcher
 	 */
 	public function __construct(
 		ISystemTagManager $tagManager,
 		ISystemTagObjectMapper $tagMapper,
 		IUserSession $userSession,
-		IGroupManager $groupManager
+		IGroupManager $groupManager,
+		EventDispatcherInterface $dispatcher
 	) {
 		$children = [
 			new SystemTagsObjectTypeCollection(
@@ -60,6 +64,20 @@ class SystemTagsRelationsCollection extends SimpleCollection {
 			),
 		];
 
+		$event = new SystemTagsEntityEvent(SystemTagsEntityEvent::EVENT_ENTITY);
+		$dispatcher->dispatch(SystemTagsEntityEvent::EVENT_ENTITY, $event);
+
+		foreach ($event->getEntityCollections() as $entity => $entityExistsFunction) {
+			$children[] = new SystemTagsObjectTypeCollection(
+				$entity,
+				$tagManager,
+				$tagMapper,
+				$userSession,
+				$groupManager,
+				$entityExistsFunction
+			);
+		}
+
 		parent::__construct('root', $children);
 	}