/* 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.bpmn.behavior;

import org.activiti.engine.impl.bpmn.helper.ScopeUtil;
import org.activiti.engine.impl.bpmn.parser.BpmnParse;
import org.activiti.engine.impl.persistence.entity.CompensateEventSubscriptionEntity;
import org.activiti.engine.impl.persistence.entity.ExecutionEntity;
import org.activiti.engine.impl.pvm.PvmScope;
import org.activiti.engine.impl.pvm.delegate.ActivityExecution;
import org.activiti.engine.impl.pvm.process.ActivityImpl;
import org.activiti.engine.impl.pvm.runtime.InterpretableExecution;
import org.activiti.engine.impl.task.TaskDefinition;


/**
 * Denotes an 'activity' in the sense of BPMN 2.0: 
 * a parent class for all tasks, subprocess and callActivity. 
 * 
 * @author Joram Barrez
 */
public class AbstractBpmnActivityBehavior extends FlowNodeActivityBehavior {
	
 

protected MultiInstanceActivityBehavior multiInstanceActivityBehavior;
  protected boolean useMixUsetask;
  protected String collectionElementVariable;
  
//  /**
//   * Subclasses that call leave() will first pass through this method, before
//   * the regular {@link FlowNodeActivityBehavior#leave(ActivityExecution)} is
//   * called. This way, we can check if the activity has loop characteristics,
//   * and delegate to the behavior if this is the case.
//   */
//  protected void leave(ActivityExecution execution) {
////    if(hasCompensationHandler(execution)) {
////      createCompensateEventSubscription(execution);
////    }
////    if (!hasLoopCharacteristics()) {
////      super.leave(execution);
////    } else if (hasMultiInstanceCharacteristics()){
////      multiInstanceActivityBehavior.leave(execution);
////    }
//	  leave(execution,null);
//  }
  
  /**
   * added by biaoping.yin
   * @param execution
   * @param destinationTaskKey
   */
  protected void leave(ActivityExecution execution) {
	    if(hasCompensationHandler(execution)) {
	      createCompensateEventSubscription(execution);
	    }
	    if (!hasLoopCharacteristics( execution)) {
//	    	if(destinationTaskKey == null || "".equals(destinationTaskKey))
//	    		super.leave(execution);
//	    	else
	    		super.leave(execution);
	    } else if (hasMultiInstanceCharacteristics( execution)){
//	    	if(destinationTaskKey == null || "".equals(destinationTaskKey))
//	    		multiInstanceActivityBehavior.leave(execution);
//	    	else
	    		multiInstanceActivityBehavior.leave(execution);
	      
	    }
	  }
  
  protected boolean hasCompensationHandler(ActivityExecution execution) {
    return execution.getActivity().getProperty(BpmnParse.PROPERTYNAME_COMPENSATION_HANDLER_ID) != null;
  }

  protected void createCompensateEventSubscription(ActivityExecution execution) {
    String compensationHandlerId = (String) execution.getActivity().getProperty(BpmnParse.PROPERTYNAME_COMPENSATION_HANDLER_ID);
    
    ExecutionEntity executionEntity = (ExecutionEntity) execution;    
    ActivityImpl compensationHandlder = executionEntity.getProcessDefinition().findActivity(compensationHandlerId);
    PvmScope scopeActivitiy = compensationHandlder.getParent(); 
    ExecutionEntity scopeExecution = ScopeUtil.findScopeExecutionForScope(executionEntity, scopeActivitiy);      

    CompensateEventSubscriptionEntity compensateEventSubscriptionEntity = CompensateEventSubscriptionEntity.createAndInsert(scopeExecution);
    compensateEventSubscriptionEntity.setActivity(compensationHandlder);        
  }

  protected boolean hasLoopCharacteristics(ActivityExecution execution) {
    return hasMultiInstanceCharacteristics(execution);
  }
  
  protected boolean hasMultiInstanceCharacteristics(ActivityExecution execution) {
	  if(useMixUsetask)
	  {
		  return execution.getTaskContext().isIsmulti();
	  }
	  else
		  return multiInstanceActivityBehavior != null;
  }

  public MultiInstanceActivityBehavior getMultiInstanceActivityBehavior() {
    return multiInstanceActivityBehavior;
  }
  
  public void setMultiInstanceActivityBehavior(MultiInstanceActivityBehavior multiInstanceActivityBehavior) {
    this.multiInstanceActivityBehavior = multiInstanceActivityBehavior;
  }
  
  @Override
  public void signal(ActivityExecution execution, String signalName, Object signalData) throws Exception {
    if("compensationDone".equals(signalName)) {
      signalCompensationDone(execution, signalData);
    } else {
      super.signal(execution, signalName, signalData);
    }
  }

  protected void signalCompensationDone(ActivityExecution execution, Object signalData) {
    // default behavior is to join compensating executions and propagate the signal if all executions 
    // have compensated
    
    // join compensating executions    
    if(execution.getExecutions().isEmpty()) {
      if(execution.getParent() != null) {
        ActivityExecution parent = execution.getParent();
        ((InterpretableExecution)execution).remove();
        ((InterpretableExecution)parent).signal("compensationDone", signalData);
      }      
    } else {      
      ((ExecutionEntity)execution).forceUpdate();  
    }
    
  }
  public boolean isUseMixUsetask() {
		return useMixUsetask;
	}
	public void setUseMixUsetask(boolean useMixUsetask) {
		this.useMixUsetask = useMixUsetask;
	}

	public String getCollectionElementVariable() {
		return collectionElementVariable;
	}

	public void setCollectionElementVariable(String collectionElementVariable) {
		this.collectionElementVariable = collectionElementVariable;
	}
}