/*
  * JBoss, Home of Professional Open Source
  * Copyright 2005, JBoss Inc., and individual contributors as indicated
  * by the @authors tag. See the copyright.txt in the distribution for a
  * full listing of individual contributors.
  *
  * This is free software; you can redistribute it and/or modify it
  * under the terms of the GNU Lesser General Public License as
  * published by the Free Software Foundation; either version 2.1 of
  * the License, or (at your option) any later version.
  *
  * This software is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this software; if not, write to the Free
  * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
  */
package org.jboss.ejb3.mdb;

import java.io.Externalizable;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Hashtable;
import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import javax.jms.Destination;
import javax.jms.JMSException;
import javax.jms.MessageProducer;
import javax.jms.ObjectMessage;
import javax.jms.Session;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import org.jboss.annotation.ejb.DeliveryMode;
import org.jboss.annotation.ejb.MessageProperties;
import org.jboss.annotation.ejb.Producer;
import org.jboss.aop.advice.Interceptor;
import org.jboss.aop.joinpoint.Invocation;
import org.jboss.aop.joinpoint.MethodInvocation;
import org.jboss.ejb3.EJB3Util;

/**
 * comment
 *
 * @author <a href="mailto:bill@jboss.org">Bill Burke</a>
 */
public class ProducerManagerImpl implements ProducerManager, Externalizable, Interceptor
{
   private static final int PERSISTENT = javax.jms.DeliveryMode.PERSISTENT;
   private static final int NON_PERSISTENT = javax.jms.DeliveryMode.NON_PERSISTENT;
   
   protected Producer producer;
   protected Destination destination;
   protected String factoryLookup;

   protected int deliveryMode = javax.jms.DeliveryMode.PERSISTENT;
   protected int timeToLive = 0;
   protected int priority = 4;
   protected HashMap methodMap;

   protected transient ConnectionFactory factory;
   protected transient Connection connection;
   protected transient Session session;
   protected transient MessageProducer msgProducer;
   protected transient String username;
   protected transient String password;
   protected transient InitialContext initialContext;
   protected Hashtable initialContextProperties;

   public void writeExternal(ObjectOutput out) throws IOException
   {
      out.writeObject(producer);
      out.writeObject(destination);
      out.writeObject(factoryLookup);
      out.writeInt(deliveryMode);
      out.writeInt(timeToLive);
      out.writeInt(priority);
      out.writeObject(methodMap);
      out.writeObject(initialContextProperties);
      if (factoryLookup == null)
      {
         out.writeObject(factory);
      }
   }

   public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException
   {
      producer = (Producer) in.readObject();
      destination = (Destination) in.readObject();
      factoryLookup = (String) in.readObject();
      deliveryMode = in.readInt();
      timeToLive = in.readInt();
      priority = in.readInt();
      methodMap = (HashMap) in.readObject();
      initialContextProperties = (Hashtable)in.readObject();
      try
      {
         initialContext = EJB3Util.getInitialContext(initialContextProperties);
      }
      catch (NamingException e)
      {
         throw new RuntimeException(e);
      }
      if (factoryLookup != null)
      {
         try
         {
            factory = (ConnectionFactory) initialContext.lookup(factoryLookup);
         }
         catch (NamingException e)
         {
            throw new RuntimeException(e);
         }
      }
      else
      {
         factory = (ConnectionFactory) in.readObject();
      }
   }

   public ProducerManagerImpl(Producer producer, Destination destination, ConnectionFactory factory, DeliveryMode deliveryMode, int timeToLive, int priority, HashMap methodMap, Hashtable initialContextProperties)
   {
      this.initialContextProperties = initialContextProperties;
      try
      {
         this.initialContext = EJB3Util.getInitialContext(initialContextProperties);
      }
      catch (NamingException e)
      {
         throw new RuntimeException(e);
      }
      this.producer = producer;
      this.destination = destination;
      this.factory = factory;
      
      int mode = deliveryMode.ordinal();
      switch (mode)
      {
         case PERSISTENT:
            this.deliveryMode = javax.jms.DeliveryMode.PERSISTENT;
            break;
         case NON_PERSISTENT:
            this.deliveryMode = javax.jms.DeliveryMode.NON_PERSISTENT;
            break;
      }
      this.timeToLive = timeToLive;
      this.priority = priority;
      this.methodMap = methodMap;
   }

   public ProducerManagerImpl(Producer producer, Destination destination, String factory, DeliveryMode deliveryMode, int timeToLive, int priority, HashMap methodMap, Hashtable initialContextProperties)
   {
      this.initialContextProperties = initialContextProperties;
      try
      {
         this.initialContext = EJB3Util.getInitialContext(initialContextProperties);
      }
      catch (NamingException e)
      {
         throw new RuntimeException(e);
      }
      this.producer = producer;
      this.destination = destination;
      this.factoryLookup = factory;
      
      int mode = deliveryMode.ordinal();
      switch (mode)
      {
         case PERSISTENT:
            this.deliveryMode = javax.jms.DeliveryMode.PERSISTENT;
            break;
         case NON_PERSISTENT:
            this.deliveryMode = javax.jms.DeliveryMode.NON_PERSISTENT;
            break;
      }
      this.timeToLive = timeToLive;
      this.priority = priority;
      this.methodMap = methodMap;
   }


   public ProducerManagerImpl()
   {
   }

   public void setUsername(String user)
   {
      this.username = user;
   }

   public void setPassword(String passwd)
   {
      this.password = passwd;
   }

   public void connect() throws JMSException
   {
      if (factory == null)
      {
         try
         {
            factory = (ConnectionFactory) initialContext.lookup(factoryLookup);
         }
         catch (NamingException e)
         {
            throw new RuntimeException(e);
         }
      }
      if (connection != null) return;
      if (username != null)
      {
         connection = factory.createConnection(username, password);
      }
      else
      {
         connection = factory.createConnection();
      }
      session = connection.createSession(producer.transacted(), producer.acknowledgeMode());
      msgProducer = session.createProducer(destination);
      msgProducer.setDeliveryMode(deliveryMode);
      msgProducer.setTimeToLive(timeToLive);
      msgProducer.setPriority(priority);
   }

   public void close() throws JMSException
   {
      msgProducer.close();
      session.close();
      connection.close();
   }

   public void commit() throws JMSException
   {
      session.commit();
   }

   public void rollback() throws JMSException
   {
      session.rollback();
   }

   public String getName()
   {
      return ProducerManager.class.getName();
   }

   public Object invoke(Invocation invocation) throws Throwable
   {
      if (session == null)
      {
         throw new RuntimeException("You must call connect() on the producer.  The JMS session has not been set");
      }
      ObjectMessage msg = session.createObjectMessage((Serializable) invocation);
      MethodInvocation mi = (MethodInvocation) invocation;
      MessageProperties props = (MessageProperties)methodMap.get(new Long(mi.getMethodHash()));
      if (props != null)
      {
         int del = (props.delivery() == DeliveryMode.PERSISTENT) ? javax.jms.DeliveryMode.PERSISTENT : javax.jms.DeliveryMode.NON_PERSISTENT;
         msgProducer.send(msg, del, props.priority(), props.timeToLive());
      }
      else
      {
         msgProducer.send(msg);
      }
      return null;
   }
}
