CSVDownloadsController.java

package edu.ucsb.cs156.frontiers.controllers;

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.charset.StandardCharsets;
import java.util.List;

import com.opencsv.bean.StatefulBeanToCsv;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;

import com.opencsv.bean.StatefulBeanToCsvBuilder;
import com.opencsv.exceptions.CsvDataTypeMismatchException;
import com.opencsv.exceptions.CsvRequiredFieldEmptyException;

import edu.ucsb.cs156.frontiers.entities.Course;
import edu.ucsb.cs156.frontiers.errors.EntityNotFoundException;
import edu.ucsb.cs156.frontiers.models.RosterStudentDTO;
import edu.ucsb.cs156.frontiers.repositories.CourseRepository;
import edu.ucsb.cs156.frontiers.services.RosterStudentDTOService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.media.Content;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.responses.ApiResponse;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;

@Tag(name = "CSV Downloads")
@RequestMapping("/api/csv")
@RestController
@Slf4j
public class CSVDownloadsController extends ApiController {

  @Autowired
  private CourseRepository courseRepository;

  @Autowired
  private RosterStudentDTOService rosterStudentDTOService;

  @Operation(summary = "Download CSV File of Roster Students", description = "Returns a CSV file as a response", responses = {
      @ApiResponse(responseCode = "200", description = "CSV file", content = @Content(mediaType = "text/csv", schema = @Schema(type = "string", format = "binary"))),
      @ApiResponse(responseCode = "500", description = "Internal Server Error")
  })
  @GetMapping(value = "/rosterstudents", produces = "text/csv")
  public ResponseEntity<StreamingResponseBody> csvForQuarter(
      @Parameter(name = "courseId", description = "course id", example = "1") @RequestParam Long courseId)
      throws EntityNotFoundException, Exception, IOException {
    Course course = courseRepository.findById(courseId)
        .orElseThrow(() -> new EntityNotFoundException(Course.class, courseId));
    StreamingResponseBody stream = (outputStream) -> {

      List<RosterStudentDTO> list = rosterStudentDTOService.getRosterStudentDTOs(courseId);
      try (Writer writer = new OutputStreamWriter(outputStream, StandardCharsets.UTF_8)) {
        try {
          StatefulBeanToCsv<RosterStudentDTO> beanToCsvWriter = rosterStudentDTOService.getStatefulBeanToCSV(writer);
          beanToCsvWriter.write(list);
        } catch (CsvDataTypeMismatchException | CsvRequiredFieldEmptyException e) {
          log.error("Error writing CSV file", e);
          throw new IOException("Error writing CSV file: " + e.getMessage());
        }
      }
    };

    return ResponseEntity.ok()
        .contentType(MediaType.parseMediaType("text/csv; charset=UTF-8"))
        .header(
            HttpHeaders.CONTENT_DISPOSITION,
            String.format("attachment;filename=%s_roster.csv", course.getCourseName()))
        .header(HttpHeaders.CONTENT_TYPE, "text/csv; charset=UTF-8")
        .header(HttpHeaders.ACCESS_CONTROL_EXPOSE_HEADERS, HttpHeaders.CONTENT_DISPOSITION)
        .body(stream);
  }

}