package com.ddf.vodsystem.controllers;

import com.ddf.vodsystem.controllers.dto.JobResponse;
import com.ddf.vodsystem.dto.APIResponse;
import com.ddf.vodsystem.dto.Job;
import com.ddf.vodsystem.exceptions.NotReadyException;
import com.ddf.vodsystem.services.JobRegistryService;
import org.springframework.core.io.FileSystemResource;
import org.springframework.core.io.Resource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.MediaTypeFactory;
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.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("api/v1/job")
public class JobController {
    private final JobRegistryService jobRegistryService;
    private static final String SUCCESS = "success";
    private static final String FILENAME_HEADER = "inline; filename=\"%s\"";

    public JobController(JobRegistryService jobRegistryService){
        this.jobRegistryService = jobRegistryService;
    }

    @GetMapping("/{uuid}")
    public ResponseEntity<APIResponse<JobResponse>> getJob(@PathVariable String uuid) {
        Job job = jobRegistryService.getJob(uuid);
        JobResponse jobResponse = new JobResponse(
                job.getUuid(),
                job.getProgressTracker().getProgress(),
                job.getProgressTracker().isComplete(),
                job.getState(),
                job.getErrorOutput(),
                job.getCreatedAt()
        );

        return ResponseEntity.ok(new APIResponse<>(
                SUCCESS,
                "Job successfully retreived",
                jobResponse
        ));
    }

    @GetMapping("/{uuid}/download")
    public ResponseEntity<Resource> downloadJobFile(@PathVariable String uuid) {
        Job job = jobRegistryService.getJob(uuid);

        if (job.getDownload() == null) {
            throw new NotReadyException("Job " + uuid + " not ready for download.");
        }

        Resource resource = new FileSystemResource(job.getDownload());

        if (!resource.exists()) {
            return ResponseEntity.notFound().build();
        }

        return ResponseEntity.ok()
                .header(HttpHeaders.CONTENT_DISPOSITION, String.format(FILENAME_HEADER, resource.getFilename()))
                .contentType(MediaTypeFactory.getMediaType(resource).orElse(MediaType.APPLICATION_OCTET_STREAM))
                .body(resource);
    }

}