D-Net

package eu.dnetlib.organizations.controller;

import java.util.Arrays;

import java.util.Collection;

import java.util.HashMap;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.TreeMap;

import java.util.stream.Collectors;

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.security.core.Authentication;

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

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

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

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

import eu.dnetlib.organizations.utils.RelationType;

@RestController

@RequestMapping("/api/organizations")

public class OrganizationController {

	@Autowired

	private OrganizationRepository organizationRepository;

	@Autowired

	private OrganizationViewRepository organizationViewRepository;

	@Autowired

	private OrganizationSimpleViewRepository organizationSimpleViewRepository;

	@Autowired

	private OpenaireSimRelRepository openaireSimRelRepository;

	@Autowired

	private ConflictViewRepository conflictViewRepository;

	@Autowired

	private DatabaseUtils databaseUtils;

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

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

		if (StringUtils.isBlank(org.getName())) {

			throw new RuntimeException("Missing field: name");

		} else if (StringUtils.isBlank(org.getCountry())) {

			throw new RuntimeException("Missing field: country");

		} else if (StringUtils.isBlank(org.getType())) {

			throw new RuntimeException("Missing field: type");

		} else if (UserInfo.isSuperAdmin(authentication) || organizationRepository.verifyAuthorizationForCountry(org.getCountry(), authentication.getName())) {

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

			return Arrays.asList(orgId);

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public OrganizationView findById(@RequestParam final String id, final Authentication authentication) {

		final OrganizationView org = organizationViewRepository.findById(id).get();

		if (UserInfo.isSuperAdmin(authentication) || organizationRepository.verifyAuthorizationForCountry(org.getCountry(), authentication.getName())) {

			return org;

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

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

		if (UserInfo.isSuperAdmin(authentication) || organizationRepository.verifyAuthorizationForId(id, authentication.getName())) {

			return organizationViewRepository.findRelations(id);

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public List<RelationByOrg> addRelation(@RequestParam final String from,

			@RequestParam final String to,

			@RequestParam final RelationType type,

			final Authentication authentication) {

		if (from.equals(to)) {

			throw new IllegalArgumentException("Invalid relation !!!");

		} else if (UserInfo.isSuperAdmin(authentication)

				|| organizationRepository.verifyAuthorizationForId(to, authentication.getName())

				|| organizationRepository.verifyAuthorizationForId(from, authentication.getName())) {

			databaseUtils.addRelation(from, to, type);

			return organizationViewRepository.findRelations(from);

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public List<RelationByOrg> deleteRelation(@RequestParam final String from,

			@RequestParam final String to,

			@RequestParam final RelationType type,

			final Authentication authentication) {

		if (from.equals(to)) {

			throw new IllegalArgumentException("Invalid relation !!!");

		} else if (UserInfo.isSuperAdmin(authentication)

				|| organizationRepository.verifyAuthorizationForId(to, authentication.getName())

				|| organizationRepository.verifyAuthorizationForId(from, authentication.getName())) {

			databaseUtils.deleteRelation(from, to, type);

			return organizationViewRepository.findRelations(from);

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public List<OpenaireSimRel> findEnrichmentsById(@RequestParam final String id, final Authentication authentication) {

		if (UserInfo.isSuperAdmin(authentication) || organizationRepository.verifyAuthorizationForId(id, authentication.getName())) {

			return openaireSimRelRepository.findByLocalId(id)

					.stream()

					.filter(s -> !s.getOaOriginalId().startsWith(OpenOrgsConstants.OPENORGS_PREFIX))

					.collect(Collectors.toList());

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public List<OpenaireSimRel> findConflictsById(@RequestParam final String id, final Authentication authentication) {

		if (UserInfo.isSuperAdmin(authentication) || organizationRepository.verifyAuthorizationForId(id, authentication.getName())) {

			return openaireSimRelRepository.findByLocalId(id)

					.stream()

					.filter(s -> s.getOaOriginalId().startsWith(OpenOrgsConstants.OPENORGS_PREFIX))

					.collect(Collectors.toList());

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public Collection<Set<OrganizationSimpleView>> findWarningsById(final Authentication authentication) {

		final Map<String, Set<OrganizationSimpleView>> res = new TreeMap<>();

		final Map<String, String> roots = new HashMap<>();

		if (UserInfo.isSuperAdmin(authentication)) {

			for (final ConflictView w : conflictViewRepository.findAll()) {

				final String root = findWarningRoot(w, roots);

				if (!res.containsKey(root)) {

					res.put(root, new LinkedHashSet<OrganizationSimpleView>());

					res.get(root).add(new OrganizationSimpleView(w.getId1(), w.getName1(), w.getType1(), w.getCity1(), w.getCountry1(), new String[] {}));

				}

				res.get(root).add(new OrganizationSimpleView(w.getId2(), w.getName2(), w.getType2(), w.getCity2(), w.getCountry2(), new String[] {}));

			}

			return res.values();

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

	private String findWarningRoot(final ConflictView w, final Map<String, String> roots) {

		if (roots.containsKey(w.getId1())) {

			return roots.get(w.getId1());

		} else if (roots.containsKey(w.getId2())) {

			return roots.get(w.getId2());

		} else {

			// id1 is the new root

			roots.put(w.getId1(), w.getId1());

			roots.put(w.getId2(), w.getId1());

			return w.getId1();

		}

	}

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

	public List<OpenaireSimRel> saveEnrichments(@RequestBody final List<OpenaireSimRel> simrels, final Authentication authentication) {

		final boolean b = UserInfo.isSuperAdmin(authentication)

				|| simrels.stream()

						.map(OpenaireSimRel::getLocalId)

						.distinct()

						.allMatch(id -> organizationRepository.verifyAuthorizationForId(id, authentication.getName()));

		if (b) {

			return openaireSimRelRepository.saveAll(simrels);

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

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

	public Page<OrganizationSimpleView> search(@PathVariable final int page,

			@PathVariable final int size,

			@RequestParam final String q,

			final Authentication authentication) {

		return UserInfo.isSuperAdmin(authentication)

				? organizationSimpleViewRepository.findByNameContainingIgnoreCase(q, PageRequest.of(page, size))

				: organizationSimpleViewRepository.findByNameForUser(q, authentication.getName(), 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,

			final Authentication authentication) {

		if (UserInfo.isSuperAdmin(authentication) || organizationRepository.verifyAuthorizationForCountry(code, authentication.getName())) {

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

		} else {

			throw new RuntimeException("User not authorized");

		}

	}

	@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,

			final Authentication authentication) {

		return UserInfo.isSuperAdmin(authentication)

				? organizationSimpleViewRepository.findByType(type, PageRequest.of(page, size))

				: organizationSimpleViewRepository.findByTypeForUser(type, authentication.getName(), PageRequest.of(page, size));

	}

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

	public List<BrowseEntry> browseCountries(final Authentication authentication) {

		return UserInfo.isSuperAdmin(authentication)

				? organizationSimpleViewRepository.browseCountries()

				: organizationSimpleViewRepository.browseCountriesForUser(authentication.getName());

	}

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

	public List<BrowseEntry> browseOrganizationTypes(final Authentication authentication) {

		return UserInfo.isSuperAdmin(authentication)

				? organizationSimpleViewRepository.browseTypes()

				: organizationSimpleViewRepository.browseTypesForUser(authentication.getName());

	}

}