From 8b56334ac96f5be01cc8430d1decf6f50d18e6c9 Mon Sep 17 00:00:00 2001
From: Ly Tuan Kiet <lyhung998877@gmail.com>
Date: Thu, 17 Oct 2024 00:52:50 +0700
Subject: [PATCH] update logic upload file and get centerstaff

---
 src/Center/center.schema.ts             |  18 ++++
 src/Minio/minio.service.ts              |   4 +-
 src/Resume/resume.schema.ts             |  24 +++++
 src/UploadedFile/uploadedfile.schema.ts | 118 ++++++++++++++++++++++++
 4 files changed, 162 insertions(+), 2 deletions(-)

diff --git a/src/Center/center.schema.ts b/src/Center/center.schema.ts
index 7d762b2..e905096 100644
--- a/src/Center/center.schema.ts
+++ b/src/Center/center.schema.ts
@@ -67,6 +67,24 @@ export class CenterSchema extends PothosSchema {
           });
         },
       }),
+      // get current center of centerstaff by providing userId
+      centerByCenterStaff: t.prismaField({
+        type: this.center(),
+        args: {
+          userId: t.arg({ type: 'String', required: true }),
+        },
+        resolve: async (query, root, args, ctx, info) => {
+          return await this.prisma.center.findFirst({
+            where: {
+              CenterStaff: {
+                some: {
+                  staffId: args.userId,
+                },
+              },
+            },
+          });
+        },
+      }),
     }));
 
     // mutation section
diff --git a/src/Minio/minio.service.ts b/src/Minio/minio.service.ts
index 81fab22..de58fcd 100644
--- a/src/Minio/minio.service.ts
+++ b/src/Minio/minio.service.ts
@@ -41,10 +41,10 @@ export class MinioService {
     return url;
   }
 
-  async deleteFile(fileName: string) {
+  async deleteFile(fileName: string, category: string) {
     return await this.minioClient.removeObject(
       this.configService.get('BUCKET_NAME') ?? 'epess',
-      fileName,
+      `${category}/${fileName}`,
     );
   }
 
diff --git a/src/Resume/resume.schema.ts b/src/Resume/resume.schema.ts
index 0924def..755a5d9 100644
--- a/src/Resume/resume.schema.ts
+++ b/src/Resume/resume.schema.ts
@@ -8,6 +8,7 @@ import {
 import { Builder } from '../Graphql/graphql.builder';
 import { PrismaService } from '../Prisma/prisma.service';
 import { MinioService } from '../Minio/minio.service';
+import { ResumeStatus } from '@prisma/client';
 @Injectable()
 export class ResumeSchema extends PothosSchema {
   constructor(
@@ -179,6 +180,29 @@ export class ResumeSchema extends PothosSchema {
           return resume;
         },
       }),
+
+      updateResumeStatus: t.prismaField({
+        type: this.resume(),
+        args: {
+          resumeId: t.arg({
+            type: 'String',
+            required: true,
+          }),
+          status: t.arg({
+            type: ResumeStatus,
+            required: true,
+          }),
+        },
+        resolve: async (query, root, args) => {
+          const { resumeId, status } = args;
+          const resume = await this.prisma.resume.update({
+            ...query,
+            where: { id: resumeId },
+            data: { status },
+          });
+          return resume;
+        },
+      }),
     }));
   }
 }
diff --git a/src/UploadedFile/uploadedfile.schema.ts b/src/UploadedFile/uploadedfile.schema.ts
index c3e31f5..719a3b4 100644
--- a/src/UploadedFile/uploadedfile.schema.ts
+++ b/src/UploadedFile/uploadedfile.schema.ts
@@ -138,6 +138,124 @@ export class UploadedFileSchema extends PothosSchema {
           return uploadedFile;
         },
       }),
+
+      multipleUpload: t.prismaField({
+        type: [this.uploadedFile()],
+        args: {
+          userId: t.arg({
+            type: 'String',
+            required: true,
+          }),
+          files: t.arg({
+            type: ['Upload'],
+            required: true,
+          }),
+          fileType: t.arg({
+            type: UploadedFileType,
+            required: true,
+          }),
+        },
+        resolve: async (query, root, args) => {
+          const user = await this.prisma.user.findUnique({
+            where: {
+              id: args.userId,
+            },
+          });
+          if (!user) {
+            throw new Error('User not found');
+          }
+          const uploadedFiles = await Promise.all(
+            args.files.map((file) =>
+              this.minioService.uploadFile(file, 'files'),
+            ),
+          );
+          // get file urls
+          const fileUrls = await Promise.all(
+            uploadedFiles.map((file) =>
+              this.minioService.getFileUrl(file.filename, 'files'),
+            ),
+          );
+          // map uploadedFiles to db
+          const dbFiles = uploadedFiles.map((file, index) => ({
+            userId: user.id,
+            fileName: file.filename,
+            type: file.mimetype,
+            fileType: args.fileType,
+            fileUrl: fileUrls[index],
+            uploadedAt: new Date(),
+          }));
+          // create files in db
+          const createdFiles =
+            await this.prisma.uploadedFile.createManyAndReturn({
+              data: dbFiles,
+            });
+          return createdFiles;
+        },
+      }),
+
+      deleteUploadedFile: t.prismaField({
+        type: this.uploadedFile(),
+        args: {
+          id: t.arg({
+            type: 'String',
+            required: true,
+          }),
+        },
+        resolve: async (query, root, args) => {
+          const file = await this.prisma.uploadedFile.findUnique({
+            where: {
+              id: args.id,
+            },
+          });
+          if (!file) {
+            throw new Error('File not found');
+          }
+          await this.minioService.deleteFile(file.fileName, 'files');
+          await this.prisma.uploadedFile.delete({
+            where: {
+              id: file.id,
+            },
+          });
+          return file;
+        },
+      }),
+
+      deleteUploadedFiles: t.prismaField({
+        type: [this.uploadedFile()],
+        args: {
+          ids: t.arg({
+            type: ['String'],
+            required: true,
+          }),
+        },
+        resolve: async (query, root, args) => {
+          const files = await this.prisma.uploadedFile.findMany({
+            where: {
+              id: {
+                in: args.ids,
+              },
+            },
+          });
+          const fileUrls = await Promise.all(
+            files.map((file) =>
+              this.minioService.getFileUrl(file.fileName, 'files'),
+            ),
+          );
+          await this.prisma.uploadedFile.deleteMany({
+            where: {
+              id: {
+                in: args.ids,
+              },
+            },
+          });
+          await Promise.all(
+            files.map((file, index) =>
+              this.minioService.deleteFile(file.fileName, 'files'),
+            ),
+          );
+          return files;
+        },
+      }),
     }));
   }
 }