package com.infoclinika.pdx.controller;

import com.infoclinika.pdx.domain.sample.SampleType;
import com.infoclinika.pdx.service.FileParser;
import com.infoclinika.pdx.service.ParsedFile;
import com.infoclinika.pdx.service.ProcessedFile;
import com.infoclinika.pdx.service.exception.ApplicationException;
import com.infoclinika.pdx.service.experiment.MicroenvironmentService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
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.ResponseStatus;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

/**
 * Handles requests to upload microenvironment data.
 */
@RestController
@RequestMapping("/api/upload")
public class MicroenvironmentUploadController {
    private static final String FILE_PARAM = "file";
    private static final String TYPE_PARAM = "type";
    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    private final FileParser fileParser;
    private final MicroenvironmentService service;

    /**
     * Constructor to provide required services.
     */
    @Autowired
    public MicroenvironmentUploadController(FileParser fileParser, MicroenvironmentService service) {
        this.fileParser = fileParser;
        this.service = service;
    }

    /**
     * Handles upload of microenvironment data file.
     *
     * @param file file to parse
     */
    @PostMapping("/microenvironment")
    @ResponseStatus(HttpStatus.OK)
    public void postFiles(@RequestParam(TYPE_PARAM) SampleType type,
                          @RequestParam(FILE_PARAM) MultipartFile file) throws ApplicationException {
        logger.debug("Request to /api/upload/microenvironment");

        final ParsedFile parsedFile = fileParser.parseFile(file);
        service.save(type, parsedFile);
    }

    /**
     * Handles upload of large microenvironment data file.
     *
     * @param file file to parse
     */
    @PostMapping("/microenvironment/batch")
    @ResponseStatus(HttpStatus.OK)
    public void postLargeFiles(@RequestParam(TYPE_PARAM) SampleType type,
                               @RequestParam(FILE_PARAM) MultipartFile file) throws ApplicationException {
        logger.debug("Request to /api/upload/microenvironment/batch");

        final ProcessedFile processedFile = fileParser.parseChunks(file);
        service.save(type, processedFile);
    }

    /**
     * Handles upload of cell type groups.
     *
     * @param file file to parse
     */
    @PostMapping("/microenvironment/group")
    @ResponseStatus(HttpStatus.OK)
    public void addCellTypeGroups(@RequestParam(FILE_PARAM) MultipartFile file) throws ApplicationException {
        logger.debug("Request to /api/upload/microenvironment/group");

        final ParsedFile parsedFile = fileParser.parseFile(file);
        service.setCellTypeGroups(parsedFile);
    }

    /**
     * Handles upload of genes content associated with Microenvironemt cell types.
     *
     * @param file file to parse
     */
    @PostMapping("/microenvironment/genes")
    @ResponseStatus(HttpStatus.OK)
    public void addAssociatedGenes(@RequestParam(FILE_PARAM) MultipartFile file) throws ApplicationException {
        logger.debug("Request to /api/upload/microenvironment/genes");

        final ParsedFile parsedFile = fileParser.parseFileWithoutShift(file);
        service.setGeneContent(parsedFile);
    }
}