✨ fix get profile

kookmin-sw · May 12, 2024 · b8cf57f · b8cf57f
1 parent 3ed6495
commit b8cf57f
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 25 deletions.
diff --git a/src/profile/dto/profile.dto.ts b/src/profile/dto/profile.dto.ts
@@ -30,6 +30,12 @@ export class ProfileDto {
     example: 'https://github.com/KMUlee',
   })
   readonly githubLink: string;
+
+  @IsString()
+  @ApiProperty({
+    example: 'https://s3.ap-northeast-2.amazonaws.com/...',
+  })
+  readonly image: string;
 }
 
 export class GetProfileDto {

diff --git a/src/profile/entities/profile.entity.ts b/src/profile/entities/profile.entity.ts
@@ -32,6 +32,9 @@ export class Profile {
   @Column()
   githubLink: string;
 
+  @Column()
+  image: string;
+
   @CreateDateColumn({ type: 'timestamptz' })
   @Exclude({ toPlainOnly: true })
   createdAt: Date;

diff --git a/src/profile/profile.controller.ts b/src/profile/profile.controller.ts
@@ -25,11 +25,6 @@ import { FileInterceptor } from '@nestjs/platform-express';
 export class ProfileController {
   constructor(private readonly profileService: ProfileService) {}
 
-  @Get('/image')
-  async getProfileImage(@GetUser() user: Payload): Promise<{ src: string }> {
-    return { src: await this.profileService.getImage(user.userId) };
-  }
-
   @Get('/:profileId')
   @ApiParam({
     name: 'profileId',
@@ -63,23 +58,6 @@ export class ProfileController {
     return await this.profileService.getProfile(user.userId);
   }
 
-  @Post('/')
-  @ApiBody({
-    description: 'Create a profile',
-    type: ProfileDto,
-  })
-  @ApiResponse({
-    status: 201,
-    description: 'Create a profile',
-    type: ProfileDto,
-  })
-  async createProfile(
-    @GetUser() user: Payload,
-    @Body() newProfile: ProfileDto,
-  ): Promise<ProfileDto> {
-    return await this.profileService.createProfile(user.userId, newProfile);
-  }
-
   @Put('/')
   @ApiBody({
     description: 'Create a profile',

diff --git a/src/profile/profile.service.ts b/src/profile/profile.service.ts
@@ -179,18 +179,36 @@ export class ProfileService {
   ): Promise<AWS.S3.ManagedUpload.SendData> {
     await this.checkAndCreateUserFolder(userId);
     const fileKey = `${userId}/profile.png`;
-
-    return this.s3
+    const upload = await this.s3
       .upload({
         Bucket: this.bucketName,
         Key: fileKey,
         Body: file.buffer,
         ACL: 'public-read',
       })
       .promise();
+    const profile = await this.profileRepository.findOne({
+      where: {
+        user: {
+          id: userId,
+        },
+      },
+      relations: ['user'],
+    });
+    try {
+      await this.profileRepository.save({
+        ...profile,
+        image: upload.Location,
+      });
+    } catch (error) {
+      console.error(`Error saving image URL: ${error}`);
+      throw new Error('Unable to save image URL.');
+    }
+
+    return upload;
   }
 
-  async getImage(userId: number) {
+  async getImage(userId: number): Promise<string> {
     const exists = await this.fileExists(userId);
 
     if (!exists) {

diff --git a/src/user/user.service.ts b/src/user/user.service.ts
@@ -45,6 +45,7 @@ export class UserService {
         title: '',
         description: '',
         githubLink: '',
+        image: '',
       });
       this.logger.log('profile created');