/*
 * Copyright 2017 Red Hat, Inc. and/or 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.jbpm.bpmn2.xml;

import java.util.Map;

import org.jbpm.workflow.core.Constraint;
import org.jbpm.workflow.core.Node;
import org.jbpm.workflow.core.impl.ConnectionRef;
import org.jbpm.workflow.core.node.Split;
import org.kie.api.definition.process.Connection;
import org.xml.sax.Attributes;

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

	public void writeNode(Node node, StringBuilder xmlDump, int metaDataType) {
		Split split = (Split) node;
		String type = null;
		switch (split.getType()) {
			case Split.TYPE_AND:
				type = "parallelGateway";
				writeNode(type, node, xmlDump, metaDataType);
				break;
            case Split.TYPE_XOR:
                type = "exclusiveGateway";
                writeNode(type, node, xmlDump, metaDataType);
                String defaultId = (String) node.getMetaData().get("Default");
                if (defaultId != null) {
                    for(Connection connection :split.getDefaultOutgoingConnections()) {
                        if(connection.getMetaData().get("UniqueId").equals(defaultId)) {
                            xmlDump.append("default=\"" +
                                    XmlBPMNProcessDumper.getUniqueNodeId(split) + "-" +
                                    XmlBPMNProcessDumper.getUniqueNodeId(connection.getTo()) +
                                    "\" ");
                            break;
                        }
                    }

                } else {
                    for (Map.Entry<ConnectionRef, Constraint> entry : split.getConstraints().entrySet()) {
                        if (entry.getValue() != null && entry.getValue().isDefault()) {
                            xmlDump.append("default=\"" +
                                           XmlBPMNProcessDumper.getUniqueNodeId(split) + "-" +
                                           XmlBPMNProcessDumper.getUniqueNodeId(node.getNodeContainer().getNode(entry.getKey().getNodeId())) +
                                           "\" ");
                            break;
                        }
                    }
                }
                break;
			case Split.TYPE_OR:
				type = "inclusiveGateway";
                writeNode(type, node, xmlDump, metaDataType);
				for (Map.Entry<ConnectionRef, Constraint> entry: split.getConstraints().entrySet()) {
					if (entry.getValue() != null && entry.getValue().isDefault()) {
						xmlDump.append("default=\"" +
							XmlBPMNProcessDumper.getUniqueNodeId(split) + "-" +
							XmlBPMNProcessDumper.getUniqueNodeId(node.getNodeContainer().getNode(entry.getKey().getNodeId())) + 
							"\" ");
						break;
					}
				}
                break;
			case Split.TYPE_XAND:
				type = "eventBasedGateway";
				writeNode(type, node, xmlDump, metaDataType);
				break;
            default:
            	type = "complexGateway";
				writeNode(type, node, xmlDump, metaDataType);
		}
		xmlDump.append("gatewayDirection=\"Diverging\" >" + EOL);
		writeExtensionElements(node, xmlDump);
		endNode(type, xmlDump);
	}

}
