org.eclipse.om2m.core/src/main/java/org/eclipse/om2m/core/controller/AEAnncController.java - om2m/org.eclipse.om2m - Git at Google

 /*******************************************************************************
  * Copyright (c) 2014, 2016 Orange.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License 2.0
  * which accompanies this distribution, and is available at
  * https://www.eclipse.org/legal/epl-2.0/
  *******************************************************************************/
 package org.eclipse.om2m.core.controller;

 import java.math.BigInteger;
 import java.util.List;

 import org.eclipse.om2m.commons.constants.Constants;
 import org.eclipse.om2m.commons.constants.MimeMediaType;
 import org.eclipse.om2m.commons.constants.Operation;
 import org.eclipse.om2m.commons.constants.ResourceStatus;
 import org.eclipse.om2m.commons.constants.ResourceType;
 import org.eclipse.om2m.commons.constants.ResponseStatusCode;
 import org.eclipse.om2m.commons.constants.ResultContent;
 import org.eclipse.om2m.commons.constants.ShortName;
 import org.eclipse.om2m.commons.entities.AccessControlPolicyEntity;
 import org.eclipse.om2m.commons.entities.AeAnncEntity;
 import org.eclipse.om2m.commons.entities.DynamicAuthorizationConsultationEntity;
 import org.eclipse.om2m.commons.entities.RemoteCSEEntity;
 import org.eclipse.om2m.commons.entities.ResourceEntity;
 import org.eclipse.om2m.commons.entities.SubscriptionEntity;
 import org.eclipse.om2m.commons.exceptions.BadRequestException;
 import org.eclipse.om2m.commons.exceptions.ConflictException;
 import org.eclipse.om2m.commons.exceptions.NotPermittedAttrException;
 import org.eclipse.om2m.commons.exceptions.ResourceNotFoundException;
 import org.eclipse.om2m.commons.resource.AE;
 import org.eclipse.om2m.commons.resource.AEAnnc;
 import org.eclipse.om2m.commons.resource.RequestPrimitive;
 import org.eclipse.om2m.commons.resource.ResponsePrimitive;
 import org.eclipse.om2m.commons.utils.Util.DateUtil;
 import org.eclipse.om2m.core.datamapper.DataMapperSelector;
 import org.eclipse.om2m.core.entitymapper.EntityMapperFactory;
 import org.eclipse.om2m.core.notifier.Notifier;
 import org.eclipse.om2m.core.redirector.Redirector;
 import org.eclipse.om2m.core.router.Patterns;
 import org.eclipse.om2m.core.urimapper.UriMapper;
 import org.eclipse.om2m.core.util.ControllerUtil;
 import org.eclipse.om2m.core.util.ControllerUtil.UpdateUtil;
 import org.eclipse.om2m.persistence.service.DAO;

 public class AEAnncController extends Controller {

 	@Override
 	public ResponsePrimitive doCreate(RequestPrimitive request) {
 		/*
 		 * AE Creation procedure Req
 		 *
 		 * @resourceName NP resourceType NP resourceID NP parentID NP
 		 * accessControlPolicyIDs O creationTime NP expirationTime O
 		 * lastModifiedTime NP labels O announceTo O announcedAttribute O
 		 * appName O app-ID M ae-ID NP pointOfAccess O ontologyRef O nodeLink NP
 		 */

 		ResponsePrimitive response = new ResponsePrimitive(request);
 		Patterns patterns = new Patterns();

 		// Get the DAO of the parent
 		DAO<ResourceEntity> dao = (DAO<ResourceEntity>) patterns.getDAO(request.getTo(), dbs);
 		if (dao == null) {
 			throw new ResourceNotFoundException("Cannot find parent resource");
 		}

 		// Get the parent entity
 		ResourceEntity parentEntity = (ResourceEntity) dao.find(transaction, request.getTo());
 		// Check the parent existence
 		if (parentEntity == null) {
 			throw new ResourceNotFoundException("Cannot find parent resource");
 		}

 		// lock parent
 		transaction.lock(parentEntity);

 		// Get lists to change in the method corresponding to specific object
 		List<AccessControlPolicyEntity> acpsToCheck = null;
 		List<AeAnncEntity> childAnncs = null;
 		List<SubscriptionEntity> subs = null;

 		// Distinguish parents

 		// Case of remoteCSE
 		if (parentEntity.getResourceType().intValue() == (ResourceType.REMOTE_CSE)) {
 			RemoteCSEEntity csr = (RemoteCSEEntity) parentEntity;
 			acpsToCheck = csr.getAccessControlPolicies();
 			childAnncs = csr.getChildAeAnncs();
 			subs = csr.getSubscriptions();
 		}
 		// case of FlexContainer
 		if (parentEntity.getResourceType().intValue() == (ResourceType.FLEXCONTAINER_ANNC)) {
 //			FlexContainerEntity fce = (FlexContainerEntity) parentEntity;
 //			acpsToCheck = fce.getAccessControlPolicies();
 //			childAnncs = null;
 //			subs = fce.getSubscriptions();
 		}

 		// case of AEAnnc
 		if (parentEntity.getResourceType().intValue() == (ResourceType.AE_ANNC)) {
 			AeAnncEntity aeAnncEntity = (AeAnncEntity) parentEntity;
 			acpsToCheck = aeAnncEntity.getAccessControlPolicies();
 			childAnncs = null;
 			subs = aeAnncEntity.getSubscriptions();
 		}

 		// other cases ?

 		// Check access control policy of the originator
 //		checkACP(acpsToCheck, request.getFrom(), Operation.CREATE);
 		checkPermissions(request, parentEntity, acpsToCheck);


 		// Check if content is present
 		if (request.getContent() == null) {
 			throw new BadRequestException("A content is requiered for AEAnnc creation");
 		}

 		// Get the java object from the representation
 		AEAnnc aeAnnc = null;
 		try {
 			if (request.getRequestContentType().equals(MimeMediaType.OBJ)) {
 				aeAnnc = (AEAnnc) request.getContent();
 			} else {
 				aeAnnc = (AEAnnc) DataMapperSelector.getDataMapperList().get(request.getRequestContentType())
 						.stringToObj((String) request.getContent());
 			}
 		} catch (ClassCastException e) {
 			throw new BadRequestException("Incorrect resource representation in content", e);
 		}
 		if (aeAnnc == null) {
 			throw new BadRequestException("Error in provided content");
 		}

 		// Check attributes
 		// @resourceName NP
 		// resourceType NP
 		// resourceID NP
 		// parentID NP
 		// creationTime NP
 		// lastModifiedTime NP
 		// expirationTime O
 		// labels O
 		// announceTo O
 		// announcedAttribute O
 		// Creating the corresponding entity
 		AeAnncEntity aeAnncEntity = new AeAnncEntity();
 		ControllerUtil.CreateUtil.fillEntityFromGenericResource(aeAnnc, aeAnncEntity);

 		// ae-ID NP
 		if (aeAnnc.getAEID() != null) {
 			throw new NotPermittedAttrException("ae-id is Not Permitted");
 		}
 		// nodeLink NP
 		if (aeAnnc.getNodeLink() != null) {
 			aeAnncEntity.setNodeLink(aeAnnc.getNodeLink());
 		}

 		if (aeAnnc.getLink() == null) {
 			throw new BadRequestException("Link is Mandatory");
 		} else {
 			aeAnncEntity.setLink(aeAnnc.getLink());
 		}

 		// app-ID M
 		if (aeAnnc.getAppID() == null) {
 			throw new BadRequestException("App ID is Mandatory");
 		} else {
 			aeAnncEntity.setAppID(aeAnnc.getAppID());
 		}
 //
 //		// requestReachability M
 //		if (ae.getRequestReachability() == null) {
 //			throw new BadRequestException("Request Reachability is Mandatory");
 //		} else {
 //			aeEntity.setRequestReachability(ae.getRequestReachability());
 //		}

 		String generatedId = generateId();
 		aeAnncEntity.setAeid(ShortName.AE_ANNC + Constants.PREFIX_SEPERATOR + generatedId);
 		// Set other parameters
 		aeAnncEntity.setResourceID("/" + Constants.CSE_ID + "/" + aeAnncEntity.getAeid());
 		aeAnncEntity.setParentCsr((RemoteCSEEntity) parentEntity);

 		if (dbs.getDAOFactory().getAeAnncDAO().find(transaction, aeAnncEntity.getResourceID()) != null) {
 			throw new ConflictException("Already registered");
 		}

 		// accessControlPolicyIDs O
 		if (!aeAnnc.getAccessControlPolicyIDs().isEmpty()) {
 			aeAnncEntity.setAccessControlPolicies(
 					ControllerUtil.buildAcpEntityList(aeAnnc.getAccessControlPolicyIDs(), transaction));
 		} else {
 			aeAnncEntity.getAccessControlPolicies().addAll(acpsToCheck);
 		}

 		// dynamicAuthorizationConsultationIDs O
 		if (!aeAnnc.getDynamicAuthorizationConsultationIDs().isEmpty()) {
 			aeAnncEntity.setDynamicAuthorizationConsultations(
 					ControllerUtil.buildDacEntityList(aeAnnc.getDynamicAuthorizationConsultationIDs(), transaction));
 		}

 		// FIXME [0001] Creation of AE with an acpi provided
 		// } else {
 		// Create the acp corresponding to the AE_ID
 //		AccessControlPolicyEntity acpEntity = new AccessControlPolicyEntity();
 //		acpEntity.setCreationTime(DateUtil.now());
 //		acpEntity.setLastModifiedTime(DateUtil.now());
 //		acpEntity.setParentID("/" + Constants.CSE_ID);
 //		acpEntity.setResourceID(
 //				"/" + Constants.CSE_ID + "/" + ShortName.ACP + Constants.PREFIX_SEPERATOR + generateId());
 //		acpEntity.setName(ShortName.ACP + ShortName.AE + Constants.PREFIX_SEPERATOR + generatedId);
 //		AccessControlRuleEntity ruleEntity = new AccessControlRuleEntity();
 //		AccessControlOriginatorEntity originatorEntity = new AccessControlOriginatorEntity(
 //				Constants.ADMIN_REQUESTING_ENTITY);
 //		ruleEntity.getAccessControlOriginators().add(originatorEntity);
 //		ruleEntity.setCreate(true);
 //		ruleEntity.setRetrieve(true);
 //		ruleEntity.setUpdate(true);
 //		ruleEntity.setDelete(true);
 //		ruleEntity.setNotify(true);
 //		ruleEntity.setDiscovery(true);
 //		acpEntity.getSelfPrivileges().add(ruleEntity);
 //		// Privileges
 //		ruleEntity = new AccessControlRuleEntity();
 //		ruleEntity.setCreate(true);
 //		ruleEntity.setRetrieve(true);
 //		ruleEntity.setUpdate(true);
 //		ruleEntity.setDelete(true);
 //		ruleEntity.setNotify(true);
 //		ruleEntity.setDiscovery(true);
 //		ruleEntity.getAccessControlOriginators().add(new AccessControlOriginatorEntity(aeAnncEntity.getAeid()));
 //		ruleEntity.getAccessControlOriginators()
 //				.add(new AccessControlOriginatorEntity(Constants.ADMIN_REQUESTING_ENTITY));
 //		acpEntity.getPrivileges().add(ruleEntity);
 //		acpEntity.setHierarchicalURI("/" + Constants.CSE_ID + "/" + Constants.CSE_NAME + "/" + acpEntity.getName());
 //		// Add the acp in the UriMapper table
 //		UriMapper.addNewUri(acpEntity.getHierarchicalURI(), acpEntity.getResourceID(),
 //				ResourceType.ACCESS_CONTROL_POLICY);
 //		dbs.getDAOFactory().getAccessControlPolicyDAO().create(transaction, acpEntity);
 		// Retrieve the acp in the database to make the link with the CSEBase
 		// resource
 //		AccessControlPolicyEntity acpDB = dbs.getDAOFactory().getAccessControlPolicyDAO().find(transaction,
 //				acpEntity.getResourceID());


 //		CSEBaseEntity cseBase = dbs.getDAOFactory().getCSEBaseDAO().find(transaction, "/" + Constants.CSE_ID);
 //		cseBase.getChildAccessControlPolicies().add(acpDB);
 //		dbs.getDAOFactory().getCSEBaseDAO().update(transaction, cseBase);
 //		// adding new acp to the acp list
 //		aeAnncEntity.getAccessControlPolicies().add(acpDB);
 //		// direct link to the generated acp
 ////		aeAnncEntity.setGeneratedAcp(acpDB);
 //		// }

 		// appName O
 		if (aeAnnc.getAppName() != null) {
 			aeAnncEntity.setAppName(aeAnnc.getAppName());
 		}
 //		// pointOfAccess O
 //		if (!ae.getPointOfAccess().isEmpty()) {
 //			aeEntity.getPointOfAccess().addAll(ae.getPointOfAccess());
 //		}
 //		// ontologyRef O
 //		if (ae.getOntologyRef() != null) {
 //			aeEntity.setOntologyRef(ae.getOntologyRef());
 //		}


 		aeAnncEntity.setParentID(parentEntity.getResourceID());
 		aeAnncEntity.setResourceType(BigInteger.valueOf(ResourceType.AE_ANNC));
 		if (aeAnnc.getName() != null) {
 			if (!patterns.checkResourceName(aeAnnc.getName())) {
 				throw new BadRequestException("Name provided is incorrect. Must be:" + patterns.ID_STRING);
 			}
 			aeAnncEntity.setName(aeAnnc.getName());
 		} else {
 			aeAnncEntity.setName(ShortName.AE_ANNC + "_" + generatedId);
 		}
 		aeAnncEntity.setHierarchicalURI(parentEntity.getHierarchicalURI() + "/" + aeAnncEntity.getName());
 		if (!UriMapper.addNewUri(aeAnncEntity.getHierarchicalURI(), aeAnncEntity.getResourceID(), ResourceType.AE_ANNC)) {
 			throw new ConflictException("Name already present in the parent collection.");
 		}


 		// Create AE in database
 		dbs.getDAOFactory().getAeAnncDAO().create(transaction, aeAnncEntity);

 		// Get the managed object from db
 		AeAnncEntity aeAnncDB = dbs.getDAOFactory().getAeAnncDAO().find(transaction, aeAnncEntity.getResourceID());

 		// Add the AE to the parentEntity list
 		childAnncs.add(aeAnncDB);
 		dao.update(transaction, parentEntity);

 		// update link with aeAnncEntity - DacEntity
 		for(DynamicAuthorizationConsultationEntity dace : aeAnncDB.getDynamicAuthorizationConsultations()) {
 			DynamicAuthorizationConsultationEntity daceFromDB = dbs.getDAOFactory().getDynamicAuthorizationDAO().find(transaction, dace.getResourceID());
 			daceFromDB.getLinkedAeAnncEntities().add(aeAnncDB);
 			dbs.getDAOFactory().getDynamicAuthorizationDAO().update(transaction, daceFromDB);
 		}

 		// Commit the DB transaction & release lock
 		transaction.commit();

 		// Create the response
 		response.setResponseStatusCode(ResponseStatusCode.CREATED);
 		// Set the location of the resource
 		setLocationAndCreationContent(request, response, aeAnncDB);

 		Notifier.notify(subs, aeAnncDB, ResourceStatus.CHILD_CREATED);

 		return response;
 	}

 	@Override
 	public ResponsePrimitive doRetrieve(RequestPrimitive request) {
 		/*
 		 * Generic retrieve procedure
 		 */
 		ResponsePrimitive response = new ResponsePrimitive(request);

 		// Check existence of the resource
 		AeAnncEntity aeAnncEntity = dbs.getDAOFactory()
 				.getAeAnncDAO().find(transaction, request.getTo());
 		if (aeAnncEntity == null){
 			throw new ResourceNotFoundException("Resource not found");
 		}

 //		checkACP(aeAnncEntity.getAccessControlPolicies(), request.getFrom(),
 //				Operation.RETRIEVE);
 		checkPermissions(request, aeAnncEntity, aeAnncEntity.getAccessControlPolicies());

 		if (ResultContent.ORIGINAL_RES.equals(request.getResultContent())) {
 			RequestPrimitive originalResourceRequest = new RequestPrimitive();
 			originalResourceRequest.setOperation(Operation.RETRIEVE);
 			originalResourceRequest.setFrom(Constants.ADMIN_REQUESTING_ENTITY);
 			originalResourceRequest.setTo(aeAnncEntity.getLink());
 			originalResourceRequest.setReturnContentType(request.getReturnContentType());
 			return Redirector.retarget(originalResourceRequest );
 		} else {
 			// Create the object used to create the representation of the resource
 			AEAnnc aeAnnc = EntityMapperFactory.getAEAnncMapper().mapEntityToResource(aeAnncEntity, request);
 			response.setContent(aeAnnc);
 		}


 		response.setResponseStatusCode(ResponseStatusCode.OK);
 		return response;
 	}

 	@Override
 	public ResponsePrimitive doUpdate(RequestPrimitive request) {
 		/*
 		 * 							Req
 		 * @resourceName			NP
 		 * resourceType				NP
 		 * resourceID				NP
 		 * parentID					NP
 		 * accessControlPolicyIDs	O
 		 * creationTime				NP
 		 * expirationTime			O
 		 * lastModifiedTime			NP
 		 * labels					NP
 		 * announceTo				O
 		 * announcedAttribute		O
 		 * appName					O
 		 * app-ID					NP
 		 * ae-ID					NP
 		 * pointOfAccess			O
 		 * ontologyRef				O
 		 * nodeLink					NP
 		 * requestReachability		O
 		 */
 		ResponsePrimitive response = new ResponsePrimitive(request);

 		// Retrieve the resource from database
 		AeAnncEntity aeAnncEntity = dbs.getDAOFactory()
 				.getAeAnncDAO().find(transaction, request.getTo());
 		if (aeAnncEntity == null){
 			throw new ResourceNotFoundException("Resource not found");
 		}

 		// lock current entity
 		transaction.lock(aeAnncEntity);

 //		checkACP(aeAnncEntity.getAccessControlPolicies(), request.getFrom(),
 //				Operation.UPDATE);
 		checkPermissions(request, aeAnncEntity, aeAnncEntity.getAccessControlPolicies());


 		// Check if content is present
 		if (request.getContent() == null){
 			throw new BadRequestException("A content is requiered for AE update");
 		}

 		// Create the java object from the resource representation
 		AEAnnc aeAnnc = null;
 		try{
 			if (request.getRequestContentType().equals(MimeMediaType.OBJ)){
 				aeAnnc = (AEAnnc) request.getContent();
 			} else {
 				aeAnnc = (AEAnnc)DataMapperSelector.getDataMapperList()
 						.get(request.getRequestContentType()).stringToObj((String)request.getContent());
 			}
 		} catch (ClassCastException e){
 			throw new BadRequestException("Incorrect resource representation in content", e);
 		}
 		if (aeAnnc == null){
 			throw new BadRequestException("Error in provided content");
 		}

 		// Check attributes

 		// NP Attributes are ignored
 		// @resourceName			NP
 		// resourceType				NP
 		// resourceID				NP
 		// parentID					NP
 		// creationTime				NP
 		// lastModifiedTime			NP
 		UpdateUtil.checkNotPermittedParameters(aeAnnc);
 		// app-ID					NP
 		if(aeAnnc.getAppID() != null){
 			throw new BadRequestException("AppID is NP");
 		}
 		// ae-ID					NP
 		if(aeAnnc.getAEID() != null){
 			throw new BadRequestException("AE ID is NP");
 		}
 		// nodeLink					NP
 		if(aeAnnc.getNodeLink() != null){
 			throw new BadRequestException("NodeLink is NP");
 		}

 		AE modifiedAttributes = new AE();
 		// labels					O
 		if(!aeAnnc.getLabels().isEmpty()){
 			aeAnncEntity.setLabelsEntitiesFromSring(aeAnnc.getLabels());
 			modifiedAttributes.getLabels().addAll(aeAnnc.getLabels());
 		}
 		// accessControlPolicyIDs	O
 		if(!aeAnnc.getAccessControlPolicyIDs().isEmpty()){
 			for(AccessControlPolicyEntity acpe : aeAnncEntity.getAccessControlPolicies()){
 				checkSelfACP(acpe, request.getFrom(), Operation.UPDATE);
 			}
 			aeAnncEntity.getAccessControlPolicies().clear();
 			aeAnncEntity.setAccessControlPolicies(ControllerUtil.buildAcpEntityList(aeAnnc.getAccessControlPolicyIDs(), transaction));
 			modifiedAttributes.getAccessControlPolicyIDs().addAll(aeAnnc.getAccessControlPolicyIDs());
 		}
 		// dynamicAuthorizationConsultationIDs O
 		if (!aeAnnc.getDynamicAuthorizationConsultationIDs().isEmpty()) {
 			aeAnncEntity.setDynamicAuthorizationConsultations(
 					ControllerUtil.buildDacEntityList(aeAnnc.getDynamicAuthorizationConsultationIDs(), transaction));

 			// update link with aeAnncEntity - DacEntity
 			for(DynamicAuthorizationConsultationEntity dace : aeAnncEntity.getDynamicAuthorizationConsultations()) {
 				DynamicAuthorizationConsultationEntity daceFromDB = dbs.getDAOFactory().getDynamicAuthorizationDAO().find(transaction, dace.getResourceID());
 				daceFromDB.getLinkedAeAnncEntities().add(aeAnncEntity);
 				dbs.getDAOFactory().getDynamicAuthorizationDAO().update(transaction, daceFromDB);
 			}
 		}
 		// expirationTime			O
 		if (aeAnnc.getExpirationTime() != null){
 			aeAnncEntity.setExpirationTime(aeAnnc.getExpirationTime());
 			modifiedAttributes.setExpirationTime(aeAnnc.getExpirationTime());
 		}
 		// appName					O
 		if(aeAnnc.getAppName() != null){
 			aeAnncEntity.setAppName(aeAnnc.getAppName());
 			modifiedAttributes.setAppName(aeAnnc.getAppName());
 		}
 		// pointOfAccess			O
 		if(!aeAnnc.getPointOfAccess().isEmpty()){
 			aeAnncEntity.getPointOfAccess().clear();
 			aeAnncEntity.getPointOfAccess().addAll(aeAnnc.getPointOfAccess());
 			modifiedAttributes.getPointOfAccess().addAll(aeAnnc.getPointOfAccess());
 		}
 		// ontologyRef				O
 		if (aeAnnc.getOntologyRef() != null){
 			aeAnncEntity.setOntologyRef(aeAnnc.getOntologyRef());
 			modifiedAttributes.setOntologyRef(aeAnnc.getOntologyRef());
 		}

 		// Last Time Modified update
 		aeAnncEntity.setLastModifiedTime(DateUtil.now());
 		modifiedAttributes.setLastModifiedTime(aeAnncEntity.getLastModifiedTime());
 		response.setContent(modifiedAttributes);
 		// Update the resource in database
 		dbs.getDAOFactory().getAeAnncDAO().update(transaction, aeAnncEntity);

 		// commit transaction & release lock
 		transaction.commit();

 		Notifier.notify(aeAnncEntity.getSubscriptions(), aeAnncEntity, ResourceStatus.UPDATED);

 		response.setResponseStatusCode(ResponseStatusCode.UPDATED);
 		return response;
 	}

 	@Override
 	public ResponsePrimitive doDelete(RequestPrimitive request) {
 		/*
 		 * Generic delete procedure
 		 */
 		ResponsePrimitive response = new ResponsePrimitive(request);

 		// Retrieve the resource from database
 		AeAnncEntity aeAnncEntity = dbs.getDAOFactory()
 				.getAeAnncDAO().find(transaction, request.getTo());
 		if (aeAnncEntity == null){
 			throw new ResourceNotFoundException("Resource not found");
 		}

 		// lock entity
 		transaction.lock(aeAnncEntity);

 //		checkACP(aeAnncEntity.getAccessControlPolicies(), request.getFrom(),
 //				Operation.DELETE);
 		checkPermissions(request, aeAnncEntity, aeAnncEntity.getAccessControlPolicies());

 		UriMapper.deleteUri(aeAnncEntity.getHierarchicalURI());

 		Notifier.notifyDeletion(aeAnncEntity.getSubscriptions(), aeAnncEntity);

 		// Delete the resource
 		dbs.getDAOFactory().getAeAnncDAO().delete(transaction, aeAnncEntity);

 		// Commit the transaction & unlock
 		transaction.commit();

 		// Return rsc
 		response.setResponseStatusCode(ResponseStatusCode.DELETED);
 		return response;
 	}

 }
	/*******************************************************************************
	* Copyright (c) 2014, 2016 Orange.
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License 2.0
	* which accompanies this distribution, and is available at
	* https://www.eclipse.org/legal/epl-2.0/
	*******************************************************************************/
	package org.eclipse.om2m.core.controller;

	import java.math.BigInteger;
	import java.util.List;

	import org.eclipse.om2m.commons.constants.Constants;
	import org.eclipse.om2m.commons.constants.MimeMediaType;
	import org.eclipse.om2m.commons.constants.Operation;
	import org.eclipse.om2m.commons.constants.ResourceStatus;
	import org.eclipse.om2m.commons.constants.ResourceType;
	import org.eclipse.om2m.commons.constants.ResponseStatusCode;
	import org.eclipse.om2m.commons.constants.ResultContent;
	import org.eclipse.om2m.commons.constants.ShortName;
	import org.eclipse.om2m.commons.entities.AccessControlPolicyEntity;
	import org.eclipse.om2m.commons.entities.AeAnncEntity;
	import org.eclipse.om2m.commons.entities.DynamicAuthorizationConsultationEntity;
	import org.eclipse.om2m.commons.entities.RemoteCSEEntity;
	import org.eclipse.om2m.commons.entities.ResourceEntity;
	import org.eclipse.om2m.commons.entities.SubscriptionEntity;
	import org.eclipse.om2m.commons.exceptions.BadRequestException;
	import org.eclipse.om2m.commons.exceptions.ConflictException;
	import org.eclipse.om2m.commons.exceptions.NotPermittedAttrException;
	import org.eclipse.om2m.commons.exceptions.ResourceNotFoundException;
	import org.eclipse.om2m.commons.resource.AE;
	import org.eclipse.om2m.commons.resource.AEAnnc;
	import org.eclipse.om2m.commons.resource.RequestPrimitive;
	import org.eclipse.om2m.commons.resource.ResponsePrimitive;
	import org.eclipse.om2m.commons.utils.Util.DateUtil;
	import org.eclipse.om2m.core.datamapper.DataMapperSelector;
	import org.eclipse.om2m.core.entitymapper.EntityMapperFactory;
	import org.eclipse.om2m.core.notifier.Notifier;
	import org.eclipse.om2m.core.redirector.Redirector;
	import org.eclipse.om2m.core.router.Patterns;
	import org.eclipse.om2m.core.urimapper.UriMapper;
	import org.eclipse.om2m.core.util.ControllerUtil;
	import org.eclipse.om2m.core.util.ControllerUtil.UpdateUtil;
	import org.eclipse.om2m.persistence.service.DAO;

	public class AEAnncController extends Controller {

	@Override
	public ResponsePrimitive doCreate(RequestPrimitive request) {
	/*
	* AE Creation procedure Req
	*
	* @resourceName NP resourceType NP resourceID NP parentID NP
	* accessControlPolicyIDs O creationTime NP expirationTime O
	* lastModifiedTime NP labels O announceTo O announcedAttribute O
	* appName O app-ID M ae-ID NP pointOfAccess O ontologyRef O nodeLink NP
	*/

	ResponsePrimitive response = new ResponsePrimitive(request);
	Patterns patterns = new Patterns();

	// Get the DAO of the parent
	DAO<ResourceEntity> dao = (DAO<ResourceEntity>) patterns.getDAO(request.getTo(), dbs);
	if (dao == null) {
	throw new ResourceNotFoundException("Cannot find parent resource");
	}

	// Get the parent entity
	ResourceEntity parentEntity = (ResourceEntity) dao.find(transaction, request.getTo());
	// Check the parent existence
	if (parentEntity == null) {
	throw new ResourceNotFoundException("Cannot find parent resource");
	}

	// lock parent
	transaction.lock(parentEntity);

	// Get lists to change in the method corresponding to specific object
	List<AccessControlPolicyEntity> acpsToCheck = null;
	List<AeAnncEntity> childAnncs = null;
	List<SubscriptionEntity> subs = null;

	// Distinguish parents

	// Case of remoteCSE
	if (parentEntity.getResourceType().intValue() == (ResourceType.REMOTE_CSE)) {
	RemoteCSEEntity csr = (RemoteCSEEntity) parentEntity;
	acpsToCheck = csr.getAccessControlPolicies();
	childAnncs = csr.getChildAeAnncs();
	subs = csr.getSubscriptions();
	}
	// case of FlexContainer
	if (parentEntity.getResourceType().intValue() == (ResourceType.FLEXCONTAINER_ANNC)) {
	// FlexContainerEntity fce = (FlexContainerEntity) parentEntity;
	// acpsToCheck = fce.getAccessControlPolicies();
	// childAnncs = null;
	// subs = fce.getSubscriptions();
	}

	// case of AEAnnc
	if (parentEntity.getResourceType().intValue() == (ResourceType.AE_ANNC)) {
	AeAnncEntity aeAnncEntity = (AeAnncEntity) parentEntity;
	acpsToCheck = aeAnncEntity.getAccessControlPolicies();
	childAnncs = null;
	subs = aeAnncEntity.getSubscriptions();
	}

	// other cases ?

	// Check access control policy of the originator
	// checkACP(acpsToCheck, request.getFrom(), Operation.CREATE);
	checkPermissions(request, parentEntity, acpsToCheck);


	// Check if content is present
	if (request.getContent() == null) {
	throw new BadRequestException("A content is requiered for AEAnnc creation");
	}

	// Get the java object from the representation
	AEAnnc aeAnnc = null;
	try {
	if (request.getRequestContentType().equals(MimeMediaType.OBJ)) {
	aeAnnc = (AEAnnc) request.getContent();
	} else {
	aeAnnc = (AEAnnc) DataMapperSelector.getDataMapperList().get(request.getRequestContentType())
	.stringToObj((String) request.getContent());
	}
	} catch (ClassCastException e) {
	throw new BadRequestException("Incorrect resource representation in content", e);
	}
	if (aeAnnc == null) {
	throw new BadRequestException("Error in provided content");
	}

	// Check attributes
	// @resourceName NP
	// resourceType NP
	// resourceID NP
	// parentID NP
	// creationTime NP
	// lastModifiedTime NP
	// expirationTime O
	// labels O
	// announceTo O
	// announcedAttribute O
	// Creating the corresponding entity
	AeAnncEntity aeAnncEntity = new AeAnncEntity();
	ControllerUtil.CreateUtil.fillEntityFromGenericResource(aeAnnc, aeAnncEntity);

	// ae-ID NP
	if (aeAnnc.getAEID() != null) {
	throw new NotPermittedAttrException("ae-id is Not Permitted");
	}
	// nodeLink NP
	if (aeAnnc.getNodeLink() != null) {
	aeAnncEntity.setNodeLink(aeAnnc.getNodeLink());
	}

	if (aeAnnc.getLink() == null) {
	throw new BadRequestException("Link is Mandatory");
	} else {
	aeAnncEntity.setLink(aeAnnc.getLink());
	}

	// app-ID M
	if (aeAnnc.getAppID() == null) {
	throw new BadRequestException("App ID is Mandatory");
	} else {
	aeAnncEntity.setAppID(aeAnnc.getAppID());
	}
	//
	// // requestReachability M
	// if (ae.getRequestReachability() == null) {
	// throw new BadRequestException("Request Reachability is Mandatory");
	// } else {
	// aeEntity.setRequestReachability(ae.getRequestReachability());
	// }

	String generatedId = generateId();
	aeAnncEntity.setAeid(ShortName.AE_ANNC + Constants.PREFIX_SEPERATOR + generatedId);
	// Set other parameters
	aeAnncEntity.setResourceID("/" + Constants.CSE_ID + "/" + aeAnncEntity.getAeid());
	aeAnncEntity.setParentCsr((RemoteCSEEntity) parentEntity);

	if (dbs.getDAOFactory().getAeAnncDAO().find(transaction, aeAnncEntity.getResourceID()) != null) {
	throw new ConflictException("Already registered");
	}

	// accessControlPolicyIDs O
	if (!aeAnnc.getAccessControlPolicyIDs().isEmpty()) {
	aeAnncEntity.setAccessControlPolicies(
	ControllerUtil.buildAcpEntityList(aeAnnc.getAccessControlPolicyIDs(), transaction));
	} else {
	aeAnncEntity.getAccessControlPolicies().addAll(acpsToCheck);
	}

	// dynamicAuthorizationConsultationIDs O
	if (!aeAnnc.getDynamicAuthorizationConsultationIDs().isEmpty()) {
	aeAnncEntity.setDynamicAuthorizationConsultations(
	ControllerUtil.buildDacEntityList(aeAnnc.getDynamicAuthorizationConsultationIDs(), transaction));
	}

	// FIXME [0001] Creation of AE with an acpi provided
	// } else {
	// Create the acp corresponding to the AE_ID
	// AccessControlPolicyEntity acpEntity = new AccessControlPolicyEntity();
	// acpEntity.setCreationTime(DateUtil.now());
	// acpEntity.setLastModifiedTime(DateUtil.now());
	// acpEntity.setParentID("/" + Constants.CSE_ID);
	// acpEntity.setResourceID(
	// "/" + Constants.CSE_ID + "/" + ShortName.ACP + Constants.PREFIX_SEPERATOR + generateId());
	// acpEntity.setName(ShortName.ACP + ShortName.AE + Constants.PREFIX_SEPERATOR + generatedId);
	// AccessControlRuleEntity ruleEntity = new AccessControlRuleEntity();
	// AccessControlOriginatorEntity originatorEntity = new AccessControlOriginatorEntity(
	// Constants.ADMIN_REQUESTING_ENTITY);
	// ruleEntity.getAccessControlOriginators().add(originatorEntity);
	// ruleEntity.setCreate(true);
	// ruleEntity.setRetrieve(true);
	// ruleEntity.setUpdate(true);
	// ruleEntity.setDelete(true);
	// ruleEntity.setNotify(true);
	// ruleEntity.setDiscovery(true);
	// acpEntity.getSelfPrivileges().add(ruleEntity);
	// // Privileges
	// ruleEntity = new AccessControlRuleEntity();
	// ruleEntity.setCreate(true);
	// ruleEntity.setRetrieve(true);
	// ruleEntity.setUpdate(true);
	// ruleEntity.setDelete(true);
	// ruleEntity.setNotify(true);
	// ruleEntity.setDiscovery(true);
	// ruleEntity.getAccessControlOriginators().add(new AccessControlOriginatorEntity(aeAnncEntity.getAeid()));
	// ruleEntity.getAccessControlOriginators()
	// .add(new AccessControlOriginatorEntity(Constants.ADMIN_REQUESTING_ENTITY));
	// acpEntity.getPrivileges().add(ruleEntity);
	// acpEntity.setHierarchicalURI("/" + Constants.CSE_ID + "/" + Constants.CSE_NAME + "/" + acpEntity.getName());
	// // Add the acp in the UriMapper table
	// UriMapper.addNewUri(acpEntity.getHierarchicalURI(), acpEntity.getResourceID(),
	// ResourceType.ACCESS_CONTROL_POLICY);
	// dbs.getDAOFactory().getAccessControlPolicyDAO().create(transaction, acpEntity);
	// Retrieve the acp in the database to make the link with the CSEBase
	// resource
	// AccessControlPolicyEntity acpDB = dbs.getDAOFactory().getAccessControlPolicyDAO().find(transaction,
	// acpEntity.getResourceID());



	// CSEBaseEntity cseBase = dbs.getDAOFactory().getCSEBaseDAO().find(transaction, "/" + Constants.CSE_ID);
	// cseBase.getChildAccessControlPolicies().add(acpDB);
	// dbs.getDAOFactory().getCSEBaseDAO().update(transaction, cseBase);
	// // adding new acp to the acp list
	// aeAnncEntity.getAccessControlPolicies().add(acpDB);
	// // direct link to the generated acp
	//// aeAnncEntity.setGeneratedAcp(acpDB);
	// // }

	// appName O
	if (aeAnnc.getAppName() != null) {
	aeAnncEntity.setAppName(aeAnnc.getAppName());
	}
	// // pointOfAccess O
	// if (!ae.getPointOfAccess().isEmpty()) {
	// aeEntity.getPointOfAccess().addAll(ae.getPointOfAccess());
	// }
	// // ontologyRef O
	// if (ae.getOntologyRef() != null) {
	// aeEntity.setOntologyRef(ae.getOntologyRef());
	// }



	aeAnncEntity.setParentID(parentEntity.getResourceID());
	aeAnncEntity.setResourceType(BigInteger.valueOf(ResourceType.AE_ANNC));
	if (aeAnnc.getName() != null) {
	if (!patterns.checkResourceName(aeAnnc.getName())) {
	throw new BadRequestException("Name provided is incorrect. Must be:" + patterns.ID_STRING);
	}
	aeAnncEntity.setName(aeAnnc.getName());
	} else {
	aeAnncEntity.setName(ShortName.AE_ANNC + "_" + generatedId);
	}
	aeAnncEntity.setHierarchicalURI(parentEntity.getHierarchicalURI() + "/" + aeAnncEntity.getName());
	if (!UriMapper.addNewUri(aeAnncEntity.getHierarchicalURI(), aeAnncEntity.getResourceID(), ResourceType.AE_ANNC)) {
	throw new ConflictException("Name already present in the parent collection.");
	}


	// Create AE in database
	dbs.getDAOFactory().getAeAnncDAO().create(transaction, aeAnncEntity);

	// Get the managed object from db
	AeAnncEntity aeAnncDB = dbs.getDAOFactory().getAeAnncDAO().find(transaction, aeAnncEntity.getResourceID());

	// Add the AE to the parentEntity list
	childAnncs.add(aeAnncDB);
	dao.update(transaction, parentEntity);

	// update link with aeAnncEntity - DacEntity
	for(DynamicAuthorizationConsultationEntity dace : aeAnncDB.getDynamicAuthorizationConsultations()) {
	DynamicAuthorizationConsultationEntity daceFromDB = dbs.getDAOFactory().getDynamicAuthorizationDAO().find(transaction, dace.getResourceID());
	daceFromDB.getLinkedAeAnncEntities().add(aeAnncDB);
	dbs.getDAOFactory().getDynamicAuthorizationDAO().update(transaction, daceFromDB);
	}

	// Commit the DB transaction & release lock
	transaction.commit();

	// Create the response
	response.setResponseStatusCode(ResponseStatusCode.CREATED);
	// Set the location of the resource
	setLocationAndCreationContent(request, response, aeAnncDB);

	Notifier.notify(subs, aeAnncDB, ResourceStatus.CHILD_CREATED);

	return response;
	}

	@Override
	public ResponsePrimitive doRetrieve(RequestPrimitive request) {
	/*
	* Generic retrieve procedure
	*/
	ResponsePrimitive response = new ResponsePrimitive(request);

	// Check existence of the resource
	AeAnncEntity aeAnncEntity = dbs.getDAOFactory()
	.getAeAnncDAO().find(transaction, request.getTo());
	if (aeAnncEntity == null){
	throw new ResourceNotFoundException("Resource not found");
	}

	// checkACP(aeAnncEntity.getAccessControlPolicies(), request.getFrom(),
	// Operation.RETRIEVE);
	checkPermissions(request, aeAnncEntity, aeAnncEntity.getAccessControlPolicies());

	if (ResultContent.ORIGINAL_RES.equals(request.getResultContent())) {
	RequestPrimitive originalResourceRequest = new RequestPrimitive();
	originalResourceRequest.setOperation(Operation.RETRIEVE);
	originalResourceRequest.setFrom(Constants.ADMIN_REQUESTING_ENTITY);
	originalResourceRequest.setTo(aeAnncEntity.getLink());
	originalResourceRequest.setReturnContentType(request.getReturnContentType());
	return Redirector.retarget(originalResourceRequest );
	} else {
	// Create the object used to create the representation of the resource
	AEAnnc aeAnnc = EntityMapperFactory.getAEAnncMapper().mapEntityToResource(aeAnncEntity, request);
	response.setContent(aeAnnc);
	}



	response.setResponseStatusCode(ResponseStatusCode.OK);
	return response;
	}

	@Override
	public ResponsePrimitive doUpdate(RequestPrimitive request) {
	/*
	* Req
	* @resourceName NP
	* resourceType NP
	* resourceID NP
	* parentID NP
	* accessControlPolicyIDs O
	* creationTime NP
	* expirationTime O
	* lastModifiedTime NP
	* labels NP
	* announceTo O
	* announcedAttribute O
	* appName O
	* app-ID NP
	* ae-ID NP
	* pointOfAccess O
	* ontologyRef O
	* nodeLink NP
	* requestReachability O
	*/
	ResponsePrimitive response = new ResponsePrimitive(request);

	// Retrieve the resource from database
	AeAnncEntity aeAnncEntity = dbs.getDAOFactory()
	.getAeAnncDAO().find(transaction, request.getTo());
	if (aeAnncEntity == null){
	throw new ResourceNotFoundException("Resource not found");
	}

	// lock current entity
	transaction.lock(aeAnncEntity);

	// checkACP(aeAnncEntity.getAccessControlPolicies(), request.getFrom(),
	// Operation.UPDATE);
	checkPermissions(request, aeAnncEntity, aeAnncEntity.getAccessControlPolicies());


	// Check if content is present
	if (request.getContent() == null){
	throw new BadRequestException("A content is requiered for AE update");
	}

	// Create the java object from the resource representation
	AEAnnc aeAnnc = null;
	try{
	if (request.getRequestContentType().equals(MimeMediaType.OBJ)){
	aeAnnc = (AEAnnc) request.getContent();
	} else {
	aeAnnc = (AEAnnc)DataMapperSelector.getDataMapperList()
	.get(request.getRequestContentType()).stringToObj((String)request.getContent());
	}
	} catch (ClassCastException e){
	throw new BadRequestException("Incorrect resource representation in content", e);
	}
	if (aeAnnc == null){
	throw new BadRequestException("Error in provided content");
	}

	// Check attributes

	// NP Attributes are ignored
	// @resourceName NP
	// resourceType NP
	// resourceID NP
	// parentID NP
	// creationTime NP
	// lastModifiedTime NP
	UpdateUtil.checkNotPermittedParameters(aeAnnc);
	// app-ID NP
	if(aeAnnc.getAppID() != null){
	throw new BadRequestException("AppID is NP");
	}
	// ae-ID NP
	if(aeAnnc.getAEID() != null){
	throw new BadRequestException("AE ID is NP");
	}
	// nodeLink NP
	if(aeAnnc.getNodeLink() != null){
	throw new BadRequestException("NodeLink is NP");
	}

	AE modifiedAttributes = new AE();
	// labels O
	if(!aeAnnc.getLabels().isEmpty()){
	aeAnncEntity.setLabelsEntitiesFromSring(aeAnnc.getLabels());
	modifiedAttributes.getLabels().addAll(aeAnnc.getLabels());
	}
	// accessControlPolicyIDs O
	if(!aeAnnc.getAccessControlPolicyIDs().isEmpty()){
	for(AccessControlPolicyEntity acpe : aeAnncEntity.getAccessControlPolicies()){
	checkSelfACP(acpe, request.getFrom(), Operation.UPDATE);
	}
	aeAnncEntity.getAccessControlPolicies().clear();
	aeAnncEntity.setAccessControlPolicies(ControllerUtil.buildAcpEntityList(aeAnnc.getAccessControlPolicyIDs(), transaction));
	modifiedAttributes.getAccessControlPolicyIDs().addAll(aeAnnc.getAccessControlPolicyIDs());
	}
	// dynamicAuthorizationConsultationIDs O
	if (!aeAnnc.getDynamicAuthorizationConsultationIDs().isEmpty()) {
	aeAnncEntity.setDynamicAuthorizationConsultations(
	ControllerUtil.buildDacEntityList(aeAnnc.getDynamicAuthorizationConsultationIDs(), transaction));

	// update link with aeAnncEntity - DacEntity
	for(DynamicAuthorizationConsultationEntity dace : aeAnncEntity.getDynamicAuthorizationConsultations()) {
	DynamicAuthorizationConsultationEntity daceFromDB = dbs.getDAOFactory().getDynamicAuthorizationDAO().find(transaction, dace.getResourceID());
	daceFromDB.getLinkedAeAnncEntities().add(aeAnncEntity);
	dbs.getDAOFactory().getDynamicAuthorizationDAO().update(transaction, daceFromDB);
	}
	}
	// expirationTime O
	if (aeAnnc.getExpirationTime() != null){
	aeAnncEntity.setExpirationTime(aeAnnc.getExpirationTime());
	modifiedAttributes.setExpirationTime(aeAnnc.getExpirationTime());
	}
	// appName O
	if(aeAnnc.getAppName() != null){
	aeAnncEntity.setAppName(aeAnnc.getAppName());
	modifiedAttributes.setAppName(aeAnnc.getAppName());
	}
	// pointOfAccess O
	if(!aeAnnc.getPointOfAccess().isEmpty()){
	aeAnncEntity.getPointOfAccess().clear();
	aeAnncEntity.getPointOfAccess().addAll(aeAnnc.getPointOfAccess());
	modifiedAttributes.getPointOfAccess().addAll(aeAnnc.getPointOfAccess());
	}
	// ontologyRef O
	if (aeAnnc.getOntologyRef() != null){
	aeAnncEntity.setOntologyRef(aeAnnc.getOntologyRef());
	modifiedAttributes.setOntologyRef(aeAnnc.getOntologyRef());
	}

	// Last Time Modified update
	aeAnncEntity.setLastModifiedTime(DateUtil.now());
	modifiedAttributes.setLastModifiedTime(aeAnncEntity.getLastModifiedTime());
	response.setContent(modifiedAttributes);
	// Update the resource in database
	dbs.getDAOFactory().getAeAnncDAO().update(transaction, aeAnncEntity);

	// commit transaction & release lock
	transaction.commit();

	Notifier.notify(aeAnncEntity.getSubscriptions(), aeAnncEntity, ResourceStatus.UPDATED);

	response.setResponseStatusCode(ResponseStatusCode.UPDATED);
	return response;
	}

	@Override
	public ResponsePrimitive doDelete(RequestPrimitive request) {
	/*
	* Generic delete procedure
	*/
	ResponsePrimitive response = new ResponsePrimitive(request);

	// Retrieve the resource from database
	AeAnncEntity aeAnncEntity = dbs.getDAOFactory()
	.getAeAnncDAO().find(transaction, request.getTo());
	if (aeAnncEntity == null){
	throw new ResourceNotFoundException("Resource not found");
	}

	// lock entity
	transaction.lock(aeAnncEntity);

	// checkACP(aeAnncEntity.getAccessControlPolicies(), request.getFrom(),
	// Operation.DELETE);
	checkPermissions(request, aeAnncEntity, aeAnncEntity.getAccessControlPolicies());

	UriMapper.deleteUri(aeAnncEntity.getHierarchicalURI());

	Notifier.notifyDeletion(aeAnncEntity.getSubscriptions(), aeAnncEntity);

	// Delete the resource
	dbs.getDAOFactory().getAeAnncDAO().delete(transaction, aeAnncEntity);

	// Commit the transaction & unlock
	transaction.commit();

	// Return rsc
	response.setResponseStatusCode(ResponseStatusCode.DELETED);
	return response;
	}

	}