feat: add benchmark creation

Author: Mikadows

src/app.module.ts

@@ -7,6 +7,7 @@ import { LoggerModule } from 'nestjs-pino';
 import { RedisModule } from 'nestjs-redis';
 import { AppController } from './app.controller';
 import { AuthModule } from './auth/auth.module';
+import { BenchmarkModule } from './benchmarks/benchmark.module';
 import { HealthController } from './health/health.controller';
 import ormconfig from './ormconfig';
 import redisconfig from './redisconfig';
@@ -17,6 +18,7 @@ import { UsersModule } from './users/users.module';
   imports: [
     ConfigModule.forRoot(),
     TypeOrmModule.forRoot(ormconfig),
+    BenchmarkModule,
     AuthModule,
     UsersModule,
     SubmissionsModule,

src/benchmarks/benchmark.controller.ts

@@ -0,0 +1,21 @@
+import { Body, Controller, Post } from '@nestjs/common';
+import { CreateBenchmarkDto } from 'src/benchmarks/dto/create-benchmark.dto';
+import { BenchmarkService } from 'src/benchmarks/benchmark.service';
+import { Benchmark } from './benchmark.entity';
+
+
+@Controller('benchmarks')
+export class BenchmarkController {
+  constructor(private readonly benchmarkService: BenchmarkService) {
+  }
+
+  @Post()
+  async createBenchmark(
+    @Body()
+      benchmark: CreateBenchmarkDto
+  ) : Promise<Benchmark>{
+    console.log(benchmark);
+    return this.benchmarkService.create(benchmark);
+  }
+
+}

src/benchmarks/benchmark.entity.ts

@@ -0,0 +1,41 @@
+/* eslint-disable @typescript-eslint/no-unused-vars */
+import { BaseEntity, Column, CreateDateColumn, Entity, ManyToOne, PrimaryGeneratedColumn } from 'typeorm';
+import { ApiProperty } from '@nestjs/swagger';
+import { User } from 'src/users/user.entity';
+
+@Entity('benchmarks')
+export class Benchmark extends BaseEntity {
+  constructor(partial: Partial<Benchmark>) {
+    super();
+    Object.assign(this, partial);
+  }
+
+  @ApiProperty()
+  @PrimaryGeneratedColumn('uuid')
+  id: string;
+
+  @ApiProperty()
+  @Column()
+  title: string;
+
+  @ApiProperty()
+  @Column()
+  subject: string;
+
+  @ApiProperty()
+  @Column()
+  giturl: string;
+
+  @ApiProperty()
+  @CreateDateColumn()
+  createdAt: Date;
+
+  @ApiProperty()
+  @Column()
+  difficulty: string;
+
+  @ManyToOne((type) => User, (user) => user.benchmarks, {eager: true})
+  @ApiProperty({type: () => User})
+  creator: User;
+
+}

src/benchmarks/benchmark.module.ts

@@ -0,0 +1,13 @@
+import { Module } from '@nestjs/common';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import { Benchmark } from './benchmark.entity';
+import { BenchmarkService } from './benchmark.service';
+import { BenchmarkController } from './benchmark.controller';
+
+@Module({
+  imports: [TypeOrmModule.forFeature([Benchmark])],
+  providers: [BenchmarkService],
+  controllers: [BenchmarkController],
+  exports: [BenchmarkService],
+})
+export class BenchmarkModule {}

src/benchmarks/benchmark.service.ts

@@ -0,0 +1,13 @@
+import { Injectable } from '@nestjs/common';
+import { CreateBenchmarkDto } from './dto/create-benchmark.dto';
+import { Benchmark } from './benchmark.entity';
+
+@Injectable()
+export class BenchmarkService {
+
+  async create(benchmarkData: CreateBenchmarkDto): Promise<Benchmark> {
+    const benchmark = new Benchmark(benchmarkData);
+    // TODO : Link benchmark to user
+    return benchmark.save();
+  }
+}

src/benchmarks/dto/create-benchmark.dto.ts

@@ -0,0 +1,24 @@
+import { IsNotEmpty, IsString } from 'class-validator';
+import { ApiProperty } from '@nestjs/swagger';
+
+export class CreateBenchmarkDto {
+  @IsNotEmpty()
+  @IsString()
+  @ApiProperty()
+  title: string;
+
+  @IsNotEmpty()
+  @IsString()
+  @ApiProperty()
+  subject: string;
+
+  @IsNotEmpty()
+  @IsString()
+  @ApiProperty()
+  giturl: string;
+
+  @IsNotEmpty()
+  @IsString()
+  @ApiProperty()
+  difficulty: string;
+}

src/migrations/1621724651623-AddBenchmarks.ts

@@ -0,0 +1,16 @@
+import {MigrationInterface, QueryRunner} from "typeorm";
+
+export class AddBenchmarks1621724651623 implements MigrationInterface {
+    name = 'AddBenchmarks1621724651623'
+
+    public async up(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`CREATE TABLE "benchmarks" ("id" uuid NOT NULL DEFAULT uuid_generate_v4(), "title" character varying NOT NULL, "subject" character varying NOT NULL, "giturl" character varying NOT NULL, "createdAt" TIMESTAMP NOT NULL DEFAULT now(), "difficulty" character varying NOT NULL, "creatorId" uuid, CONSTRAINT "PK_7e1ca1c910680770da116e0b95c" PRIMARY KEY ("id"))`);
+        await queryRunner.query(`ALTER TABLE "benchmarks" ADD CONSTRAINT "FK_f7dfba1f69546feced443cd097c" FOREIGN KEY ("creatorId") REFERENCES "users"("id") ON DELETE NO ACTION ON UPDATE NO ACTION`);
+    }
+
+    public async down(queryRunner: QueryRunner): Promise<void> {
+        await queryRunner.query(`ALTER TABLE "benchmarks" DROP CONSTRAINT "FK_f7dfba1f69546feced443cd097c"`);
+        await queryRunner.query(`DROP TABLE "benchmarks"`);
+    }
+
+}

src/users/user.entity.ts

@@ -1,6 +1,7 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
 import { ApiProperty } from '@nestjs/swagger';
 import { Exclude } from 'class-transformer';
+import { Benchmark } from 'src/benchmarks/benchmark.entity';
 import { Submission } from 'src/submissions/submission.entity';
 import {
   BaseEntity,
@@ -61,6 +62,9 @@ export class User extends BaseEntity {
   @OneToMany((type) => Submission, (submission) => submission.user)
   submissions: Submission[];
 
+  @OneToMany((type) => Benchmark, (benchmark) => benchmark.creator)
+  benchmarks: Benchmark[];
+
   @BeforeInsert()
   @BeforeUpdate()
   async hashPassword(): Promise<void> {