D-Net

package eu.dnetlib.organizations.controller;

import java.util.Arrays;

import java.util.List;

import java.util.UUID;

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

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

import eu.dnetlib.organizations.model.OpenaireSimRel;

import eu.dnetlib.organizations.model.Organization;

import eu.dnetlib.organizations.model.OtherIdentifier;

import eu.dnetlib.organizations.model.OtherName;

import eu.dnetlib.organizations.model.Type;

import eu.dnetlib.organizations.model.Url;

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

import eu.dnetlib.organizations.repository.OpenaireSimRelRepository;

import eu.dnetlib.organizations.repository.OrganizationRepository;

import eu.dnetlib.organizations.repository.OtherIdentifierRepository;

import eu.dnetlib.organizations.repository.OtherNameRepository;

import eu.dnetlib.organizations.repository.TypeRepository;

import eu.dnetlib.organizations.repository.UrlRepository;

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

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

@RestController

@RequestMapping("/api/organizations")

public class OrganizationController {

	@Autowired

	private OrganizationViewRepository organizationViewRepository;

	@Autowired

	private OrganizationSimpleViewRepository organizationSimpleViewRepository;

	@Autowired

	private OpenaireSimRelRepository openaireSimRelRepository;

	@Autowired

	private AcronymRepository acronymRepository;

	@Autowired

	private OrganizationRepository organizationRepository;

	@Autowired

	private OtherIdentifierRepository otherIdentifierRepository;

	@Autowired

	private OtherNameRepository otherNameRepository;

	@Autowired

	private TypeRepository typeRepository;

	@Autowired

	private UrlRepository urlRepository;

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

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

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

		return Arrays.asList(orgId);

	}

	private String insertOrUpdateOrganization(final OrganizationView org, final boolean update) {

		final String orgId = update ? org.getId() : "openorgs____::" + DigestUtils.md5Hex(UUID.randomUUID().toString());

		if (update) {

			acronymRepository.deleteByOrgId(orgId);

			otherNameRepository.deleteByOrgId(orgId);

			otherIdentifierRepository.deleteByOrgId(orgId);

			typeRepository.deleteByOrgId(orgId);

			urlRepository.deleteByOrgId(orgId);

		}

		organizationRepository.save(new Organization(orgId, org.getName(), org.getLat(), org.getLng(), org.getCity(), org.getCountry()));

		org.getAcronyms().forEach(s -> acronymRepository.save(new Acronym(orgId, s)));

		org.getOtherNames().forEach(n -> otherNameRepository.save(new OtherName(orgId, n.getName(), n.getLang())));

		org.getOtherIdentifiers().forEach(id -> otherIdentifierRepository.save(new OtherIdentifier(orgId, id.getId(), id.getType())));

		org.getTypes().forEach(t -> typeRepository.save(new Type(orgId, t)));

		org.getUrls().forEach(u -> urlRepository.save(new Url(orgId, u)));

		if (update) {

			// SET LAST MODIFICATION DATE AND USER

		} else {

			// SET LAST CREATION AND MODIFICATION 5DATE AND USER

		}

		return 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 = "/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();

	}

}