FM2-636: Enhancements in Existing FHIR Task Resource

api/src/main/java/org/openmrs/module/fhir2/FhirConstants.java

@@ -267,12 +267,16 @@ private FhirConstants() {
 
 	public static final String STATUS_SEARCH_HANDLER = "status.search.handler";
 
+	public static final String TASK_CODE_SEARCH_HANDLER = "task.code.search.handler";
+
 	public static final String FULFILLER_STATUS_SEARCH_HANDLER = "fullfillerStatus.search.handler";
 
 	public static final String BASED_ON_REFERENCE_SEARCH_HANDLER = "based.on.reference.search.handler";
 
 	public static final String OWNER_REFERENCE_SEARCH_HANDLER = "owner.reference.search.handler";
 
+	public static final String FOR_REFERENCE_SEARCH_HANDLER = "for.reference.search.handler";
+
 	public static final String RESULT_SEARCH_HANDLER = "result.search.handler";
 
 	public static final String COMMON_SEARCH_HANDLER = "common.search.handler";

api/src/main/java/org/openmrs/module/fhir2/api/dao/impl/FhirTaskDaoImpl.java

@@ -50,6 +50,13 @@ protected void setupSearchParams(Criteria criteria, SearchParameterMap theParams
 					entry.getValue().forEach(
 					    param -> handleReference(criteria, (ReferenceAndListParam) param.getParam(), "ownerReference", "o"));
 					break;
+				case FhirConstants.FOR_REFERENCE_SEARCH_HANDLER:
+					entry.getValue().forEach(
+					    param -> handleReference(criteria, (ReferenceAndListParam) param.getParam(), "forReference", "f"));
+					break;
+				case FhirConstants.TASK_CODE_SEARCH_HANDLER:
+					entry.getValue().forEach(code -> handleTaskCodeConcept(criteria, (TokenAndListParam) code.getParam()));
+					break;
 				case FhirConstants.STATUS_SEARCH_HANDLER:
 					entry.getValue()
 					        .forEach(param -> handleStatus((TokenAndListParam) param.getParam()).ifPresent(criteria::add));
@@ -108,4 +115,14 @@ private void handleReference(Criteria criteria, ReferenceAndListParam reference,
 			return Optional.empty();
 		}).ifPresent(criteria::add);
 	}
+
+	private void handleTaskCodeConcept(Criteria criteria, TokenAndListParam code) {
+		if (code != null) {
+			if (lacksAlias(criteria, "tc")) {
+				criteria.createAlias("taskCode", "tc");
+			}
+
+			handleCodeableConcept(criteria, code, "tc", "tcm", "tcrt").ifPresent(criteria::add);
+		}
+	}
 }

api/src/main/java/org/openmrs/module/fhir2/api/search/param/TaskSearchParams.java

@@ -44,17 +44,23 @@ public class TaskSearchParams extends BaseResourceSearchParams {
 
 	private ReferenceAndListParam ownerReference;
 
+	private ReferenceAndListParam forReference;
+
+	private TokenAndListParam taskCode;
+
 	private TokenAndListParam status;
 
 	@Builder
 	public TaskSearchParams(ReferenceAndListParam basedOnReference, ReferenceAndListParam ownerReference,
-	    TokenAndListParam status, TokenAndListParam id, DateRangeParam lastUpdated, SortSpec sort,
-	    HashSet<Include> includes) {
+	    ReferenceAndListParam forReference, TokenAndListParam taskCode, TokenAndListParam status, TokenAndListParam id,
+	    DateRangeParam lastUpdated, SortSpec sort, HashSet<Include> includes) {
 
 		super(id, lastUpdated, sort, includes, null);
 
 		this.basedOnReference = basedOnReference;
 		this.ownerReference = ownerReference;
+		this.forReference = forReference;
+		this.taskCode = taskCode;
 		this.status = status;
 
 	}
@@ -63,6 +69,8 @@ public TaskSearchParams(ReferenceAndListParam basedOnReference, ReferenceAndList
 	public SearchParameterMap toSearchParameterMap() {
 		return baseSearchParameterMap().addParameter(FhirConstants.BASED_ON_REFERENCE_SEARCH_HANDLER, getBasedOnReference())
 		        .addParameter(FhirConstants.OWNER_REFERENCE_SEARCH_HANDLER, getOwnerReference())
+		        .addParameter(FhirConstants.FOR_REFERENCE_SEARCH_HANDLER, getForReference())
+		        .addParameter(FhirConstants.TASK_CODE_SEARCH_HANDLER, getTaskCode())
 		        .addParameter(FhirConstants.STATUS_SEARCH_HANDLER, getStatus());
 	}
 }

api/src/main/java/org/openmrs/module/fhir2/api/translators/impl/ServiceRequestTranslatorImpl.java

@@ -140,7 +140,7 @@ private Reference determineServiceRequestPerformer(String orderUuid) {
 		IBundleProvider results = taskService.searchForTasks(new TaskSearchParams(
 		        new ReferenceAndListParam()
 		                .addAnd(new ReferenceOrListParam().add(new ReferenceParam("ServiceRequest", null, orderUuid))),
-		        null, null, null, null, null, null));
+		        null, null, null, null, null, null, null, null));
 
 		Collection<Task> serviceRequestTasks = results.getResources(START_INDEX, END_INDEX).stream().map(p -> (Task) p)
 		        .collect(Collectors.toList());

api/src/main/java/org/openmrs/module/fhir2/api/translators/impl/TaskTranslatorImpl.java

@@ -21,9 +21,12 @@
 
 import lombok.AccessLevel;
 import lombok.Setter;
+import org.hl7.fhir.r4.model.Annotation;
 import org.hl7.fhir.r4.model.Identifier;
+import org.hl7.fhir.r4.model.Period;
 import org.hl7.fhir.r4.model.Task;
 import org.openmrs.module.fhir2.FhirConstants;
+import org.openmrs.module.fhir2.api.translators.ConceptTranslator;
 import org.openmrs.module.fhir2.api.translators.ReferenceTranslator;
 import org.openmrs.module.fhir2.api.translators.TaskInputTranslator;
 import org.openmrs.module.fhir2.api.translators.TaskOutputTranslator;
@@ -45,6 +48,9 @@ public class TaskTranslatorImpl implements TaskTranslator {
 	@Autowired
 	private TaskOutputTranslator taskOutputTranslator;
 
+	@Autowired
+	private ConceptTranslator conceptTranslator;
+
 	@Override
 	public Task toFhirResource(@Nonnull FhirTask openmrsTask) {
 		notNull(openmrsTask, "The openmrsTask object should not be null");
@@ -116,6 +122,30 @@ private void setFhirTaskFields(FhirTask openmrsTask, Task fhirTask) {
 			        .filter(Objects::nonNull).collect(Collectors.toList()));
 		}
 
+		if (openmrsTask.getTaskCode() != null) {
+			fhirTask.setCode(conceptTranslator.toFhirResource(openmrsTask.getTaskCode()));
+		}
+
+		if (openmrsTask.getPartOfReferences() != null && !openmrsTask.getPartOfReferences().isEmpty()) {
+			fhirTask.setPartOf(openmrsTask.getPartOfReferences().stream().map(referenceTranslator::toFhirResource)
+			        .collect(Collectors.toList()));
+		}
+
+		if (openmrsTask.getExecutionStartTime() != null || openmrsTask.getExecutionEndTime() != null) {
+			Period period = new Period();
+			if (openmrsTask.getExecutionStartTime() != null) {
+				period.setStart(openmrsTask.getExecutionStartTime());
+			}
+			if (openmrsTask.getExecutionEndTime() != null) {
+				period.setEnd(openmrsTask.getExecutionEndTime());
+			}
+			fhirTask.setExecutionPeriod(period);
+		}
+
+		if (openmrsTask.getComment() != null) {
+			fhirTask.addNote(new Annotation().setText(openmrsTask.getComment()));
+		}
+
 		fhirTask.setAuthoredOn(openmrsTask.getDateCreated());
 
 		if (openmrsTask.getDateChanged() != null) {
@@ -183,6 +213,28 @@ private void setOpenmrsTaskFields(FhirTask openmrsTask, Task fhirTask) {
 			        .filter(Objects::nonNull).collect(Collectors.toSet()));
 		}
 
+		if (fhirTask.hasCode()) {
+			openmrsTask.setTaskCode(conceptTranslator.toOpenmrsType(fhirTask.getCode()));
+		}
+
+		if (fhirTask.hasExecutionPeriod()) {
+			if (fhirTask.getExecutionPeriod().hasStart()) {
+				openmrsTask.setExecutionStartTime(fhirTask.getExecutionPeriod().getStart());
+			}
+			if (fhirTask.getExecutionPeriod().hasEnd()) {
+				openmrsTask.setExecutionEndTime(fhirTask.getExecutionPeriod().getEnd());
+			}
+		}
+
+		if (fhirTask.hasPartOf()) {
+			openmrsTask.setPartOfReferences(
+			    fhirTask.getPartOf().stream().map(referenceTranslator::toOpenmrsType).collect(Collectors.toSet()));
+		}
+
+		if (fhirTask.hasNote()) {
+			openmrsTask.setComment(fhirTask.getNoteFirstRep().getText());
+		}
+
 		openmrsTask.setName(FhirConstants.TASK + "/" + fhirTask.getId());
 
 	}

api/src/main/java/org/openmrs/module/fhir2/model/FhirTask.java

@@ -19,16 +19,19 @@
 import javax.persistence.Id;
 import javax.persistence.JoinColumn;
 import javax.persistence.JoinTable;
+import javax.persistence.ManyToOne;
 import javax.persistence.OneToMany;
 import javax.persistence.OneToOne;
 import javax.persistence.Table;
 
+import java.util.Date;
 import java.util.Set;
 
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 import lombok.NoArgsConstructor;
 import org.openmrs.BaseOpenmrsMetadata;
+import org.openmrs.Concept;
 
 @Data
 @NoArgsConstructor
@@ -129,4 +132,37 @@ public enum TaskIntent {
 	@OneToOne(cascade = CascadeType.ALL)
 	@JoinColumn(name = "location_reference_id", referencedColumnName = "reference_id")
 	private FhirReference locationReference;
+
+	/**
+	 * Type of Task
+	 */
+	@ManyToOne
+	@JoinColumn(name = "task_code", referencedColumnName = "concept_id")
+	protected Concept taskCode;
+
+	/**
+	 * Task that this particular task is part of.
+	 */
+	@OneToMany(cascade = CascadeType.ALL)
+	@JoinTable(name = "fhir_task_part_of_reference", joinColumns = @JoinColumn(name = "task_id"), inverseJoinColumns = @JoinColumn(name = "reference_id"))
+	private Set<FhirReference> partOfReferences;
+
+	/**
+	 * Actual start time of the execution
+	 */
+	@Column(name = "execution_start_time")
+	private Date executionStartTime;
+
+	/**
+	 * Actual end time of the execution
+	 */
+	@Column(name = "execution_end_time")
+	private Date executionEndTime;
+
+	/**
+	 * Comment made about the task
+	 */
+	@Column(name = "comment")
+	private String comment;
+
 }

api/src/main/java/org/openmrs/module/fhir2/providers/r3/TaskFhirResourceProvider.java

@@ -96,6 +96,8 @@ public OperationOutcome deleteTask(@IdParam IdType id) {
 	public IBundleProvider searchTasks(
 	        @OptionalParam(name = Task.SP_BASED_ON, chainWhitelist = { "" }) ReferenceAndListParam basedOnReference,
 	        @OptionalParam(name = Task.SP_OWNER, chainWhitelist = { "" }) ReferenceAndListParam ownerReference,
+	        @OptionalParam(name = Task.SP_SUBJECT, chainWhitelist = { "" }) ReferenceAndListParam forReference,
+	        @OptionalParam(name = Task.SP_CODE) TokenAndListParam taskCode,
 	        @OptionalParam(name = Task.SP_STATUS) TokenAndListParam status,
 	        @OptionalParam(name = Task.SP_RES_ID) TokenAndListParam id,
 	        @OptionalParam(name = "_lastUpdated") DateRangeParam lastUpdated, @Sort SortSpec sort,
@@ -105,7 +107,7 @@ public IBundleProvider searchTasks(
 		if (CollectionUtils.isEmpty(includes)) {
 			includes = null;
 		}
-		return new SearchQueryBundleProviderR3Wrapper(fhirTaskService.searchForTasks(
-		    new TaskSearchParams(basedOnReference, ownerReference, status, id, lastUpdated, sort, includes)));
+		return new SearchQueryBundleProviderR3Wrapper(fhirTaskService.searchForTasks(new TaskSearchParams(basedOnReference,
+		        ownerReference, forReference, taskCode, status, id, lastUpdated, sort, includes)));
 	}
 }

api/src/main/java/org/openmrs/module/fhir2/providers/r4/TaskFhirResourceProvider.java

@@ -103,6 +103,8 @@ public OperationOutcome deleteTask(@IdParam IdType id) {
 	public IBundleProvider searchTasks(
 	        @OptionalParam(name = Task.SP_BASED_ON, chainWhitelist = { "" }) ReferenceAndListParam basedOnReference,
 	        @OptionalParam(name = Task.SP_OWNER, chainWhitelist = { "" }) ReferenceAndListParam ownerReference,
+	        @OptionalParam(name = Task.SP_SUBJECT, chainWhitelist = { "" }) ReferenceAndListParam forReference,
+	        @OptionalParam(name = Task.SP_CODE) TokenAndListParam taskCode,
 	        @OptionalParam(name = Task.SP_STATUS) TokenAndListParam status,
 	        @OptionalParam(name = Task.SP_RES_ID) TokenAndListParam id,
 	        @OptionalParam(name = "_lastUpdated") DateRangeParam lastUpdated, @Sort SortSpec sort,
@@ -111,7 +113,7 @@ public IBundleProvider searchTasks(
 		if (CollectionUtils.isEmpty(includes)) {
 			includes = null;
 		}
-		return service.searchForTasks(
-		    new TaskSearchParams(basedOnReference, ownerReference, status, id, lastUpdated, sort, includes));
+		return service.searchForTasks(new TaskSearchParams(basedOnReference, ownerReference, forReference, taskCode, status,
+		        id, lastUpdated, sort, includes));
 	}
 }

api/src/main/resources/liquibase.xml

@@ -1067,4 +1067,47 @@
             <column name="attribute_type_id" />
         </createIndex>
     </changeSet>
+
+    <changeSet id="alter_fhir_task_20230903" author="r.kalai">
+        <preConditions onFail="MARK_RAN" onError="WARN">
+            <tableExists tableName="fhir_task"/>
+        </preConditions>
+        <addColumn tableName="fhir_task">
+            <column name="execution_start_time" type="datetime"/>
+            <column name="execution_end_time" type="datetime"/>
+            <column name="task_code" type="int"/>
+            <column name="comment" type="varchar(255)"/>
+        </addColumn>
+        <addForeignKeyConstraint baseTableName="fhir_task" baseColumnNames="task_code"
+                                 constraintName="task_code_fk"
+                                 referencedTableName="concept"
+                                 referencedColumnNames="concept_id"/>
+    </changeSet>
+
+    <changeSet id="add_part_of_reference_join_table_20200311" author="r.kalai">
+        <preConditions onFail="MARK_RAN" onError="WARN">
+            <not>
+                <tableExists tableName="fhir_task_part_of_reference"/>
+            </not>
+            <tableExists tableName="fhir_task"/>
+            <tableExists tableName="fhir_reference"/>
+        </preConditions>
+        <createTable tableName="fhir_task_part_of_reference">
+            <column name="task_id" type="int">
+                <constraints nullable="false"/>
+            </column>
+            <column name="reference_id" type="int">
+                <constraints nullable="false"/>
+            </column>
+        </createTable>
+        <addForeignKeyConstraint baseTableName="fhir_task_part_of_reference" baseColumnNames="task_id"
+                                 constraintName="task_part_of_fk"
+                                 referencedTableName="fhir_task"
+                                 referencedColumnNames="task_id"/>
+        <addForeignKeyConstraint baseTableName="fhir_task_part_of_reference" baseColumnNames="reference_id"
+                                 constraintName="reference_part_of_fk"
+                                 referencedTableName="fhir_reference"
+                                 referencedColumnNames="reference_id"/>
+    </changeSet>
+
 </databaseChangeLog>