From cf39092c98e16ae67335003777e213fb82f3ffbd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matej=20Ba=C4=8Do?= <matejbaco2000@gmail.com>
Date: Wed, 22 Mar 2023 19:52:43 +0100
Subject: [PATCH] Add patch task for scheduleUpdatedAt

---
 src/Appwrite/Migration/Version/V17.php        | 11 ---
 src/Appwrite/Platform/Services/Tasks.php      |  2 +
 .../PatchDeleteScheduleUpdatedAtAttribute.php | 70 +++++++++++++++++++
 3 files changed, 72 insertions(+), 11 deletions(-)
 create mode 100644 src/Appwrite/Platform/Tasks/PatchDeleteScheduleUpdatedAtAttribute.php

diff --git a/src/Appwrite/Migration/Version/V17.php b/src/Appwrite/Migration/Version/V17.php
index 5ee897ae19..66a02662d1 100644
--- a/src/Appwrite/Migration/Version/V17.php
+++ b/src/Appwrite/Migration/Version/V17.php
@@ -58,17 +58,6 @@ class V17 extends Migration
                     }
 
                     break;
-                case 'functions':
-                    try {
-                        /**
-                         * Delete 'scheduleUpdatedAt' attribute
-                         */
-                        $this->projectDB->deleteAttribute($id, 'scheduleUpdatedAt');
-                        $this->projectDB->deleteCachedCollection($id);
-                    } catch (\Throwable $th) {
-                        Console::warning("'scheduleUpdatedAt' from {$id}: {$th->getMessage()}");
-                    }
-                    break;
 
                 default:
                     break;
diff --git a/src/Appwrite/Platform/Services/Tasks.php b/src/Appwrite/Platform/Services/Tasks.php
index 00cf3f89c5..a0c5d3a547 100644
--- a/src/Appwrite/Platform/Services/Tasks.php
+++ b/src/Appwrite/Platform/Services/Tasks.php
@@ -13,6 +13,7 @@ use Appwrite\Platform\Tasks\SDKs;
 use Appwrite\Platform\Tasks\Specs;
 use Appwrite\Platform\Tasks\SSL;
 use Appwrite\Platform\Tasks\Hamster;
+use Appwrite\Platform\Tasks\PatchDeleteScheduleUpdatedAtAttribute;
 use Appwrite\Platform\Tasks\Usage;
 use Appwrite\Platform\Tasks\Vars;
 use Appwrite\Platform\Tasks\Version;
@@ -33,6 +34,7 @@ class Tasks extends Service
             ->addAction(Install::getName(), new Install())
             ->addAction(Maintenance::getName(), new Maintenance())
             ->addAction(PatchCreateMissingSchedules::getName(), new PatchCreateMissingSchedules())
+            ->addAction(PatchDeleteScheduleUpdatedAtAttribute::getName(), new PatchDeleteScheduleUpdatedAtAttribute())
             ->addAction(Schedule::getName(), new Schedule())
             ->addAction(Migrate::getName(), new Migrate())
             ->addAction(SDKs::getName(), new SDKs())
diff --git a/src/Appwrite/Platform/Tasks/PatchDeleteScheduleUpdatedAtAttribute.php b/src/Appwrite/Platform/Tasks/PatchDeleteScheduleUpdatedAtAttribute.php
new file mode 100644
index 0000000000..4ee213303c
--- /dev/null
+++ b/src/Appwrite/Platform/Tasks/PatchDeleteScheduleUpdatedAtAttribute.php
@@ -0,0 +1,70 @@
+<?php
+
+namespace Appwrite\Platform\Tasks;
+
+use Utopia\Platform\Action;
+use Utopia\CLI\Console;
+use Utopia\Database\Query;
+use Utopia\Database\Database;
+use Utopia\Database\Validator\Authorization;
+
+class PatchDeleteScheduleUpdatedAtAttribute extends Action
+{
+    public static function getName(): string
+    {
+        return 'patch-delete-schedule-updated-at-attribute';
+    }
+
+    public function __construct()
+    {
+        $this
+            ->desc('Ensure function collections do not have scheduleUpdatedAt attribute')
+            ->inject('dbForConsole')
+            ->inject('getProjectDB')
+            ->callback(fn (Database $dbForConsole, callable $getProjectDB) => $this->action($dbForConsole, $getProjectDB));
+    }
+
+    /**
+     * Iterate over every function on every project to make sure there is a schedule. If not, recreate the schedule.
+     */
+    public function action(Database $dbForConsole, callable $getProjectDB): void
+    {
+        Authorization::disable();
+        Authorization::setDefaultStatus(false);
+
+        Console::title('PatchDeleteScheduleUpdatedAtAttribute V1');
+        Console::success(APP_NAME . ' PatchDeleteScheduleUpdatedAtAttribute v1 has started');
+
+        $limit = 100;
+        $projectCursor = null;
+        while (true) {
+            $projectsQueries = [Query::limit($limit)];
+            if ($projectCursor !== null) {
+                $projectsQueries[] = Query::cursorAfter($projectCursor);
+            }
+            $projects = $dbForConsole->find('projects', $projectsQueries);
+
+            if (count($projects) === 0) {
+                break;
+            }
+
+            foreach ($projects as $project) {
+                Console::log("Checking Project " . $project->getAttribute('name') . " (" . $project->getId() . ")");
+                $dbForProject = $getProjectDB($project);
+
+                try {
+                    /**
+                     * Delete 'scheduleUpdatedAt' attribute
+                     */
+                    $dbForProject->deleteAttribute('functions', 'scheduleUpdatedAt');
+                    $dbForProject->deleteCachedCollection('functions');
+                    Console::success("'scheduleUpdatedAt' deleted.");
+                } catch (\Throwable $th) {
+                    Console::warning("'scheduleUpdatedAt' errored: {$th->getMessage()}");
+                }
+            }
+
+            $projectCursor = $projects[array_key_last($projects)];
+        }
+    }
+}