argos/dmp-backend/src/main/java/rest/entities/DMPs.java

package rest.entities;


import java.util.*;
import java.util.stream.Collectors;

import javax.transaction.Transactional;

import models.criteria.DataRepositoryCriteria;
import models.criteria.OrganisationCriteria;
import models.criteria.ResearcherCriteria;
import models.criteria.ServiceCriteria;
import models.dmp.DataManagementPlan;
import models.dmp.DataManagementPlanTableRequest;
import models.helpers.DataTableData;
import models.project.ProjectTableRequest;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.web.bind.annotation.CrossOrigin;
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.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import dao.entities.DMPDao;
import dao.entities.DMPProfileDao;
import dao.entities.DataRepositoryDao;
import dao.entities.DatasetDao;
import dao.entities.DatasetProfileDao;
import dao.entities.DatasetProfileRulesetDao;
import dao.entities.DatasetProfileViewstyleDao;
import dao.entities.OrganisationDao;
import dao.entities.ProjectDao;
import dao.entities.RegistryDao;
import dao.entities.ResearcherDao;
import dao.entities.ServiceDao;
import dao.entities.UserInfoDao;
import entities.DMP;
import entities.Dataset;
import entities.Project;
import entities.UserInfo;
import entities.responses.IDLabelPair;
import helpers.SerializerProvider;
import managers.DataManagementPlanManager;
import managers.ProjectManager;
import utilities.builders.DomainModelConverter;


@RestController
@CrossOrigin
public class DMPs  {
	
	@Autowired private DataRepositoryDao dataRepositoryDao;
	@Autowired private DatasetDao datasetDao;
	@Autowired private DatasetProfileDao datasetProfileDao;
	@Autowired private DatasetProfileRulesetDao datasetProfileRulesetDao;
	@Autowired private DatasetProfileViewstyleDao datasetProfileViewstyleDao;
	@Autowired private DMPDao dMPDao;
	@Autowired private DMPProfileDao dMPProfileDao;
	@Autowired private OrganisationDao organisationDao;
	@Autowired private ProjectDao projectDao;
	@Autowired private RegistryDao registryDao;
	@Autowired private ResearcherDao researcherDao;
	@Autowired private ServiceDao serviceDao;
	@Autowired private UserInfoDao userInfoDao;
	

	
	// FETCH BY DMP(S)
	
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmps/getPaged" }, consumes = "application/json", produces="application/json")
	public @ResponseBody ResponseEntity<DataTableData<models.dmp.DataManagementPlan>> getPaged(@RequestBody DataManagementPlanTableRequest dataManagementPlanTableRequest) {
		try {
			DataTableData<models.dmp.DataManagementPlan> dataTable = new DataManagementPlanManager().getPaged(dMPDao, dataManagementPlanTableRequest);
			
			return ResponseEntity.status(HttpStatus.OK).body(dataTable);

		} catch (Exception ex) {			
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(null);
		}
	}
	
	@RequestMapping(method = RequestMethod.GET, value = { "/dmps/getSingle/{id}" }, produces="application/json")
	public @ResponseBody ResponseEntity<models.dmp.DataManagementPlan> getPaged(@PathVariable String id) {
		try {
			models.dmp.DataManagementPlan project = new DataManagementPlanManager().getSingle(dMPDao, id);
			return ResponseEntity.status(HttpStatus.OK).body(project);

		} catch (Exception ex) {			
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(null);
		}
	}
	
	@Transactional
	@RequestMapping(method = RequestMethod.POST, value = { "/dmps/add" }, consumes = "application/json", produces="application/json")
	public @ResponseBody ResponseEntity<entities.DMP> addDmp(@RequestBody models.dmp.DataManagementPlan dataManagementPlan) {
		entities.DMP createdProject = dMPDao.update(dataManagementPlan.toDataModel());
		return ResponseEntity.status(HttpStatus.CREATED).body(createdProject);
	}
	
	
	////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
	
	@RequestMapping(method = RequestMethod.GET, value = { "/dmps" }, produces="text/plain")
	public @ResponseBody ResponseEntity<Object> listDMPs(){
		try {
			List<UUID> allIDs = dMPDao.listAllIDs();
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(allIDs));
		}
		catch(Exception ex) {
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Serialization issue: "+ex.getMessage());
		}
	}

	@RequestMapping(method = RequestMethod.GET, value = { "/dmps/{id}" }, produces="application/json")
	public @ResponseBody ResponseEntity<Object> getDMP(@PathVariable("id") String id){
		try {
			DMP dmp = dMPDao.read(UUID.fromString(id));
			DataManagementPlan dataManagementPlan = new DataManagementPlan();
			dataManagementPlan.fromDataModel(dmp);
			return ResponseEntity.status(HttpStatus.OK).body(dataManagementPlan);
		}
		catch(Exception ex) {
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.BAD_REQUEST).body("Erroneous input: "+ex.getMessage());
		}
	}

	@RequestMapping(method = RequestMethod.GET, value = { "/dmp/listDMPLabelID" }, produces="text/plain")
	public @ResponseBody ResponseEntity<Object> listDmpLabelID(){
		try {
			List<IDLabelPair> allIDLabels = dMPDao.listAllIDsLabels();
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(allIDLabels));
		}
		catch(Exception ex) {
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Serialization issue: "+ex.getMessage());
		}
	}
	
	
	/**
	 * This should be called on extreme cases. It's computationally intensive
	 */
	@RequestMapping(method = RequestMethod.GET, value = { "/dmp/getAll" }, produces="application/json")
	public @ResponseBody ResponseEntity<Object> getAllDMPs(){
		
		try {
			List<DMP> allDMPs = dMPDao.getAll();
			return new ResponseEntity<Object>(SerializerProvider.toJson(allDMPs), HttpStatus.OK);	
		}
		catch(Exception ex) {
			ex.printStackTrace();
			return new ResponseEntity<>(null, HttpStatus.INTERNAL_SERVER_ERROR);
		}
	}
	
	
	@Transactional
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/create" }, consumes = "application/json", produces="application/json")
	public @ResponseBody ResponseEntity<Object> createDMP(@RequestBody DMP dmp) {
		try {
			DMP createdDmp = dMPDao.update(dmp);
			return ResponseEntity.status(HttpStatus.CREATED).body(SerializerProvider.toJson(createdDmp));
		} catch (Exception e) {
			e.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("{\"msg\":\"Could not create DMP!\"");
		}
		
	}
	
	@Transactional
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/update" }, consumes = "application/json", produces="application/json")
	public @ResponseBody ResponseEntity<Object> updateDMP(@RequestBody DataManagementPlan dataManagementPlan) {
		
		DMP newDmp = dataManagementPlan.toDataModel();
		if(newDmp.getOrganisations()!=null&&!newDmp.getOrganisations().isEmpty()){
			for(entities.Organisation organisation: newDmp.getOrganisations()){
				OrganisationCriteria criteria = new OrganisationCriteria();
				criteria.setLike(organisation.getReference());
				List<entities.Organisation> entries = this.organisationDao.listBy(criteria);
				if(entries!=null&&!entries.isEmpty())organisation.setId(entries.get(0).getId());
				else organisation = this.organisationDao.create(organisation);
			}
		}

		if(newDmp.getResearchers()!=null&&!newDmp.getResearchers().isEmpty()){
			for(entities.Researcher researcher : newDmp.getResearchers()){
				ResearcherCriteria criteria = new ResearcherCriteria();
				criteria.setLike(researcher.getReference());
				List<entities.Researcher> entries = this.researcherDao.listBy(criteria);
				if(entries!=null&&!entries.isEmpty())researcher.setId(entries.get(0).getId());
				else researcher = this.researcherDao.create(researcher);
			}
		}
		
		DMP previousDmp = dMPDao.read(dataManagementPlan.getId());
		previousDmp.setResearchers(newDmp.getResearchers());
		previousDmp.setOrganisations(newDmp.getOrganisations());
		previousDmp.setLabel(dataManagementPlan.getLabel());
		previousDmp.setVersion(dataManagementPlan.getVersion());
		previousDmp.setStatus((short)dataManagementPlan.getStatus());
		//if(!previousDmp.getProject().getId().equals(newDmp.getProject().getId()))previousDmp.setProject(projectDao.read(newDmp.getProject().getId()));
		
		try {
			DMP updatedDMP = dMPDao.update(previousDmp);
			return ResponseEntity.status(HttpStatus.CREATED).body(SerializerProvider.toJson(updatedDMP));
		} catch (Exception e) {
			e.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("{\"msg\":\"Could not update DMP!\"");
		}
		
	}
	
	
	
	
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/getdatasets" }, consumes = "application/json", produces="application/json")
	public @ResponseBody ResponseEntity<Object> getDatasetsOfDMP(@RequestBody DMP dmp) {
		try {
			List<Dataset> datasets = datasetDao.getDatasetsOfDmp(dmp.getId());
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(datasets));
		} catch (Exception e) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("{\"msg\":\"Could not get datasets of DMP!\"");
		}
		
	}
	
	
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/delete" }, consumes = "application/json", produces="text/plain")
	public @ResponseBody ResponseEntity<Object> delete(@RequestBody DMP dmp) {
		
		DMP d = new DMP();
		d.setId(dmp.getId());
		try {
			dMPDao.delete(d);
			return ResponseEntity.status(HttpStatus.CREATED).body("DELETED!");
		} catch (Exception e) {
			e.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("{\"msg\":\"Could not Delete DMP!\"");
		}
		
	}
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/softdelete" }, consumes = "application/json", produces="text/plain")
	public @ResponseBody ResponseEntity<Object> softDelete(@RequestBody DMP dmp) {
		
		try{
			DMP d = dMPDao.read(dmp.getId());
			d.setStatus(new Short("-1"));
			dMPDao.update(d);
			return ResponseEntity.status(HttpStatus.OK).body("{\"msg\":\"deleted DMP!\"");
		} catch (Exception e) {
			e.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("{\"msg\":\"Could not soft delete DMP!\"");
		}
		
	}
	
	
	
	
	////////////////////////////////
	//// USER - RELATED ACTIONS ////
	////////////////////////////////
	
	@RequestMapping(method = RequestMethod.GET, value = { "/dmp/getofuser" }, produces="text/plain")
	public @ResponseBody ResponseEntity<Object> getDmpsOfUser(){
		
		String userID = null;
		try {
			userID = SecurityContextHolder.getContext().getAuthentication().getPrincipal().toString();
		} catch(NullPointerException ex) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("You have not logged in. You shouldn't be here");
		}
		
		try {
			//List<DMP> nonDeleted = userInfoDao.getDmpsOfUser(userID);
			List<DMP> nonDeleted = dMPDao.getDMPsOfUser(userID);
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(nonDeleted));
		}
		catch(Exception ex) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Serialization issue: "+ex.getMessage());
		}
		
	}
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/createofuser" }, produces="text/plain", consumes = "application/json")
	public @ResponseBody ResponseEntity<Object> createDmpOfUser(@RequestBody DataManagementPlan dataManagementPlan){
		
		
		String userID = null;
		try {
			userID = SecurityContextHolder.getContext().getAuthentication().getPrincipal().toString();
		} catch(NullPointerException ex) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("You have not logged in. You shouldn't be here");
		}
		
		UserInfo userInfo = userInfoDao.read(UUID.fromString(userID));
		
		if(userInfo==null) //this should normally never happer
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("There's no such a user on the system. You shouldn't be here");
		
		try {
			DMP dmp = dataManagementPlan.toDataModel();

			if(dmp.getOrganisations()!=null&&!dmp.getOrganisations().isEmpty()){
				for(entities.Organisation organisation: dmp.getOrganisations()){
					OrganisationCriteria criteria = new OrganisationCriteria();
					criteria.setLike(organisation.getReference());
					List<entities.Organisation> entries = this.organisationDao.listBy(criteria);
					if(entries!=null&&!entries.isEmpty())organisation.setId(entries.get(0).getId());
					else organisation = this.organisationDao.create(organisation);
				}
			}

			if(dmp.getResearchers()!=null&&!dmp.getResearchers().isEmpty()){
				for(entities.Researcher researcher : dmp.getResearchers()){
					ResearcherCriteria criteria = new ResearcherCriteria();
					criteria.setLike(researcher.getReference());
					List<entities.Researcher> entries = this.researcherDao.listBy(criteria);
					if(entries!=null&&!entries.isEmpty())researcher.setId(entries.get(0).getId());
					else researcher = this.researcherDao.create(researcher);
				}
			}
			dmp.setId(null);
			
			dmp.setCreator(userInfo);
			dmp.setProject(this.projectDao.read(dataManagementPlan.getProject().getId()));
			Set<UserInfo> users = new HashSet<UserInfo>();
			users.add(userInfo);
			dmp.setUsers(users);
			
			dmp.setCreated(new Date());
			dmp.setModified(new Date());
			dmp.setStatus(new Short("0"));
			
			DMP newdmp = dMPDao.create(dmp);
			
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(newdmp));	
		}
		catch(Exception ex) {
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Serialization issue: "+ex.getMessage());
		}
		
	}
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/cloneforuser" }, produces="text/plain", consumes = "application/json")
	public @ResponseBody ResponseEntity<Object> cloneDmpOfUser(@RequestBody DMP dmp){
		
		String userID = null;
		try {
			userID = SecurityContextHolder.getContext().getAuthentication().getPrincipal().toString();
		} catch(NullPointerException ex) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("You have not logged in. You shouldn't be here");
		}
		
		UserInfo userInfo = userInfoDao.read(UUID.fromString(userID));
		
		if(userInfo==null) //this should normally never happer
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("There's no such a user on the system. You shouldn't be here");
		
		DMP clone = dMPDao.read(dmp.getId());
		
		
		try {
			
			Set<UserInfo> users = new HashSet<UserInfo>();
			users.add(userInfo);
			clone.setUsers(users);
			
			clone.setCreated(new Date());
			clone.setModified(new Date());
			clone.setStatus(new Short("0"));
			
			String cloneLabel = dmp.getLabel();
			if(cloneLabel==null || cloneLabel.isEmpty()) //if the provided label is null or empty, use parent's label + "_clone"
				cloneLabel = clone.getLabel()+"_clone";
			clone.setVersion(clone.getVersion()+1);
			clone.setLabel(cloneLabel);
			clone.setPrevious(clone.getId());
			
			
			clone.setId(null);
			
			clone = dMPDao.create(clone);
			
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(clone));	
		}
		catch(Exception ex) {
			ex.printStackTrace();
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Serialization issue: "+ex.getMessage());
		}
		
	}
	
	
	@RequestMapping(method = RequestMethod.POST, value = { "/dmp/adduser" }, produces="text/plain")
	public @ResponseBody ResponseEntity<Object> addUserToDmp(@RequestBody DMP dmp){
		
		
		String userID = null;
		try {
			userID = SecurityContextHolder.getContext().getAuthentication().getPrincipal().toString();
		} catch(NullPointerException ex) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("You have not logged in. You shouldn't be here");
		}
		
		final UserInfo userInfo = userInfoDao.read(UUID.fromString(userID));
		
		if(userInfo==null) //this should normally never happer
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("There's no such a user on the system. You shouldn't be here");
		
		if(dmp==null || dmp.getId()==null)
			return ResponseEntity.status(HttpStatus.BAD_REQUEST).body("DMP is null or dmp has null id");
		
		try {
			
			DMP existingDMP = dMPDao.read(dmp.getId());
			
			Set<UserInfo> users = existingDMP.getUsers().parallelStream().filter(user -> user.getId().toString() != userInfo.getId().toString()).collect(Collectors.toSet());
			
			users.add(userInfo);
			dmp.setUsers(users);
			
			DMP updateddmp = dMPDao.update(dmp);
			
			return ResponseEntity.status(HttpStatus.OK).body(SerializerProvider.toJson(updateddmp));	
		}
		catch(Exception ex) {
			return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body("Serialization issue: "+ex.getMessage());
		}
		
	}
	
	
	
	
	private static void addNullAndForeignElems(DMP existing, DMP newone) {
		
		newone.setModified(new Date());
		if(newone.getStatus()==null)
			newone.setStatus(existing.getStatus());
		if(newone.getCreated()==null)
			newone.setCreated(existing.getCreated());
		
		newone.setDataset(existing.getDataset());
		newone.setOrganisations(existing.getOrganisations());
		newone.setResearchers(existing.getResearchers());
		newone.setUsers(existing.getUsers());
	}
	
	
}