@forklaunch/implementation-iam-base 0.8.24 → 1.0.0

package/lib/eject/services/organization.service.ts

@@ -11,10 +11,11 @@ import {
   UpdateOrganizationDto
 } from '@forklaunch/interfaces-iam/types';
 import { AnySchemaValidator } from '@forklaunch/validator';
-import { EntityManager } from '@mikro-orm/core';
+import { EntityManager, FilterQuery, InferEntity } from '@mikro-orm/core';
 import { OrganizationDtos } from '../domain/types/iamDto.types';
 import { OrganizationEntities } from '../domain/types/iamEntities.types';
 import { OrganizationMappers } from '../domain/types/organization.mapper.types';
+import { Organization } from '../persistence/entities';
 
 export class BaseOrganizationService<
   SchemaValidator extends AnySchemaValidator,
@@ -87,14 +88,15 @@ export class BaseOrganizationService<
     if (em) {
       await em.persist(organization);
     } else {
-      await this.em.persistAndFlush(organization);
+      await this.em.persist(organization).flush();
     }
 
     return this.mappers.OrganizationMapper.toDto(organization);
   }
 
   async getOrganization(
-    idDto: IdDto,
+    idDto: IdDto &
+      FilterQuery<InferEntity<MapperEntities['OrganizationMapper']>>,
     em?: EntityManager
   ): Promise<MapperDomains['OrganizationMapper']> {
     if (this.evaluatedTelemetryOptions.logging) {
@@ -102,15 +104,12 @@ export class BaseOrganizationService<
     }
 
     const organization = await (em ?? this.em).findOneOrFail(
-      'Organization',
-      idDto,
-      {
-        populate: ['id', '*']
-      }
+      this.mappers.OrganizationMapper.entity as typeof Organization,
+      idDto
     );
 
     return this.mappers.OrganizationMapper.toDto(
-      organization as MapperEntities['OrganizationMapper']
+      organization as InferEntity<MapperEntities['OrganizationMapper']>
     );
   }
 
@@ -136,21 +135,31 @@ export class BaseOrganizationService<
     if (em) {
       await em.persist(updatedOrganization);
     } else {
-      await this.em.persistAndFlush(updatedOrganization);
+      await this.em.persist(updatedOrganization).flush();
     }
 
     return this.mappers.OrganizationMapper.toDto(updatedOrganization);
   }
 
-  async deleteOrganization(idDto: IdDto, em?: EntityManager): Promise<void> {
+  async deleteOrganization(
+    idDto: IdDto &
+      FilterQuery<InferEntity<MapperEntities['OrganizationMapper']>>,
+    em?: EntityManager
+  ): Promise<void> {
     if (this.evaluatedTelemetryOptions.logging) {
       this.openTelemetryCollector.info('Deleting organization', idDto);
     }
 
     if (em) {
-      await em.nativeDelete('Organization', idDto);
+      await em.nativeDelete(
+        this.mappers.OrganizationMapper.entity as typeof Organization,
+        idDto
+      );
     } else {
-      await this.em.nativeDelete('Organization', idDto);
+      await this.em.nativeDelete(
+        this.mappers.OrganizationMapper.entity as typeof Organization,
+        idDto
+      );
     }
   }
 }

package/lib/eject/services/permission.service.ts

@@ -15,7 +15,8 @@ import {
   UpdatePermissionDto
 } from '@forklaunch/interfaces-iam/types';
 import { AnySchemaValidator } from '@forklaunch/validator';
-import { EntityManager } from '@mikro-orm/core';
+import { EntityManager, InferEntity } from '@mikro-orm/core';
+import { Permission } from '../persistence/entities';
 import { PermissionDtos } from '../domain/types/iamDto.types';
 import { PermissionEntities } from '../domain/types/iamEntities.types';
 import { PermissionMappers } from '../domain/types/permission.mapper.types';
@@ -63,25 +64,25 @@ export class BasePermissionService<
 
   // start: global helper functions
   private async updateRolesWithPermissions(
-    roles: MapperEntities['RoleEntityMapper'][],
-    permissions: MapperEntities['PermissionMapper'][]
-  ): Promise<MapperEntities['RoleEntityMapper'][]> {
+    roles: InferEntity<MapperEntities['RoleEntityMapper']>[],
+    permissions: InferEntity<MapperEntities['PermissionMapper']>[]
+  ): Promise<InferEntity<MapperEntities['RoleEntityMapper']>[]> {
     return Promise.all(
       roles.map(async (role) => {
-        permissions.forEach((permission) => role.permissions.add(permission));
+        permissions.forEach((permission) => role.permissions?.add(permission));
         return role;
       })
     );
   }
 
   private async removePermissionsFromRoles(
-    roles: MapperEntities['RoleEntityMapper'][],
-    permissions: MapperEntities['PermissionMapper'][]
-  ): Promise<MapperEntities['RoleEntityMapper'][]> {
+    roles: InferEntity<MapperEntities['RoleEntityMapper']>[],
+    permissions: InferEntity<MapperEntities['PermissionMapper']>[]
+  ): Promise<InferEntity<MapperEntities['RoleEntityMapper']>[]> {
     return Promise.all(
       roles.map(async (role) => {
         permissions.forEach((permission) =>
-          role.permissions.remove(permission)
+          role.permissions?.remove(permission)
         );
         return role;
       })
@@ -91,16 +92,16 @@ export class BasePermissionService<
   private async getBatchRoles(
     roles?: IdsDto,
     em?: EntityManager
-  ): Promise<MapperEntities['RoleEntityMapper'][]> {
+  ): Promise<InferEntity<MapperEntities['RoleEntityMapper']>[]> {
     return roles
       ? await Promise.all(
           (await this.roleServiceFactory().getBatchRoles(roles, em)).map(
             async (role) => {
               return (em ?? this.em).merge(
-                await this.mappers.RoleEntityMapper.toEntity(
+                (await this.mappers.RoleEntityMapper.toEntity(
                   role,
                   em ?? this.em
-                )
+                )) as InferEntity<MapperEntities['RoleEntityMapper']>
               );
             }
           )
@@ -114,18 +115,21 @@ export class BasePermissionService<
     permission,
     addToRoles
   }: {
-    permission: MapperEntities['PermissionMapper'];
-    addToRoles: MapperEntities['RoleEntityMapper'][];
+    permission: InferEntity<MapperEntities['PermissionMapper']>;
+    addToRoles: InferEntity<MapperEntities['RoleEntityMapper']>[];
   }): Promise<{
-    permission: MapperEntities['PermissionMapper'];
-    roles: MapperEntities['RoleEntityMapper'][];
+    permission: InferEntity<MapperEntities['PermissionMapper']>;
+    roles: InferEntity<MapperEntities['RoleEntityMapper']>[];
   }> {
-    let roles: MapperEntities['RoleEntityMapper'][] = [];
+    let roles: InferEntity<MapperEntities['RoleEntityMapper']>[] = [];
     if (addToRoles) {
       roles = await this.updateRolesWithPermissions(addToRoles, [permission]);
     }
 
-    return { permission, roles };
+    return {
+      permission,
+      roles
+    };
   }
 
   private async extractCreatePermissionEntityToEntityData(
@@ -133,8 +137,8 @@ export class BasePermissionService<
     em?: EntityManager,
     ...args: unknown[]
   ): Promise<{
-    permission: MapperEntities['PermissionMapper'];
-    addToRoles: MapperEntities['RoleEntityMapper'][];
+    permission: InferEntity<MapperEntities['PermissionMapper']>;
+    addToRoles: InferEntity<MapperEntities['RoleEntityMapper']>[];
   }> {
     return {
       permission: (em ?? this.em).merge(
@@ -143,7 +147,7 @@ export class BasePermissionService<
           em ?? this.em,
           ...args
         )
-      ),
+      ) as InferEntity<MapperEntities['PermissionMapper']>,
       addToRoles: permissionDto.addToRolesIds
         ? await this.getBatchRoles({ ids: permissionDto.addToRolesIds }, em)
         : []
@@ -173,7 +177,7 @@ export class BasePermissionService<
     if (em) {
       await em.persist([permission, ...roles]);
     } else {
-      await this.em.persistAndFlush([permission, ...roles]);
+      await this.em.persist([permission, ...roles]).flush();
     }
 
     return this.mappers.PermissionMapper.toDto(permission);
@@ -189,8 +193,11 @@ export class BasePermissionService<
         permissionDtos
       );
     }
-    const rolesCache: Record<string, MapperEntities['RoleEntityMapper']> = {};
-    const permissions: MapperEntities['PermissionMapper'][] = [];
+    const rolesCache: Record<
+      string,
+      InferEntity<MapperEntities['RoleEntityMapper']>
+    > = {};
+    const permissions: InferEntity<MapperEntities['PermissionMapper']>[] = [];
     for (const createPermissionEntity of permissionDtos) {
       const { permission, roles } = await this.createPermissionEntity(
         await this.extractCreatePermissionEntityToEntityData(
@@ -200,15 +207,15 @@ export class BasePermissionService<
       );
       await Promise.all(
         roles.map(async (role) => {
-          if (!role.permissions.isInitialized()) {
-            return role.permissions.init();
+          if (!role.permissions?.isInitialized()) {
+            return role.permissions?.init();
           }
         })
       );
       await Promise.all(
         roles.map(async (role) => {
-          if (!role.permissions.isInitialized()) {
-            return role.permissions.init();
+          if (!role.permissions?.isInitialized()) {
+            return role.permissions?.init();
           }
         })
       );
@@ -217,9 +224,9 @@ export class BasePermissionService<
           rolesCache[role.id] &&
           role.permissions !== rolesCache[role.id].permissions
         ) {
-          role.permissions.getItems().forEach((permission) => {
-            if (!rolesCache[role.id].permissions.contains(permission)) {
-              rolesCache[role.id].permissions.add(permission);
+          role.permissions?.getItems().forEach((permission) => {
+            if (!rolesCache[role.id].permissions?.contains(permission)) {
+              rolesCache[role.id].permissions?.add(permission);
             }
           });
         } else {
@@ -233,7 +240,7 @@ export class BasePermissionService<
     if (em) {
       await em.persist(entities);
     } else {
-      await this.em.persistAndFlush(entities);
+      await this.em.persist(entities).flush();
     }
 
     return Promise.all(
@@ -250,9 +257,12 @@ export class BasePermissionService<
     if (this.evaluatedTelemetryOptions.logging) {
       this.openTelemetryCollector.info('Getting permission', idDto);
     }
-    const permission = await (em ?? this.em).findOneOrFail('Permission', idDto);
+    const permission = await (em ?? this.em).findOneOrFail(
+      this.mappers.PermissionMapper.entity as typeof Permission,
+      idDto
+    );
     return this.mappers.PermissionMapper.toDto(
-      permission as MapperEntities['PermissionMapper']
+      permission as InferEntity<MapperEntities['PermissionMapper']>
     );
   }
 
@@ -265,12 +275,15 @@ export class BasePermissionService<
     }
     return Promise.all(
       (
-        await (em ?? this.em).find('Permission', {
-          id: { $in: idsDto.ids }
-        })
+        await (em ?? this.em).find(
+          this.mappers.PermissionMapper.entity as typeof Permission,
+          {
+            id: { $in: idsDto.ids }
+          }
+        )
       ).map((permission) =>
         this.mappers.PermissionMapper.toDto(
-          permission as MapperEntities['PermissionMapper']
+          permission as InferEntity<MapperEntities['PermissionMapper']>
         )
       )
     );
@@ -282,8 +295,8 @@ export class BasePermissionService<
     em?: EntityManager,
     ...args: unknown[]
   ): Promise<{
-    permission: MapperEntities['PermissionMapper'];
-    roles: MapperEntities['RoleEntityMapper'][];
+    permission: InferEntity<MapperEntities['PermissionMapper']>;
+    roles: InferEntity<MapperEntities['RoleEntityMapper']>[];
   }> => {
     const permission = await this.mappers.UpdatePermissionMapper.toEntity(
       permissionDto,
@@ -297,17 +310,21 @@ export class BasePermissionService<
       ? await this.getBatchRoles({ ids: permissionDto.removeFromRolesIds }, em)
       : [];
 
-    let roles: MapperEntities['RoleEntityMapper'][] = [];
+    let roles: InferEntity<MapperEntities['RoleEntityMapper']>[] = [];
 
     roles = roles.concat(
-      await this.updateRolesWithPermissions(addToRoles, [permission])
+      await this.updateRolesWithPermissions(addToRoles, [
+        permission as InferEntity<MapperEntities['PermissionMapper']>
+      ])
     );
     roles = roles.concat(
-      await this.removePermissionsFromRoles(removeFromRoles, [permission])
+      await this.removePermissionsFromRoles(removeFromRoles, [
+        permission as InferEntity<MapperEntities['PermissionMapper']>
+      ])
     );
 
     return {
-      permission,
+      permission: permission as InferEntity<MapperEntities['PermissionMapper']>,
       roles
     };
   };
@@ -326,7 +343,7 @@ export class BasePermissionService<
     if (em) {
       await em.persist(entities);
     } else {
-      await this.em.persistAndFlush(entities);
+      await this.em.persist(entities).flush();
     }
 
     return this.mappers.PermissionMapper.toDto(permission);
@@ -342,8 +359,11 @@ export class BasePermissionService<
         permissionDtos
       );
     }
-    const rolesCache: Record<string, MapperEntities['RoleEntityMapper']> = {};
-    const permissions: MapperEntities['PermissionMapper'][] = [];
+    const rolesCache: Record<
+      string,
+      InferEntity<MapperEntities['RoleEntityMapper']>
+    > = {};
+    const permissions: InferEntity<MapperEntities['PermissionMapper']>[] = [];
     await (em ?? this.em).transactional(async (em) => {
       permissionDtos.map(async (updatePermissionDto) => {
         const { permission, roles } =
@@ -353,9 +373,9 @@ export class BasePermissionService<
             rolesCache[role.id] &&
             role.permissions !== rolesCache[role.id].permissions
           ) {
-            role.permissions.getItems().forEach((permission) => {
-              if (!rolesCache[role.id].permissions.contains(permission)) {
-                rolesCache[role.id].permissions.add(permission);
+            role.permissions?.getItems().forEach((permission) => {
+              if (!rolesCache[role.id].permissions?.contains(permission)) {
+                rolesCache[role.id].permissions?.add(permission);
               }
             });
           } else {
@@ -369,7 +389,7 @@ export class BasePermissionService<
       if (em) {
         await em.persist(entities);
       } else {
-        await this.em.persistAndFlush(entities);
+        await this.em.persist(entities).flush();
       }
     });
 
@@ -384,7 +404,10 @@ export class BasePermissionService<
     if (this.evaluatedTelemetryOptions.logging) {
       this.openTelemetryCollector.info('Deleting permission', idDto);
     }
-    await (em ?? this.em).nativeDelete('Permission', idDto);
+    await (em ?? this.em).nativeDelete(
+      this.mappers.PermissionMapper.entity as typeof Permission,
+      idDto
+    );
   }
 
   async deleteBatchPermissions(
@@ -394,8 +417,11 @@ export class BasePermissionService<
     if (this.evaluatedTelemetryOptions.logging) {
       this.openTelemetryCollector.info('Deleting batch permissions', idsDto);
     }
-    await (em ?? this.em).nativeDelete('Permission', {
-      id: { $in: idsDto.ids }
-    });
+    await (em ?? this.em).nativeDelete(
+      this.mappers.PermissionMapper.entity as typeof Permission,
+      {
+        id: { $in: idsDto.ids }
+      }
+    );
   }
 }

package/lib/eject/services/role.service.ts

@@ -5,7 +5,8 @@ import {
   TelemetryOptions
 } from '@forklaunch/core/http';
 import { RoleService } from '@forklaunch/interfaces-iam/interfaces';
-import { EntityManager } from '@mikro-orm/core';
+import { EntityManager, FilterQuery, InferEntity } from '@mikro-orm/core';
+import { Role } from '../persistence/entities';
 
 import { IdDto, IdsDto } from '@forklaunch/common';
 import {
@@ -73,7 +74,7 @@ export class BaseRoleService<
     if (em) {
       await em.persist(role);
     } else {
-      await this.em.persistAndFlush(role);
+      await this.em.persist(role).flush();
     }
 
     return this.mappers.RoleMapper.toDto(role);
@@ -97,7 +98,7 @@ export class BaseRoleService<
     if (em) {
       await em.persist(roles);
     } else {
-      await this.em.persistAndFlush(roles);
+      await this.em.persist(roles).flush();
     }
 
     return Promise.all(
@@ -110,11 +111,17 @@ export class BaseRoleService<
       this.openTelemetryCollector.info('Getting role', { id });
     }
 
-    const role = await (em ?? this.em).findOneOrFail('Role', id, {
-      populate: ['id', '*']
-    });
+    const role = await (em ?? this.em).findOneOrFail(
+      this.mappers.RoleMapper.entity as typeof Role,
+      id as FilterQuery<InferEntity<MapperEntities['RoleMapper']>>,
+      {
+        populate: ['id', '*']
+      }
+    );
 
-    return this.mappers.RoleMapper.toDto(role as MapperEntities['RoleMapper']);
+    return this.mappers.RoleMapper.toDto(
+      role as InferEntity<MapperEntities['RoleMapper']>
+    );
   }
 
   async getBatchRoles({ ids }: IdsDto, em?: EntityManager): Promise<RoleDto[]> {
@@ -125,7 +132,7 @@ export class BaseRoleService<
     return Promise.all(
       (
         await (em ?? this.em).find(
-          'Role',
+          this.mappers.RoleMapper.entity as typeof Role,
           {
             id: { $in: ids }
           },
@@ -134,7 +141,9 @@ export class BaseRoleService<
           }
         )
       ).map((role) =>
-        this.mappers.RoleMapper.toDto(role as MapperEntities['RoleMapper'])
+        this.mappers.RoleMapper.toDto(
+          role as InferEntity<MapperEntities['RoleMapper']>
+        )
       )
     );
   }
@@ -157,7 +166,7 @@ export class BaseRoleService<
     if (em) {
       await em.persist(role);
     } else {
-      await this.em.persistAndFlush(role);
+      await this.em.persist(role).flush();
     }
 
     return this.mappers.RoleMapper.toDto(role);
@@ -181,12 +190,10 @@ export class BaseRoleService<
     if (em) {
       await em.persist(roles);
     } else {
-      await this.em.persistAndFlush(roles);
+      await this.em.persist(roles).flush();
     }
     return Promise.all(
-      roles.map((role) =>
-        this.mappers.RoleMapper.toDto(role as MapperEntities['RoleMapper'])
-      )
+      roles.map((role) => this.mappers.RoleMapper.toDto(role))
     );
   }
 
@@ -195,7 +202,10 @@ export class BaseRoleService<
       this.openTelemetryCollector.info('Deleting role', idDto);
     }
 
-    await (em ?? this.em).nativeDelete('Role', idDto);
+    await (em ?? this.em).nativeDelete(
+      this.mappers.RoleMapper.entity as typeof Role,
+      idDto
+    );
   }
 
   async deleteBatchRoles(idsDto: IdsDto, em?: EntityManager): Promise<void> {
@@ -203,6 +213,11 @@ export class BaseRoleService<
       this.openTelemetryCollector.info('Deleting batch roles', idsDto);
     }
 
-    await (em ?? this.em).nativeDelete('Role', { id: { $in: idsDto.ids } });
+    await (em ?? this.em).nativeDelete(
+      this.mappers.RoleMapper.entity as typeof Role,
+      {
+        id: { $in: idsDto.ids }
+      }
+    );
   }
 }