From e3244361bada8bc79384f49e127a9a310f05ae89 Mon Sep 17 00:00:00 2001
From: Christopher Ng <chrng8@gmail.com>
Date: Fri, 4 Feb 2022 20:49:11 +0000
Subject: Allow registration of migrators

Signed-off-by: Christopher Ng <chrng8@gmail.com>
---
 .../AppFramework/Bootstrap/RegistrationContext.php | 25 ++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'lib/private/AppFramework/Bootstrap')

diff --git a/lib/private/AppFramework/Bootstrap/RegistrationContext.php b/lib/private/AppFramework/Bootstrap/RegistrationContext.php
index b40d3356d1a..7b4d24036e8 100644
--- a/lib/private/AppFramework/Bootstrap/RegistrationContext.php
+++ b/lib/private/AppFramework/Bootstrap/RegistrationContext.php
@@ -52,6 +52,7 @@ use OCP\Notification\INotifier;
 use OCP\Profile\ILinkAction;
 use OCP\Search\IProvider;
 use OCP\Support\CrashReport\IReporter;
+use OCP\UserMigration\IMigrator as IUserMigrator;
 use Psr\Log\LoggerInterface;
 use Throwable;
 
@@ -78,6 +79,9 @@ class RegistrationContext {
 	/** @var ServiceRegistration<IRoomBackend>[] */
 	private $calendarRoomBackendRegistrations = [];
 
+	/** @var ServiceRegistration<IUserMigrator>[] */
+	private $userMigrators = [];
+
 	/** @var ServiceFactoryRegistration[] */
 	private $services = [];
 
@@ -293,6 +297,13 @@ class RegistrationContext {
 					$class
 				);
 			}
+
+			public function registerUserMigrator(string $migratorClass): void {
+				$this->context->registerUserMigrator(
+					$this->appId,
+					$migratorClass
+				);
+			}
 		};
 	}
 
@@ -412,6 +423,13 @@ class RegistrationContext {
 		);
 	}
 
+	/**
+	 * @psalm-param class-string<IUserMigrator> $migratorClass
+	 */
+	public function registerUserMigrator(string $appId, string $migratorClass): void {
+		$this->userMigrators[] = new ServiceRegistration($appId, $migratorClass);
+	}
+
 	/**
 	 * @param App[] $apps
 	 */
@@ -687,4 +705,11 @@ class RegistrationContext {
 	public function getCalendarRoomBackendRegistrations(): array {
 		return $this->calendarRoomBackendRegistrations;
 	}
+
+	/**
+	 * @return ServiceRegistration<IUserMigrator>[]
+	 */
+	public function getUserMigrators(): array {
+		return $this->userMigrators;
+	}
 }
-- 
cgit v1.2.3