reloop-v2/backend/src/problem-sets/problem-sets.controller.ts

import {
  Body,
  Controller,
  Delete,
  Get,
  Param,
  ParseIntPipe,
  Post,
  Query,
  UseGuards,
} from '@nestjs/common';
import {
  IsArray,
  IsInt,
  IsNumber,
  IsOptional,
  IsString,
  Max,
  Min,
  ValidateNested,
} from 'class-validator';
import { Type } from 'class-transformer';
import { JwtAuthGuard } from '../auth/jwt-auth.guard';
import { CurrentUser } from '../auth/current-user.decorator';
import { AuthUser } from '../auth/jwt.strategy';
import { ProblemSetsService } from './problem-sets.service';

class ListProblemSetsQuery {
  @IsOptional()
  @IsString()
  subjectName?: string;

  @IsOptional()
  @Type(() => Number)
  @IsInt()
  @Min(1900)
  @Max(2100)
  year?: number;
}

class CreateFromPdfProblemDto {
  @IsInt()
  number: number;

  @IsString()
  imageUrl: string;

  @IsOptional()
  @Type(() => Number)
  @IsInt()
  @Min(1)
  @Max(5)
  correctAnswer?: number;

  @IsOptional()
  @IsNumber()
  @Min(0)
  @Max(1)
  difficulty?: number;
}

class CreateFromPdfDto {
  @IsString()
  title: string;

  @IsOptional()
  @IsString()
  subjectName?: string;

  @IsArray()
  @ValidateNested({ each: true })
  @Type(() => CreateFromPdfProblemDto)
  problems: CreateFromPdfProblemDto[];
}

@Controller('problem-sets')
@UseGuards(JwtAuthGuard)
export class ProblemSetsController {
  constructor(private readonly svc: ProblemSetsService) {}

  @Get()
  list(@Query() q: ListProblemSetsQuery) {
    return this.svc.list(q);
  }

  @Get('my-uploads')
  myUploads(@CurrentUser() user: AuthUser) {
    return this.svc.listMyUploads(user.id);
  }

  @Get(':id')
  one(@Param('id', ParseIntPipe) id: number) {
    return this.svc.getOne(id);
  }

  @Post('create-from-pdf')
  createFromPdf(
    @CurrentUser() user: AuthUser,
    @Body() dto: CreateFromPdfDto,
  ) {
    return this.svc.createFromPdf(user.id, dto);
  }

  @Delete(':id')
  deleteUpload(
    @CurrentUser() user: AuthUser,
    @Param('id', ParseIntPipe) id: number,
  ) {
    return this.svc.deleteUpload(user.id, id);
  }
}