package ru.ulstu.file;

import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
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.multipart.MultipartFile;
import ru.ulstu.configuration.Constants;
import ru.ulstu.core.model.response.Response;
import ru.ulstu.file.model.FileData;
import ru.ulstu.file.service.FileService;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;

import static java.nio.charset.StandardCharsets.UTF_8;
import static ru.ulstu.file.FileController.URL;

@RestController
@RequestMapping(URL)
public class FileController {
    public static final String URL = Constants.API_1_0 + "files";

    private final FileService fileService;

    public FileController(FileService fileService) {
        this.fileService = fileService;
    }

    @GetMapping("/download-tmp/{tmp-file-name}")
    public ResponseEntity<byte[]> getFile(@PathVariable("tmp-file-name") String tmpFileName) throws IOException {
        HttpHeaders headers = new HttpHeaders();
        headers.add("Content-Disposition", "attachment; filename='"
                + URLEncoder.encode(fileService.getTmpFileName(tmpFileName), UTF_8.toString()) + "'");
        return new ResponseEntity<>(fileService.getTmpFile(tmpFileName), headers, HttpStatus.OK);
    }

    @GetMapping("/download/{file-id}")
    public ResponseEntity<byte[]> getFile(@PathVariable("file-id") Integer fileId) throws UnsupportedEncodingException {
        HttpHeaders headers = new HttpHeaders();
        FileData fileData = fileService.getFile(fileId);
        headers.add("Content-Disposition", "attachment; filename='" +
                URLEncoder.encode(fileData.getName(), UTF_8.toString()) + "'");
        return new ResponseEntity<>(fileData.getData(), headers, HttpStatus.OK);
    }

    @PostMapping("/uploadTmpFile")
    public Response<String> upload(@RequestParam("file") MultipartFile multipartFile) throws IOException {
        return new Response(fileService.uploadToTmpDir(multipartFile));
    }
}