Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import java.io.IOException;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Set;

import org.apache.struts2.rest.DefaultHttpHeaders;

import org.apache.struts2.rest.HttpHeaders;

import org.apache.thrift.TException;

import in.shop2020.model.v1.catalog.InventoryServiceException;

import in.shop2020.model.v1.catalog.Item;

import in.shop2020.model.v1.order.BillingInfo;

import in.shop2020.model.v1.order.LineItem;

import in.shop2020.model.v1.order.OrderInfo;

import in.shop2020.model.v1.order.Payment;

import in.shop2020.model.v1.order.PaymentInfo;

import in.shop2020.model.v1.order.PaymentStatus;

import in.shop2020.model.v1.order.Shipment;

import in.shop2020.model.v1.order.ShipmentInfo;

import in.shop2020.model.v1.order.Transaction;

import in.shop2020.model.v1.order.TransactionServiceException;

import in.shop2020.model.v1.order.TransactionStatus;

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

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

import in.shop2020.model.v1.shoppingcart.LineStatus;

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

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

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

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

import in.shop2020.payments.PaymentException;

import in.shop2020.thrift.clients.CatalogServiceClient;

import in.shop2020.thrift.clients.LogisticsServiceClient;

import in.shop2020.thrift.clients.PaymentServiceClient;

import in.shop2020.thrift.clients.ShoppingCartClient;

import in.shop2020.thrift.clients.TransactionServiceClient;

import in.shop2020.thrift.clients.UserContextServiceClient;

import in.shop2020.utils.Logger;

public class PayresponseController extends BaseController{

	public static String AMOUNT = "amount";

	public static String TRACKID = "trackid";

	public static String TRACKID_CANCELLED = "trackId";

	public static String RESULT = "result";

	public static String AUTH = "auth";

	public static String TRANID = "tranid";

	public static String PAYMENTID = "paymentId";

	public static String REF = "ref";

	public static String POSTDATE = "postdate";

	public static String CANCELED = "CANCELED";

	public static String APPROVED = "APPROVED";

	public static String CAPTURED = "CAPTURED";

	PaymentServiceClient pclient = null;

	LogisticsServiceClient lsc = null;

	ShoppingCartClient cl = null;

	UserContextServiceClient usc = null;

	CatalogServiceClient csc = null;

	TransactionServiceClient tsc = null;

	String amount;

	String trackId;

	String result;

	String postdate;

	String auth;

	String ref;

	String tranId;

	String paymentId;

	long transaction_id = 0;

	String message = "Unable to process the payment. PLease try Again.";

	in.shop2020.payments.Payment pmnt = null;

	public PayresponseController(){

		super();

		try {

			pclient = new PaymentServiceClient();

			lsc = new LogisticsServiceClient();

			cl = new ShoppingCartClient();

			usc = new UserContextServiceClient();

			csc = new CatalogServiceClient();

			tsc = new TransactionServiceClient();

		} catch (Exception e) {

			Logger.log("Could not initialize the paymentservice client", this);

		}

	}

	public HttpHeaders show() throws IOException, SecurityException{

		//need to ignore id here.

		amount = this.request.getParameter(AMOUNT);

		paymentId = this.request.getParameter(PAYMENTID);

		tranId = this.request.getParameter(TRANID);

		result = this.request.getParameter(RESULT);

		if (result.equalsIgnoreCase(CANCELED)){

			trackId = this.request.getParameter(TRACKID_CANCELLED);

		}else{

			trackId = this.request.getParameter(TRACKID);

		}

		auth = this.request.getParameter(AUTH);

		ref = this.request.getParameter(REF);

		postdate = this.request.getParameter(POSTDATE);

		//update the payment info object

		try {

			pclient.getClient().addBankDetails(Long.parseLong(trackId), paymentId, tranId, "", "", postdate, auth, ref);

			message = "Payment processed. Creating transaction. In case you see this message, please call us with ref no "+ trackId;

			Transaction t = getTransaction();

			transaction_id = tsc.getClient().createTransaction(t);

			message = "Your order id is "+ transaction_id;

			cl.getClient().commitCart(pmnt.getCart_id());

			message = "Error commiting the cart, but your order has been placed. Please refer to "+ transaction_id;

		} catch (NumberFormatException e) {

			Logger.log(e.toString(), this);

		} catch (PaymentException e) {

			Logger.log(e.toString(), this);

		} catch (TException e) {

			Logger.log(e.toString(), this);

		} catch (TransactionServiceException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		return new DefaultHttpHeaders(result);

	}

	private Transaction getTransaction(){

		try {

			pmnt = pclient.getClient().getPayment(Long.parseLong(trackId));

		} catch (NumberFormatException e1) {

			// TODO Auto-generated catch block

			e1.printStackTrace();

		} catch (PaymentException e1) {

			// TODO Auto-generated catch block

			e1.printStackTrace();

		} catch (TException e1) {

			// TODO Auto-generated catch block

			e1.printStackTrace();

		}

		Transaction t = new Transaction();

		PaymentInfo paymentInfo = new PaymentInfo();

		t.setPaymentInfo(paymentInfo);

		t.setShoppingCartid(pmnt.getCart_id());

		t.setCustomer_id(pmnt.getUser_id());

		t.setCreatedOn(pmnt.getBank_ack_timestamp());

		t.setTransactionStatus(TransactionStatus.INIT);

		t.setStatusDescription("New order");

		//create payment

		Payment payment = new Payment();

		payment.setAmount(Double.parseDouble(amount));

		payment.setBank_tx_id(tranId);

		payment.setMerchant_tx_id(trackId);

		payment.setStatus(PaymentStatus.SUCCESS);

		payment.setMode("credit card");

		payment.setCompletionTimestamp(pmnt.getBank_ack_timestamp());

		payment.setSubmissionTimestamp(pmnt.getInit_timestamp());

		Map<Long, Payment> paymentMap = new HashMap<Long, Payment>();

		paymentMap.put(payment.getCompletionTimestamp(), payment);

		paymentInfo.setPayments(paymentMap);

		Cart c = null;

		try {

			c = cl.getClient().getCart(pmnt.getCart_id());

		} catch (ShoppingCartException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (TException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (Exception e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		//Shipping info

		ShipmentInfo shipmentInfo = new ShipmentInfo();

		t.setShipmentInfo(shipmentInfo);

		Shipment shipment = new Shipment();

		try {

			shipment.setAirwayBillNo(lsc.getClient().getEmptyAWB(1));

		} catch (TException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		try {

			shipment.setAddress(getAddress(usc.getClient().getPrimaryInfo(pmnt.getUser_id(), false).getAddresses(),c.getAddressId()));

		} catch (UserContextException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		} catch (TException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}

		shipment.setInsurance(0.0);

		shipment.setProvider("Aramex");

		shipment.setTrackingId(shipment.getAirwayBillNo());

		shipment.setWeight(0.0);

		shipment.setValue(Double.parseDouble(amount));

		shipment.setLineItems(null);

		shipmentInfo.addToShipments(shipment);

		//billinginfo will be added later by warehouse

		OrderInfo orderInfo = new OrderInfo();

		t.setOrderInfo(orderInfo);

		List<Line> lines = c.getLines();

		for(Line line : lines){

			if(line.getLineStatus() == LineStatus.LINE_ACTIVE){

				//line is active

				LineItem lineItem = new LineItem();

				Item item = null;

				try {

					item = csc.getClient().getItemByCatalogId(line.getItemId());

				} catch (InventoryServiceException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				} catch (TException e) {

					// TODO Auto-generated catch block

					e.printStackTrace();

				}

				in.shop2020.model.v1.order.Item oitem = new in.shop2020.model.v1.order.Item();

				oitem.setId(item.getCatalogItemId());

				//get price for it 

				long warehouse = getWarehouse();

				oitem.setPrice(item.getPrice().get(warehouse));

				oitem.setWeight(item.getWeight());

				lineItem.setItem(oitem);

				lineItem.setAddedOn(pmnt.getBank_ack_timestamp());

				orderInfo.addToLineitems(lineItem);

			}

		}

		return t;

	}

	private String getAddress(Set<Address> address, long id){

		for(Address a : address){

			if(a.getId() == id){

				//Prepare String

				String add = a.getName()+",\n"+a.getLine1()+",\n"+a.getLine2()+",\n Landmark"+a.getLandmark()+",/n"+a.getCity()+",\n"+a.getState()+",\n"+a.getCountry()+",\n"+a.getPin()+",\n Phone :- "+a.getPhone();

				return add;

			}

		}

		return "";

	}

	private long getWarehouse(){

		return 1l;

	}

	public String getMessage(){

		return this.message;

	}

	public long getTransactionId(){

		return this.transaction_id;

	}

}