D-Net

package eu.dnetlib.repo.manager.service.controllers;

import com.fasterxml.jackson.core.JsonProcessingException;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.fasterxml.jackson.databind.SerializationFeature;

import eu.dnetlib.domain.data.Repository;

import eu.dnetlib.domain.data.RepositoryInterface;

import eu.dnetlib.domain.enabling.Vocabulary;

import eu.dnetlib.repo.manager.service.domain.RequestFilter;

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

import eu.dnetlib.repo.manager.service.utils.Converter;

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

import gr.uoa.di.driver.enabling.vocabulary.VocabularyLoader;

import org.apache.commons.codec.digest.DigestUtils;

import org.apache.log4j.Logger;

import org.json.JSONArray;

import org.json.JSONException;

import org.json.JSONObject;

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

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

import org.springframework.core.ParameterizedTypeReference;

import org.springframework.http.HttpEntity;

import org.springframework.http.HttpHeaders;

import org.springframework.http.HttpMethod;

import org.springframework.http.ResponseEntity;

import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;

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

import org.springframework.stereotype.Component;

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

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

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

import org.springframework.web.client.RestClientException;

import org.springframework.web.client.RestTemplate;

import org.springframework.web.util.UriComponents;

import org.springframework.web.util.UriComponentsBuilder;

import javax.annotation.PostConstruct;

import java.sql.Timestamp;

import java.util.*;

import java.util.concurrent.ConcurrentHashMap;

import java.util.stream.Collectors;

@Component

public class RepositoryApiImpl implements RepositoryApi {

    @Value("${api.baseAddress}")

    private String baseAddress;

    private RestTemplate restTemplate = null;

    private HttpHeaders httpHeaders;

    private final String[] vocabularyNames = {"dnet:countries", "dnet:datasource_typologies", "dnet:compatibilityLevel"};

    private static final Logger LOGGER = Logger.getLogger(RepositoryApiImpl.class);

    @Value("${services.repomanager.usageStatisticsDiagramsBaseURL}")

    private String usageStatisticsDiagramsBaseURL;

    @Value("${services.repomanager.usageStatisticsNumbersBaseURL}")

    private String usageStatisticsNumbersBaseURL;

    @Autowired

    private VocabularyLoader vocabularyLoader;

    @Autowired

    private PiWikApi piWikApi;

    @Autowired

    private EmailUtils emailUtils;

    private Map<String, Vocabulary> vocabularyMap = new ConcurrentHashMap<String, Vocabulary>();

    private Map<String, String> countriesMap = new HashMap<>();

    private Map<String, String> inverseCountriesMap = new HashMap<>();

    private static Map<String,List<String>> dataSourceClass = new HashMap<String,List<String>>(){{

        put("opendoar",new ArrayList<String>(){{ add("pubsrepository::institutional");

                                                    add("pubsrepository::thematic");

                                                    add("pubsrepository::unknown");

                                                    add("pubsrepository::mock");

        }});

        put("re3data",new ArrayList<String>(){{ add("datarepository::unknown");

        }});

        put("journal",new ArrayList<String>(){{ add("pubsrepository::journal");

        }});

        put("aggregator",new ArrayList<String>(){{ add("aggregator::pubsrepository::institutional");

                                                        add("aggregator::pubsrepository::journals");

                                                        add("aggregator::datarepository");

                                                        add("aggregator::pubsrepository::unknown");

        }});

    }};

    private static Map<String,String> invertedDataSourceClass = new HashMap<String,String>(){{

        put("pubsrepository::institutional","opendoar");

        put("pubsrepository::thematic","opendoar");

        put("pubsrepository::unknown","opendoar");

        put("pubsrepository::mock","opendoar");

        put("datarepository::unknown","re3data");

        put("pubsrepository::journal","journal");

        put("aggregator::pubsrepository::institutional","aggregator");

        put("aggregator::pubsrepository::journals","aggregator");

        put("aggregator::datarepository","aggregator");

        put("aggregator::pubsrepository::unknown","aggregator");

    }};

    @PostConstruct

    private void init() {

        LOGGER.debug("Initialization method of repository api!");

        restTemplate = new RestTemplate();

        restTemplate.getMessageConverters().add(new MappingJackson2HttpMessageConverter());

        httpHeaders = new HttpHeaders();

        httpHeaders.set("Content-Type", "application/json");

        for (String vocName : vocabularyNames) {

            vocabularyMap.put(vocName, vocabularyLoader.getVocabulary(vocName, Locale.ENGLISH, Locale.ROOT));

        }

        Country[] countries = getCountries();

        for (Country c : countries) {

            countriesMap.put(c.getName(), c.getCode());

            inverseCountriesMap.put(c.getCode(), c.getName());

        }

    }

    @Override

    public Country[] getCountries() {

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/countries")

                .build().encode();

        return restTemplate.getForObject(uriComponents.toUri(), Country[].class);

    }

    @Override

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

                                                     @PathVariable("mode") String mode,

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

        LOGGER.debug("Getting repositories by country!");

        int page = 0;

        int size = 100;

        List<Repository> resultSet = new ArrayList<>();

        String countryCode = countriesMap.get(country);

        String filterKey = "UNKNOWN";

        if (mode.equalsIgnoreCase("opendoar"))

            filterKey = "openaire____::opendoar";

        else if (mode.equalsIgnoreCase("re3data"))

            filterKey = "openaire____::re3data";

        LOGGER.debug("Country code equals : " + countryCode);

        LOGGER.debug("Filter mode equals : " + filterKey);

        UriComponents uriComponents = searchDatasource(String.valueOf(page),String.valueOf(size));

        RequestFilter requestFilter = new RequestFilter();

        requestFilter.setCountry(countryCode);

        try{

            String rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

            JSONArray jsonArray = (JSONArray) new JSONObject(rs).get("datasourceInfo");

            while (jsonArray.length() > 0 ) {

                resultSet.addAll(this.getRepositoriesByMode(filterKey, Converter.jsonToRepositoryList(new JSONObject(rs))));

                page += 1;

                uriComponents = searchDatasource(String.valueOf(page),String.valueOf(size));

                rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

                jsonArray = (JSONArray) new JSONObject(rs).get("datasourceInfo");

            }

            for (Repository r : resultSet)

                this.updateRepositoryInfo(r);

            return resultSet;

        }catch (Exception e){

            LOGGER.debug("Exception on getRepositoriesByCountry" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    private Repository updateRepositoryInfo(Repository r) throws JSONException {

        /*

        * from datasource class

        * we get the datasource type form the inverted map

        * */

        r.setDatasourceType(getRepositoryType(r.getDatasourceClass()));

        r.setInterfaces(this.getRepositoryInterface(r.getId()));

        r.setPiwikInfo(piWikApi.getPiwikSiteForRepo(r.getId()));

        r.setCountryName(getCountryName(r.getCountryCode()));

        return r;

    }

    private Collection<Repository> getRepositoriesByMode(String mode, List<Repository> rs) {

        List<Repository> reps = new ArrayList<>();

        for (Repository r : rs) {

            if (r.getCollectedFrom() != null && r.getCollectedFrom().equals(mode))

                reps.add(r);

        }

        return reps;

    }

    @Override

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

    public List<Repository> getRepositoriesOfUser(@PathVariable("userEmail") String userEmail,

                                                  @PathVariable("page") String page,

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

        LOGGER.debug("Retreiving repositories of user : " + userEmail );

        UriComponents uriComponents = searchDatasource(page,size);

        RequestFilter requestFilter = new RequestFilter();

        requestFilter.setRegisteredby(userEmail);

        try{

            String rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

            List<Repository> repos = Converter.jsonToRepositoryList(new JSONObject(rs));

            for (Repository r : repos)

                this.updateRepositoryInfo(r);

            return repos;

        }catch (Exception e){

            LOGGER.debug("Exception on getRepositoriesOfUser" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    @Override

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

        LOGGER.debug("Retreiving repositories with id : " + id );

        Repository repo = null;

        UriComponents uriComponents = searchDatasource("0","100");

        RequestFilter requestFilter = new RequestFilter();

        requestFilter.setId(id);

        try{

            String rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

            JSONArray jsonArray = (JSONArray) new JSONObject(rs).get("datasourceInfo");

            if(jsonArray.length() == 0)

                throw new ResourceNotFoundException();

            repo = Converter.jsonToRepositoryObject(jsonArray.getJSONObject(0));

            return updateRepositoryInfo(repo);

        }catch (JSONException e){

            LOGGER.debug("Exception on getRepositoryById" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    @Override

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

        LOGGER.debug("Retreiving aggregations for repository with id : " + id );

        UriComponents uriComponents = searchDatasource("0","100");

        RequestFilter requestFilter = new RequestFilter();

        requestFilter.setId(id);

        List<AggregationDetails> aggregationHistory = new ArrayList<>();

        try {

            String rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

            JSONObject repository = new JSONObject(rs);

            aggregationHistory.addAll(Converter.getAggregationHistoryFromJson(repository.getJSONArray("datasourceInfo").getJSONObject(0)));

            return aggregationHistory.size() == 0? aggregationHistory : aggregationHistory.stream()

                                                    .sorted(Comparator.comparing(AggregationDetails::getDate).reversed())

                                                    .limit(20)

                                                    .collect(Collectors.toList());

        } catch (JSONException e) {

            LOGGER.debug("Exception on getRepositoryAggregations" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    @Override

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

        LOGGER.debug("Retreiving aggregations (by year) for repository with id : " + id );

        UriComponents uriComponents = searchDatasource("0","100");

        RequestFilter requestFilter = new RequestFilter();

        requestFilter.setId(id);

        List<AggregationDetails> aggregationHistory = new ArrayList<>();

        Map<String, List<AggregationDetails>> aggregationByYear = new HashMap<>();

        try {

            String rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

            JSONObject repository = new JSONObject(rs);

            aggregationHistory.addAll(Converter.getAggregationHistoryFromJson(repository.getJSONArray("datasourceInfo").getJSONObject(0)));

            return aggregationHistory.size() == 0? aggregationByYear:createYearMap(aggregationHistory);

        } catch (JSONException e) {

            LOGGER.debug("Exception on getRepositoryAggregations" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    private Map<String,List<AggregationDetails>> createYearMap(List<AggregationDetails> aggregationHistory) {

        Map<String, List<AggregationDetails>> aggregationByYear;

        aggregationHistory = aggregationHistory.stream()

                            .sorted(Comparator.comparing(AggregationDetails::getDate).reversed())

                            .collect(Collectors.toList());

       return aggregationHistory.stream()

                            .collect(Collectors.groupingBy(AggregationDetails::getYear));

    }

    @Override

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

                                                  @PathVariable("page") String page,

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

        LOGGER.debug("Retreiving  repositories with official name : " + name );

        UriComponents uriComponents = searchDatasource("0","100");

        RequestFilter requestFilter = new RequestFilter();

        requestFilter.setOfficialname(name);

        try{

            String rs = restTemplate.postForObject(uriComponents.toUri(),requestFilter, String.class);

            List<Repository> repos = Converter.jsonToRepositoryList(new JSONObject(rs));

            for (Repository r : repos)

                updateRepositoryInfo(r);

            return repos;

        }catch (Exception e){

            LOGGER.debug("Exception on getRepositoriesByName" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    @Override

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

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/api/")

                .path("/{id}")

                .build().expand(id).encode();

        try{

            String rs = restTemplate.getForObject(uriComponents.toUri(), String.class);

            return Converter.jsonToRepositoryInterfaceList(new JSONObject(rs));

        }catch (Exception e ){

            LOGGER.debug("Exception on getRepositoryInterface" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    @Override

    @PreAuthorize("hasRole('ROLE_USER') and #repository.registeredBy == authentication.userInfo.email")

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

                                    @RequestBody Repository repository) throws Exception {

        repository = this.setRepositoryFeatures(datatype,repository);

        LOGGER.debug("storing " + datatype + " repository with id: " + repository.getId());

        this.storeRepository(repository);

        return repository;

    }

    @Override

    @PreAuthorize("hasRole('ROLE_USER') and #repository.registeredBy == authentication.userInfo.email")

    public Repository updateRepository(@RequestBody Repository repository) throws JSONException {

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/update/")

                .build()

                .encode();

        try {

            String json_repository = Converter.repositoryObjectToJson(repository);

            HttpEntity<String> httpEntity = new HttpEntity <String> (json_repository,httpHeaders);

            restTemplate.postForObject(uriComponents.toUri(),httpEntity,String.class);

            return repository;

        }catch (Exception e){

            LOGGER.debug("Exception on updateRepository" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    private Repository setRepositoryFeatures(String datatype, Repository repository) {

        //TODO update map

        repository.setCountryCode(countriesMap.get(repository.getCountryName()));

        repository.setActivationId(UUID.randomUUID().toString());

        repository.setCollectedFrom("infrastruct_::openaire");

        if (datatype.equals("journal")) {

            repository.setId("openaire____::issn" + repository.getIssn());

            repository.setNamespacePrefix("issn" + repository.getIssn());

        }else if (datatype.equals("aggregator")) {

            repository.setId("openaire____::" + com.unboundid.util.Base64.encode(repository.getOfficialName()));

            repository.setNamespacePrefix(DigestUtils.md5Hex(repository.getOfficialName()).substring(0,12));

        }

        return repository;

    }

    private void updateInterface(String datatype,RepositoryInterface iFace) {

        //TODO call update base url

        //((DatasourceManagerService) this.dmService.getService()).updateBaseUrl(repo.getId(), iFace.getId(), iFace.getBaseUrl());

        if (!iFace.getAccessSet().isEmpty()) {

            LOGGER.debug("set not empty: " + iFace.getAccessSet());

            //TODO call update method for access params

            // ((DatasourceManagerService) this.dmService.getService()).updateAccessParam(repo.getId(), iFace.getId(), "set", iFace.getAccessSet(), false);

        } else {

            //TODO call deleteAccessParamOrExtraField

            //((DatasourceManagerService) this.dmService.getService()).deleteAccessParamOrExtraField(repo.getId(), iFace.getId(), "set");

        }

        //TODO update content description

        //((DatasourceManagerService) this.dmService.getService()).updateContentDescription(repo.getId(), iFace.getId(), "metadata");

        if (datatype.equals("re3data")) {

            //TODO call update access params

            //  ((DatasourceManagerService) this.dmService.getService()).updateAccessParam(repo.getId(), iFace.getId(), "format", "oai_datacite", false);

            iFace.setAccessFormat("oai_datacite");

        } else {

            //TODO call update access params

            //((DatasourceManagerService) this.dmService.getService()).updateAccessParam(repo.getId(), iFace.getId(), "format", "oai_dc", false);

            iFace.setAccessFormat("oai_dc");

        }

    }

    private void storeRepository(Repository repository) throws JSONException {

        Date utilDate = new Date();

        Timestamp date = new Timestamp(utilDate.getTime());

        repository.setDateOfCollection(date);

        repository.setAggregator("OPENAIRE");

        repository.setCountryCode(countriesMap.get(repository.getCountryName()));

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/add/")

                .build()

                .encode();

        String json_repository = Converter.repositoryObjectToJson(repository);

        ObjectMapper mapper = new ObjectMapper();

        mapper.enable(SerializationFeature.INDENT_OUTPUT);

        try {

            LOGGER.debug(mapper.writeValueAsString(json_repository));

        } catch (JsonProcessingException e) {

            e.printStackTrace();

        }

        HttpEntity<String> httpEntity = new HttpEntity <String> (json_repository,httpHeaders);

        restTemplate.postForObject(uriComponents.toUri(),httpEntity,String.class);

    }

    @Override

    @PreAuthorize("hasRole('ROLE_USER') and #registeredBy == authentication.userInfo.email")

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

                                          @RequestParam("registeredBy") String registeredBy){

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/api/")

                .path("/{id}")

                .build().expand(id).encode();

        LOGGER.debug(uriComponents.toUri());

        restTemplate.delete(uriComponents.toUri());

    }

    @Override

    @PreAuthorize("hasRole('ROLE_USER') and #registeredBy == authentication.userInfo.email")

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

                                                      @RequestParam("repoId") String repoId,

                                                      @RequestParam("registeredBy") String registeredBy,

                                                      @RequestBody RepositoryInterface repositoryInterface) throws JSONException,ResourceNotFoundException {

        try {

            Repository e = this.getRepositoryById(repoId);

            repositoryInterface = createRepositoryInterface(e,repositoryInterface,datatype);

            String json_interface = Converter.repositoryInterfaceObjectToJson(e,repositoryInterface);

            UriComponents uriComponents = UriComponentsBuilder

                    .fromHttpUrl(baseAddress + "/ds/api/add/")

                    .build()

                    .encode();

            HttpEntity<String> httpEntity = new HttpEntity <String> (json_interface,httpHeaders);

            restTemplate.postForObject(uriComponents.toUri(),httpEntity,String.class);

            return repositoryInterface;

        } catch (JSONException e) {

            LOGGER.debug("Exception on addRepositoryInterface" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    private RepositoryInterface createRepositoryInterface(Repository repo, RepositoryInterface iFace, String datatype) {

        iFace.setContentDescription("metadata");

        iFace.setCompliance("UNKNOWN");

        if (datatype.equals("re3data"))

            iFace.setAccessFormat("oai_datacite");

        else

            iFace.setAccessFormat("oai_dc");

        if (repo.getDatasourceClass() != null && !repo.getDatasourceClass().isEmpty())

            iFace.setTypology(repo.getDatasourceClass());

        else if (datatype.equalsIgnoreCase("journal"))

            iFace.setTypology("pubsrepository::journal");

        else if (datatype.equalsIgnoreCase("aggregator"))

            iFace.setTypology("aggregator::pubsrepository::unknown");

        else if (datatype.equalsIgnoreCase("opendoar"))

            iFace.setTypology("pubsrepository::unknown");

        else if (datatype.equalsIgnoreCase("re3data"))

            iFace.setTypology("datarepository::unknown");

        iFace.setRemovable(true);

        iFace.setAccessProtocol("oai");

        iFace.setMetadataIdentifierPath("//*[local-name()='header']/*[local-name()='identifier']");

        iFace.setId("api_________::" + repo.getId() + "::" + UUID.randomUUID().toString().substring(0, 8));

        if (iFace.getAccessSet().isEmpty()) {

            LOGGER.debug("set is empty: " + iFace.getAccessSet());

            iFace.removeAccessSet();

        }

        return iFace;

    }

    @Override

    public List<String> getDnetCountries() {

        LOGGER.debug("Getting dnet-countries!");

        return Converter.readFile("countries.txt");

    }

    @Override

    public List<String> getTypologies() {

        return Converter.readFile("typologies.txt");

    }

    @Override

    public List<Timezone> getTimezones() {

        List<String> timezones =  Converter.readFile("timezones.txt");

        return Converter.toTimezones(timezones);

    }

    @Override

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

    public List<String> getUrlsOfUserRepos(@PathVariable("user_email") String userEmail,

                                           @PathVariable("page") String page,

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

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/api/baseurl/")

                .path("/{page}/{size}")

                .build().expand(page,size).encode();

        try{

            RequestFilter requestFilter = new RequestFilter();

            requestFilter.setRegisteredby(userEmail);

            return Arrays.asList(restTemplate.postForObject(uriComponents.toUri(),requestFilter, String[].class));

        }catch (Exception e){

            LOGGER.debug("Exception on addRepositoryInterface" , e);

            emailUtils.reportException(e);

            throw e;

        }

    }

    @Override

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

        List<String> resultSet = new ArrayList<>();

        for (Map.Entry<String, String> entry : this.getVocabulary("dnet:datasource_typologies").getAsMap().entrySet()) {

            if (mode.equalsIgnoreCase("aggregator")) {

                if (entry.getKey().contains("aggregator"))

                    resultSet.add(entry.getValue());

            } else if (mode.equalsIgnoreCase("journal")) {

                if (entry.getKey().contains("journal"))

                    resultSet.add(entry.getValue());

            } else if (mode.equalsIgnoreCase("opendoar")) {

                if (entry.getKey().contains("pubsrepository"))

                    resultSet.add(entry.getValue());

            } else if (mode.equalsIgnoreCase("re3data")) {

                if (entry.getKey().contains("datarepository"))

                    resultSet.add(entry.getValue());

            }

        }

        return resultSet;

    }

    private Vocabulary getVocabulary(String vocName) {

        if (!vocabularyMap.containsKey(vocName)) {

            vocabularyMap.put(vocName, vocabularyLoader.getVocabulary(vocName, Locale.ENGLISH, Locale.ROOT));

        }

        return vocabularyMap.get(vocName);

    }

    @Override

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

        LOGGER.debug("Getting compatibility classes for mode: " + mode);

        Map<String, String> retMap = new HashMap<String, String>();

        Map<String, String> compatibilityClasses = this.getVocabulary("dnet:compatibilityLevel").getAsMap();

        boolean foundData = false;

        for (Map.Entry<String, String> entry : compatibilityClasses.entrySet()) {

            if (mode.equalsIgnoreCase(Constants.REPOSITORY_MODE_ALL))

                return compatibilityClasses;

            else if (mode.equalsIgnoreCase(Constants.REPOSITORY_MODE_RE3DATA)) {

                if (entry.getKey().matches("^openaire[1-9].0_data$")) {

                    retMap.put(entry.getKey(), entry.getValue());

                    foundData = true;

                }

            } else {

                if (entry.getKey().matches("^openaire[1-9].0$") || entry.getKey().equals("driver"))

                    retMap.put(entry.getKey(), entry.getValue());

            }

        }

        //TODO TO BE REMOVED WHEN VOCABULARIES ARE UPDATED

        if (mode.equalsIgnoreCase(Constants.REPOSITORY_MODE_RE3DATA) && !foundData)

            retMap.put("openaire2.0_data", "OpenAIRE Data (funded, referenced datasets)");

        return retMap;

    }

    @Override

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

        LOGGER.debug("Getting datasource classes for mode: " + mode);

        Map<String, String> retMap = new HashMap<String, String>();

        for (Map.Entry<String, String> entry : this.getVocabulary("dnet:datasource_typologies").getAsMap().entrySet()) {

            if (mode.equalsIgnoreCase("aggregator")) {

                if (entry.getKey().contains("aggregator"))

                    retMap.put(entry.getKey(), entry.getValue());

            } else if (mode.equalsIgnoreCase("journal")) {

                if (entry.getKey().contains("journal"))

                    retMap.put(entry.getKey(), entry.getValue());

            } else if (mode.equalsIgnoreCase("opendoar")) {

                if (entry.getKey().contains("pubsrepository"))

                    retMap.put(entry.getKey(), entry.getValue());

            } else if (mode.equalsIgnoreCase("re3data")) {

                if (entry.getKey().contains("datarepository"))

                    retMap.put(entry.getKey(), entry.getValue());

            }

        }

        return filterResults(retMap,mode);

    }

    private Map<String,String> filterResults(Map<String, String> map,String mode) {

        HashMap<String,String> filteredMap = new HashMap<>();

        for(String key:map.keySet())

            if(dataSourceClass.get(mode).contains(key))

                filteredMap.put(key,map.get(key));

        return filteredMap;

    }

    @Override

    public String getCountryName(String countryCode) {

        return inverseCountriesMap.get(countryCode);

    }

    @Override

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

        try {

            MetricsInfo metricsInfo = new MetricsInfo();

            metricsInfo.setDiagramsBaseURL(this.usageStatisticsDiagramsBaseURL);

            metricsInfo.setMetricsNumbers(getMetricsNumbers(getOpenAIREId(repoId)));

            return metricsInfo;

        } catch (Exception e) {

            LOGGER.error("Error while getting metrics info for repository: ", e);

            emailUtils.reportException(e);

            throw new RepositoryServiceException("General error", RepositoryServiceException.ErrorCode.GENERAL_ERROR);

        }

    }

    @Override

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

        if(mode.equals("opendoar"))

            return Collections.singletonMap("lastCollectionDate", getRepositoryInterface("openaire____::"+mode).get(0).getLastCollectionDate());

        else

            /*

            * first api of re3data has null value on collection date

            * */

            return Collections.singletonMap("lastCollectionDate", getRepositoryInterface("openaire____::"+mode).get(1).getLastCollectionDate());

    }

    @Override

    @PreAuthorize("hasRole('ROLE_USER') and #registeredBy == authentication.userInfo.email")

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

                                                         @RequestParam("registeredBy") String registeredBy,

                                                         @RequestBody RepositoryInterface repositoryInterface) throws JSONException {

        this.updateBaseUrl(repoId,repositoryInterface.getId(),repositoryInterface.getBaseUrl());

        this.updateCompliance(repoId,repositoryInterface.getId(),repositoryInterface.getCompliance());

        this.updateValidationSet(repoId,repositoryInterface.getId(),repositoryInterface.getAccessSet());

        return repositoryInterface;

    }

    private void updateValidationSet(String repositoryId, String repositoryInterfaceId, String validationSet) {

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/api/oaiset")

                .queryParam("dsId",repositoryId)

                .queryParam("apiId",repositoryInterfaceId)

                .queryParam("oaiSet",validationSet)

                .build().encode();

        restTemplate.postForObject(uriComponents.toUri(),null,String.class);

    }

    private void updateBaseUrl(String repositoryId, String repositoryInterfaceId, String baseUrl) {

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/api/baseurl")

                .queryParam("dsId",repositoryId)

                .queryParam("apiId",repositoryInterfaceId)

                .queryParam("baseUrl",baseUrl)

                .build().encode();

        restTemplate.postForObject(uriComponents.toUri(),null,String.class);

    }

    private void updateCompliance(String repositoryId, String repositoryInterfaceId,String compliance) {

        UriComponents uriComponents = UriComponentsBuilder

                .fromHttpUrl(baseAddress + "/ds/api/compliance")

                .queryParam("dsId",repositoryId)

                .queryParam("apiId",repositoryInterfaceId)

                .queryParam("compliance",compliance)

                .build().encode();

        restTemplate.postForObject(uriComponents.toUri(),null,String.class);