D-Net

package eu.dnetlib.repo.manager.controllers;

import eu.dnetlib.domain.data.Repository;

import eu.dnetlib.domain.data.RepositoryInterface;

import eu.dnetlib.repo.manager.domain.*;

import eu.dnetlib.repo.manager.exception.ResourceNotFoundException;

import eu.dnetlib.repo.manager.service.RepositoryServiceImpl;

import io.swagger.annotations.Api;

import org.apache.log4j.Logger;

import org.json.JSONException;

import org.mitre.openid.connect.model.OIDCAuthenticationToken;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.http.MediaType;

import org.springframework.security.access.prepost.PreAuthorize;

import org.springframework.security.core.Authentication;

import org.springframework.security.core.context.SecurityContextHolder;

import org.springframework.web.bind.annotation.*;

import java.io.IOException;

import java.util.List;

import java.util.Map;

@RestController

@RequestMapping(value = "/repository")

@Api(description = "Repository API", tags = {"repository"})

public class RepositoryController {

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

    @Autowired

    private RepositoryServiceImpl repositoryService;

    @RequestMapping(value = "/getCountries", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public Country[] getCountries() {

        return repositoryService.getCountries();

    }

    @RequestMapping(value = "/getRepositoriesByCountry/{country}/{mode}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<RepositorySnippet> getRepositoriesByCountry(@PathVariable("country") String country,

                                                            @PathVariable("mode") String mode,

                                                            @RequestParam(value = "managed", required = false) Boolean managed) throws JSONException, IOException {

        return repositoryService.getRepositoriesByCountry(country, mode, managed);

    }

    @RequestMapping(value = "/getRepositoriesOfUser/{page}/{size}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_USER')")

    public List<RepositorySnippet> getRepositoriesSnippetOfUser(

            @PathVariable("page") String page,

            @PathVariable("size") String size) throws JSONException, IOException {

        return repositoryService.getRepositoriesSnippetOfUser(((OIDCAuthenticationToken) SecurityContextHolder.getContext().getAuthentication()).getUserInfo().getEmail(), page, size);

    }

    @RequestMapping(value = "/user/repositories/{page}/{size}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_USER')")

    public List<Repository> getRepositoriesOfUser(

            @PathVariable("page") String page,

            @PathVariable("size") String size) throws JSONException, IOException {

        return repositoryService.getRepositoriesOfUser(((OIDCAuthenticationToken) SecurityContextHolder.getContext().getAuthentication()).getUserInfo().getEmail(), page, size);

    }

    @RequestMapping(value = "/searchRegisteredRepositories/{page}/{size}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

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

    public List<RepositorySnippet> searchRegisteredRepositories(@RequestParam(name = "country", required = false) String country,

                                                                @RequestParam(name = "typology", required = false) String typology,

                                                                @RequestParam(name = "englishName", required = false) String englishName,

                                                                @RequestParam(name = "officialName", required = false) String officialName,

                                                                @RequestParam("requestSortBy") String requestSortBy,

                                                                @RequestParam("order") String order,

                                                                @PathVariable("page") int page,

                                                                @PathVariable("size") int pageSize) throws Exception {

        return repositoryService.searchRegisteredRepositories(country, typology, englishName, officialName, requestSortBy, order, page, pageSize);

    }

    // TODO: Antonis K - Replace here the registeredBy

    @RequestMapping(value = "/getRepositoryById/{id}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or @aaiUserRoleService.isMemberOf(#id)")

    public Repository getRepositoryById(@PathVariable("id") String id) throws JSONException, ResourceNotFoundException {

        Repository repo = repositoryService.getRepositoryById(id);

        if (repo != null)

            logger.info("Returning repository " + repo.getId() + " registered by " + repo.getRegisteredBy());

        else

            logger.info("Requested repository " + id + " not found");

        return repo;

    }

    @RequestMapping(value = "/getRepositoryAggregations/{id}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<AggregationDetails> getRepositoryAggregations(@PathVariable("id") String id) throws JSONException {

        return repositoryService.getRepositoryAggregations(id, 0, 20);

    }

    @RequestMapping(value = "/getRepositoryAggregationsByYear/{id}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public Map<String, List<AggregationDetails>> getRepositoryAggregationsByYear(@PathVariable("id") String id) throws JSONException {

        return repositoryService.getRepositoryAggregationsByYear(id);

    }

    @RequestMapping(value = "/getRepositoriesByName/{name:.+}/{page}/{size}/", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<Repository> getRepositoriesByName(@PathVariable("name") String name,

                                                  @PathVariable("page") String page,

                                                  @PathVariable("size") String size) throws JSONException {

        return repositoryService.getRepositoriesByName(name, page, size);

    }

    @RequestMapping(value = "/getRepositoryInterface/{id}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or @aaiUserRoleService.isMemberOf(#id)")

    public List<RepositoryInterface> getRepositoryInterface(@PathVariable("id") String id) throws JSONException {

        return repositoryService.getRepositoryInterface(id);

    }

    @RequestMapping(value = "/addRepository", method = RequestMethod.POST,

            consumes = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

//    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or (hasRole(@aaiUserRoleService.convertRepoIdToRoleId(#repository.id)) or hasRole(@aaiUserRoleService.convertRepoIdToRoleId(returnObject.id)))")

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

    public Repository addRepository(@RequestParam("datatype") String datatype,

                                    @RequestBody Repository repository) throws Exception {

        // TODO:

        //  1) add repository

        //  2) get repository id and create new role

        //  3) assign new role to authenticated user

        return repositoryService.addRepository(datatype, repository);

    }

    @RequestMapping(value = "/getDnetCountries", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<String> getDnetCountries() {

        return repositoryService.getDnetCountries();

    }

    @RequestMapping(value = "/getTypologies", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<String> getTypologies() {

        return repositoryService.getTypologies();

    }

    @RequestMapping(value = "/getTimezones", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<Timezone> getTimezones() {

        return repositoryService.getTimezones();

    }

    @RequestMapping(value = "/updateRepository", method = RequestMethod.POST,

            consumes = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or @aaiUserRoleService.isMemberOf(#repository.id)")

    public Repository updateRepository(@RequestBody Repository repository, Authentication authentication) throws Exception {

        return repositoryService.updateRepository(repository, authentication);

    }

    @RequestMapping(value = "/deleteInterface/", method = RequestMethod.DELETE)

    @PreAuthorize("@aaiUserRoleService.isMemberOf(#id)")

    public void deleteRepositoryInterface(@RequestParam("id") String id,

                                          @RequestParam("registeredBy") String registeredBy) {

        repositoryService.deleteRepositoryInterface(id, registeredBy);

    }

    @RequestMapping(value = "/addInterface", method = RequestMethod.POST,

            consumes = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or @aaiUserRoleService.isMemberOf(#repoId)")

    public RepositoryInterface addRepositoryInterface(@RequestParam("datatype") String datatype,

                                                      @RequestParam("repoId") String repoId,

                                                      @RequestParam("registeredBy") String registeredBy,

                                                      @RequestParam(value = "comment", required = false) String comment,

                                                      @RequestBody RepositoryInterface repositoryInterface) throws Exception {

        return repositoryService.addRepositoryInterface(datatype, repoId, registeredBy, comment, repositoryInterface);

    }

    @RequestMapping(value = "/getUrlsOfUserRepos/{page}/{size}/", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_USER')")

    public List<String> getUrlsOfUserRepos(@PathVariable("page") String page, @PathVariable("size") String size) {

        return repositoryService.getUrlsOfUserRepos(((OIDCAuthenticationToken) SecurityContextHolder.getContext().getAuthentication()).getUserInfo().getEmail(), page, size);

    }

    @RequestMapping(value = "/getDatasourceVocabularies/{mode}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public List<String> getDatasourceVocabularies(@PathVariable("mode") String mode) {

        return repositoryService.getDatasourceVocabularies(mode);

    }

    @RequestMapping(value = "/getCompatibilityClasses/{mode}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public Map<String, String> getCompatibilityClasses(@PathVariable("mode") String mode) {

        return repositoryService.getCompatibilityClasses(mode);

    }

    @RequestMapping(value = "/getDatasourceClasses/{mode}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public Map<String, String> getDatasourceClasses(@PathVariable("mode") String mode) {

        return repositoryService.getDatasourceClasses(mode);

    }

    @RequestMapping(value = "/getMetricsInfoForRepository/{repoId}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public MetricsInfo getMetricsInfoForRepository(@PathVariable("repoId") String repoId) throws RepositoryServiceException {

        return repositoryService.getMetricsInfoForRepository(repoId);

    }

    @RequestMapping(value = "/getListLatestUpdate/{mode}", method = RequestMethod.GET,

            produces = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    public Map<String, String> getListLatestUpdate(@PathVariable("mode") String mode) throws JSONException {

        return repositoryService.getListLatestUpdate(mode);

    }

    @RequestMapping(value = "/updateRepositoryInterface", method = RequestMethod.POST,

            consumes = MediaType.APPLICATION_JSON_VALUE)

    @ResponseBody

    @PreAuthorize("hasRole('ROLE_ADMIN') or hasRole('ROLE_PROVIDE_ADMIN') or @aaiUserRoleService.isMemberOf(#repoId)")

    public RepositoryInterface updateRepositoryInterface(@RequestParam("repoId") String repoId,

                                                         @RequestParam("registeredBy") String registeredBy,

                                                         @RequestParam(value = "comment", required = false) String comment,

                                                         @RequestBody RepositoryInterface repositoryInterface) throws Exception {

        return repositoryService.updateRepositoryInterface(repoId, registeredBy, comment, repositoryInterface);

    }

}