src/modules/labs/controllers/labs.controller.ts

import {
  Controller,
  Get,
  Post,
  Put,
  Delete,
  Patch,
  Body,
  Param,
  Query,
  Req,
  UseGuards,
  ParseIntPipe,
  HttpCode,
  HttpStatus,
  UseInterceptors,
  UploadedFile,
} from '@nestjs/common';
import {
  ApiTags,
  ApiOperation,
  ApiParam,
  ApiBody,
  ApiResponse,
  ApiBearerAuth,
  ApiConsumes,
} from '@nestjs/swagger';
import { FileInterceptor } from '@nestjs/platform-express';
import { JwtAuthGuard } from '../../auth/guards/jwt-auth.guard';
import { RolesGuard } from '../../auth/guards/roles.guard';
import { Roles } from '../../auth/roles.decorator';
import { RoleName } from '../../auth/entities/role.entity';
import { LabsService } from '../services/labs.service';
import {
  CreateLabDto,
  UpdateLabDto,
  SubmitLabDto,
  GradeLabSubmissionDto,
  CreateInstructionDto,
  UpdateInstructionDto,
  MarkLabAttendanceDto,
  LabQueryDto,
  UploadLabInstructionDto,
  UploadLabTaMaterialDto,
  UploadLabSubmissionDto,
} from '../dto';
import { LabStatus } from '../enums';
import { Lab } from '../entities/lab.entity';
import { LabInstruction } from '../entities/lab-instruction.entity';

@ApiTags('Labs')
@ApiBearerAuth('JWT-auth')
@Controller('api/labs')
@UseGuards(JwtAuthGuard, RolesGuard)
export class LabsController {
  constructor(private readonly labsService: LabsService) {}

  // ============ LABS CRUD ============

  @Get()
  @ApiOperation({
    summary: 'List labs',
    description: 'List all labs with optional filtering by course and status. Supports pagination.',
  })
  @ApiResponse({ status: 200, description: 'Labs retrieved successfully' })
  async findAll(@Query() query: LabQueryDto) {
    return this.labsService.findAll(query);
  }

  @Post()
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Create lab',
    description: 'Create a new lab assignment. Requires INSTRUCTOR, TA, ADMIN, or IT_ADMIN role.',
  })
  @ApiBody({ type: CreateLabDto })
  @ApiResponse({ status: 201, description: 'Lab created successfully' })
  @ApiResponse({ status: 403, description: 'Forbidden - Insufficient role' })
  async create(@Body() dto: CreateLabDto, @Req() req: any) {
    const userId = req.user.userId || req.user.id;
    return this.labsService.create(dto, userId);
  }

  @Get(':id')
  @ApiOperation({
    summary: 'Get lab by ID',
    description: 'Get lab details including instructions.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiResponse({ status: 200, description: 'Lab retrieved successfully' })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async findOne(@Param('id', ParseIntPipe) id: number) {
    return this.labsService.findById(id);
  }

  @Put(':id')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @ApiOperation({
    summary: 'Update lab',
    description: 'Update lab details. Requires INSTRUCTOR, TA, ADMIN, or IT_ADMIN role.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiBody({ type: UpdateLabDto })
  @ApiResponse({ status: 200, description: 'Lab updated successfully' })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async update(@Param('id', ParseIntPipe) id: number, @Body() dto: UpdateLabDto) {
    return this.labsService.update(id, dto);
  }

  @Delete(':id')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @HttpCode(HttpStatus.NO_CONTENT)
  @ApiOperation({
    summary: 'Delete lab',
    description: 'Delete a lab. Requires INSTRUCTOR, TA, ADMIN, or IT_ADMIN role.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiResponse({ status: 204, description: 'Lab deleted successfully' })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async remove(@Param('id', ParseIntPipe) id: number) {
    return this.labsService.remove(id);
  }

  @Patch(':id/status')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @ApiOperation({
    summary: 'Change lab status',
    description: `
Change the status of a lab (publish, close, or archive).

### Status Values
- \`draft\`: Not visible to students
- \`published\`: Visible to students, accepting submissions
- \`closed\`: No longer accepting submissions
- \`archived\`: Hidden from all views
    `,
  })
  @ApiParam({ name: 'id', description: 'Lab ID', type: Number })
  @ApiBody({
    schema: {
      properties: {
        status: {
          type: 'string',
          enum: ['draft', 'published', 'closed', 'archived'],
          example: 'published',
        },
      },
    },
  })
  @ApiResponse({ status: 200, description: 'Lab status updated', type: Lab })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async changeStatus(
    @Param('id', ParseIntPipe) id: number,
    @Body('status') status: LabStatus,
  ): Promise<Lab> {
    return this.labsService.changeStatus(id, status);
  }

  // ============ INSTRUCTIONS ============

  @Get(':id/instructions')
  @ApiOperation({
    summary: 'Get lab instructions',
    description: 'Get all instructions for a lab, ordered by order_index.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiResponse({ status: 200, description: 'Instructions retrieved' })
  async getInstructions(@Param('id', ParseIntPipe) id: number) {
    return this.labsService.getInstructions(id);
  }

  @Post(':id/instructions')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Add instruction to lab',
    description: 'Add a step-by-step instruction to a lab. Supports markdown text and file attachments.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiBody({ type: CreateInstructionDto })
  @ApiResponse({ status: 201, description: 'Instruction added' })
  async addInstruction(@Param('id', ParseIntPipe) id: number, @Body() dto: CreateInstructionDto) {
    return this.labsService.addInstruction(id, dto);
  }

  @Patch(':id/instructions/:instructionId')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @HttpCode(HttpStatus.OK)
  @ApiOperation({
    summary: 'Update lab instruction',
    description: 'Update instruction text, order index, or file attachment. Requires INSTRUCTOR, TA, ADMIN, or IT_ADMIN role.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiParam({ name: 'instructionId', description: 'Instruction ID', example: 1 })
  @ApiBody({ type: UpdateInstructionDto })
  @ApiResponse({ status: 200, description: 'Instruction updated' })
  @ApiResponse({ status: 404, description: 'Instruction not found' })
  async updateInstruction(
    @Param('id', ParseIntPipe) labId: number,
    @Param('instructionId', ParseIntPipe) instructionId: number,
    @Body() dto: UpdateInstructionDto,
  ): Promise<LabInstruction> {
    return this.labsService.updateInstruction(labId, instructionId, dto);
  }

  @Delete(':id/instructions/:instructionId')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @HttpCode(HttpStatus.OK)
  @ApiOperation({
    summary: 'Delete lab instruction',
    description: 'Delete an instruction from a lab. Requires INSTRUCTOR, TA, ADMIN, or IT_ADMIN role.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiParam({ name: 'instructionId', description: 'Instruction ID', example: 1 })
  @ApiResponse({ status: 200, description: 'Instruction deleted' })
  @ApiResponse({ status: 404, description: 'Instruction not found' })
  async deleteInstruction(
    @Param('id', ParseIntPipe) labId: number,
    @Param('instructionId', ParseIntPipe) instructionId: number,
  ): Promise<void> {
    return this.labsService.deleteInstruction(labId, instructionId);
  }

  // ============ SUBMISSIONS ============

  @Post(':id/submit')
  @Roles(RoleName.STUDENT)
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Submit lab work',
    description: 'Submit lab work as a student. Can include text and/or file attachment. Auto-detects late submissions.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiBody({ type: SubmitLabDto })
  @ApiResponse({ status: 201, description: 'Lab submitted successfully' })
  async submit(@Param('id', ParseIntPipe) id: number, @Body() dto: SubmitLabDto, @Req() req: any) {
    const userId = req.user.userId || req.user.id;
    return this.labsService.submit(id, userId, dto);
  }

  @Get(':id/submissions')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @ApiOperation({
    summary: 'List lab submissions',
    description: 'List all submissions for a lab. Requires INSTRUCTOR, TA, ADMIN, or IT_ADMIN role.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiResponse({ status: 200, description: 'Submissions retrieved' })
  async getSubmissions(@Param('id', ParseIntPipe) id: number) {
    return this.labsService.getSubmissions(id);
  }

  @Get(':id/submissions/my')
  @Roles(RoleName.STUDENT)
  @ApiOperation({
    summary: 'Get my lab submission',
    description: 'Get the current student\'s submission for a lab.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiResponse({ status: 200, description: 'Student submission retrieved' })
  async getMySubmission(@Param('id', ParseIntPipe) id: number, @Req() req: any) {
    const userId = req.user.userId || req.user.id;
    return this.labsService.getMySubmission(id, userId);
  }

  @Patch(':id/submissions/:subId/grade')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @ApiOperation({
    summary: 'Grade lab submission',
    description: 'Grade a lab submission with score, feedback, and status. Creates a grade record in the gradebook.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiParam({ name: 'subId', description: 'Submission ID', example: 1 })
  @ApiBody({ type: GradeLabSubmissionDto })
  @ApiResponse({ status: 200, description: 'Submission graded' })
  @ApiResponse({ status: 404, description: 'Submission not found' })
  async gradeSubmission(
    @Param('id', ParseIntPipe) id: number,
    @Param('subId', ParseIntPipe) subId: number,
    @Body() dto: GradeLabSubmissionDto,
    @Req() req: any,
  ) {
    const graderId = req.user.userId || req.user.id;
    return this.labsService.gradeSubmission(id, subId, dto, graderId);
  }

  // ============ ATTENDANCE ============

  @Post(':id/attendance')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Mark lab attendance',
    description: 'Mark or update a student\'s attendance for a lab session.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiBody({ type: MarkLabAttendanceDto })
  @ApiResponse({ status: 201, description: 'Attendance marked' })
  async markAttendance(
    @Param('id', ParseIntPipe) id: number,
    @Body() dto: MarkLabAttendanceDto,
    @Req() req: any,
  ) {
    const markedBy = req.user.userId || req.user.id;
    return this.labsService.markAttendance(id, dto, markedBy);
  }

  @Get(':id/attendance')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @ApiOperation({
    summary: 'Get lab attendance',
    description: 'Get attendance records for a lab session.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiResponse({ status: 200, description: 'Attendance records retrieved' })
  async getAttendance(@Param('id', ParseIntPipe) id: number) {
    return this.labsService.getAttendance(id);
  }

  // ============ GOOGLE DRIVE UPLOADS ============

  @Post(':id/instructions/upload')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @UseInterceptors(FileInterceptor('file'))
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Upload lab instruction to Google Drive',
    description: 'Upload a lab instruction file (PDF, DOCX, etc.) directly to Google Drive. Creates folder structure automatically.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiConsumes('multipart/form-data')
  @ApiBody({
    schema: {
      type: 'object',
      required: ['file'],
      properties: {
        file: {
          type: 'string',
          format: 'binary',
          description: 'Instruction file (PDF, DOCX, etc.)',
        },
        title: {
          type: 'string',
          description: 'Instruction title',
          example: 'Lab 1 - Getting Started Guide',
        },
        orderIndex: {
          type: 'integer',
          description: 'Order index for instruction steps',
          example: 1,
        },
      },
    },
  })
  @ApiResponse({ status: 201, description: 'Instruction uploaded successfully' })
  @ApiResponse({ status: 400, description: 'No file provided' })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async uploadInstruction(
    @Param('id', ParseIntPipe) id: number,
    @UploadedFile() file: Express.Multer.File,
    @Body() dto: UploadLabInstructionDto,
    @Req() req: any,
  ) {
    if (!file) {
      throw new Error('No file provided');
    }
    const userId = req.user.userId || req.user.id;
    return this.labsService.uploadInstructionToDrive(
      id,
      file,
      dto.title,
      dto.orderIndex || 0,
      userId,
    );
  }

  @Post(':id/ta-materials/upload')
  @Roles(RoleName.INSTRUCTOR, RoleName.TA, RoleName.ADMIN, RoleName.IT_ADMIN)
  @UseInterceptors(FileInterceptor('file'))
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Upload TA material to Google Drive',
    description: 'Upload TA-only material (answer keys, grading rubrics, etc.) to Google Drive.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiConsumes('multipart/form-data')
  @ApiBody({
    schema: {
      type: 'object',
      required: ['file'],
      properties: {
        file: {
          type: 'string',
          format: 'binary',
          description: 'TA material file',
        },
        title: {
          type: 'string',
          description: 'Material title',
          example: 'Lab 1 Answer Key',
        },
        materialType: {
          type: 'string',
          description: 'Type of TA material',
          example: 'answer_key',
          enum: ['answer_key', 'grading_rubric', 'solution', 'notes'],
        },
      },
    },
  })
  @ApiResponse({ status: 201, description: 'TA material uploaded successfully' })
  @ApiResponse({ status: 400, description: 'No file provided' })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async uploadTaMaterial(
    @Param('id', ParseIntPipe) id: number,
    @UploadedFile() file: Express.Multer.File,
    @Body() dto: UploadLabTaMaterialDto,
    @Req() req: any,
  ) {
    if (!file) {
      throw new Error('No file provided');
    }
    const userId = req.user.userId || req.user.id;
    return this.labsService.uploadTaMaterialToDrive(
      id,
      file,
      dto.title,
      dto.materialType,
      userId,
    );
  }

  @Post(':id/submissions/upload')
  @Roles(RoleName.STUDENT)
  @UseInterceptors(FileInterceptor('file'))
  @HttpCode(HttpStatus.CREATED)
  @ApiOperation({
    summary: 'Upload lab submission to Google Drive',
    description: 'Upload lab submission file directly to Google Drive. Creates student folder automatically. Auto-detects late submissions.',
  })
  @ApiParam({ name: 'id', description: 'Lab ID', example: 1 })
  @ApiConsumes('multipart/form-data')
  @ApiBody({
    schema: {
      type: 'object',
      required: ['file'],
      properties: {
        file: {
          type: 'string',
          format: 'binary',
          description: 'Submission file',
        },
        submissionText: {
          type: 'string',
          description: 'Optional submission notes/comments',
          example: 'Completed all tasks as instructed.',
        },
      },
    },
  })
  @ApiResponse({ status: 201, description: 'Submission uploaded successfully' })
  @ApiResponse({ status: 400, description: 'No file provided' })
  @ApiResponse({ status: 404, description: 'Lab not found' })
  async uploadSubmission(
    @Param('id', ParseIntPipe) id: number,
    @UploadedFile() file: Express.Multer.File,
    @Body() dto: UploadLabSubmissionDto,
    @Req() req: any,
  ) {
    if (!file) {
      throw new Error('No file provided');
    }
    const userId = req.user.userId || req.user.id;
    return this.labsService.uploadSubmissionToDrive(
      id,
      file,
      dto.submissionText,
      userId,
    );
  }
}