From ad1d514fb2b5caae2fb2d236259b86f2be13fb5b Mon Sep 17 00:00:00 2001
From: Bradley Schofield <ionicisere@gmail.com>
Date: Thu, 25 Jan 2024 13:05:43 +0000
Subject: [PATCH] Create V21.php

---
 src/Appwrite/Migration/Version/V21.php | 148 +++++++++++++++++++++++++
 1 file changed, 148 insertions(+)
 create mode 100644 src/Appwrite/Migration/Version/V21.php

diff --git a/src/Appwrite/Migration/Version/V21.php b/src/Appwrite/Migration/Version/V21.php
new file mode 100644
index 0000000000..77badb2d3e
--- /dev/null
+++ b/src/Appwrite/Migration/Version/V21.php
@@ -0,0 +1,148 @@
+<?php
+
+namespace Appwrite\Migration\Version;
+
+use Appwrite\Migration\Migration;
+use Utopia\CLI\Console;
+use Utopia\Database\Database;
+use Utopia\Database\Document;
+use Utopia\Database\Helpers\ID;
+
+class V20 extends Migration
+{
+    public function execute(): void
+    {
+        /**
+         * Disable SubQueries for Performance.
+         */
+        foreach (['subQueryIndexes', 'subQueryPlatforms', 'subQueryDomains', 'subQueryKeys', 'subQueryWebhooks', 'subQuerySessions', 'subQueryTokens', 'subQueryMemberships', 'subQueryVariables'] as $name) {
+            Database::addFilter(
+                $name,
+                fn () => null,
+                fn () => []
+            );
+        }
+        Console::log('Migrating Project: ' . $this->project->getAttribute('name') . ' (' . $this->project->getId() . ')');
+        $this->projectDB->setNamespace("_{$this->project->getInternalId()}");
+
+        Console::info('Migrating Collections');
+        $this->migrateCollections();
+
+        Console::info('Migrating Documents');
+        $this->forEachDocument([$this, 'fixDocument']);
+    }
+
+    /**
+     * Migrate all Collections.
+     *
+     * @return void
+     * @throws \Throwable
+     * @throws Exception
+     */
+    private function migrateCollections(): void
+    {
+        foreach ($this->collections as $collection) {
+            $id = $collection['$id'];
+
+            Console::log("Migrating Collection \"{$id}\"");
+
+            $this->projectDB->setNamespace("_{$this->project->getInternalId()}");
+
+            switch ($id) {
+                case '_metadata':
+                    $this->createCollection('providers');
+                    $this->createCollection('messages');
+                    $this->createCollection('topics');
+                    $this->createCollection('subscribers');
+                    $this->createCollection('targets');
+
+                    break;
+                case 'users':
+                    // Create targets attribute
+                    try {
+                        $this->createAttributeFromCollection($this->projectDB, $id, 'targets');
+                        $this->projectDB->deleteCachedCollection($id);
+                    } catch (\Throwable $th) {
+                        Console::warning("'targets' from {$id}: {$th->getMessage()}");
+                    }
+                    break;
+                case 'projects':
+                    // Rename providers to oAuthProviders
+                    try {
+                        $this->projectDB->renameAttribute($id, 'providers', 'oAuthProviders');
+                        $this->projectDB->deleteCachedCollection($id);
+                    } catch (\Throwable $th) {
+                        Console::warning("'oAuthProviders' from {$id}: {$th->getMessage()}");
+                    }
+                    break;
+                case 'schedules':
+                    try {
+                        $this->createAttributeFromCollection($this->projectDB, $id, 'resourceCollection');
+                        $this->projectDB->deleteCachedCollection($id);
+                    } catch (\Throwable $th) {
+                        Console::warning("'schedules' from {$id}: {$th->getMessage()}");
+                    }
+                    break;
+                case 'webhooks':
+                    try {
+                        $this->createAttributeFromCollection($this->projectDB, $id, 'enabled');
+                        $this->createAttributeFromCollection($this->projectDB, $id, 'logs');
+                        $this->createAttributeFromCollection($this->projectDB, $id, 'attempts');
+                        $this->projectDB->deleteCachedCollection($id);
+                    } catch (\Throwable $th) {
+                        Console::warning("'webhooks' from {$id}: {$th->getMessage()}");
+                    }
+                    break;
+                default:
+                    break;
+            }
+
+            usleep(50000);
+        }
+    }
+
+    /**
+     * Fix run on each document
+     *
+     * @param Document $document
+     * @return Document
+     */
+    protected function fixDocument(Document $document): Document
+    {
+        switch ($document->getCollection()) {
+            case 'projects':
+                /**
+                 * Bump version number.
+                 */
+                $document->setAttribute('version', '1.5.0');
+                break;
+            case 'schedules':
+                $document->setAttribute('resourceCollection', 'functions');
+                break;
+            case 'users':
+                if ($document->getAttribute('email', '') !== '') {
+                    $target = new Document([
+                        '$id' => ID::unique(),
+                        'userId' => $document->getId(),
+                        'userInternalId' => $document->getInternalId(),
+                        'providerType' => MESSAGE_TYPE_EMAIL,
+                        'identifier' => $document->getAttribute('email'),
+                    ]);
+                    $this->projectDB->createDocument('targets', $target);
+                }
+
+                if ($document->getAttribute('phone', '') !== '') {
+                    $target = new Document([
+                        '$id' => ID::unique(),
+                        'userId' => $document->getId(),
+                        'userInternalId' => $document->getInternalId(),
+                        'providerType' => MESSAGE_TYPE_SMS,
+                        'identifier' => $document->getAttribute('phone'),
+                    ]);
+                    $this->projectDB->createDocument('targets', $target);
+                }
+                break;
+        }
+        return $document;
+    }
+}
\ No newline at end of file