From babbc490f2b573c19fc7dd290ffe0a3e183523bb Mon Sep 17 00:00:00 2001
From: Alexander Onnikov <Alexander.Onnikov@xored.com>
Date: Wed, 10 Apr 2024 12:59:51 +0700
Subject: [PATCH] EQMS-669 Add role creation utilities (#5273)

Signed-off-by: Alexander Onnikov <Alexander.Onnikov@xored.com>
---
 .../spaceTypes/editor/CreateRole.svelte       | 29 +------
 plugins/setting/package.json                  |  1 +
 plugins/setting/src/utils.ts                  | 84 +++++++++++++++----
 3 files changed, 74 insertions(+), 40 deletions(-)

diff --git a/plugins/setting-resources/src/components/spaceTypes/editor/CreateRole.svelte b/plugins/setting-resources/src/components/spaceTypes/editor/CreateRole.svelte
index 45339f0d80..5dd40f74a8 100644
--- a/plugins/setting-resources/src/components/spaceTypes/editor/CreateRole.svelte
+++ b/plugins/setting-resources/src/components/spaceTypes/editor/CreateRole.svelte
@@ -14,11 +14,9 @@
 -->
 <script lang="ts">
   import presentation, { getClient } from '@hcengineering/presentation'
-  import setting from '@hcengineering/setting'
+  import { createSpaceTypeRole } from '@hcengineering/setting'
   import { Modal, ModernEditbox } from '@hcengineering/ui'
-  import core, { AttachedData, Attribute, PropertyType, Ref, Role, SpaceType } from '@hcengineering/core'
-  import { ArrOf, TypeRef } from '@hcengineering/model'
-  import { getEmbeddedLabel } from '@hcengineering/platform'
+  import { AttachedData, Role, SpaceType } from '@hcengineering/core'
 
   import settingRes from '../../../plugin'
   import { clearSettingsStore } from '../../../store'
@@ -37,28 +35,7 @@
       return
     }
 
-    const roleId = await client.addCollection(
-      core.class.Role,
-      core.space.Model,
-      type._id,
-      type._class,
-      'roles',
-      roleData
-    )
-
-    // Create role as an attribute of space type's mixin
-    await client.createDoc(
-      core.class.Attribute,
-      core.space.Model,
-      {
-        name: roleId,
-        attributeOf: type.targetClass,
-        type: ArrOf(TypeRef(core.class.Account)),
-        label: getEmbeddedLabel(`Role: ${name}`),
-        editor: setting.component.RoleAssignmentEditor
-      },
-      `role-${roleId}` as Ref<Attribute<PropertyType>>
-    )
+    await createSpaceTypeRole(client, type, roleData)
   }
 
   $: canSave = roleData.name.trim() !== ''
diff --git a/plugins/setting/package.json b/plugins/setting/package.json
index 826d117a57..5ee349736d 100644
--- a/plugins/setting/package.json
+++ b/plugins/setting/package.json
@@ -34,6 +34,7 @@
   "dependencies": {
     "@hcengineering/platform": "^0.6.9",
     "@hcengineering/core": "^0.6.28",
+    "@hcengineering/model": "^0.6.7",
     "@hcengineering/templates": "^0.6.7",
     "@hcengineering/ui": "^0.6.11"
   },
diff --git a/plugins/setting/src/utils.ts b/plugins/setting/src/utils.ts
index 0662ab7f50..cace8ab6b8 100644
--- a/plugins/setting/src/utils.ts
+++ b/plugins/setting/src/utils.ts
@@ -13,14 +13,29 @@
 // limitations under the License.
 //
 
-import core, { Class, ClassifierKind, Data, Doc, Ref, SpaceType, TxOperations } from '@hcengineering/core'
+import core, {
+  AttachedData,
+  Attribute,
+  Class,
+  ClassifierKind,
+  Data,
+  PropertyType,
+  Ref,
+  Role,
+  Space,
+  SpaceType,
+  TxOperations
+} from '@hcengineering/core'
+import { ArrOf, TypeRef } from '@hcengineering/model'
 import { getEmbeddedLabel } from '@hcengineering/platform'
+import setting from './index'
 
-export async function createSpaceType (
+export async function createSpaceType<T extends SpaceType> (
   client: TxOperations,
-  data: Omit<Data<SpaceType>, 'targetClass'>,
-  _id: Ref<SpaceType>
-): Promise<Ref<SpaceType>> {
+  data: Omit<Data<T>, 'targetClass'>,
+  _id: Ref<T>,
+  _class: Ref<Class<T>> = core.class.SpaceType
+): Promise<Ref<T>> {
   const descriptorObj = client.getModel().findObject(data.descriptor)
   if (descriptorObj === undefined) {
     throw new Error('Descriptor is not found in the model')
@@ -29,7 +44,7 @@ export async function createSpaceType (
   const baseClassClazz = client.getHierarchy().getClass(descriptorObj.baseClass)
   // NOTE: it is important for this id to be consistent when re-creating the same
   // space type with the same id as it will happen during every migration if type is created by the system
-  const spaceTypeMixinId = `${_id}:type:mixin` as Ref<Class<Doc>>
+  const spaceTypeMixinId = `${_id}:type:mixin` as Ref<Class<Space>>
   await client.createDoc(
     core.class.Mixin,
     core.space.Model,
@@ -43,15 +58,56 @@ export async function createSpaceType (
   )
 
   return await client.createDoc(
-    core.class.SpaceType,
+    _class,
     core.space.Model,
-    {
-      shortDescription: data.shortDescription,
-      descriptor: data.descriptor,
-      roles: data.roles,
-      name: data.name,
-      targetClass: spaceTypeMixinId
-    },
+    { ...data, targetClass: spaceTypeMixinId } as unknown as Data<T>,
     _id
   )
 }
+
+export async function createSpaceTypeRole (
+  client: TxOperations,
+  type: Pick<SpaceType, '_id' | '_class' | 'targetClass'>,
+  data: AttachedData<Role>,
+  _id?: Ref<Role> | undefined
+): Promise<Ref<Role>> {
+  const name = data.name.trim()
+
+  const roleId = await client.addCollection(
+    core.class.Role,
+    core.space.Model,
+    type._id,
+    type._class,
+    'roles',
+    data,
+    _id
+  )
+
+  await client.createDoc(
+    core.class.Attribute,
+    core.space.Model,
+    {
+      name: roleId,
+      attributeOf: type.targetClass,
+      type: ArrOf(TypeRef(core.class.Account)),
+      label: getEmbeddedLabel(`Role: ${name}`),
+      editor: setting.component.RoleAssignmentEditor
+    },
+    `role-${roleId}` as Ref<Attribute<PropertyType>>
+  )
+
+  return roleId
+}
+
+export async function createSpaceTypeRoles (
+  tx: TxOperations,
+  typeId: Ref<SpaceType>,
+  roles: Pick<Role, '_id' | 'name' | 'permissions'>[]
+): Promise<void> {
+  const spaceType = await tx.findOne(core.class.SpaceType, { _id: typeId })
+  if (spaceType === undefined) return
+
+  for (const { _id, name, permissions } of roles) {
+    await createSpaceTypeRole(tx, spaceType, { name, permissions }, _id)
+  }
+}