/*
 * The contents of this file are subject to the terms
 * of the Common Development and Distribution License
 * (the "License").  You may not use this file except
 * in compliance with the License.
 *
 * You can obtain a copy of the license at
 * http://www.opensource.org/licenses/cddl1.php
 * See the License for the specific language governing
 * permissions and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL
 * HEADER in each file and include the License file at
 * http://www.opensource.org/licenses/cddl1.php.  If 
 * applicable, add the following below this CDDL HEADER, 
 * with the fields enclosed by brackets "[]" replaced 
 * with your own identifying information: 
 * Portions Copyright [yyyy]
 * [name of copyright owner]
 */ 

/*
 * $(@)BillingEngineBean.java $Revision: 1.1.1.1 $ $Date: 2006/03/15 13:12:10 $
 * 
 * Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
 */
/**

 *

 * @version: 1.0

 * @date: Sept 19, 2002

 *

 */

package com.sun.sjc.idtv.vod.server.billing;



import java.sql.*;

import java.util.*;

import javax.sql.*;

import javax.ejb.*;

import javax.naming.*;

import com.sun.sjc.idtv.vod.shared.data.*;



/**

 * Implementation bean for the <code>BillingEngine</code> interface, <code>BillingEngine</code>

 * defines all possible business methods for the bean.

 *

 * @see BillingEngine

 * @see BillingEngineHome

 */

public class BillingEngineBean implements SessionBean {



    private  javax.ejb.SessionContext m_ctx = null; 

    public static final int QUERYTIMEOUT = 60;

    private DataSource dataSource;



    /**

     * Sets the session context. Required by EJB spec.

     * @param ctx A SessionContext object.

     */

    public void setSessionContext(javax.ejb.SessionContext ctx) { 

        m_ctx = ctx; 

	try {

	    InitialContext ic = new InitialContext();

	    //dataSource = (DataSource) ic.lookup("java:comp/env/jdbc/vod");

	    dataSource = (DataSource) ic.lookup("jdbc/sample");

	} catch (Exception ex) {

	    throw new EJBException("Unable to connect to database. " +

				   ex.getMessage());

	}

    }



    /**

     * Creates a bean. Required by EJB spec.

     * @exception throws CreateException.

     */

    public void ejbCreate() throws java.rmi.RemoteException, javax.ejb.CreateException { 

    } 



    /**

     * Removes the bean. Required by EJB spec.

     */

    public void ejbRemove() { 

    } 



    /**

     * Loads the state of the bean from secondary storage. Required by EJB spec.

     */

    public void ejbActivate() { 

    } 



    /**

     * Serializes the state of the bean to secondary storage. Required by EJB spec.

     */

    public void ejbPassivate() { 

    } 



    /**

     * Required by EJB spec.

     */

    public void BillingEngine() { 

    } 





    /**

     * Rent a movie.

     * @param subscriber the subscriber.

     * @param movie the movie.

     * @return true if the account balance allows the rental

     * @exception RemoteException

     */

    public boolean rentMovie(Subscriber subscriber, Movie movie) throws java.rmi.RemoteException, SQLException {



	// check if already rented

	boolean currentrental = isRental(subscriber, movie);

	float balance = 0;

	if (!currentrental) {

	    // check account balance

	    balance = getAccountBalance(subscriber);

	    if (balance < movie.stdprice) {

		return false;

	    }

	}



	// log

	insertCDR(subscriber, movie);



	// bill

	if (!currentrental) {

	    setAccountBalance(subscriber, balance - movie.stdprice);

	}

	return true;

    }



    private boolean isRental(Subscriber subscriber, Movie movie) throws java.rmi.RemoteException, SQLException {



       // get db connection from pool

       Connection conn = dataSource.getConnection();



       // check if this movie is currently rented

       PreparedStatement stmt = conn.prepareStatement("SELECT * FROM subscrentals WHERE subscrid = ? AND movieid = ? AND starttimest > ? AND elapsedviewtime > ?");



       stmt.setQueryTimeout(QUERYTIMEOUT);

       if (subscriber.ismaster) {

	   stmt.setLong(1, subscriber.masterid);

       } else {

	   stmt.setLong(1, subscriber.id);

       }

       stmt.setLong(2, movie.id);

       Timestamp yesterday = new Timestamp(System.currentTimeMillis() - (24 * 60 * 60 * 1000)); // 24 hours watch window

       stmt.setTimestamp(3, yesterday);

       stmt.setLong(4, -1); // unlimited viewtime for now

       ResultSet rs = stmt.executeQuery();

       

       boolean isrental = rs.next();



       stmt.close();

       conn.close();

       return isrental;

    }



    private void insertCDR(Subscriber subscriber, Movie movie) throws SQLException {



       // get db connection from pool

       Connection conn = dataSource.getConnection();



       // insert a CDR

       PreparedStatement stmt = conn.prepareStatement("INSERT INTO billingcdr VALUES (?, ?, ?, ?)");



       stmt.setQueryTimeout(QUERYTIMEOUT);

       stmt.setLong(1, subscriber.masterid);

       stmt.setLong(2, movie.id);

       Timestamp now = new Timestamp(System.currentTimeMillis());

       stmt.setTimestamp(3, now);

       stmt.setFloat(4, movie.stdprice);

       stmt.executeUpdate();

       stmt.close();



       // rent it

       stmt = conn.prepareStatement("INSERT INTO subscrentals VALUES (?, ?, ?, ?, ?)");



       stmt.setQueryTimeout(QUERYTIMEOUT);

       if (subscriber.ismaster) {

	   stmt.setLong(1, subscriber.masterid);

       } else {

	   stmt.setLong(1, subscriber.id);

       }

       stmt.setLong(2, movie.id);

       stmt.setTimestamp(3, now);

       stmt.setLong(4, 0);

       stmt.setString(5, "");

       stmt.executeUpdate();

       stmt.close();



       conn.close();

       return;

    }



    /**

     * Get the account balance for a given subscriber.

     * @param subscriber the subscriber.

     * @return account balance

     * @exception RemoteException

     */

    public float getAccountBalance(Subscriber subscriber) throws java.rmi.RemoteException, SQLException {



       // get db connection from pool

       Connection conn = dataSource.getConnection();



       // get account balance

       PreparedStatement stmt = conn.prepareStatement("SELECT subscracct.balance FROM subscracct WHERE (subscracct.subscrid = ?)");



       stmt.setQueryTimeout(QUERYTIMEOUT);

       stmt.setLong(1, subscriber.masterid);

       ResultSet rs = stmt.executeQuery();

       float balance = 0;

       if (rs.next()) {

	   balance = rs.getFloat(1);

       }

       rs.close();

       stmt.close();

       conn.close();

       return balance;

    }



    /**

     * Set the account balance for a given subscriber.

     * @param subscriber the subscriber.

     * @exception RemoteException

     */

    public void setAccountBalance(Subscriber subscriber, float balance) throws java.rmi.RemoteException, SQLException {



       // get db connection from pool

       Connection conn = dataSource.getConnection();



       // set account balance

       PreparedStatement stmt = conn.prepareStatement("UPDATE subscracct SET balance = ? WHERE (subscracct.subscrid = ?)");



       stmt.setQueryTimeout(QUERYTIMEOUT);

       stmt.setFloat(1, balance);

       stmt.setLong(2, subscriber.masterid);

       stmt.executeUpdate();

       stmt.close();

       conn.close();

       return;

    }



    /**

     * Get the account movement history for a given subscriber.

     * @param subscriber the subscriber.

     * @param start start date for the history.

     * @param end end date for the history.

     * @return movement history

     * @exception RemoteException

     */

    public String[] getHistory(Subscriber subscriber, java.util.Date start, java.util.Date end) throws java.rmi.RemoteException, SQLException {



	// get db connection from pool

       Connection conn = dataSource.getConnection();



       // get CDRs

       //PreparedStatement stmt = conn.prepareStatement("SELECT * FROM billingcdr WHERE (billingcdr.subscrid = ?) AND (billingcdr.purchasedate >= ?) AND (billingcdr.purchasedate < ?) ORDER BY billingcdr.purchasedate");

       PreparedStatement stmt = conn.prepareStatement("SELECT * FROM billingcdr WHERE (billingcdr.subscrid = ?) ORDER BY billingcdr.purchasedate");



       stmt.setQueryTimeout(QUERYTIMEOUT);

       stmt.setLong(1, subscriber.masterid);

       //stmt.setTimestamp(2, new Timestamp(start.getTime()));

       //stmt.setTimestamp(3, new Timestamp(end.getTime()));

       ResultSet rs = stmt.executeQuery();

       Vector v = new Vector();

       while (rs.next()) {

	   String cdr = "" + rs.getString(1) + " " + rs.getString(2) + " " + rs.getString(3);

	   v.addElement(cdr);

       }

       rs.close();

       stmt.close();

       conn.close();

       String[] cdrs = new String[v.size()];

       for (int i=0; i<v.size(); i++ ) {

	   cdrs[i] = (String) v.elementAt(i);

       }

       return cdrs;

    }

}