feat(be): implement image upload

.devcontainer/devcontainer.json

@@ -7,7 +7,7 @@
     "database",
     "cache",
     "rabbitmq",
-    "testcase",
+    "storage",
     "test-database"
   ],
   "workspaceFolder": "/workspace",

.env.stage

@@ -22,17 +22,26 @@ RABBITMQ_CONSUMER_CONNECTION_NAME=iris-consumer
 RABBITMQ_CONSUMER_TAG=consumer
 RABBITMQ_PRODUCER_CONNECTION_NAME=iris-producer
 
+# Storage
+STORAGE_BUCKET_ENDPOINT_URL=http://127.0.0.1:9000
 # Testcase Endpoint
 TESTCASE_BUCKET_NAME=test-bucket
-TESTCASE_ENDPOINT_URL=http://127.0.0.1:9000
+# TESTCASE_ENDPOINT_URL=http://127.0.0.1:9000
 TESTCASE_ACCESS_KEY=skku
 TESTCASE_SECRET_KEY=skku1234
 
+# Media Upload Endpoint
+MEDIA_BUCKET_NAME=image-bucket
+# MEDIA_BUCKET_BASE_URL=http://127.0.0.1:9000/image-bucket/
+MEDIA_ACCESS_KEY=skku
+MEDIA_SECRET_KEY=skku1234
+
 REDIS_HOST=127.0.0.1
 REDIS_PORT=6380
 
 DATABASE_URL=postgresql://postgres:[email protected]:5433/skkuding?schema=public
 TEST_DATABASE_URL=postgresql://postgres:[email protected]:5434/skkuding?schema=public
 
+
 # TODO: Add information where each of these variables are used
 # TODO: I want to edit values after the container is created...

.gitpod.yml

@@ -23,7 +23,7 @@ ports:
   - port: 15672 # RabbitMQ Dashboard
     onOpen: ignore
 
-  - port: 30000 # Testcase Server
+  - port: 30000 # Storage Server
     onOpen: ignore
 
 tasks:

apps/backend/apps/admin/src/problem/model/image.output.ts

@@ -0,0 +1,7 @@
+import { Field, ObjectType } from '@nestjs/graphql'
+
+@ObjectType({ description: 'image' })
+export class ImageSource {
+  @Field(() => String)
+  src: string
+}

apps/backend/apps/admin/src/problem/problem.module.ts

@@ -1,4 +1,5 @@
 import { Module } from '@nestjs/common'
+import { ConfigModule } from '@nestjs/config'
 import { StorageModule } from '@admin/storage/storage.module'
 import { ProblemTagResolver, TagResolver } from './problem-tag.resolver'
 import {
@@ -9,7 +10,7 @@ import {
 import { ProblemService } from './problem.service'
 
 @Module({
-  imports: [StorageModule],
+  imports: [StorageModule, ConfigModule],
   providers: [
     ProblemResolver,
     ProblemTagResolver,

apps/backend/apps/admin/src/problem/problem.resolver.ts

@@ -19,12 +19,14 @@ import {
 } from '@nestjs/graphql'
 import {
   ContestProblem,
+  Image,
   Problem,
   ProblemTag,
   ProblemTestcase,
   WorkbookProblem
 } from '@generated'
 import { Prisma } from '@prisma/client'
+import { PrismaClientKnownRequestError } from '@prisma/client/runtime/library'
 import { AuthenticatedRequest } from '@libs/auth'
 import { OPEN_SPACE_ID } from '@libs/constants'
 import {
@@ -33,6 +35,7 @@ import {
   UnprocessableDataException
 } from '@libs/exception'
 import { CursorValidationPipe, GroupIDPipe, RequiredIntPipe } from '@libs/pipe'
+import { ImageSource } from './model/image.output'
 import {
   CreateProblemInput,
   UploadFileInput,
@@ -105,6 +108,43 @@ export class ProblemResolver {
     }
   }
 
+  @Mutation(() => ImageSource)
+  async uploadImage(
+    @Args('input') input: UploadFileInput,
+    @Context('req') req: AuthenticatedRequest
+  ) {
+    try {
+      return await this.problemService.uploadImage(input, req.user.id)
+    } catch (error) {
+      if (error instanceof UnprocessableDataException) {
+        throw error.convert2HTTPException()
+      }
+      this.logger.error(error)
+      throw new InternalServerErrorException()
+    }
+  }
+
+  @Mutation(() => Image)
+  async deleteImage(
+    @Args('filename') filename: string,
+    @Context('req') req: AuthenticatedRequest
+  ) {
+    try {
+      return await this.problemService.deleteImage(filename, req.user.id)
+    } catch (error) {
+      if (error instanceof UnprocessableDataException) {
+        throw error.convert2HTTPException()
+      } else if (
+        error instanceof PrismaClientKnownRequestError &&
+        error.code == 'P2025'
+      ) {
+        throw new NotFoundException(error.message)
+      }
+      this.logger.error(error)
+      throw new InternalServerErrorException()
+    }
+  }
+
   @Query(() => [Problem])
   async getProblems(
     @Args(

apps/backend/apps/admin/src/problem/problem.service.spec.ts

@@ -11,7 +11,7 @@ import {
   UnprocessableDataException
 } from '@libs/exception'
 import { PrismaService } from '@libs/prisma'
-import { S3Provider } from '@admin/storage/s3.provider'
+import { S3MediaProvider, S3Provider } from '@admin/storage/s3.provider'
 import { StorageService } from '@admin/storage/storage.service'
 import {
   exampleContest,
@@ -96,6 +96,7 @@ describe('ProblemService', () => {
         StorageService,
         ConfigService,
         S3Provider,
+        S3MediaProvider,
         { provide: CACHE_MANAGER, useValue: { del: () => null } }
       ]
     }).compile()