Dev

DEV

Author: Distrect

packages/api/src/database/entity/gantt/gant.entity.ts

@@ -1,12 +1,6 @@
 import ProjectEntity from '@entities/project/project.entity';
 import TaskEntity from '@entities/task/task.entity';
-import {
-  Column,
-  Entity,
-  ManyToOne,
-  OneToMany,
-  PrimaryGeneratedColumn,
-} from 'typeorm';
+import { Entity, OneToMany, OneToOne, PrimaryGeneratedColumn } from 'typeorm';
 
 @Entity()
 export default class GanttEntity {
@@ -16,6 +10,6 @@ export default class GanttEntity {
   @OneToMany(() => TaskEntity, (taskEntity) => taskEntity.gantt)
   tasks: TaskEntity[];
 
-  @ManyToOne(() => ProjectEntity, (projectEntity) => projectEntity.gantt)
+  @OneToOne(() => ProjectEntity, (projectEntity) => projectEntity.gantt)
   project: ProjectEntity;
 }

packages/api/src/database/entity/gantt/gantt.dao.ts

@@ -0,0 +1,31 @@
+import GanttEntity from '@entities/gantt/gant.entity';
+import { IGetGanttDATA } from '@entities/gantt/gantt.interface';
+import { Injectable } from '@nestjs/common';
+import { InjectRepository } from '@nestjs/typeorm';
+import { RecordNotFoundError } from 'src/shared/errors';
+import { Repository } from 'typeorm';
+
+@Injectable()
+export default class GanttDAO {
+  constructor(
+    @InjectRepository(GanttEntity)
+    private ganttRepository: Repository<GanttEntity>,
+  ) {}
+
+  public async getGanttWithTasks(args: IGetGanttDATA) {
+    const gantWithTaks = await this.ganttRepository.findOne({
+      where: args,
+      relations: { tasks: true },
+    });
+
+    if (gantWithTaks === null)
+      throw new RecordNotFoundError('Gantt record not found');
+
+    return gantWithTaks;
+  }
+
+  public async deleteGantt(args: IGetGanttDATA) {
+    await this.getGanttWithTasks(args);
+    await this.ganttRepository.delete(args);
+  }
+}

packages/api/src/database/entity/gantt/gantt.entity.module.ts

@@ -0,0 +1,11 @@
+import { Module } from '@nestjs/common';
+import GanttDAO from '@entities/gantt/gantt.dao';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import GanttEntity from '@entities/gantt/gant.entity';
+
+@Module({
+  imports: [TypeOrmModule.forFeature([GanttEntity])],
+  providers: [GanttDAO],
+  exports: [GanttDAO],
+})
+export default class GanttEntityModule {}

packages/api/src/database/entity/gantt/gantt.interface.ts

@@ -0,0 +1,17 @@
+import ProjectEntity from '@entities/project/project.entity';
+
+export type IGetGanttDATA =
+  | {
+      ganttID: number;
+    }
+  | {
+      project: {
+        projectID: number;
+      };
+    }
+  | {
+      ganttID: number;
+      project: {
+        projectID: number;
+      };
+    };

packages/api/src/database/entity/project/project.entity.ts

@@ -48,6 +48,7 @@ export default class ProjectEntity {
   )
   teamMembers: TeamMemberEntity[];
 
-  @OneToMany(() => GanttEntity, (ganttEntity) => ganttEntity.project)
+  @OneToOne(() => GanttEntity, (ganttEntity) => ganttEntity.project)
+  @JoinColumn({ name: 'ganttFID', referencedColumnName: 'ganttID' })
   gantt: GanttEntity;
 }

packages/api/src/database/entity/task/task.entity.module.ts

@@ -0,0 +1,11 @@
+import { Module } from '@nestjs/common';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import TaskDAO from '@entities/task/task.dao';
+import TaskEntity from '@entities/task/task.entity';
+
+@Module({
+  imports: [TypeOrmModule.forFeature([TaskEntity])],
+  providers: [TaskDAO],
+  exports: [TaskDAO],
+})
+export default class TaskEntityModule {}

packages/api/src/database/entity/task/task.entity.ts

@@ -1,5 +1,5 @@
 import GanttEntity from '@entities/gantt/gant.entity';
-import { PredecessorType } from './../../../../../shared/interfaces/project/task.interface';
+import { PredecessorType } from 'shared/interfaces';
 import {
   Entity,
   PrimaryGeneratedColumn,

packages/api/src/database/entity/task/task.interface.ts

@@ -1,6 +1,8 @@
-import GanttEntity from '@entities/gantt/gant.entity';
 import TaskEntity from '@entities/task/task.entity';
-import { PredecessorType } from 'shared/interfaces/project/task.interface';
+
+type TaskKeys = keyof TaskEntity;
+
+export interface ITask extends Pick<TaskEntity, TaskKeys> {}
 
 export type IGetTaskDATA =
   | {

packages/api/src/modules/gantt/gantt.feature.controller.ts

@@ -0,0 +1,12 @@
+import { Controller } from '@nestjs/common';
+import GanttFeatureService from './gantt.feature.service';
+import { ITasChangeDTO } from './gantt.feature.interface';
+
+@Controller('')
+export default class GanttFeatureController {
+  constructor(private ganttFeatureService: GanttFeatureService) {}
+
+  public async applyChanges(data: ITasChangeDTO[]) {
+    return await this.ganttFeatureService.applyChanges(data);
+  }
+}

packages/api/src/modules/gantt/gantt.feature.interface.ts

@@ -0,0 +1,10 @@
+import { ITask } from './../../database/entity/task/task.interface';
+
+type ChangeState = 'Unchanged' | 'Deleted' | 'Updated' | 'Created';
+
+export interface ITasChangeDTO
+  extends Omit<ITask, 'taskID' | 'gantt' | 'createdAt' | 'updatedAt'> {
+  taskID?: number;
+  ganttID?: number;
+  state: ChangeState;
+}

packages/api/src/modules/gantt/gantt.feature.module.ts

@@ -0,0 +1,11 @@
+import { Module } from '@nestjs/common';
+import GanttFeatureService from './gantt.feature.service';
+import GanttFeatureController from './gantt.feature.controller';
+import DatabaseModule from 'src/database/database.module';
+
+@Module({
+  imports: [DatabaseModule],
+  providers: [GanttFeatureService],
+  controllers: [GanttFeatureController],
+})
+export default class GanttFeatureModule {}

packages/api/src/modules/gantt/gantt.feature.service.ts

@@ -0,0 +1,61 @@
+import { Injectable } from '@nestjs/common';
+import GanttDAO from '@entities/gantt/gantt.dao';
+import TaskDAO from '@entities/task/task.dao';
+import { ITasChangeDTO } from './gantt.feature.interface';
+
+@Injectable()
+export default class GanttFeatureService {
+  constructor(
+    private ganttDAO: GanttDAO,
+    private taskDAO: TaskDAO,
+  ) {}
+
+  public async applyChanges(data: ITasChangeDTO[]) {
+    const updatedTasks: any[] = [];
+    const deletedTasks: any[] = [];
+    const createdTasks: any[] = [];
+
+    for (const task of data) {
+      if (task.state === 'Unchanged') continue;
+
+      switch (task.state) {
+        case 'Created':
+          const { taskID, ...rest } = task;
+          createdTasks.push(rest);
+          break;
+
+        case 'Deleted':
+          if (task.taskID === undefined)
+            throw new Error("Data missing 'taskID' attribute");
+
+          const gantt = task.ganttID
+            ? { gantt: { ganttID: task.ganttID } }
+            : {};
+
+          deletedTasks.push({
+            taskID: task.taskID,
+            gantt,
+          });
+          break;
+
+        case 'Updated':
+          if (task.taskID === undefined)
+            throw new Error("Data missing 'taskID' attribute");
+
+          updatedTasks.push(task);
+
+        default:
+          throw new Error('Unknown error');
+      }
+    }
+
+    const [createdTaskEntities, updatedTaskEntites, deletedTaskEntites] =
+      await Promise.all([
+        this.taskDAO.bulkCreateTaskRecords(createdTasks),
+        this.taskDAO.bulkUpdateTaskRecords(updatedTasks),
+        this.taskDAO.bulkDeleteTaskRecords(deletedTasks),
+      ]).then((res) => res);
+
+    return { createdTaskEntities, updatedTaskEntites, deletedTaskEntites };
+  }
+}