resource-registry/src/main/java/org/gcube/informationsystem/resourceregistry/instances/model/entities/FacetManagement.java

package org.gcube.informationsystem.resourceregistry.instances.model.entities;

import org.gcube.com.fasterxml.jackson.databind.JsonNode;
import org.gcube.informationsystem.base.reference.AccessType;
import org.gcube.informationsystem.model.reference.entities.Facet;
import org.gcube.informationsystem.resourceregistry.api.exceptions.AlreadyPresentException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.NotFoundException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.ResourceRegistryException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.contexts.ContextException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.entities.facet.FacetAlreadyPresentException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.entities.facet.FacetAvailableInAnotherContextException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.entities.facet.FacetNotFoundException;
import org.gcube.informationsystem.resourceregistry.api.exceptions.types.SchemaViolationException;
import org.gcube.informationsystem.resourceregistry.contexts.security.SecurityContext.PermissionMode;
import org.gcube.informationsystem.resourceregistry.instances.model.Operation;
import org.gcube.informationsystem.resourceregistry.instances.model.relations.ConsistsOfManagement;
import org.gcube.informationsystem.types.reference.entities.FacetType;

import com.arcadedb.graph.Edge;
import com.arcadedb.graph.Vertex;
import com.arcadedb.graph.Vertex.DIRECTION;
import com.arcadedb.remote.RemoteDatabase;

/**
 * @author Luca Frosini (ISTI - CNR)
 */
public class FacetManagement extends EntityManagement<Facet, FacetType> {
	
	protected ConsistsOfManagement consistsOfManagement;
	protected ResourceManagement resourceManagement;
	
	public FacetManagement() {
		super(AccessType.FACET);
	}
	
	@Override
	protected FacetNotFoundException getSpecificNotFoundException(NotFoundException e) {
		return new FacetNotFoundException(e.getMessage(), e.getCause());
	}
	
	@Override
	public FacetAvailableInAnotherContextException getSpecificAvailableInAnotherContextException(String message) {
		return new FacetAvailableInAnotherContextException(message);
	}
	
	@Override
	protected FacetAlreadyPresentException getSpecificAlreadyPresentException(String message) {
		return new FacetAlreadyPresentException(message);
	}
	
	
	@Override
	public JsonNode createCompleteJsonNode() throws ResourceRegistryException {
		return serializeSelfAsJsonNode();
	}
	
	@Override
	protected Vertex reallyCreate() throws FacetAlreadyPresentException, ResourceRegistryException {
		return createVertex();
	}
	
	@Override
	protected Vertex reallyUpdate() throws FacetNotFoundException, ResourceRegistryException {
		Vertex facet = getElement();
		facet = (Vertex) updateProperties(documentType, facet, jsonNode, ignoreKeys, ignoreStartWithKeys);
		return facet;
	}
	
	@Override
	protected void reallyRemoveFromContext()
			throws ContextException, ResourceRegistryException {
		if(entryPoint) {
			Edge edge = getElement().getEdges(DIRECTION.IN).iterator().next();
			consistsOfManagement = new ConsistsOfManagement();
			consistsOfManagement.setElement(edge);
			consistsOfManagement.setTargetEntityManagement(this);
			// Not needed consistsOfManagement.setSourceSecurityContext(sourceSecurityContext);
			consistsOfManagement.setTargetSecurityContext(targetSecurityContext);
			
			
			affectedInstances.put(uuid, consistsOfManagement.serializeAsAffectedInstance());
			
			Vertex vertex = getElement().getVertices(DIRECTION.IN).iterator().next();
			resourceManagement = new ResourceManagement();
			resourceManagement.setElement(vertex);
			// Not needed resourceManagement.setSourceSecurityContext(sourceSecurityContext);
			resourceManagement.setTargetSecurityContext(targetSecurityContext);
			resourceManagement.addToRelationManagements(consistsOfManagement);
		}
		super.reallyRemoveFromContext();
	}
	
	@Override
	protected void reallyDelete() throws FacetNotFoundException, ResourceRegistryException {
		if(entryPoint) {
			Edge edge = getElement().getEdges(DIRECTION.IN).iterator().next();
			consistsOfManagement = new ConsistsOfManagement();
			consistsOfManagement.setElement(edge);
			consistsOfManagement.setTargetEntityManagement(this);
			affectedInstances.put(uuid, consistsOfManagement.serializeAsAffectedInstance());
			
			Vertex vertex = getElement().getVertices(DIRECTION.IN).iterator().next();
			resourceManagement = new ResourceManagement();
			resourceManagement.setElement(vertex);
			resourceManagement.addToRelationManagements(consistsOfManagement);
		}
		
		affectedInstances.put(uuid, serializeAsAffectedInstance());
		getElement().delete();
	}
	
	protected void checkResource() throws SchemaViolationException, ResourceRegistryException {
		if(!entryPoint) {
			return;
		}
		
		if(operation.isSafe()) {
			/* 
			 * The sanity check is not required for a safe operation.
			 */
			return;
		}
		
		if(operation == Operation.UPDATE) {
			// an update to the Facet only modify the Facet properties, no need to check the source resource
			return;
		}
		RemoteDatabase targetSecurityContextDatabase = null;
		try {
			if(resourceManagement==null) {
				Vertex vertex = getElement().getVertices(DIRECTION.IN).iterator().next();
				ResourceManagement resourceManagement = new ResourceManagement();
				resourceManagement.setElement(vertex);
			}
			switch (operation) {
				case CREATE: case DELETE:
					resourceManagement.setWorkingContext(getWorkingContext());
					resourceManagement.setDatabase(database);
					break;
					
				case ADD_TO_CONTEXT:
					resourceManagement.setSourceSecurityContext(sourceSecurityContext);
					resourceManagement.setTargetSecurityContext(targetSecurityContext);
					resourceManagement.setWorkingContext(targetSecurityContext);
					targetSecurityContextDatabase  = targetSecurityContext.getRemoteDatabase(PermissionMode.READER);
					resourceManagement.setDatabase(targetSecurityContextDatabase);
					break;
				
				case REMOVE_FROM_CONTEXT:
					// Not needed resourceManagement.setSourceSecurityContext(sourceSecurityContext);
					resourceManagement.setTargetSecurityContext(targetSecurityContext);
					resourceManagement.setWorkingContext(targetSecurityContext);
					targetSecurityContextDatabase  = targetSecurityContext.getRemoteDatabase(PermissionMode.READER);
					resourceManagement.setDatabase(targetSecurityContextDatabase );
					break;
					
				default:
					// You should not be here
					return;
			}
			
			resourceManagement.setOperation(operation);
			
			resourceManagement.sanityCheck();
			
		}catch (ResourceRegistryException e) {
			throw e;
		}catch (Exception e) {
			throw new ResourceRegistryException(e);
		}finally {
			if(targetSecurityContextDatabase !=null) {
				targetSecurityContextDatabase.close();
//				database.activateOnCurrentThread();
			}
		}
	}
	
	@Override
	public void sanityCheck() throws SchemaViolationException, ResourceRegistryException {
		super.sanityCheck();
		checkResource();
	}
	
	@Override
	public String create() throws AlreadyPresentException, ResourceRegistryException {
		throw new SchemaViolationException("You cannot create a stand alone Facet");
	}
	
	public Vertex internalCreate() throws AlreadyPresentException, ResourceRegistryException {
		if(entryPoint && operation == Operation.CREATE) {
			throw new SchemaViolationException("You cannot create a stand alone Facet");
		}
		return super.internalCreate();
	}
	
}