Add metadata upload

Author: ArtemKolodko

src/app.controller.ts

@@ -20,10 +20,11 @@ import {GetTradesDto} from "./dto/trade.dto";
 import {AddUserDto, GetUsersDto} from "./dto/user.dto";
 import {UserService} from "./user/user.service";
 import {FileInterceptor} from "@nestjs/platform-express";
-import { Storage } from '@google-cloud/storage'
 import * as path from "node:path";
 import {GcloudService} from "./gcloud/gcloud.service";
 const serviceKey = path.join(__dirname, './keys.json')
+import { v4 as uuidv4 } from 'uuid';
+import {AddTokenMetadataDto} from "./dto/metadata.dto";
 
 @SkipThrottle()
 @ApiTags('app')
@@ -97,10 +98,35 @@ export class AppController {
 
   @Post('/uploadImage')
   @UseInterceptors(FileInterceptor('file'))
-  async uploadFile(@UploadedFile() uploadedFile: Express.Multer.File, @Headers() headers) {
+  async uploadImage(@UploadedFile() uploadedFile: Express.Multer.File, @Headers() headers) {
     const userAddress = headers['meta_user_address']
-    const publicImageUrl = await this.gCloudService.uploadImage(userAddress, uploadedFile)
-    this.logger.log(`Image uploaded, publicUrl=${publicImageUrl}, userAddress=${userAddress}`)
-    return publicImageUrl
+    const uuid = uuidv4()
+    const imageUrl = await this.gCloudService.uploadImage(uploadedFile, uuid)
+    this.logger.log(`Image uploaded, imageUrl=${imageUrl}, userAddress=${userAddress}`)
+    return imageUrl
+  }
+
+  @Post('/metadata')
+  async addMetadata(@Body() dto: AddTokenMetadataDto) {
+    let uuid = ''
+
+    if(!dto.image) {
+      throw new BadRequestException('Image property is missing')
+    }
+
+    const imageItems = dto.image.split('/')
+    if(imageItems.length > 0) {
+      uuid = imageItems[imageItems.length - 1].split('.')[0]
+    } else {
+      throw new BadRequestException('Invalid image url')
+    }
+
+    if(!uuid) {
+      throw new BadRequestException('Failed to get uuid')
+    }
+
+    const metadataUrl = await this.gCloudService.uploadMetadata(dto, uuid)
+    this.logger.log(`Metadata uploaded, url=${metadataUrl}, content: ${JSON.stringify(dto)}`)
+    return metadataUrl
   }
 }

src/app.service.ts

@@ -52,7 +52,7 @@ export class AppService {
         this.tokenFactoryContract = new this.web3.eth.Contract(TokenFactoryABI, contractAddress);
         this.bootstrap().then(
           () => {
-              // this.eventsTrackingLoop()
+              this.eventsTrackingLoop()
           }
         )
         this.logger.log(`App service started`)

src/dto/metadata.dto.ts

@@ -0,0 +1,30 @@
+import { ApiProperty } from '@nestjs/swagger';
+import {IsString} from 'class-validator';
+import {Transform, Type} from "class-transformer";
+
+export class AddTokenMetadataDto {
+  @ApiProperty({ type: String, required: true })
+  @Type(() => String)
+  @IsString()
+  userAddress: string;
+
+  @ApiProperty({ type: String, required: true })
+  @Type(() => String)
+  @IsString()
+  name: string;
+
+  @ApiProperty({ type: String, required: true })
+  @Type(() => String)
+  @IsString()
+  ticker: string;
+
+  @ApiProperty({ type: String, required: true })
+  @Type(() => String)
+  @IsString()
+  description: string;
+
+  @ApiProperty({ type: String, required: true })
+  @Type(() => String)
+  @IsString()
+  image: string;
+}

src/gcloud/gcloud.service.ts

@@ -1,6 +1,7 @@
 import {Injectable, Logger} from '@nestjs/common';
 import {ConfigService} from "@nestjs/config";
 import {Storage} from "@google-cloud/storage";
+import {AddTokenMetadataDto} from "../dto/metadata.dto";
 
 @Injectable()
 export class GcloudService {
@@ -15,11 +16,19 @@ export class GcloudService {
     })
   }
 
-  public async uploadImage(userAddress: string, uploadedFile: Express.Multer.File) {
+  public async uploadImage(uploadedFile: Express.Multer.File, filename: string) {
     const bucket = this.storage.bucket('pump-fun-metadata')
-    const storageFileUrl = `images/${userAddress}/test.jpg`
+    const storageFileUrl = `images/${filename}.jpg`
     const file = bucket.file(storageFileUrl)
     await file.save(uploadedFile.buffer)
     return `https://storage.googleapis.com/pump-fun-metadata/${storageFileUrl}`
   }
+
+  public async uploadMetadata(dto: AddTokenMetadataDto, filename: string) {
+    const bucket = this.storage.bucket('pump-fun-metadata')
+    const storageFileUrl = `metadata/${filename}.json`
+    const file = bucket.file(storageFileUrl)
+    await file.save(JSON.stringify(dto))
+    return `https://storage.googleapis.com/pump-fun-metadata/${storageFileUrl}`
+  }
 }