package com.titan.processpayment;
import com.titan.customer.*;

import java.sql.*;
import java.rmi.RemoteException;
import javax.ejb.SessionContext;

import javax.naming.InitialContext;
import javax.sql.DataSource;
import javax.ejb.EJBException;
import javax.ejb.CreateException;
import javax.naming.NamingException;

public class ProcessPaymentBean implements javax.ejb.SessionBean {

    final public static String CASH = "CASH";
    final public static String CREDIT = "CREDIT";
    final public static String CHECK = "CHECK";
        
    public SessionContext context;

    public void ejbCreate() throws CreateException {
    }
    
    public boolean byCash(CustomerRemote customer, 
                          double amount)
    throws PaymentException{
        return process(getCustomerID(customer), amount, CASH, null, -1, null, null);
    }
    
    public boolean byCheck(CustomerRemote customer, 
                           CheckDO check, double amount)
    throws PaymentException{
        int minCheckNumber = getMinCheckNumber();
        if (check.checkNumber > minCheckNumber) {
          return process(getCustomerID(customer), amount, CHECK, 
                       check.checkBarCode, check.checkNumber, null, null);
        }
        else {
           throw new PaymentException("Check number is too low. Must be at least "+minCheckNumber);
        }
    }

    public boolean byCredit(CustomerRemote customer, 
                            CreditCardDO card, double amount)
    throws PaymentException {
        if (card.expiration.before(new java.util.Date())) {
           throw new PaymentException("Expiration date has passed");
        }
        else {
           return process(getCustomerID(customer), amount, 
                          CREDIT, null, -1, card.number, new java.sql.Date(card.expiration.getTime()));
        }
    }

    private boolean process(Integer customerID, double amount,
                            String type, String checkBarCode, 
                            int checkNumber, String creditNumber, 
                            java.sql.Date creditExpDate)
        throws PaymentException{

		System.out.println("process() with customerID="+customerID+" amount="+amount);
        Connection con = null;
        PreparedStatement ps = null;

        try {
            con = getConnection();
            ps = con.prepareStatement
               ("INSERT INTO \"PaymentBeanTable\" (\"_customer_id\", \"amount\", \"type\", \"checkBarCode\", \"checkNumber\", \"creditNumber\", \"creditExpDate\")"+
                  " VALUES (?,?,?,?,?,?,?)");
            ps.setInt(1,customerID.intValue());
            ps.setDouble(2,amount);
            ps.setString(3,type);
            ps.setString(4,checkBarCode);
            ps.setInt(5,checkNumber);
            ps.setString(6,creditNumber);
            ps.setDate(7,creditExpDate);
            int retVal = ps.executeUpdate();
            if (retVal!=1) {
                throw new EJBException("Payment insert failed");
            }         
            return true;
        } catch(SQLException sql) {
            throw new EJBException(sql);
        } finally {
             try {
                 if (ps != null) ps.close();
                 if (con!= null) con.close();
             } catch(SQLException se){se.printStackTrace();}
        }
    }

    public void ejbActivate() {}
    public void ejbPassivate() {}
    public void ejbRemove() {}
    public void setSessionContext(SessionContext ctx) {
        context = ctx;
    }

    private Integer getCustomerID(CustomerRemote customer) {
        try {
            return (Integer)customer.getPrimaryKey();
        } catch(RemoteException re) {
            throw new EJBException(re);
        }
    }

	private Connection getConnection() throws SQLException {
		try {
			InitialContext jndiCntx = new InitialContext();
			DataSource ds = (DataSource)
				jndiCntx.lookup("java:comp/env/jdbc/Titan");
			return ds.getConnection();
		} catch(NamingException ne){throw new EJBException(ne);}
	}

	private int getMinCheckNumber() {
		try {
			InitialContext jndiCntx = new InitialContext( );
			Integer value = (Integer)
				jndiCntx.lookup("java:comp/env/minCheckNumber");
			return value.intValue();
		} catch(NamingException ne){throw new EJBException(ne);}
	}

}