package eu.eudat.models.dataset;

import eu.eudat.models.DataModel;
import eu.eudat.models.datasetprofile.DatasetProfileListingModel;
import eu.eudat.models.dmp.AssociatedProfiles;
import eu.eudat.models.dmp.DataManagementPlan;
import eu.eudat.utilities.builders.DomainModelConverter;

import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.UUID;
import java.util.stream.Collectors;

public class Dataset implements DataModel<eu.eudat.entities.Dataset>{
	private UUID id;
    private String label;
    private String reference;
    private String uri;
    private String description;
    private short status;
    private String properties;
    private Date created;
    private DataManagementPlan dmp;
    private DatasetProfileListingModel profile;
    private List<Registry> registries;
	private List<Service> services;
	private List<DataRepository> dataRepositories;

	public UUID getId() {
		return id;
	}
	public void setId(UUID id) {
		this.id = id;
	}
	public String getLabel() {
		return label;
	}
	public void setLabel(String label) {
		this.label = label;
	}
	public String getReference() {
		return reference;
	}
		
	public void setReference(String reference) {
		this.reference = reference;
	}
	public String getUri() {
		return uri;
	}
	public void setUri(String uri) {
		this.uri = uri;
	}
	public short getStatus() {
		return status;
	}
	public void setStatus(short status) {
		this.status = status;
	}
	public String getProperties() {
		return properties;
	}
	public void setProperties(String properties) {
		this.properties = properties;
	}
	public List<Registry> getRegistries() {
		return registries;
	}
	public void setRegistries(List<Registry> registries) {
		this.registries = registries;
	}
	public List<Service> getServices() {
		return services;
	}
	public void setServices(List<Service> services) {
		this.services = services;
	}
	public List<DataRepository> getDataRepositories() {
		return dataRepositories;
	}
	public void setDataRepositories(List<DataRepository> dataRepositories) {
		this.dataRepositories = dataRepositories;
	}

	public Date getCreated() {
		return created;
	}

	public void setCreated(Date created) {
		this.created = created;
	}

	public String getDescription() {
		return description;
	}
	public void setDescription(String description) {
		this.description = description;
	}

	public DataManagementPlan getDmp() {
		return dmp;
	}

	public void setDmp(DataManagementPlan dmp) {
		this.dmp = dmp;
	}

	public DatasetProfileListingModel getProfile() {
		return profile;
	}

	public void setProfile(DatasetProfileListingModel profile) {
		this.profile = profile;
	}

	public void fromDataModel(eu.eudat.entities.Dataset entity) throws IllegalAccessException, InstantiationException{
		this.id = entity.getId();
		this.label = entity.getLabel();
		this.properties = entity.getProperties();
		this.reference = entity.getReference();
		this.description = entity.getDescription();
		this.profile = new DatasetProfileListingModel();
		this.profile.fromDataModel(entity.getProfile());
		this.registries = new DomainModelConverter<eu.eudat.entities.Registry,Registry>().fromDataModel(entity.getRegistries().stream().collect(Collectors.toList()), Registry.class);
		this.dataRepositories = new DomainModelConverter<eu.eudat.entities.DataRepository,DataRepository>().fromDataModel(entity.getDataRepositories().stream().collect(Collectors.toList()), DataRepository.class);
		this.services = new DomainModelConverter<eu.eudat.entities.Service,Service>().fromDataModel(entity.getServices().stream().collect(Collectors.toList()), Service.class);
		this.created = entity.getCreated();
	}
	
	public eu.eudat.entities.Dataset toDataModel(){
		eu.eudat.entities.Dataset entity = new eu.eudat.entities.Dataset();
		entity.setId(this.id);
		entity.setLabel(this.label);
		entity.setReference(this.reference);
		entity.setUri(this.uri);
		entity.setProperties(this.properties);
		entity.setStatus(this.status);
		entity.setDmp(dmp.toDataModel());
		entity.setDescription(this.description);
		entity.setCreated(this.created!=null?this.created:new Date());
		entity.setModified(new Date());
		entity.setProfile(profile.toDataModel());  ///TODO
		if(!this.registries.isEmpty()){
			entity.setRegistries(new HashSet<eu.eudat.entities.Registry>());
			for(Registry registry:this.registries){
				entity.getRegistries().add(registry.toDataModel());
			}
		}
				
		if(!this.dataRepositories.isEmpty()){
			entity.setDataRepositories(new HashSet<eu.eudat.entities.DataRepository>());
			for(DataRepository dataRepository:this.dataRepositories){
				entity.getDataRepositories().add(dataRepository.toDataModel());
			}
		}
		
		if(!this.services.isEmpty()){
			entity.setServices(new HashSet<eu.eudat.entities.Service>());
			for(Service service:this.services){
				entity.getServices().add(service.toDataModel());
			}
		}
		return entity;
	}
	
}