Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import java.util.List;

import in.shop2020.model.v1.user.Cart;

import in.shop2020.model.v1.user.Line;

import in.shop2020.model.v1.user.ShoppingCartException;

import in.shop2020.payments.PaymentException;

import in.shop2020.payments.PaymentService.Client;

import in.shop2020.serving.controllers.BaseController;

import in.shop2020.serving.utils.Utils;

import in.shop2020.thrift.clients.PaymentServiceClient;

import in.shop2020.thrift.clients.UserContextServiceClient;

import org.apache.log4j.Logger;

import org.apache.struts2.convention.annotation.InterceptorRef;

import org.apache.struts2.convention.annotation.InterceptorRefs;

import org.apache.struts2.convention.annotation.Result;

import org.apache.struts2.convention.annotation.Results;

import org.apache.thrift.TException;

@InterceptorRefs({

    @InterceptorRef("myDefault"),

    @InterceptorRef("login")

})

@Results({

    @Result(name="payredirect", type="redirectAction", 

			params = {"actionName" , "${url}", "paymentid", "${pid}", "txnid", "${txn}"}),

    @Result(name="shipping-redirect", type="redirectAction", 

    		params = {"actionName" , "shipping"})

})

public class OrderController extends BaseController {

	private static final long serialVersionUID = 1L;

	private static Logger log = Logger.getLogger(Class.class);

	private String id;

	private long txnId = 0;

	//FIXME right now only one PG. Once we will have more, need to fix it.

	private String paymentUrl="hdfc-pay";

	private int gatewayId=1;

	private long paymentId;

	private double amount;

	public OrderController(){

		super();

	}

    // GET /order/ orderid

    public String show() {

    	log.info("id=" + id);

    	htmlSnippets.put("MYACCOUNT_HEADER", pageLoader.getMyaccountHeaderHtml());

    	htmlSnippets.put("ORDER_DETAILS", pageLoader.getOrderDetailsHtml(Long.parseLong(id)));

    	return "show";

    }

    // POST /order/

    public String create(){

    	String addressIdString = this.request.getParameter("addressid");

    	if(addressIdString == null){

    		addActionError("Please specify shipping address to continue.");

    		return "shipping-redirect";

    	}

    	long addressId = Long.parseLong(addressIdString);

    	long currentCartId = userinfo.getCartId();

    	try{

    		if(!createOrdersAndPayment(addressId, currentCartId)){

    			addActionError("We are experiencing problem. Please try later.");

    			return "shipping-redirect";

    		}

    	}catch (Exception e) {

    		addActionError("We are experiencing problem. Please try later.");

    		log.error("Exception in createOrders function. Something want wrong.");

    		e.printStackTrace();

    		return "shipping-redirect";

		}

		return "payredirect";

    }

	public String getId(){

		return id;

	}

	public void setId(String id){

		this.id = id;

	}

	public String getMyaccountHeaderSnippet(){

		return htmlSnippets.get("MYACCOUNT_HEADER");

	}

	public String getOrderDetailsSnippet(){

		return htmlSnippets.get("ORDER_DETAILS");

	}

	public String getUrl(){

		return this.paymentUrl;

	}

	public long getPid(){

		return this.paymentId;

	}

	public long getTxn(){

		return this.txnId;

	}

	public double getAmount(){

		return this.amount;

	}

	private double getPaymentAmount(long cartId) throws ShoppingCartException, TException{

		double totalAmount = 0;

		Cart cart;

		UserContextServiceClient userContextServiceClient = null;

		try {

			userContextServiceClient = new UserContextServiceClient();

		} catch (Exception e) {

			e.printStackTrace();

		}

		in.shop2020.model.v1.user.UserContextService.Client userClient = userContextServiceClient.getClient();

		cart = userClient.getCart(cartId);

		List<Line> lineItems = cart.getLines(); 

		for (Line line : lineItems) {

			long productId = line.getItemId();

			totalAmount =  totalAmount + line.getQuantity() * Utils.getItemPrice(productId);

		}

		return totalAmount;

	}

	/**

	 * 

	 * @param addressId

	 * @param currentCartId

	 * @return

	 */

	private boolean createOrdersAndPayment(long addressId, long currentCartId){

		UserContextServiceClient userServiceClient = null;

		try {

			userServiceClient = new UserContextServiceClient();

		} catch (Exception e) {

			e.printStackTrace();

			return false;

		}

		in.shop2020.model.v1.user.UserContextService.Client userClient = userServiceClient.getClient();

		try {

			userClient.addAddressToCart(currentCartId, addressId);

		} catch (ShoppingCartException e1) {

			log.error("Not able to set address in the cart." + e1.getId() + e1.getMessage());

			e1.printStackTrace();

			return false;

		} catch (TException e1) {

			log.error("Thrift exception while setting address in cart." + e1.getMessage());

			e1.printStackTrace();

			return false;

		}

		try {

			if(!userClient.validateCart(currentCartId)){

				addActionError("Your cart has been updated.");

				return false;

			}

		} catch (ShoppingCartException e1) {

			log.error("Error while validating shopping cart." + e1.getId() + e1.getMessage());

			e1.printStackTrace();

			return false;

		} catch (TException e) {

			log.error("Thrift exception while validating cart." + e.getMessage());

			e.printStackTrace();

			return false;

		}

		try {

			txnId = userClient.createOrders(currentCartId);

		} catch (ShoppingCartException e1) {

			log.error("Error while creating orders from cart." + e1.getId() + e1.getMessage());

			e1.printStackTrace();

			return false;

		} catch (TException e) {

			log.error("Thrift exception while creating orders from cart." + e.getMessage());

			e.printStackTrace();

			return false;

		}

		PaymentServiceClient paymentServiceClient = null;

		try {

			paymentServiceClient = new PaymentServiceClient();

		} catch (Exception e) {

			log.error("Error while getting payment client");

			e.printStackTrace();

			return false;

		}

		try {

			amount = getPaymentAmount(userinfo.getCartId());

		} catch (ShoppingCartException e1) {

			log.error("Not able to fetch payment amount from cart id." + e1.getId() + e1.getMessage());

			e1.printStackTrace();

			return false;

		} catch (TException e1) {

			log.error("Not able to fetch payment amount." + e1.getMessage());

			e1.printStackTrace();

			return false;

		}

		Client paymentClient = paymentServiceClient.getClient();

		try {

			this.paymentId = paymentClient.createPayment(userinfo.getUserId(), amount, gatewayId, txnId);

		} catch (PaymentException e1) {

			log.error("Not able to create payment object." + e1.getError_code() + e1.getMessage());

			e1.printStackTrace();

			return false;

		} catch (TException e) {

			log.error("Not able to create payment object." + e.getMessage());

			e.printStackTrace();

			return false;

		}

		return true;

	}

}