From 2ed5d9dce7a05c1517021ef5d5d6b4329f9aee97 Mon Sep 17 00:00:00 2001
From: hannathkadher <h.kadher@syncrow.ae>
Date: Wed, 20 Nov 2024 16:16:17 +0400
Subject: [PATCH] Added space product relation

---
 libs/common/src/database/database.module.ts   |  2 ++
 .../product/entities/product.entity.ts        |  4 +++
 .../space/entities/space-product.entity.ts    | 32 +++++++++++++++++++
 .../modules/space/entities/space.entity.ts    |  4 +++
 .../space/repositories/space.repository.ts    |  8 +++++
 5 files changed, 50 insertions(+)
 create mode 100644 libs/common/src/modules/space/entities/space-product.entity.ts

diff --git a/libs/common/src/database/database.module.ts b/libs/common/src/database/database.module.ts
index 868ef5a..92f3db6 100644
--- a/libs/common/src/database/database.module.ts
+++ b/libs/common/src/database/database.module.ts
@@ -23,6 +23,7 @@ import { CommunityEntity } from '../modules/community/entities';
 import { DeviceStatusLogEntity } from '../modules/device-status-log/entities';
 import { SceneEntity, SceneIconEntity } from '../modules/scene/entities';
 import { SceneDeviceEntity } from '../modules/scene-device/entities';
+import { SpaceProductEntity } from '../modules/space/entities/space-product.entity';
 
 @Module({
   imports: [
@@ -49,6 +50,7 @@ import { SceneDeviceEntity } from '../modules/scene-device/entities';
           SpaceEntity,
           SubspaceEntity,
           SpaceTypeEntity,
+          SpaceProductEntity,
           UserSpaceEntity,
           DeviceUserPermissionEntity,
           UserRoleEntity,
diff --git a/libs/common/src/modules/product/entities/product.entity.ts b/libs/common/src/modules/product/entities/product.entity.ts
index 6553dc1..a7cf957 100644
--- a/libs/common/src/modules/product/entities/product.entity.ts
+++ b/libs/common/src/modules/product/entities/product.entity.ts
@@ -2,6 +2,7 @@ import { Column, Entity, OneToMany } from 'typeorm';
 import { ProductDto } from '../dtos';
 import { AbstractEntity } from '../../abstract/entities/abstract.entity';
 import { DeviceEntity } from '../../device/entities';
+import { SpaceProductEntity } from '../../space/entities/space-product.entity';
 
 @Entity({ name: 'product' })
 export class ProductEntity extends AbstractEntity<ProductDto> {
@@ -21,6 +22,9 @@ export class ProductEntity extends AbstractEntity<ProductDto> {
   })
   public prodType: string;
 
+  @OneToMany(() => SpaceProductEntity, (spaceProduct) => spaceProduct.product)
+  spaceProducts: SpaceProductEntity[];
+
   @OneToMany(
     () => DeviceEntity,
     (devicesProductEntity) => devicesProductEntity.productDevice,
diff --git a/libs/common/src/modules/space/entities/space-product.entity.ts b/libs/common/src/modules/space/entities/space-product.entity.ts
new file mode 100644
index 0000000..3dd6cbe
--- /dev/null
+++ b/libs/common/src/modules/space/entities/space-product.entity.ts
@@ -0,0 +1,32 @@
+import { Column, Entity, ManyToOne, JoinColumn } from 'typeorm';
+import { SpaceEntity } from './space.entity';
+import { AbstractEntity } from '../../abstract/entities/abstract.entity';
+import { ProductEntity } from '../../product/entities';
+
+@Entity({ name: 'space-product' })
+export class SpaceProductEntity extends AbstractEntity<SpaceProductEntity> {
+  @ManyToOne(() => SpaceEntity, (space) => space.spaceProducts, {
+    nullable: false,
+    onDelete: 'CASCADE',
+  })
+  @JoinColumn({ name: 'space_id' })
+  space: SpaceEntity;
+
+  @ManyToOne(() => ProductEntity, (product) => product.spaceProducts, {
+    nullable: false,
+    onDelete: 'CASCADE',
+  })
+  @JoinColumn({ name: 'product_id' })
+  product: ProductEntity;
+
+  @Column({
+    nullable: false,
+    type: 'int',
+  })
+  productCount: number;
+
+  constructor(partial: Partial<SpaceProductEntity>) {
+    super();
+    Object.assign(this, partial);
+  }
+}
diff --git a/libs/common/src/modules/space/entities/space.entity.ts b/libs/common/src/modules/space/entities/space.entity.ts
index 196a522..837b875 100644
--- a/libs/common/src/modules/space/entities/space.entity.ts
+++ b/libs/common/src/modules/space/entities/space.entity.ts
@@ -12,6 +12,7 @@ import { UserSpaceEntity } from '../../user/entities';
 import { DeviceEntity } from '../../device/entities';
 import { CommunityEntity } from '../../community/entities';
 import { SubspaceEntity } from './subspace.entity';
+import { SpaceProductEntity } from './space-product.entity';
 
 @Entity({ name: 'space-type' })
 export class SpaceTypeEntity extends AbstractEntity<SpaceTypeDto> {
@@ -83,6 +84,9 @@ export class SpaceEntity extends AbstractEntity<SpaceDto> {
   )
   devices: DeviceEntity[];
 
+  @OneToMany(() => SpaceProductEntity, (spaceProduct) => spaceProduct.space)
+  spaceProducts: SpaceProductEntity[];
+
   constructor(partial: Partial<SpaceEntity>) {
     super();
     Object.assign(this, partial);
diff --git a/libs/common/src/modules/space/repositories/space.repository.ts b/libs/common/src/modules/space/repositories/space.repository.ts
index 5e0962c..d34fc24 100644
--- a/libs/common/src/modules/space/repositories/space.repository.ts
+++ b/libs/common/src/modules/space/repositories/space.repository.ts
@@ -1,6 +1,7 @@
 import { DataSource, Repository } from 'typeorm';
 import { Injectable } from '@nestjs/common';
 import { SpaceEntity, SpaceTypeEntity, SubspaceEntity } from '../entities';
+import { SpaceProductEntity } from '../entities/space-product.entity';
 
 @Injectable()
 export class SpaceRepository extends Repository<SpaceEntity> {
@@ -22,3 +23,10 @@ export class SubspaceRepository extends Repository<SubspaceEntity> {
     super(SubspaceEntity, dataSource.createEntityManager());
   }
 }
+
+@Injectable()
+export class SpaceProductRepository extends Repository<SpaceProductEntity> {
+  constructor(private dataSource: DataSource) {
+    super(SpaceProductEntity, dataSource.createEntityManager());
+  }
+}