uoa-repository-manager-service/src/main/java/eu/dnetlib/repo/manager/controllers/PiWikController.java

package eu.dnetlib.repo.manager.controllers;

import eu.dnetlib.domain.data.PiwikInfo;
import eu.dnetlib.repo.manager.domain.OrderByField;
import eu.dnetlib.repo.manager.domain.OrderByType;
import eu.dnetlib.repo.manager.domain.Paging;
import eu.dnetlib.repo.manager.service.PiWikServiceImpl;
import eu.dnetlib.repo.manager.shared.RepositoryServiceException;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import org.apache.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.FileSystemResource;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Files;
import java.nio.file.Path;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

@RestController
@RequestMapping(value = "/piwik")
@Api(description = "Piwik API",  tags = {"piwik"})
public class PiWikController {

    private static final Logger logger = Logger
            .getLogger(PiWikController.class);

    @Autowired
    private PiWikServiceImpl piWikService;


    @RequestMapping(value = "/getPiwikSiteForRepo/{repositoryId}" , method = RequestMethod.GET,produces = MediaType.APPLICATION_JSON_VALUE)
    @ResponseBody
    public PiwikInfo getPiwikSiteForRepo(@PathVariable("repositoryId") String repositoryId) {
        return piWikService.getPiwikSiteForRepo(repositoryId);
    }

    @RequestMapping(value = "/savePiwikInfo" , method = RequestMethod.POST,produces = MediaType.APPLICATION_JSON_VALUE)
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or (hasRole('ROLE_USER') " +
            "and #piwikInfo.requestorEmail == authentication.userInfo.email)")
    public PiwikInfo savePiwikInfo(@RequestBody PiwikInfo piwikInfo) {
        return piWikService.savePiwikInfo(piwikInfo);
    }

    @RequestMapping(value = "/getPiwikSitesForRepos" , method = RequestMethod.GET,produces = MediaType.APPLICATION_JSON_VALUE)
    @ApiImplicitParams({
            @ApiImplicitParam(name = "from", dataType = "number", paramType = "query"),
            @ApiImplicitParam(name = "quantity", dataType = "number", paramType = "query"),
            @ApiImplicitParam(name = "order", dataType = "eu.dnetlib.repo.manager.domain.OrderByType", paramType = "query"),
            @ApiImplicitParam(name = "searchField", dataType = "eu.dnetlib.repo.manager.domain.OrderByField", paramType = "query"),
            @ApiImplicitParam(name = "orderField", dataType = "string", paramType = "query"),
    })
    public Paging<PiwikInfo> getPiwikSitesForRepos(
            @RequestParam(value = "from",required=false,defaultValue = "0") int from,
            @RequestParam(value = "quantity",required=false,defaultValue = "100") int quantity,
            @RequestParam(value = "order",required=false,defaultValue = "DSC") OrderByType orderType,
            @RequestParam(value = "orderField", required = false, defaultValue = "REPOSITORY_NAME") OrderByField orderField,
            @RequestParam(value = "searchField", required = false, defaultValue = "") String searchField

    ){
        Paging<PiwikInfo> results = new Paging<>();
        List<PiwikInfo> returning = piWikService.getPiwikSitesForRepos(orderField,orderType,from,quantity,searchField);
        results.setFrom(from);
        results.setTo(from + returning.size());
        results.setTotal(piWikService.getPiwikSitesTotals());
        results.setResults(returning);
        return results;
    }
    @ApiImplicitParams({
            @ApiImplicitParam(name = "from", dataType = "number", paramType = "query"),
            @ApiImplicitParam(name = "quantity", dataType = "number", paramType = "query"),
            @ApiImplicitParam(name = "order", dataType = "eu.dnetlib.repo.manager.domain.OrderByType", paramType = "query"),
            @ApiImplicitParam(name = "searchField", dataType = "eu.dnetlib.repo.manager.domain.OrderByField", paramType = "query"),
            @ApiImplicitParam(name = "orderField", dataType = "string", paramType = "query"),
    })
    @RequestMapping(value = "/getPiwikSitesForRepos/csv" , method = RequestMethod.GET,produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
    @ResponseBody
    public FileSystemResource getPiwikSitesForReposToCsv(
            @RequestParam(value = "from",required=false,defaultValue = "0") int from,
            @RequestParam(value = "quantity",required=false,defaultValue = "100") int quantity,
            @RequestParam(value = "order",required=false,defaultValue = "DSC") OrderByType orderType,
            @RequestParam(value = "orderField", required = false, defaultValue = "REPOSITORY_NAME") OrderByField orderField,
            @RequestParam(value = "searchField", required = false, defaultValue = "") String searchField,
            HttpServletResponse response,
            HttpServletRequest request
    ) throws IOException {

        Path p = Files.createTempFile("exportingCsv-", new Date().toString());
        List<PiwikInfo> returning = piWikService.getPiwikSitesForRepos(orderField,orderType,from,quantity,searchField);
        try (PrintWriter writer = new PrintWriter(p.toFile())) {

            StringBuilder sb = new StringBuilder();
            sb.append(" Repository ID , Repository name, Country, Site ID, Authentication token, Creation date, Requestor full name, Requestor email, Validated, Validation date, Comment \n");

            for(PiwikInfo piwikInfo : returning){
                sb.append(
                        (piwikInfo.getRepositoryId() == null ? "," : piwikInfo.getRepositoryId()+ ",")+
                                (piwikInfo.getRepositoryName() == null ? "," : piwikInfo.getRepositoryName()+ ",")+
                                (piwikInfo.getCountry() == null ? "," : piwikInfo.getCountry()+ ",")+
                                (piwikInfo.getSiteId() == null ? "," : piwikInfo.getSiteId()+ ",") +
                                (piwikInfo.getAuthenticationToken() == null ? "," : piwikInfo.getAuthenticationToken()+ ",")+
                                (piwikInfo.getCreationDate() == null ? "," : piwikInfo.getCreationDate().toString()+ ",") +
                                (piwikInfo.getRequestorName() == null ? "," : piwikInfo.getRequestorName()+ ",") +
                                (piwikInfo.getRequestorEmail() == null ? "," : piwikInfo.getRequestorEmail()+ ",")+
                                piwikInfo.isValidated() + "," +
                                (piwikInfo.getValidationDate() == null ? "," : piwikInfo.getValidationDate().toString()+ ",") +
                                (piwikInfo.getComment() == null ? "\n" : piwikInfo.getComment()+ "\n")

                );
            }
            writer.write(sb.toString());

        } catch (FileNotFoundException e) {
            logger.error(e.getMessage());
        }


        String mimeType = request.getServletContext().getMimeType(p.toFile().getAbsolutePath());
        if (mimeType == null) {
            mimeType = "application/octet-stream";
        }
        response.setContentType(mimeType);
        response.setContentLength((int) p.toFile().length());


        String headerKey = "Content-Disposition";
        SimpleDateFormat sdfDate = new SimpleDateFormat("ddMMyyyy");
        Date now = new Date();
        String strDate = sdfDate.format(now);
        String headerValue = String.format("attachment; filename=\"csv-%s.csv\"",
                strDate);
        response.setHeader(headerKey, headerValue);


        return new FileSystemResource(p.toFile());

    }


    @RequestMapping(value = "/approvePiwikSite/{repositoryId}" , method = RequestMethod.GET)
    @ResponseBody
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN')")
    public ResponseEntity<Object> approvePiwikSite(@PathVariable("repositoryId") String repositoryId) {
        return piWikService.approvePiwikSite(repositoryId);
    }

    @RequestMapping(value = "/getOpenaireId/{repositoryId}" , method = RequestMethod.GET,produces = MediaType.APPLICATION_JSON_VALUE)
    @ResponseBody
    public String getOpenaireId(String repositoryid){
        return piWikService.getOpenaireId(repositoryid);
    }

    @RequestMapping(value = "/markPiwikSiteAsValidated/{repositoryId}" , method = RequestMethod.POST,
            consumes = MediaType.APPLICATION_JSON_VALUE)
    @ResponseBody
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN')")
    public ResponseEntity<Object> markPiwikSiteAsValidated(@PathVariable("repositoryId") String repositoryId) throws RepositoryServiceException {
        return piWikService.markPiwikSiteAsValidated(repositoryId);
    }

    @RequestMapping(value = "/enableMetricsForRepository", method = RequestMethod.POST,
            consumes = MediaType.APPLICATION_JSON_VALUE)
    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or (hasRole('ROLE_USER') and #piwikInfo.requestorEmail == authentication.userInfo.email)")
    public PiwikInfo enableMetricsForRepository(@RequestParam("officialName") String officialName,
                                                @RequestParam("repoWebsite") String repoWebsite,
                                                @RequestBody PiwikInfo piwikInfo) throws RepositoryServiceException {
        return piWikService.enableMetricsForRepository(officialName, repoWebsite, piwikInfo);
    }


}