/*
 * SoapUI, Copyright (C) 2004-2016 SmartBear Software 
 *
 * Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent 
 * versions of the EUPL (the "Licence"); 
 * You may not use this work except in compliance with the Licence. 
 * You may obtain a copy of the Licence at: 
 * 
 * http://ec.europa.eu/idabc/eupl 
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the Licence is 
 * distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
 * express or implied. See the Licence for the specific language governing permissions and limitations 
 * under the Licence. 
 */

package com.eviware.soapui.impl.wsdl.submit.transports.jms.util;

import com.eviware.soapui.SoapUI;
import com.eviware.soapui.impl.wsdl.submit.transports.jms.JMSEndpoint;
import com.eviware.soapui.impl.wsdl.support.MessageExchangeModelItem;
import com.eviware.soapui.model.ModelItem;
import com.eviware.soapui.model.iface.MessageExchange;
import com.eviware.soapui.model.iface.Request;
import com.eviware.soapui.support.StringUtils;
import com.eviware.soapui.support.types.StringToStringMap;
import com.eviware.soapui.support.xml.XmlUtils;

import javax.jms.BytesMessage;
import javax.jms.JMSException;
import javax.jms.MapMessage;
import java.util.Enumeration;

public class JMSUtils {

    private static boolean checkIfJMS(Request request) {
        try {
            String endpoint = request.getEndpoint();
            return StringUtils.hasContent(endpoint) && endpoint.startsWith(JMSEndpoint.JMS_ENDPOINT_PREFIX);
        } catch (NullPointerException e) {
            SoapUI.logError(e);
        }
        return false;
    }

    private static boolean checkIfJMS(MessageExchangeModelItem messageExchange) {
        try {
            MessageExchange me = ((MessageExchangeModelItem) messageExchange).getMessageExchange();
            if (me != null) {
                StringToStringMap strmap = me.getProperties();
                if (strmap != null && strmap.containsKey("Endpoint")) {
                    String r = me.getProperty("Endpoint");
                    return r != null && r.startsWith(JMSEndpoint.JMS_ENDPOINT_PREFIX);
                } else {
                    return false;
                }
            } else {
                return false;
            }
        } catch (NullPointerException e) {
            SoapUI.logError(e);
        }
        return false;
    }

    public static boolean checkIfJMS(ModelItem modelItem) {
        if (modelItem instanceof Request) {
            return checkIfJMS((Request) modelItem);
        } else {
            if (modelItem instanceof MessageExchangeModelItem) {
                return checkIfJMS((MessageExchangeModelItem) modelItem);
            }
        }
        return false;
    }

    public static String extractMapMessagePayloadToString(MapMessage mapMessage) throws JMSException {
        StringBuffer sb = new StringBuffer();

        Enumeration<?> mapNames = mapMessage.getMapNames();

        while (mapNames.hasMoreElements()) {
            String key = (String) mapNames.nextElement();
            String value = mapMessage.getString(key);
            sb.append(key + ": " + value);
        }

        return sb.toString();
    }

    public static String extractMapMessagePayloadToXML(MapMessage mapMessage) throws JMSException {
        StringBuffer sb = new StringBuffer("<message>\n");

        Enumeration<?> mapNames = mapMessage.getMapNames();

        while (mapNames.hasMoreElements()) {
            String key = (String) mapNames.nextElement();
            String value = mapMessage.getString(key);
            sb.append("<" + key + ">" + XmlUtils.entitize(value) + "</" + key + ">\n");
        }
        sb.append("</message>");
        return sb.toString();
    }

    public static byte[] extractByteArrayFromMessage(BytesMessage message) throws JMSException {
        byte[] bytes = new byte[(int) message.getBodyLength()];
        message.readBytes(bytes);
        return bytes;
    }
}