Improve version handling for user_migration

Signed-off-by: Côme Chilliet <come.chilliet@nextcloud.com>

1 files changed, 9 insertions, 1 deletions

diff --git a/lib/public/UserMigration/TMigratorBasicVersionHandling.php b/lib/public/UserMigration/TMigratorBasicVersionHandling.php
index cb51e34c4b9..6b695ec45d9 100644
--- a/lib/public/UserMigration/TMigratorBasicVersionHandling.php
+++ b/lib/public/UserMigration/TMigratorBasicVersionHandling.php
@@ -32,6 +32,8 @@ namespace OCP\UserMigration;
 trait TMigratorBasicVersionHandling {
 	protected int $version = 1;
 
+	protected bool $mandatory = false;
+
 	/**
 	 * {@inheritDoc}
 	 */
@@ -42,7 +44,13 @@ trait TMigratorBasicVersionHandling {
 	/**
 	 * {@inheritDoc}
 	 */
-	public function canImport(int $version): bool {
+	public function canImport(
+		IImportSource $importSource,
+		?int $version
+	): bool {
+		if ($version === null) {
+			return !$this->mandatory;
+		}
 		return ($this->version >= $version);
 	}
 }