FormDataResource.java

/* 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.rest.service.api.form;

import java.util.HashMap;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.activiti.engine.ActivitiException;
import org.activiti.engine.ActivitiIllegalArgumentException;
import org.activiti.engine.ActivitiObjectNotFoundException;
import org.activiti.engine.FormService;
import org.activiti.engine.form.FormData;
import org.activiti.engine.runtime.ProcessInstance;
import org.activiti.rest.service.api.RestResponseFactory;
import org.activiti.rest.service.api.runtime.process.ProcessInstanceResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

/**
 * @author Tijs Rademakers
 */
@RestController
public class FormDataResource {
  
  @Autowired
  protected RestResponseFactory restResponseFactory;
  
  @Autowired
  protected FormService formService;

  @RequestMapping(value="/form/form-data", method = RequestMethod.GET, produces="application/json")
  public FormDataResponse getFormData(@RequestParam(value="taskId", required=false) String taskId,
      @RequestParam(value="processDefinitionId", required=false) String processDefinitionId, HttpServletRequest request) {
    
    if (taskId == null && processDefinitionId == null) {
      throw new ActivitiIllegalArgumentException("The taskId or processDefinitionId parameter has to be provided");
    }
    
    if (taskId != null && processDefinitionId != null) {
      throw new ActivitiIllegalArgumentException("Not both a taskId and a processDefinitionId parameter can be provided");
    }
    
    FormData formData = null;
    String id = null;
    if (taskId != null) {
      formData = formService.getTaskFormData(taskId);
      id = taskId;
    } else {
      formData = formService.getStartFormData(processDefinitionId);
      id = processDefinitionId;
    }
    
    if (formData == null) {
      throw new ActivitiObjectNotFoundException("Could not find a form data with id '" + id + "'.", FormData.class);
    }
    
    return restResponseFactory.createFormDataResponse(formData);
  }
  
  @RequestMapping(value="/form/form-data", method = RequestMethod.POST, produces="application/json")
  public ProcessInstanceResponse submitForm(@RequestBody SubmitFormRequest submitRequest, 
      HttpServletRequest request, HttpServletResponse response) {
    
    if (submitRequest == null) {
      throw new ActivitiException("A request body was expected when executing the form submit.");
    }

    if (submitRequest.getTaskId() == null && submitRequest.getProcessDefinitionId() == null) {
      throw new ActivitiIllegalArgumentException("The taskId or processDefinitionId property has to be provided");
    }
    
    Map<String, String> propertyMap = new HashMap<String, String>();
    if (submitRequest.getProperties() != null) {
      for (RestFormProperty formProperty : submitRequest.getProperties()) {
        propertyMap.put(formProperty.getId(), formProperty.getValue());
      }
    }
    
    if (submitRequest.getTaskId() != null) {
      formService.submitTaskFormData(submitRequest.getTaskId(), propertyMap);
      response.setStatus(HttpStatus.NO_CONTENT.value());
      return null;
      
    } else {
      ProcessInstance processInstance = null;
      if (submitRequest.getBusinessKey() != null) {
        processInstance = formService.submitStartFormData(submitRequest.getProcessDefinitionId(), 
            submitRequest.getBusinessKey(), propertyMap);
      } else {
        processInstance = formService.submitStartFormData(submitRequest.getProcessDefinitionId(), propertyMap);
      }
      return restResponseFactory.createProcessInstanceResponse(processInstance);
    }
  }
}