CohortMemberREsource

package org.openmrs.module.webservices.rest.web.resource;import java.util.ArrayList;import java.util.List;import org.openmrs.Cohort;import org.openmrs.Patient;import org.openmrs.annotation.Handler;import org.openmrs.api.context.Context;import org.openmrs.module.webservices.rest.SimpleObject;import org.openmrs.module.webservices.rest.web.ConversionUtil;import org.openmrs.module.webservices.rest.web.RequestContext;import org.openmrs.module.webservices.rest.web.annotation.SubResource;import org.openmrs.module.webservices.rest.web.representation.DefaultRepresentation;import org.openmrs.module.webservices.rest.web.representation.RefRepresentation;import org.openmrs.module.webservices.rest.web.representation.Representation;import org.openmrs.module.webservices.rest.web.resource.impl.DelegatingResourceDescription;import org.openmrs.module.webservices.rest.web.resource.impl.DelegatingSubResource;import org.openmrs.module.webservices.rest.web.response.ObjectNotFoundException;import org.openmrs.module.webservices.rest.web.response.ResourceDoesNotSupportOperationException;import org.openmrs.module.webservices.rest.web.response.ResponseException;/** * Sub-resource for cohort members */@SubResource(parent = CohortResource.class, parentProperty = "REMOVE-THIS-PROPERTY", path = "patients")@Handler(supports = Patient.class, order = 0)public class CohortMemberResource extends DelegatingSubResource<Patient, Cohort, CohortResource> {		@Override	public List<Patient> doGetAll(Cohort parent, RequestContext context) throws ResponseException {		List<Patient> members = new ArrayList<Patient>();		members.addAll(Context.getPatientSetService().getPatients(parent.getMemberIds()));		return members;	}		@Override	public Cohort getParent(Patient instance) {		return parent;	}		@Override	public void setParent(Patient instance, Cohort parent) {		this.parent = parent;	}		@Override	protected void delete(Patient delegate, String reason, RequestContext context) throws ResponseException {		removeMemberFromCohort(delegate);	}		@Override	public Patient getByUniqueId(String uniqueId) {		return Context.getPatientService().getPatientByUuid(uniqueId);	}		@Override	public DelegatingResourceDescription getRepresentationDescription(Representation rep) {		if (rep instanceof RefRepresentation) {			DelegatingResourceDescription description = new DelegatingResourceDescription();			description.addProperty("uuid");			description.addProperty("uri", findMethod("getUri"));			description.addProperty("display", findMethod("getDisplayString"));			return description;		} else if (rep instanceof DefaultRepresentation) {			DelegatingResourceDescription description = new DelegatingResourceDescription();			description.addProperty("uuid");			description.addProperty("gender");			description.addProperty("age");			description.addProperty("birthdate");			description.addProperty("birthdateEstimated");			description.addProperty("dead");			description.addProperty("deathDate");			description.addProperty("causeOfDeath", Representation.REF);			description.addProperty("preferredName", "personName", Representation.REF);			description.addProperty("personAddress", Representation.REF);			description.addProperty("activeIdentifiers", Representation.REF);			description.addProperty("activeAttributes", Representation.REF);			description.addProperty("uri", findMethod("getUri"));			return description;		}		return null;	}		@Override	protected Patient newDelegate() {		return new Patient();	}		@Override	public void purge(Patient delegate, RequestContext context) throws ResponseException {		throw new ResourceDoesNotSupportOperationException();	}		/**	 * @should add patient to cohort	 */	@Override	protected Patient save(Patient delegate) {		addMemberToCohort(delegate);		return delegate;	}		private Cohort parent = null;		public void addMemberToCohort(Patient patient) {		getParent(patient).addMember(patient.getId());		Context.getCohortService().saveCohort(getParent(patient));	}		public void removeMemberFromCohort(Patient patient) {		getParent(patient).removeMember(patient.getId());		Context.getCohortService().saveCohort(getParent(patient));	}		@Override	public Object create(String parentUniqueId, SimpleObject post, RequestContext context) throws ResponseException {		Cohort parent = Context.getCohortService().getCohortByUuid(parentUniqueId);		Patient delegate = newDelegate();		setParent(delegate, parent);		setConvertedProperties(delegate, post);		delegate = save(delegate);		return ConversionUtil.convertToRepresentation(delegate, Representation.DEFAULT);	}		@Override	public void delete(String parentUniqueId, String uuid, String reason, RequestContext context) throws ResponseException {	    // TODO Same as create	}	}