/*
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you 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.apache.synapse.config.xml;

import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMNamespace;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.synapse.SynapseException;
import org.apache.synapse.util.xpath.SynapseXPath;

/**
 * 
 */
public class SynapsePathSerializer {

    private static final Log log = LogFactory.getLog(SynapsePathSerializer.class);

    public static OMElement serializePath(SynapsePath path, OMElement elem, String attribName) {

        OMNamespace nullNS = elem.getOMFactory()
            .createOMNamespace(XMLConfigConstants.NULL_NAMESPACE, "");

        if (path != null) {

            elem.addAttribute(elem.getOMFactory().createOMAttribute(
                    attribName, nullNS, path.toString()));

            serializeNamespaces(elem, path);

        } else {
            handleException("Couldn't find the xpath in the SynapseXPath");
        }

        return elem;
    }

    public static OMElement serializePath(SynapsePath path, String expression,
                                           OMElement elem, String attribName) {

        OMNamespace nullNS = elem.getOMFactory()
            .createOMNamespace(XMLConfigConstants.NULL_NAMESPACE, "");

        if (path != null && expression != null) {

            if(path.getPathType() == SynapsePath.JSON_PATH) {
                if (expression.contains("json-eval(")) {
                    elem.addAttribute(elem.getOMFactory().createOMAttribute(
                            attribName, nullNS, expression));
                } else {
                    elem.addAttribute(elem.getOMFactory().createOMAttribute(
                            attribName, nullNS, "json-eval(" + expression + ")"));
                }
            } else if(path.getPathType() == SynapsePath.X_PATH) {
                elem.addAttribute(elem.getOMFactory().createOMAttribute(
                        attribName, nullNS, expression));
            } else if (path.getPathType() == SynapsePath.SYNAPSE_EXPRESSIONS_PATH) {
                if (expression.startsWith("{${") && expression.endsWith("}}")) {
                    elem.addAttribute(elem.getOMFactory().createOMAttribute(
                            attribName, nullNS, expression));
                } else if (expression.startsWith("{") && expression.endsWith("}")) {
                    elem.addAttribute(elem.getOMFactory().createOMAttribute(
                            attribName, nullNS, "{${" + expression.substring(1, expression.length() - 1) + "}}"));
                } else {
                    elem.addAttribute(elem.getOMFactory().createOMAttribute(
                            attribName, nullNS, "{${" + expression + "}"));
                }
            }

            serializeNamespaces(elem, path);

        } else {
            handleException("Couldn't find the xpath in the SynapseXPath");
        }

        return elem;
    }
    
	public static OMElement serializeTextPath(SynapsePath path, String expression,
			OMElement elem, String attribName) {

		if (path != null && expression != null) {

            if(path.getPathType() == SynapsePath.JSON_PATH) {
                if (expression.contains("json-eval(")) {
                    // JSON evaluation expression as a value. eg: {json-eval($.)}
                    elem.setText(expression);
                } else {
                    elem.setText("json-eval(" + expression + ")");
                }
            } else if(path.getPathType() == SynapsePath.X_PATH) {
                elem.setText(expression);
            }

			serializeNamespaces(elem, path);

		} else {
			handleException("Couldn't find the xpath in the SynapseXPath");
		}

		return elem;
	}

    public static OMElement serializePathWithBraces(SynapsePath path, OMElement elem, String attribName) {

        OMNamespace nullNS = elem.getOMFactory()
                .createOMNamespace(XMLConfigConstants.NULL_NAMESPACE, "");

        if (path != null) {

            elem.addAttribute(elem.getOMFactory().createOMAttribute(
                    attribName, nullNS, '{'+path.toString()+'}'));

            serializeNamespaces(elem, path);

        } else {
            handleException("Couldn't find the xpath in the SynapseXPath");
        }

        return elem;
    }


    private static void serializeNamespaces(OMElement elem, SynapsePath path) {

        for (Object o : path.getNamespaces().keySet()) {
            String prefix = (String) o;
            String uri = path.getNamespaceContext().translateNamespacePrefixToUri(prefix);
            if (!XMLConfigConstants.SYNAPSE_NAMESPACE.equals(uri)) {
                elem.declareNamespace(uri, prefix);
            }
        }
    }

    private static void handleException(String message) {
        log.error(message);
        throw new SynapseException(message); 
    }
}
