From aac81144fe56015bae909393c2b4d3bdea9c8da2 Mon Sep 17 00:00:00 2001
From: Alexander Onnikov <Alexander.Onnikov@xored.com>
Date: Tue, 12 Mar 2024 15:31:10 +0700
Subject: [PATCH] UBERF-5953 Remove CollaboratorDocument class from Mongo
 (#4930)

Signed-off-by: Alexander Onnikov <Alexander.Onnikov@xored.com>
---
 models/document/src/migration.ts | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/models/document/src/migration.ts b/models/document/src/migration.ts
index 59b4576ce0..440ed5d15c 100644
--- a/models/document/src/migration.ts
+++ b/models/document/src/migration.ts
@@ -149,6 +149,16 @@ async function migrateWrongDomainContent (client: MigrationClient): Promise<void
   )
 }
 
+async function migrateDeleteCollaboratorDocument (client: MigrationClient): Promise<void> {
+  await client.deleteMany(DOMAIN_ATTACHMENT, { _class: 'document:class:CollaboratorDocument' as Ref<Class<Doc>> })
+  await client.deleteMany(DOMAIN_DOCUMENT, { _class: 'document:class:CollaboratorDocument' as Ref<Class<Doc>> })
+  await client.deleteMany(DOMAIN_TX, {
+    _class: core.class.TxCollectionCUD,
+    collection: 'attachments',
+    'tx.objectClass': 'document:class:CollaboratorDocument' as Ref<Class<Doc>>
+  })
+}
+
 async function setNoParent (client: MigrationClient): Promise<void> {
   await client.update(
     DOMAIN_DOCUMENT,
@@ -218,6 +228,10 @@ export const documentOperation: MigrateOperation = {
       {
         state: 'wrongDomainContent',
         func: migrateWrongDomainContent
+      },
+      {
+        state: 'deleteCollaboratorDocument',
+        func: migrateDeleteCollaboratorDocument
       }
     ])
   },