content: /*
   * Copyright 2017-2025 Hyland Software, Inc. and its affiliates.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *     http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.activiti.cloud.services.query.events.handlers;
  
  import jakarta.persistence.EntityManager;
  import java.util.Date;
  import java.util.Optional;
  import java.util.Set;
  import java.util.function.Predicate;
  import java.util.stream.Stream;
  import org.activiti.api.process.model.ProcessInstance;
  import org.activiti.api.process.model.events.ProcessRuntimeEvent;
  import org.activiti.api.task.model.Task.TaskStatus;
  import org.activiti.cloud.api.model.shared.events.CloudRuntimeEvent;
  import org.activiti.cloud.api.process.model.events.CloudProcessCompletedEvent;
  import org.activiti.cloud.api.task.model.impl.CloudTaskImpl;
  import org.activiti.cloud.api.task.model.impl.events.CloudTaskCancelledEventImpl;
  import org.activiti.cloud.services.query.model.ProcessInstanceEntity;
  import org.activiti.cloud.services.query.model.QueryException;
  import org.activiti.cloud.services.query.model.TaskEntity;
  
  public class ProcessCompletedEventHandler implements QueryEventHandler {
  
      private final EntityManager entityManager;
      private final TaskCancelledEventHandler taskCancelledEventHandler;
  
      public ProcessCompletedEventHandler(
          EntityManager entityManager,
          TaskCancelledEventHandler taskCancelledEventHandler
      ) {
          this.entityManager = entityManager;
          this.taskCancelledEventHandler = taskCancelledEventHandler;
      }
  
      @Override
      public void handle(CloudRuntimeEvent<?, ?> event) {
          CloudProcessCompletedEvent completedEvent = (CloudProcessCompletedEvent) event;
          String processInstanceId = completedEvent.getEntity().getId();
          Optional<ProcessInstanceEntity> findResult = Optional.ofNullable(
              entityManager.find(ProcessInstanceEntity.class, processInstanceId)
          );
          if (findResult.isPresent()) {
              ProcessInstanceEntity processInstanceEntity = findResult.get();
              processInstanceEntity.setStatus(ProcessInstance.ProcessInstanceStatus.COMPLETED);
              processInstanceEntity.setLastModified(new Date(completedEvent.getTimestamp()));
              processInstanceEntity.setCompletedDate(new Date(completedEvent.getTimestamp()));
              entityManager.persist(processInstanceEntity);
              callCancelledEventHandlerToCancelRemainingTasks(processInstanceEntity);
          } else {
              throw new QueryException("Unable to find process instance with the given id: " + processInstanceId);
          }
      }
  
      private void callCancelledEventHandlerToCancelRemainingTasks(ProcessInstanceEntity processInstanceEntity) {
          Predicate<TaskEntity> cancellableTasks = task ->
              TaskStatus.ASSIGNED.equals(task.getStatus()) || TaskStatus.CREATED.equals(task.getStatus());
  
          Stream
              .ofNullable(processInstanceEntity.getTasks())
              .flatMap(Set::stream)
              .filter(cancellableTasks)
              .map(task -> {
                  CloudTaskImpl cloudTask = new CloudTaskImpl();
                  cloudTask.setId(task.getId());
                  return new CloudTaskCancelledEventImpl(cloudTask);
              })
              .forEach(taskCancelledEventHandler::handle);
      }
  
      @Override
      public String getHandledEvent() {
          return ProcessRuntimeEvent.ProcessEvents.PROCESS_COMPLETED.name();
      }
  }
  