/*
 * Copyright  1999-2004 The Apache Software Foundation.
 *
 *  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.apache.savan.eventing.client;

import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMAttribute;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMNamespace;
import org.apache.axiom.soap.*;
import org.apache.axis2.addressing.AddressingConstants;
import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.addressing.EndpointReferenceHelper;
import org.apache.axis2.client.Options;
import org.apache.axis2.client.ServiceClient;
import org.apache.axis2.databinding.types.Duration;
import org.apache.axis2.databinding.utils.ConverterUtil;
import org.apache.savan.eventing.EventingConstants;
import org.apache.savan.subscription.ExpirationBean;
import org.apache.savan.util.CommonUtil;

import javax.xml.namespace.QName;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;

public class EventingClient {

    ServiceClient serviceClient = null;
    HashMap subscriptionDataMap = null;

    public EventingClient(ServiceClient serviceClient) {
        this.serviceClient = serviceClient;
        subscriptionDataMap = new HashMap();
    }

    public void subscribe(EventingClientBean bean, String subscriptionID) throws Exception {

        Options options = serviceClient.getOptions();
        if (options == null) {
            options = new Options();
            serviceClient.setOptions(options);
        }

        String SOAPVersion = options.getSoapVersionURI();
        if (SOAPVersion == null)
            SOAPVersion = SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI;

        SOAPEnvelope envelope = createSubscriptionEnvelope(bean, SOAPVersion);

        String oldAction = options.getAction();
        String action = EventingConstants.Actions.Subscribe;
        options.setAction(action);
        OMElement subscriptionResponse =
                serviceClient.sendReceive(envelope.getBody().getFirstElement());
        SubscriptionResponseData subscriptionResponseData =
                getSubscriptionResponseData(subscriptionResponse);

        subscriptionDataMap.put(subscriptionID, subscriptionResponseData);

        options.setAction(oldAction);
    }

    public void renewSubscription(Date newExpirationTime, String subscriptionID) throws Exception {
        String expirationString = ConverterUtil.convertToString(newExpirationTime);
        renewSubscription(expirationString, subscriptionID);
    }

    public void renewSubscription(Duration duration, String subscriptionID) throws Exception {
        String expirationString = ConverterUtil.convertToString(duration);
        renewSubscription(expirationString, subscriptionID);
    }

    private void renewSubscription(String expirationString, String subscriptionID)
            throws Exception {
        SubscriptionResponseData data =
                (SubscriptionResponseData)subscriptionDataMap.get(subscriptionID);
        EndpointReference managerEPR = data.getSubscriptionManager();
        if (managerEPR == null)
            throw new Exception("Manager EPR is not set");

        Options options = serviceClient.getOptions();
        if (options == null) {
            options = new Options();
            serviceClient.setOptions(options);
        }

        String SOAPVersion = options.getSoapVersionURI();
        if (SOAPVersion == null)
            SOAPVersion = SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI;

        SOAPEnvelope envelope = createRenewSubscriptionEnvelope(expirationString, SOAPVersion);

        String oldAction = options.getAction();
        String action = EventingConstants.Actions.Renew;
        options.setAction(action);

        EndpointReference oldTo = serviceClient.getOptions().getTo();
        options.setTo(managerEPR);

        OMElement renewResponse = serviceClient.sendReceive(envelope.getBody().getFirstElement());

        options.setAction(oldAction);
        options.setTo(oldTo);
    }

    public void unsubscribe(String subscriptionID) throws Exception {
        SubscriptionResponseData data =
                (SubscriptionResponseData)subscriptionDataMap.get(subscriptionID);
        EndpointReference managerEPR = data.getSubscriptionManager();
        if (managerEPR == null)
            throw new Exception("Manager EPR is not set");

        Options options = serviceClient.getOptions();
        if (options == null) {
            options = new Options();
            serviceClient.setOptions(options);
        }

        String SOAPVersion = options.getSoapVersionURI();
        if (SOAPVersion == null)
            SOAPVersion = SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI;

        SOAPEnvelope envelope = createUnsubscriptionEnvelope(SOAPVersion);

        String oldAction = options.getAction();
        String action = EventingConstants.Actions.Unsubscribe;
        options.setAction(action);

        EndpointReference oldTo = serviceClient.getOptions().getTo();
        options.setTo(managerEPR);

        OMElement unsubscribeResponse =
                serviceClient.sendReceive(envelope.getBody().getFirstElement());
        //TODO process unsubscriber response

        options.setAction(oldAction);
        options.setTo(oldTo);
    }

    public SubscriptionStatus getSubscriptionStatus(String subscriptionID) throws Exception {
        SubscriptionResponseData data =
                (SubscriptionResponseData)subscriptionDataMap.get(subscriptionID);
        EndpointReference managerEPR = data.getSubscriptionManager();
        if (managerEPR == null)
            throw new Exception("Manager EPR is not set");

        Options options = serviceClient.getOptions();
        if (options == null) {
            options = new Options();
            serviceClient.setOptions(options);
        }

        String SOAPVersion = options.getSoapVersionURI();
        if (SOAPVersion == null)
            SOAPVersion = SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI;

        SOAPEnvelope envelope = createGetStatusEnvelope(SOAPVersion);

        String oldAction = options.getAction();
        String action = EventingConstants.Actions.GetStatus;
        options.setAction(action);

        EndpointReference oldTo = serviceClient.getOptions().getTo();
        options.setTo(managerEPR);

        OMElement getStatusResponse =
                serviceClient.sendReceive(envelope.getBody().getFirstElement());
        SubscriptionStatus subscriptionStatus = getSubscriptionStatus(getStatusResponse);

        options.setAction(oldAction);
        options.setTo(oldTo);

        return subscriptionStatus;
    }

    private SubscriptionResponseData getSubscriptionResponseData(OMElement responseMessagePayload)
            throws Exception {
        SubscriptionResponseData data = new SubscriptionResponseData();

        OMElement subscriberManagerElement = responseMessagePayload.getFirstChildWithName(new QName(
                EventingConstants.EVENTING_NAMESPACE,
                EventingConstants.ElementNames.SubscriptionManager));
        EndpointReference managerEPR = EndpointReferenceHelper.fromOM(subscriberManagerElement);
        data.setSubscriptionManager(managerEPR);

        OMElement expiresElement = responseMessagePayload.getFirstChildWithName(new QName(
                EventingConstants.EVENTING_NAMESPACE, EventingConstants.ElementNames.Expires));
        if (expiresElement != null) {
            String text = expiresElement.getText().trim();

            ExpirationBean expirationBean = new ExpirationBean();
            if (CommonUtil.isDuration(text)) {
                expirationBean.setDuration(true);
                Duration duration = ConverterUtil.convertToDuration(text);
                expirationBean.setDurationValue(duration);
            } else {
                expirationBean.setDuration(false);
                Date date = ConverterUtil.convertToDateTime(text).getTime();
                expirationBean.setDateValue(date);
            }

            data.setExpiration(expirationBean);
        }

        return data;
    }

    private SubscriptionStatus getSubscriptionStatus(OMElement getStatusResponseElement)
            throws Exception {
        SubscriptionStatus subscriptionStatus = new SubscriptionStatus();

        OMElement expiresElementElement = getStatusResponseElement.getFirstChildWithName(new QName(
                EventingConstants.EVENTING_NAMESPACE, EventingConstants.ElementNames.Expires));
        if (expiresElementElement != null) {
            String valueStr = expiresElementElement.getText();
//			long expires = Long.parseLong(valueStr);
            subscriptionStatus.setExpirationValue(valueStr);
        }

        return subscriptionStatus;
    }

    private SOAPEnvelope createSubscriptionEnvelope(EventingClientBean bean, String SOAPVersion)
            throws Exception {
        SOAPFactory factory = null;

        if (SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP11Factory();
        else if (SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP12Factory();
        else throw new Exception("Unknown SOAP version");

        SOAPEnvelope envelope = factory.getDefaultEnvelope();
        SOAPBody body = envelope.getBody();

        OMNamespace ens = factory.createOMNamespace(EventingConstants.EVENTING_NAMESPACE,
                                                    EventingConstants.EVENTING_PREFIX);

        OMElement subscriptionElement =
                factory.createOMElement(EventingConstants.ElementNames.Subscribe, ens);

        EndpointReference endToEPR = bean.getEndToEPR();
        if (bean.getEndToEPR() != null) {
            OMElement endToElement = EndpointReferenceHelper.toOM(
                    subscriptionElement.getOMFactory(), endToEPR, new QName(
                    EventingConstants.EVENTING_NAMESPACE, EventingConstants.ElementNames.EndTo,
                    EventingConstants.EVENTING_PREFIX),
                                                                  AddressingConstants.Submission.WSA_NAMESPACE);
            subscriptionElement.addChild(endToElement);
        }

        EndpointReference deliveryEPR = bean.getDeliveryEPR();
        if (deliveryEPR == null)
            throw new Exception("Delivery EPR is not set");

        OMElement deliveryElement =
                factory.createOMElement(EventingConstants.ElementNames.Delivery, ens);
        OMElement notifyToElement = EndpointReferenceHelper.toOM(subscriptionElement.getOMFactory(),
                                                                 deliveryEPR, new QName(
                EventingConstants.EVENTING_NAMESPACE, EventingConstants.ElementNames.NotifyTo,
                EventingConstants.EVENTING_PREFIX), AddressingConstants.Submission.WSA_NAMESPACE);

        deliveryElement.addChild(notifyToElement);
        subscriptionElement.addChild(deliveryElement);

        if (bean.getExpirationTime() != null || bean.getExpirationDuration() != null) {
            String timeString = null;

            //if time is set it will be taken. Otherwise duration will be taken.
            if (bean.getExpirationTime() != null) {
                Date date = bean.getExpirationTime();
                Calendar calendar = Calendar.getInstance();
                calendar.setTime(date);
                timeString = ConverterUtil.convertToString(calendar);
            } else if (bean.getExpirationDuration() != null) {
                Duration duration = bean.getExpirationDuration();
                timeString = ConverterUtil.convertToString(duration);
            }

            OMElement expiresElement =
                    factory.createOMElement(EventingConstants.ElementNames.Expires, ens);
            expiresElement.setText(timeString);
            subscriptionElement.addChild(expiresElement);
        }

        if (bean.getFilter() != null) {
            String filter = bean.getFilter();
            String dialect = bean.getFilterDialect();

            OMElement filterElement =
                    factory.createOMElement(EventingConstants.ElementNames.Filter, ens);
            OMAttribute dialectAttr = factory.createOMAttribute(
                    EventingConstants.ElementNames.Dialect, null, dialect);
            filterElement.addAttribute(dialectAttr);
            filterElement.setText(filter);

            subscriptionElement.addChild(filterElement);
        }

        body.addChild(subscriptionElement);

        return envelope;
    }

    private SOAPEnvelope createRenewSubscriptionEnvelope(String expiresString, String SOAPVersion)
            throws Exception {
        SOAPFactory factory = null;

        if (SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP11Factory();
        else if (SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP12Factory();
        else throw new Exception("Unknown SOAP version");

        SOAPEnvelope envelope = factory.getDefaultEnvelope();
        SOAPBody body = envelope.getBody();

        OMNamespace ens = factory.createOMNamespace(EventingConstants.EVENTING_NAMESPACE,
                                                    EventingConstants.EVENTING_PREFIX);
        OMElement renewElement = factory.createOMElement(EventingConstants.ElementNames.Renew, ens);
        OMElement expiresElement =
                factory.createOMElement(EventingConstants.ElementNames.Expires, ens);
        expiresElement.setText(expiresString);
        renewElement.addChild(expiresElement);

        body.addChild(renewElement);

        return envelope;
    }

    private SOAPEnvelope createUnsubscriptionEnvelope(String SOAPVersion) throws Exception {
        SOAPFactory factory = null;

        if (SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP11Factory();
        else if (SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP12Factory();
        else throw new Exception("Unknown SOAP version");

        SOAPEnvelope envelope = factory.getDefaultEnvelope();
        SOAPBody body = envelope.getBody();

        OMNamespace ens = factory.createOMNamespace(EventingConstants.EVENTING_NAMESPACE,
                                                    EventingConstants.EVENTING_PREFIX);
        OMElement unsubscribeElement =
                factory.createOMElement(EventingConstants.ElementNames.Unsubscribe, ens);
        body.addChild(unsubscribeElement);

        return envelope;
    }

    private SOAPEnvelope createGetStatusEnvelope(String SOAPVersion) throws Exception {
        SOAPFactory factory = null;

        if (SOAP11Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP11Factory();
        else if (SOAP12Constants.SOAP_ENVELOPE_NAMESPACE_URI.equals(SOAPVersion))
            factory = OMAbstractFactory.getSOAP12Factory();
        else throw new Exception("Unknown SOAP version");

        SOAPEnvelope envelope = factory.getDefaultEnvelope();
        SOAPBody body = envelope.getBody();

        OMNamespace ens = factory.createOMNamespace(EventingConstants.EVENTING_NAMESPACE,
                                                    EventingConstants.EVENTING_PREFIX);
        OMElement getStatusElement =
                factory.createOMElement(EventingConstants.ElementNames.GetStatus, ens);
        body.addChild(getStatusElement);

        return envelope;
    }

}