fix: Move converters into the bundle package [DHIS2-18093]

dhis-2/dhis-services/dhis-service-tracker/src/main/java/org/hisp/dhis/tracker/imports/bundle/TrackerObjectsMapper.java

@@ -0,0 +1,347 @@
+/*
+ * Copyright (c) 2004-2024, University of Oslo
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ * Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ *
+ * Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ * Neither the name of the HISP project nor the names of its contributors may
+ * be used to endorse or promote products derived from this software without
+ * specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+package org.hisp.dhis.tracker.imports.bundle;
+
+import static org.apache.commons.collections4.CollectionUtils.isNotEmpty;
+import static org.hisp.dhis.program.EnrollmentStatus.COMPLETED;
+import static org.hisp.dhis.relationship.RelationshipEntity.PROGRAM_INSTANCE;
+import static org.hisp.dhis.relationship.RelationshipEntity.PROGRAM_STAGE_INSTANCE;
+import static org.hisp.dhis.relationship.RelationshipEntity.TRACKED_ENTITY_INSTANCE;
+
+import java.util.Date;
+import java.util.Optional;
+import java.util.stream.Collectors;
+import javax.annotation.Nonnull;
+import org.hisp.dhis.category.CategoryOptionCombo;
+import org.hisp.dhis.dataelement.DataElement;
+import org.hisp.dhis.event.EventStatus;
+import org.hisp.dhis.eventdatavalue.EventDataValue;
+import org.hisp.dhis.note.Note;
+import org.hisp.dhis.organisationunit.OrganisationUnit;
+import org.hisp.dhis.program.Enrollment;
+import org.hisp.dhis.program.Event;
+import org.hisp.dhis.program.Program;
+import org.hisp.dhis.program.ProgramStage;
+import org.hisp.dhis.program.UserInfoSnapshot;
+import org.hisp.dhis.relationship.Relationship;
+import org.hisp.dhis.relationship.RelationshipItem;
+import org.hisp.dhis.relationship.RelationshipKey;
+import org.hisp.dhis.relationship.RelationshipType;
+import org.hisp.dhis.trackedentity.TrackedEntity;
+import org.hisp.dhis.trackedentity.TrackedEntityType;
+import org.hisp.dhis.tracker.imports.domain.DataValue;
+import org.hisp.dhis.tracker.imports.preheat.TrackerPreheat;
+import org.hisp.dhis.tracker.imports.util.RelationshipKeySupport;
+import org.hisp.dhis.user.User;
+import org.hisp.dhis.user.UserDetails;
+import org.hisp.dhis.util.DateUtils;
+
+/**
+ * TrackerObjectsMapper maps tracker domain objects to Hibernate objects so they can be persisted in
+ * the DB. This class provides static methods to convert imported domain objects such as {@link
+ * TrackedEntity}, {@link Enrollment}, {@link Event}, and {@link Relationship} into their
+ * corresponding database entities. It gets existing records from the database through the preheat
+ * and maps the incoming data accordingly, ensuring that all necessary fields are populated
+ * correctly. All the values that should be set by the system are set here (eg. createdAt,
+ * updatedBy...)
+ */
+public class TrackerObjectsMapper {
+  private TrackerObjectsMapper() {
+    throw new IllegalStateException("Utility class");
+  }
+
+  public static @Nonnull TrackedEntity map(
+      @Nonnull TrackerPreheat preheat,
+      @Nonnull org.hisp.dhis.tracker.imports.domain.TrackedEntity trackedEntity,
+      @Nonnull UserDetails user) {
+    TrackedEntity dbTrackedEntity = preheat.getTrackedEntity(trackedEntity.getTrackedEntity());
+
+    Date now = new Date();
+
+    if (dbTrackedEntity == null) {
+      dbTrackedEntity = new TrackedEntity();
+      dbTrackedEntity.setUid(trackedEntity.getTrackedEntity());
+      dbTrackedEntity.setCreated(now);
+      dbTrackedEntity.setCreatedByUserInfo(UserInfoSnapshot.from(user));
+      dbTrackedEntity.setStoredBy(trackedEntity.getStoredBy());
+    }
+
+    dbTrackedEntity.setLastUpdated(now);
+    dbTrackedEntity.setLastUpdatedByUserInfo(UserInfoSnapshot.from(user));
+    dbTrackedEntity.setCreatedAtClient(DateUtils.fromInstant(trackedEntity.getCreatedAtClient()));
+    dbTrackedEntity.setLastUpdatedAtClient(
+        DateUtils.fromInstant(trackedEntity.getUpdatedAtClient()));
+
+    OrganisationUnit organisationUnit = preheat.getOrganisationUnit(trackedEntity.getOrgUnit());
+    dbTrackedEntity.setOrganisationUnit(organisationUnit);
+    TrackedEntityType trackedEntityType =
+        preheat.getTrackedEntityType(trackedEntity.getTrackedEntityType());
+    dbTrackedEntity.setTrackedEntityType(trackedEntityType);
+
+    dbTrackedEntity.setPotentialDuplicate(trackedEntity.isPotentialDuplicate());
+    dbTrackedEntity.setInactive(trackedEntity.isInactive());
+    dbTrackedEntity.setGeometry(trackedEntity.getGeometry());
+
+    return dbTrackedEntity;
+  }
+
+  public static @Nonnull Enrollment map(
+      @Nonnull TrackerPreheat preheat,
+      @Nonnull org.hisp.dhis.tracker.imports.domain.Enrollment enrollment,
+      @Nonnull UserDetails user) {
+    Enrollment dbEnrollment = preheat.getEnrollment(enrollment.getEnrollment());
+
+    Date now = new Date();
+
+    if (dbEnrollment == null) {
+      dbEnrollment = new Enrollment();
+      dbEnrollment.setUid(enrollment.getEnrollment());
+      dbEnrollment.setCreated(now);
+      dbEnrollment.setStoredBy(enrollment.getStoredBy());
+      dbEnrollment.setCreatedByUserInfo(UserInfoSnapshot.from(user));
+    }
+
+    dbEnrollment.setLastUpdated(now);
+    dbEnrollment.setLastUpdatedByUserInfo(UserInfoSnapshot.from(user));
+    dbEnrollment.setCreatedAtClient(DateUtils.fromInstant(enrollment.getCreatedAtClient()));
+    dbEnrollment.setLastUpdatedAtClient(DateUtils.fromInstant(enrollment.getUpdatedAtClient()));
+
+    OrganisationUnit organisationUnit = preheat.getOrganisationUnit(enrollment.getOrgUnit());
+    dbEnrollment.setOrganisationUnit(organisationUnit);
+    Program program = preheat.getProgram(enrollment.getProgram());
+    dbEnrollment.setProgram(program);
+    TrackedEntity trackedEntity = preheat.getTrackedEntity(enrollment.getTrackedEntity());
+    dbEnrollment.setTrackedEntity(trackedEntity);
+
+    Date enrollmentDate = DateUtils.fromInstant(enrollment.getEnrolledAt());
+    Date occurredDate = DateUtils.fromInstant(enrollment.getOccurredAt());
+    dbEnrollment.setEnrollmentDate(enrollmentDate);
+    dbEnrollment.setOccurredDate(occurredDate != null ? occurredDate : enrollmentDate);
+
+    dbEnrollment.setFollowup(enrollment.isFollowUp());
+    dbEnrollment.setGeometry(enrollment.getGeometry());
+
+    if (enrollment.getStatus() != dbEnrollment.getStatus()) {
+      dbEnrollment.setStatus(enrollment.getStatus());
+      switch (dbEnrollment.getStatus()) {
+        case ACTIVE -> {
+          dbEnrollment.setCompletedDate(null);
+          dbEnrollment.setCompletedBy(null);
+        }
+        case COMPLETED -> {
+          dbEnrollment.setCompletedDate(now);
+          dbEnrollment.setCompletedBy(user.getUsername());
+        }
+        case CANCELLED -> {
+          dbEnrollment.setCompletedDate(now);
+          dbEnrollment.setCompletedBy(null);
+        }
+      }
+    }
+
+    if (isNotEmpty(enrollment.getNotes())) {
+      dbEnrollment
+          .getNotes()
+          .addAll(
+              enrollment.getNotes().stream()
+                  .map(note -> map(preheat, note, user))
+                  .collect(Collectors.toSet()));
+    }
+    return dbEnrollment;
+  }
+
+  public static @Nonnull Event map(
+      @Nonnull TrackerPreheat preheat,
+      @Nonnull org.hisp.dhis.tracker.imports.domain.Event event,
+      @Nonnull UserDetails user) {
+    Event dbEvent = preheat.getEvent(event.getEvent());
+
+    Date now = new Date();
+
+    if (dbEvent == null) {
+      dbEvent = new Event();
+      dbEvent.setUid(event.getEvent());
+      dbEvent.setCreated(now);
+      dbEvent.setStoredBy(event.getStoredBy());
+      dbEvent.setCreatedByUserInfo(UserInfoSnapshot.from(user));
+    }
+    dbEvent.setLastUpdated(now);
+    dbEvent.setLastUpdatedByUserInfo(UserInfoSnapshot.from(user));
+    dbEvent.setCreatedAtClient(DateUtils.fromInstant(event.getCreatedAtClient()));
+    dbEvent.setLastUpdatedAtClient(DateUtils.fromInstant(event.getUpdatedAtClient()));
+
+    OrganisationUnit organisationUnit = preheat.getOrganisationUnit(event.getOrgUnit());
+    dbEvent.setOrganisationUnit(organisationUnit);
+    Program program = preheat.getProgram(event.getProgram());
+    dbEvent.setEnrollment(getEnrollment(preheat, event.getEnrollment(), program));
+    ProgramStage programStage = preheat.getProgramStage(event.getProgramStage());
+    dbEvent.setProgramStage(programStage);
+
+    dbEvent.setOccurredDate(DateUtils.fromInstant(event.getOccurredAt()));
+    dbEvent.setScheduledDate(DateUtils.fromInstant(event.getScheduledAt()));
+    if (program.isRegistration()
+        && dbEvent.getScheduledDate() == null
+        && dbEvent.getOccurredDate() != null) {
+      dbEvent.setScheduledDate(dbEvent.getOccurredDate());
+    }
+
+    dbEvent.setGeometry(event.getGeometry());
+
+    EventStatus currentStatus = event.getStatus();
+    EventStatus previousStatus = dbEvent.getStatus();
+    if (currentStatus != previousStatus && currentStatus == EventStatus.COMPLETED) {
+      dbEvent.setCompletedDate(now);
+      dbEvent.setCompletedBy(user.getUsername());
+    }
+
+    if (currentStatus != EventStatus.COMPLETED) {
+      dbEvent.setCompletedDate(null);
+      dbEvent.setCompletedBy(null);
+    }
+    dbEvent.setStatus(currentStatus);
+
+    if (event.getAttributeOptionCombo().isNotBlank()) {
+      dbEvent.setAttributeOptionCombo(
+          preheat.getCategoryOptionCombo(event.getAttributeOptionCombo()));
+    } else {
+      dbEvent.setAttributeOptionCombo(preheat.getDefault(CategoryOptionCombo.class));
+    }
+
+    if (Boolean.TRUE.equals(programStage.isEnableUserAssignment())
+        && event.getAssignedUser() != null
+        && !event.getAssignedUser().isEmpty()) {
+      Optional<User> assignedUser =
+          preheat.getUserByUsername(event.getAssignedUser().getUsername());
+      assignedUser.ifPresent(dbEvent::setAssignedUser);
+    }
+
+    // TODO(DHIS2-18223): Remove data value mapping and fix changelog logic
+    for (DataValue dataValue : event.getDataValues()) {
+      DataElement dataElement = preheat.getDataElement(dataValue.getDataElement());
+
+      EventDataValue eventDataValue = new EventDataValue();
+      eventDataValue.setDataElement(dataElement.getUid());
+      eventDataValue.setCreated(DateUtils.fromInstant(dataValue.getCreatedAt()));
+      eventDataValue.setCreatedByUserInfo(UserInfoSnapshot.from(user));
+      eventDataValue.setValue(dataValue.getValue());
+      eventDataValue.setLastUpdated(now);
+      eventDataValue.setProvidedElsewhere(dataValue.isProvidedElsewhere());
+      eventDataValue.setLastUpdatedByUserInfo(UserInfoSnapshot.from(user));
+
+      dbEvent.getEventDataValues().add(eventDataValue);
+    }
+
+    if (isNotEmpty(event.getNotes())) {
+      dbEvent
+          .getNotes()
+          .addAll(
+              event.getNotes().stream()
+                  .map(note -> map(preheat, note, user))
+                  .collect(Collectors.toSet()));
+    }
+
+    return dbEvent;
+  }
+
+  public static @Nonnull Relationship map(
+      @Nonnull TrackerPreheat preheat,
+      @Nonnull org.hisp.dhis.tracker.imports.domain.Relationship relationship,
+      @Nonnull UserDetails user) {
+    Date now = new Date();
+    Relationship dbRelationship = new org.hisp.dhis.relationship.Relationship();
+    dbRelationship.setUid(relationship.getRelationship());
+    dbRelationship.setCreated(now);
+    dbRelationship.setLastUpdated(now);
+    dbRelationship.setLastUpdatedBy(preheat.getUserByUid(user.getUid()).orElse(null));
+    dbRelationship.setCreatedAtClient(DateUtils.fromInstant(relationship.getCreatedAtClient()));
+
+    RelationshipType relationshipType =
+        preheat.getRelationshipType(relationship.getRelationshipType());
+    dbRelationship.setRelationshipType(relationshipType);
+
+    // FROM
+    RelationshipItem fromItem = new org.hisp.dhis.relationship.RelationshipItem();
+    fromItem.setRelationship(dbRelationship);
+    switch (relationshipType.getFromConstraint().getRelationshipEntity()) {
+      case TRACKED_ENTITY_INSTANCE ->
+          fromItem.setTrackedEntity(
+              preheat.getTrackedEntity(relationship.getFrom().getTrackedEntity()));
+      case PROGRAM_INSTANCE ->
+          fromItem.setEnrollment(preheat.getEnrollment(relationship.getFrom().getEnrollment()));
+      case PROGRAM_STAGE_INSTANCE ->
+          fromItem.setEvent(preheat.getEvent(relationship.getFrom().getEvent()));
+    }
+    dbRelationship.setFrom(fromItem);
+
+    // TO
+    RelationshipItem toItem = new org.hisp.dhis.relationship.RelationshipItem();
+    toItem.setRelationship(dbRelationship);
+    switch (relationshipType.getToConstraint().getRelationshipEntity()) {
+      case TRACKED_ENTITY_INSTANCE ->
+          toItem.setTrackedEntity(
+              preheat.getTrackedEntity(relationship.getTo().getTrackedEntity()));
+      case PROGRAM_INSTANCE ->
+          toItem.setEnrollment(preheat.getEnrollment(relationship.getTo().getEnrollment()));
+      case PROGRAM_STAGE_INSTANCE ->
+          toItem.setEvent(preheat.getEvent(relationship.getTo().getEvent()));
+    }
+    dbRelationship.setTo(toItem);
+
+    RelationshipKey relationshipKey =
+        RelationshipKeySupport.getRelationshipKey(relationship, relationshipType);
+    dbRelationship.setKey(relationshipKey.asString());
+    dbRelationship.setInvertedKey(relationshipKey.inverseKey().asString());
+
+    return dbRelationship;
+  }
+
+  private static @Nonnull Note map(
+      @Nonnull TrackerPreheat preheat,
+      @Nonnull org.hisp.dhis.tracker.imports.domain.Note note,
+      @Nonnull UserDetails user) {
+    Date now = new Date();
+
+    Note dbNote = new Note();
+    dbNote.setUid(note.getNote());
+    dbNote.setCreated(now);
+    dbNote.setLastUpdated(now);
+    dbNote.setLastUpdatedBy(preheat.getUserByUid(user.getUid()).orElse(null));
+    dbNote.setCreator(note.getStoredBy());
+    dbNote.setNoteText(note.getValue());
+
+    return dbNote;
+  }
+
+  private static Enrollment getEnrollment(
+      TrackerPreheat preheat, String enrollment, Program program) {
+    return switch (program.getProgramType()) {
+      case WITH_REGISTRATION -> preheat.getEnrollment(enrollment);
+      case WITHOUT_REGISTRATION -> preheat.getEnrollmentsWithoutRegistration(program.getUid());
+    };
+  }
+}