content: /*
   * Copyright 2010-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.engine.impl.cmd;
  
  import com.fasterxml.jackson.databind.JsonNode;
  import com.fasterxml.jackson.databind.node.ObjectNode;
  import java.io.Serializable;
  import java.util.Collection;
  import java.util.HashMap;
  import java.util.Map;
  import java.util.Map.Entry;
  import org.activiti.bpmn.model.BpmnModel;
  import org.activiti.bpmn.model.SubProcess;
  import org.activiti.bpmn.model.ValuedDataObject;
  import org.activiti.engine.ActivitiIllegalArgumentException;
  import org.activiti.engine.ActivitiObjectNotFoundException;
  import org.activiti.engine.DynamicBpmnConstants;
  import org.activiti.engine.impl.DataObjectImpl;
  import org.activiti.engine.impl.context.Context;
  import org.activiti.engine.impl.interceptor.Command;
  import org.activiti.engine.impl.interceptor.CommandContext;
  import org.activiti.engine.impl.persistence.entity.ExecutionEntity;
  import org.activiti.engine.impl.persistence.entity.TaskEntity;
  import org.activiti.engine.impl.persistence.entity.VariableInstance;
  import org.activiti.engine.impl.util.ProcessDefinitionUtil;
  import org.activiti.engine.runtime.DataObject;
  import org.activiti.engine.task.Task;
  
  public class GetTaskDataObjectsCmd implements Command<Map<String, DataObject>>, Serializable {
  
      private static final long serialVersionUID = 1L;
      protected String taskId;
      protected Collection<String> variableNames;
      protected String locale;
      protected boolean withLocalizationFallback;
  
      public GetTaskDataObjectsCmd(String taskId, Collection<String> variableNames) {
          this.taskId = taskId;
          this.variableNames = variableNames;
      }
  
      public GetTaskDataObjectsCmd(
          String taskId,
          Collection<String> variableNames,
          String locale,
          boolean withLocalizationFallback
      ) {
          this.taskId = taskId;
          this.variableNames = variableNames;
          this.locale = locale;
          this.withLocalizationFallback = withLocalizationFallback;
      }
  
      public Map<String, DataObject> execute(CommandContext commandContext) {
          if (taskId == null) {
              throw new ActivitiIllegalArgumentException("taskId is null");
          }
  
          TaskEntity task = commandContext.getTaskEntityManager().findById(taskId);
  
          if (task == null) {
              throw new ActivitiObjectNotFoundException("task " + taskId + " doesn't exist", Task.class);
          }
  
          Map<String, DataObject> dataObjects = null;
          Map<String, VariableInstance> variables = null;
          if (variableNames == null) {
              variables = task.getVariableInstances();
          } else {
              variables = task.getVariableInstances(variableNames, false);
          }
  
          if (variables != null) {
              dataObjects = new HashMap<>(variables.size());
  
              for (Entry<String, VariableInstance> entry : variables.entrySet()) {
                  VariableInstance variableEntity = entry.getValue();
  
                  String localizedName = null;
                  String localizedDescription = null;
  
                  ExecutionEntity executionEntity = commandContext
                      .getExecutionEntityManager()
                      .findById(variableEntity.getExecutionId());
                  while (!executionEntity.isScope()) {
                      executionEntity = executionEntity.getParent();
                  }
  
                  BpmnModel bpmnModel = ProcessDefinitionUtil.getBpmnModel(executionEntity.getProcessDefinitionId());
                  ValuedDataObject foundDataObject = null;
                  if (executionEntity.getParentId() == null) {
                      for (ValuedDataObject dataObject : bpmnModel.getMainProcess().getDataObjects()) {
                          if (dataObject.getName().equals(variableEntity.getName())) {
                              foundDataObject = dataObject;
                              break;
                          }
                      }
                  } else {
                      SubProcess subProcess = (SubProcess) bpmnModel.getFlowElement(executionEntity.getActivityId());
                      for (ValuedDataObject dataObject : subProcess.getDataObjects()) {
                          if (dataObject.getName().equals(variableEntity.getName())) {
                              foundDataObject = dataObject;
                              break;
                          }
                      }
                  }
  
                  if (locale != null && foundDataObject != null) {
                      ObjectNode languageNode = Context.getLocalizationElementProperties(
                          locale,
                          foundDataObject.getId(),
                          task.getProcessDefinitionId(),
                          withLocalizationFallback
                      );
  
                      if (languageNode != null) {
                          JsonNode nameNode = languageNode.get(DynamicBpmnConstants.LOCALIZATION_NAME);
                          if (nameNode != null) {
                              localizedName = nameNode.asText();
                          }
                          JsonNode descriptionNode = languageNode.get(DynamicBpmnConstants.LOCALIZATION_DESCRIPTION);
                          if (descriptionNode != null) {
                              localizedDescription = descriptionNode.asText();
                          }
                      }
                  }
  
                  if (foundDataObject != null) {
                      dataObjects.put(
                          variableEntity.getName(),
                          new DataObjectImpl(
                              variableEntity.getName(),
                              variableEntity.getValue(),
                              foundDataObject.getDocumentation(),
                              foundDataObject.getType(),
                              localizedName,
                              localizedDescription,
                              foundDataObject.getId()
                          )
                      );
                  }
              }
          }
  
          return dataObjects;
      }
  }
  