From 2269b3a3ea2413d52b2d69ad09c7d7842075a2e5 Mon Sep 17 00:00:00 2001
From: Kraiem Taha Yassine <tahayk2@gmail.com>
Date: Fri, 17 Nov 2023 16:42:49 +0100
Subject: [PATCH] feat(DB): rollback scripts (#1675)

* feat(DB): rollback scripts
---
 .../rollback_dbs/clickhouse/1.15.0/1.15.0.sql | 15 +++++
 .../rollback_dbs/postgresql/1.15.0/1.15.0.sql | 64 +++++++++++++++++++
 .../rollback_dbs/postgresql/1.15.0/1.15.0.sql | 57 +++++++++++++++++
 3 files changed, 136 insertions(+)
 create mode 100644 ee/scripts/schema/db/rollback_dbs/clickhouse/1.15.0/1.15.0.sql
 create mode 100644 ee/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql
 create mode 100644 scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql

diff --git a/ee/scripts/schema/db/rollback_dbs/clickhouse/1.15.0/1.15.0.sql b/ee/scripts/schema/db/rollback_dbs/clickhouse/1.15.0/1.15.0.sql
new file mode 100644
index 000000000..63fb1c91a
--- /dev/null
+++ b/ee/scripts/schema/db/rollback_dbs/clickhouse/1.15.0/1.15.0.sql
@@ -0,0 +1,15 @@
+CREATE OR REPLACE FUNCTION openreplay_version AS() -> 'v1.14.0-ee';
+
+ALTER TABLE experimental.events
+    DROP COLUMN IF EXISTS transfer_size;
+
+ALTER TABLE experimental.events
+    DROP COLUMN IF EXISTS selector;
+
+ALTER TABLE experimental.events
+    DROP COLUMN IF EXISTS coordinate;
+
+ALTER TABLE experimental.sessions
+    DROP COLUMN IF EXISTS timezone;
+
+DROP TABLE IF EXISTS experimental.ios_events;
\ No newline at end of file
diff --git a/ee/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql b/ee/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql
new file mode 100644
index 000000000..6cfa00033
--- /dev/null
+++ b/ee/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql
@@ -0,0 +1,64 @@
+\set previous_version 'v1.15.0-ee'
+\set next_version 'v1.14.0-ee'
+SELECT openreplay_version()                       AS current_version,
+       openreplay_version() = :'previous_version' AS valid_previous,
+       openreplay_version() = :'next_version'     AS is_next
+\gset
+
+\if :valid_previous
+\echo valid previous DB version :'previous_version', starting DB downgrade to :'next_version'
+BEGIN;
+SELECT format($fn_def$
+CREATE OR REPLACE FUNCTION openreplay_version()
+    RETURNS text AS
+$$
+SELECT '%1$s'
+$$ LANGUAGE sql IMMUTABLE;
+$fn_def$, :'next_version')
+\gexec
+
+--
+ALTER TABLE IF EXISTS events_common.requests
+    DROP COLUMN IF EXISTS transfer_size;
+
+ALTER TABLE IF EXISTS public.sessions
+    DROP COLUMN IF EXISTS timezone;
+
+ALTER TABLE IF EXISTS public.projects
+    DROP COLUMN IF EXISTS platform;
+
+DROP TABLE IF EXISTS events_common.crashes;
+
+DROP TABLE IF EXISTS public.crashes_ios;
+
+DROP SCHEMA IF EXISTS events_ios CASCADE;
+
+ALTER TABLE IF EXISTS public.users
+    DROP COLUMN IF EXISTS jwt_refresh_jti,
+    DROP COLUMN IF EXISTS jwt_refresh_iat;
+
+ALTER TABLE IF EXISTS events.clicks
+    DROP COLUMN IF EXISTS x,
+    DROP COLUMN IF EXISTS y;
+
+ALTER TABLE IF EXISTS public.metrics
+    DROP COLUMN IF EXISTS card_info;
+
+
+DROP TABLE IF EXISTS public.assist_events;
+
+DROP TABLE IF EXISTS public.assist_events_aggregates;
+
+
+DROP TABLE IF EXISTS public.assist_events_aggregates_logs;
+
+ALTER TABLE IF EXISTS public.users
+    DROP COLUMN IF EXISTS settings;
+
+COMMIT;
+
+\elif :is_next
+\echo old version detected :'next_version', nothing to do
+\else
+\warn skipping DB downgrade of :'next_version', expected previous version :'previous_version', found :'current_version'
+\endif
\ No newline at end of file
diff --git a/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql b/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql
new file mode 100644
index 000000000..54c645f4b
--- /dev/null
+++ b/scripts/schema/db/rollback_dbs/postgresql/1.15.0/1.15.0.sql
@@ -0,0 +1,57 @@
+\set previous_version 'v1.15.0'
+\set next_version 'v1.14.0'
+SELECT openreplay_version()                       AS current_version,
+       openreplay_version() = :'previous_version' AS valid_previous,
+       openreplay_version() = :'next_version'     AS is_next
+\gset
+
+\if :valid_previous
+\echo valid previous DB version :'previous_version', starting DB downgrade to :'next_version'
+BEGIN;
+SELECT format($fn_def$
+CREATE OR REPLACE FUNCTION openreplay_version()
+    RETURNS text AS
+$$
+SELECT '%1$s'
+$$ LANGUAGE sql IMMUTABLE;
+$fn_def$, :'next_version')
+\gexec
+
+--
+ALTER TABLE IF EXISTS events_common.requests
+    DROP COLUMN IF EXISTS transfer_size;
+
+ALTER TABLE IF EXISTS public.sessions
+    DROP COLUMN IF EXISTS timezone;
+
+ALTER TABLE IF EXISTS public.projects
+    DROP COLUMN IF EXISTS platform;
+
+DROP TABLE IF EXISTS events_common.crashes;
+
+DROP TABLE IF EXISTS public.crashes_ios;
+
+DROP SCHEMA IF EXISTS events_ios CASCADE;
+
+ALTER TABLE IF EXISTS public.users
+    DROP COLUMN IF EXISTS jwt_refresh_jti,
+    DROP COLUMN IF EXISTS jwt_refresh_iat;
+
+ALTER TABLE IF EXISTS events.clicks
+    DROP COLUMN IF EXISTS x,
+    DROP COLUMN IF EXISTS y;
+
+
+ALTER TABLE IF EXISTS public.metrics
+    DROP COLUMN IF EXISTS card_info;
+
+ALTER TABLE IF EXISTS public.users
+    DROP COLUMN IF EXISTS settings;
+
+COMMIT;
+
+\elif :is_next
+\echo old version detected :'next_version', nothing to do
+\else
+\warn skipping DB downgrade of :'next_version', expected previous version :'previous_version', found :'current_version'
+\endif
\ No newline at end of file