org.drools.bpmn2.xml.EndNodeHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of drools-bpmn2
There is a newer version: 5.1.1
/**
 * Copyright 2010 JBoss Inc
 *
 * 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.drools.bpmn2.xml;

import java.util.List;

import org.drools.compiler.xml.XmlDumper;
import org.drools.workflow.core.DroolsAction;
import org.drools.workflow.core.Node;
import org.drools.workflow.core.impl.DroolsConsequenceAction;
import org.drools.workflow.core.node.EndNode;
import org.xml.sax.Attributes;

public class EndNodeHandler extends AbstractNodeHandler {
    
    protected Node createNode(Attributes attrs) {
        throw new IllegalArgumentException("Reading in should be handled by end event handler");
    }
    
    @SuppressWarnings("unchecked")
	public Class generateNodeFor() {
        return EndNode.class;
    }

    public void writeNode(Node node, StringBuilder xmlDump, boolean includeMeta) {
		EndNode endNode = (EndNode) node;
		writeNode("endEvent", endNode, xmlDump, includeMeta);
		if (endNode.isTerminate()) {
    		xmlDump.append(">" + EOL);
            xmlDump.append("        " + EOL);
    		endNode("endEvent", xmlDump);
		} else {
		    List actions = endNode.getActions(EndNode.EVENT_NODE_ENTER);
		    if (actions != null && !actions.isEmpty()) {
		        if (actions.size() == 1) {
		            DroolsConsequenceAction action = (DroolsConsequenceAction) actions.get(0);
		            String s = action.getConsequence();
		            if (s.startsWith("org.drools.process.instance.impl.WorkItemImpl workItem = new org.drools.process.instance.impl.WorkItemImpl();")) {
		                xmlDump.append(">" + EOL);
                        String variable = (String) endNode.getMetaData("MappingVariable");
                        if (variable != null) {
                            xmlDump.append(
                                "      " + EOL + 
                                "      " + EOL + 
                                "        " + XmlDumper.replaceIllegalChars(variable) + "" + EOL + 
                                "        " + XmlBPMNProcessDumper.getUniqueNodeId(endNode) + "_Input" + EOL + 
                                "      " + EOL + 
                                "      " + EOL + 
                                "        " + XmlBPMNProcessDumper.getUniqueNodeId(endNode) + "_Input" + EOL + 
                                "      " + EOL);
                        }
                        xmlDump.append("      " + EOL);
                        endNode("endEvent", xmlDump);
		            } else if (s.startsWith("kcontext.getKnowledgeRuntime().signalEvent(\"")) {
                        xmlDump.append(">" + EOL);
		                s = s.substring(44);
		                String type = s.substring(0, s.indexOf("\""));
		                s = s.substring(s.indexOf(",") + 2);
		                String variable = null;
		                if (!s.startsWith("null")) {
		                    variable = s.substring(0, s.indexOf(")"));
	                        xmlDump.append(
                                "      " + EOL + 
                                "      " + EOL + 
                                "        " + XmlDumper.replaceIllegalChars(variable) + "" + EOL + 
                                "        " + XmlBPMNProcessDumper.getUniqueNodeId(endNode) + "_Input" + EOL + 
                                "      " + EOL + 
                                "      " + EOL + 
                                "        " + XmlBPMNProcessDumper.getUniqueNodeId(endNode) + "_Input" + EOL + 
                                "      " + EOL);
	                    }
		                if (type.startsWith("Compensate-")) {
			                xmlDump.append("      " + EOL);
		                } else {
		                	xmlDump.append("      " + EOL);
		                }
		                endNode("endEvent", xmlDump);
		            } else if (s.startsWith("kcontext.getProcessInstance().signalEvent(\"")) {
		            	xmlDump.append(">" + EOL);
		            	s = s.substring(43);
		                String type = s.substring(0, s.indexOf("\""));
		                xmlDump.append("      " + EOL);
		                endNode("endEvent", xmlDump);
		            } else {
		                throw new IllegalArgumentException("Unknown action " + s);
		            }
		        }
		    } else {
		        endNode(xmlDump);
		    }
		}
	}

}