D-Net

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.service.RepositoryService;

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.PostAuthorize;

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;

    @Autowired

    private RepositoryService repositoryService;

    @RequestMapping(value = "/getPiwikSiteForRepo/{repositoryId}" , method = RequestMethod.GET,produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or (@repositoryService.getRepositoryById(#repositoryId).registeredBy==authentication.userInfo.email and hasRole('ROLE_USER'))")

    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 (@repositoryService.getRepositoryById(#piwikInfo.repositoryId).registeredBy==authentication.userInfo.email and hasRole('ROLE_USER'))")

    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 = "orderField", dataType = "eu.dnetlib.repo.manager.domain.OrderByField", paramType = "query"),

            @ApiImplicitParam(name = "searchField", 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(searchField));

        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 = "10000") 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,0,10000,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");//dd/MM/yyyy

        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

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN')")

    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);

    }

}