D-Net

package eu.dnetlib.organizations.controller;

import java.security.Principal;

import java.util.Arrays;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

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

import org.springframework.data.domain.Page;

import org.springframework.data.domain.PageRequest;

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

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

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

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

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

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

import eu.dnetlib.organizations.model.OpenaireSimRel;

import eu.dnetlib.organizations.model.utils.BrowseEntry;

import eu.dnetlib.organizations.model.utils.RelationByOrg;

import eu.dnetlib.organizations.model.view.OrganizationSimpleView;

import eu.dnetlib.organizations.model.view.OrganizationView;

import eu.dnetlib.organizations.repository.OpenaireSimRelRepository;

import eu.dnetlib.organizations.repository.readonly.OrganizationSimpleViewRepository;

import eu.dnetlib.organizations.repository.readonly.OrganizationViewRepository;

import eu.dnetlib.organizations.utils.DatabaseUtils;

import eu.dnetlib.organizations.utils.RelationType;

@RestController

@RequestMapping("/api/organizations")

public class OrganizationController {

	@Autowired

	private OrganizationViewRepository organizationViewRepository;

	@Autowired

	private OrganizationSimpleViewRepository organizationSimpleViewRepository;

	@Autowired

	private OpenaireSimRelRepository openaireSimRelRepository;

	@Autowired

	private DatabaseUtils databaseUtils;

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

	public List<String> save(@RequestBody final OrganizationView org, final Principal principal) {

		final String user = principal != null ? principal.getName() : "anonymous";

		final String orgId = databaseUtils.insertOrUpdateOrganization(org, user, StringUtils.isNotBlank(org.getId()));

		return Arrays.asList(orgId);

	}

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

	public OrganizationView findById(@RequestParam final String id) {

		return organizationViewRepository.findById(id).get();

	}

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

	public List<RelationByOrg> findRelationsById(@RequestParam final String id) {

		return organizationViewRepository.findRelations(id);

	}

	@RequestMapping(value = "/relations", method = RequestMethod.PUT)

	public List<RelationByOrg> addRelation(@RequestParam final String from, @RequestParam final String to, @RequestParam final RelationType type) {

		if (from.equals(to)) { throw new IllegalArgumentException("Invalid relation !!!"); }

		databaseUtils.addRelation(from, to, type);

		return organizationViewRepository.findRelations(from);

	}

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

	public List<RelationByOrg> deleteRelation(@RequestParam final String from, @RequestParam final String to, @RequestParam final RelationType type) {

		if (from.equals(to)) { throw new IllegalArgumentException("Invalid relation !!!"); }

		databaseUtils.deleteRelation(from, to, type);

		return organizationViewRepository.findRelations(from);

	}

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

	public List<OpenaireSimRel> findSimilaritiesById(@RequestParam final String id) {

		return openaireSimRelRepository.findByLocalId(id);

	}

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

	public Page<OrganizationSimpleView> search(@PathVariable final int page, @PathVariable final int size, @RequestParam final String q) {

		return organizationSimpleViewRepository.findByNameContainingIgnoreCase(q, PageRequest.of(page, size));

	}

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

	public Page<OrganizationSimpleView> findByCountry(@PathVariable final String code, @PathVariable final int page, @PathVariable final int size) {

		return organizationSimpleViewRepository.findByCountry(code, PageRequest.of(page, size));

	}

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

	public Page<OrganizationSimpleView> findByType(@PathVariable final String type, @PathVariable final int page, @PathVariable final int size) {

		return organizationSimpleViewRepository.findByType(type, PageRequest.of(page, size));

	}

	@RequestMapping(value = "/browse/countries", method = RequestMethod.GET)

	public List<BrowseEntry> browseCountries() {

		return organizationSimpleViewRepository.browseCountries();

	}

	@RequestMapping(value = "/browse/types", method = RequestMethod.GET)

	public List<BrowseEntry> browseOrganizationTypes() {

		return organizationSimpleViewRepository.browseTypes();

	}

}