validate UID

utopia-php · Oct 9, 2024 · b230c8b · b230c8b
1 parent b37050f
commit b230c8b
Showing 1 changed file with 36 additions and 1 deletion.
diff --git a/src/Migration/Destinations/Appwrite.php b/src/Migration/Destinations/Appwrite.php
@@ -25,6 +25,7 @@
 use Utopia\Database\Query;
 use Utopia\Database\Validator\Index as IndexValidator;
 use Utopia\Database\Validator\Structure;
+use Utopia\Database\Validator\UID;
 use Utopia\Migration\Destination;
 use Utopia\Migration\Exception;
 use Utopia\Migration\Resource;
@@ -302,14 +303,25 @@ public function importDatabaseResource(Resource $resource, bool $isLast): Resour
     /**
      * @throws AuthorizationException
      * @throws StructureException
-     * @throws DatabaseException
+     * @throws DatabaseException|Exception
      */
     protected function createDatabase(Database $resource): bool
     {
         $resourceId = $resource->getId() == 'unique()'
             ? ID::unique()
             : $resource->getId();
 
+        $validator = new UID();
+
+        if ($validator->isValid($resourceId)) {
+            throw new Exception(
+                resourceName: $resource->getName(),
+                resourceGroup: $resource->getGroup(),
+                resourceId: $resource->getId(),
+                message: $validator->getDescription(),
+            );
+        }
+
         $resource->setId($resourceId);
 
         $database = $this->database->createDocument('databases', new UtopiaDocument([
@@ -354,6 +366,17 @@ protected function createCollection(Collection $resource): bool
             ? ID::unique()
             : $resource->getId();
 
+        $validator = new UID();
+
+        if ($validator->isValid($resourceId)) {
+            throw new Exception(
+                resourceName: $resource->getName(),
+                resourceGroup: $resource->getGroup(),
+                resourceId: $resource->getId(),
+                message: $validator->getDescription(),
+            );
+        }
+
         $resource->setId($resourceId);
 
         $database = $this->database->getDocument(
@@ -865,9 +888,21 @@ protected function createIndex(Index $resource): bool
      * @throws AuthorizationException
      * @throws DatabaseException
      * @throws StructureException
+     * @throws Exception
      */
     protected function createDocument(Document $resource, bool $isLast): bool
     {
+        $validator = new UID();
+
+        if ($validator->isValid($resource->getId())) {
+            throw new Exception(
+                resourceName: $resource->getName(),
+                resourceGroup: $resource->getGroup(),
+                resourceId: $resource->getId(),
+                message: $validator->getDescription(),
+            );
+        }
+
         // Check if document has already been created
         $exists = \array_key_exists(
             $resource->getId(),