/*
 * Copyright (c) 2000, 2020 Oracle and/or its affiliates. All rights reserved.
 * Copyright (c) 2021 Contributors to the Eclipse Foundation
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.messaging.ums.service;

import com.sun.messaging.xml.MessageTransformer;
import com.sun.messaging.ums.common.MessageUtil;

import java.util.Properties;

import jakarta.jms.BytesMessage;
import jakarta.jms.JMSException;
import jakarta.jms.Message;
import jakarta.jms.MessageConsumer;
import jakarta.jms.TextMessage;
import jakarta.xml.soap.MessageFactory;
import jakarta.xml.soap.SOAPConstants;
import jakarta.xml.soap.SOAPMessage;

import com.sun.messaging.ums.common.Constants;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import jakarta.jms.MessageProducer;

public class ReceiveServiceImpl implements ReceiveService {

    // private Lock lock = null;
    private Properties props = null;
    private ClientPool cache = null;
    // default receive timeout
    private long receiveTimeout = 7000;
    private Logger logger = UMSServiceImpl.logger;
    private MessageFactory soapMF = null;

    private static final String UMS_DMQ = "UMS.DMQ";

    public ReceiveServiceImpl(String provider, ClientPool cache, CacheSweeper sweeper, Properties p) throws JMSException {

        this.props = p;

        // lock = new Lock();

        // cache = new ClientPool(provider, SERVICE_NAME, props, lock);

        // sweeper.addClientPool(cache);

        this.cache = cache;

        String tmp = props.getProperty(Constants.IMQ_RECEIVE_TIMEOUT, Constants.IMQ_RECEIVE_TIMEOUT_DEFAULT_VALUE);
        receiveTimeout = Long.parseLong(tmp);

        initSOAPMessageFactory();

    }

    private void initSOAPMessageFactory() throws JMSException {

        try {
            soapMF = MessageFactory.newInstance(SOAPConstants.DEFAULT_SOAP_PROTOCOL);
        } catch (Exception ex) {

            JMSException jmse = new JMSException(ex.getMessage());
            jmse.setLinkedException(ex);

            throw jmse;
        }
    }

    @Override
    public SOAPMessage receive(SOAPMessage request) throws JMSException {
        SOAPMessage reply = null;
        Client client = null;

        try {

            /**
             * String clientId = MessageUtil.getServiceClientId(request);
             *
             * if (clientId == null) { user = MessageUtil.getServiceAttribute(request, "user"); pass =
             * MessageUtil.getServiceAttribute(request, "password"); }
             *
             * client = cache.getClient(clientId, user, pass);
             **/

            client = cache.getClient(request);

            // Object syncObj = lock.getLock(client.getId());
            Object syncObj = client.getLock();

            // Object syncObj = lock.getLock(clientId);

            MessageConsumer consumer = null;

            Message jmsMessage = null;

            if (UMSServiceImpl.debug) {
                logger.info("Receiving message ...");
            }

            synchronized (syncObj) {

                // client = cache.getClient(clientId);

                boolean isTopic = MessageUtil.isServiceTopicDomain(request);

                String destName = MessageUtil.getServiceDestinationName(request);

                consumer = client.getConsumer(isTopic, destName);

                long timeout = getTimeout(request);

                jmsMessage = consumer.receive(timeout);
            }

            // logger.info ("ReceiveService received message ...");

            if (jmsMessage == null) {

                if (UMSServiceImpl.debug) {
                    logger.info("No messages received ...");
                }

            } else if (jmsMessage instanceof BytesMessage) {

                reply = MessageTransformer.SOAPMessageFromJMSMessage(jmsMessage, soapMF);

                if (UMSServiceImpl.debug) {
                    logger.info("received soap message : " + reply);
                }
                // reply.writeTo(System.out);

            } else {
                // XXX I18N
                logger.warning("received message is not a JMS BytesMessage type: " + jmsMessage);
                this.sendToDMQ(client, jmsMessage);
            }

        } catch (Throwable ex) {

            ex.printStackTrace();

            if (ex instanceof JMSException) {
                throw (JMSException) ex;
            } else {
                JMSException jmse = new JMSException(ex.getMessage());
                if (ex instanceof Exception) {
                    jmse.setLinkedException((Exception) ex);
                }
                throw jmse;
            }

        } finally {
            cache.returnClient(client);
        }

        return reply;
    }

    private long getTimeout(SOAPMessage sm) {
        long timeout = 30000;

        try {
            timeout = MessageUtil.getServiceReceiveTimeout(sm);

            if (timeout <= 0 || timeout >= 60000) {
                timeout = this.receiveTimeout;

                // XXX I18N
                logger.info("timeout value is between 0 (exclusive) and 60000 (inclusive) milli secs.");
            }

        } catch (Exception e) {
            logger.log(Level.WARNING, e.getMessage(), e);
        }

        return timeout;
    }

    /**
     * receive a TextMessage from MQ.
     *
     * @param clientId
     * @param destName
     * @param isTopic
     * @return
     * @throws jakarta.jms.JMSException
     */
    public String receiveText(String sid, String destName, boolean isTopic, long timeout, Map map) throws JMSException {
        String reply = null;

        Client client = null;

        try {

            /**
             * if (clientId == null) {
             *
             * String[] ua = (String[]) map.get("user");
             *
             * if (ua != null && ua.length == 1) { user = ua[0]; }
             *
             * String[] pa = (String[]) map.get("password"); if (pa != null && pa.length == 1) { pass = pa[0]; } }
             *
             * client = cache.getClient(clientId, user, pass);
             **/
            client = cache.getClient(sid, map);
            // Object syncObj = lock.getLock(client.getId());
            Object syncObj = client.getLock();

            MessageConsumer consumer = null;

            Message jmsMessage = null;

            if (UMSServiceImpl.debug) {
                logger.info("ReceiveService receiving Text message ...");
            }

            synchronized (syncObj) {

                consumer = client.getConsumer(isTopic, destName);

                jmsMessage = consumer.receive(timeout);
            }

            if (UMSServiceImpl.debug) {
                logger.info("ReceiveService received message ..." + jmsMessage);
            }

            if (jmsMessage == null) {

                if (UMSServiceImpl.debug) {
                    logger.info("No messages received ...");
                }

            } else if (jmsMessage instanceof TextMessage) {

                reply = ((TextMessage) jmsMessage).getText();

                if (UMSServiceImpl.debug) {
                    logger.info("received text message : " + reply);
                }

            } else {

                // XXX I18N
                logger.warning("received message is not a TextMessage type, message=" + jmsMessage);
                this.sendToDMQ(client, jmsMessage);
            }

        } catch (Exception ex) {

            logger.log(Level.WARNING, ex.getMessage(), ex);

            if (ex instanceof JMSException) {
                throw (JMSException) ex;
            } else {
                JMSException jmse = new JMSException(ex.getMessage());
                jmse.setLinkedException(ex);
                throw jmse;
            }

        } finally {
            cache.returnClient(client);
        }

        return reply;
    }

    /**
     * Messages unable to process is sent to UMS_DMQ
     *
     * @param client
     * @param message
     */
    public void sendToDMQ(Client client, Message message) throws JMSException {

        jakarta.jms.Destination dmq = cache.getJMSDestination(UMS_DMQ, false);
        MessageProducer producer = client.getProducer();

        producer.send(dmq, message);

        // if (UMSServiceImpl.debug) {
        logger.info("Message sent to DMQ, destination=" + UMS_DMQ + ", message=" + message);
        // }

    }

    @Override
    public void close() {

        try {

            // sweeper.removeClientPool(cache);

            this.cache.close();
        } catch (Exception e) {
            logger.log(Level.WARNING, e.getMessage(), e);
        }
    }

}
