Image uploading for moderators

Author: enchantinggg4

src/app.module.ts

@@ -35,7 +35,6 @@ import * as redisStore from "cache-manager-redis-store";
 import { MetaController } from "./rest/meta/meta.controller";
 import { HttpCacheInterceptor } from "./utils/cache-key-track";
 import { GetReportsAvailableQuery } from "./gateway/queries/GetReportsAvailable/get-reports-available.query";
-import { MulterModule } from "@nestjs/platform-express";
 import { ScheduleModule } from "@nestjs/schedule";
 import { MainService } from "./main.service";
 import { Entities } from "./db.config";
@@ -119,6 +118,8 @@ import * as TelegramBot from "node-telegram-bot-api";
 import { TelegramNotificationService } from "./rest/notification/telegram-notification.service";
 import { NewTicketMessageCreatedHandler } from "./rest/notification/event-handler/new-ticket-message-created.handler";
 import { MetricsService } from "./metrics.service";
+import { StorageController } from "./rest/storage/storage.controller";
+import { StorageMapper } from "./rest/storage/storage.mapper";
 
 const OPENAPI_GENERATED: Provider[] = [
   {
@@ -240,9 +241,6 @@ const OPENAPI_GENERATED: Provider[] = [
       rootPath: join(__dirname, "./upload"),
       serveRoot: "/static/",
     }),
-    MulterModule.register({
-      dest: "./dist/upload",
-    }),
     JwtModule.registerAsync({
       useFactory(config: ConfigService): JwtModuleOptions {
         return {
@@ -304,6 +302,8 @@ const OPENAPI_GENERATED: Provider[] = [
     NotificationController,
     FeedbackController,
     AdminFeedbackController,
+
+    StorageController,
   ],
   providers: [
     ...OPENAPI_GENERATED,
@@ -354,6 +354,7 @@ const OPENAPI_GENERATED: Provider[] = [
     LobbyMapper,
     NotificationMapper,
     FeedbackMapper,
+    StorageMapper,
 
     UserRepository,
     UserCreatedHandler,

src/main.ts

@@ -37,9 +37,7 @@ async function bootstrap() {
     .addBearerAuth()
     .build();
 
-  const document = SwaggerModule.createDocument(app, options, {
-    deepScanRoutes: true,
-  });
+  const document = SwaggerModule.createDocument(app, options);
   SwaggerModule.setup("api", app, document);
 
   app.connectMicroservice({

src/rest/storage/storage.controller.ts

@@ -0,0 +1,98 @@
+import {
+  Controller,
+  Get,
+  Post,
+  Query,
+  UploadedFile,
+  UseInterceptors,
+} from "@nestjs/common";
+import { ReqLoggingInterceptor } from "../../middleware/req-logging.interceptor";
+import { ApiBody, ApiConsumes, ApiQuery, ApiTags } from "@nestjs/swagger";
+import { ModeratorGuard, WithUser } from "../../utils/decorator/with-user";
+import { FileInterceptor } from "@nestjs/platform-express";
+import { InjectS3, S3 } from "nestjs-s3";
+import {
+  ListObjectsV2CommandInput,
+  PutObjectCommandInput,
+} from "@aws-sdk/client-s3";
+import { ConfigService } from "@nestjs/config";
+import { UploadedImageDto, UploadedImagePageDto } from "./storage.dto";
+import { StorageMapper } from "./storage.mapper";
+
+interface IFile {
+  fieldname: string;
+  originalname: string;
+  encoding: string;
+  mimetype: string;
+  buffer: Buffer;
+  size: number;
+}
+
+@UseInterceptors(ReqLoggingInterceptor)
+@Controller("storage")
+@ApiTags("storage")
+export class StorageController {
+  constructor(
+    @InjectS3() private readonly s3: S3,
+    private readonly config: ConfigService,
+    private readonly mapper: StorageMapper,
+  ) {}
+
+  @ApiBody({
+    schema: {
+      type: "object",
+      properties: {
+        file: {
+          type: "string",
+          format: "binary",
+        },
+      },
+    },
+  })
+  @ApiConsumes("multipart/form-data")
+  @ModeratorGuard()
+  @WithUser()
+  @UseInterceptors(FileInterceptor("file"))
+  @Post("upload")
+  public async uploadImage(
+    @UploadedFile() file: IFile,
+  ): Promise<UploadedImageDto> {
+    const Key = this.config.get("s3.uploadPrefix") + file.originalname;
+
+    const putObjectCommandInput: PutObjectCommandInput = {
+      Bucket: this.config.get("s3.bucket"),
+      Key,
+      Body: file.buffer,
+      ContentType: file.mimetype,
+      ACL: "public-read",
+
+      Metadata: {
+        originalName: file.originalname,
+      },
+    };
+
+    await this.s3.putObject(putObjectCommandInput);
+
+    return this.mapper.mapS3Item(Key);
+  }
+
+  @ApiQuery({
+    name: "ctoken",
+    required: false,
+  })
+  @Get()
+  public async getUploadedFiles(
+    @Query("ctoken") ctoken?: string,
+  ): Promise<UploadedImagePageDto> {
+    const response = await this.s3.listObjectsV2({
+      Bucket: this.config.get("s3.bucket"),
+      Prefix: this.config.get("s3.uploadPrefix"),
+      ContinuationToken: ctoken,
+    } satisfies ListObjectsV2CommandInput);
+
+    return {
+      items: response.Contents.map((it) => it.Key).map(this.mapper.mapS3Item),
+      ctoken: response.ContinuationToken,
+    };
+  }
+}

src/rest/storage/storage.dto.ts

@@ -0,0 +1,8 @@
+export class UploadedImageDto {
+  url: string;
+}
+
+export class UploadedImagePageDto {
+  items: UploadedImageDto[];
+  ctoken?: string;
+}

src/rest/storage/storage.mapper.ts

@@ -0,0 +1,12 @@
+import { Injectable } from "@nestjs/common";
+import { ConfigService } from "@nestjs/config";
+import { UploadedImageDto } from "./storage.dto";
+
+@Injectable()
+export class StorageMapper {
+  constructor(private readonly config: ConfigService) {}
+
+  public mapS3Item = (key: string): UploadedImageDto => ({
+    url: `${this.config.get("api.s3root")}/${this.config.get("s3.bucket")}/${key}`,
+  });
+}