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 org.apache.struts2.convention.annotation.Result;
import org.apache.struts2.convention.annotation.Results;
import org.apache.thrift.TException;

import in.shop2020.model.v1.order.LineItem;
import in.shop2020.model.v1.order.Order;
import in.shop2020.model.v1.order.TransactionServiceException;
import in.shop2020.model.v1.order.TransactionStatus;
import in.shop2020.model.v1.user.ShoppingCartException;
import in.shop2020.payments.PaymentException;
import in.shop2020.payments.PaymentStatus;
import in.shop2020.serving.utils.Utils;
import in.shop2020.thrift.clients.PaymentServiceClient;
import in.shop2020.thrift.clients.TransactionServiceClient;
import in.shop2020.thrift.clients.UserContextServiceClient;
import in.shop2020.utils.Logger;

@Results({
    @Result(name="redirect", type="redirectAction", 
                params = {"actionName" , "login"}),
    @Result(name="shipping-redirect", type="redirectAction", 
                params = {"actionName" , "shipping"})
                
})

public class PayResponseController extends BaseController{
        private static final long serialVersionUID = 1L;

        private enum PaymentReturnStatus{
                CAPTURED("CAPTURED"),
                NOT_CAPTURED ("NOT CAPTURED"),
                CANCELLED ("CANCELLED"),
                DENIED_BY_RISK("DENIED BY RISK"),
                HOST_TIMEOUT("HOST TIMEOUT");
                private String value;
                PaymentReturnStatus(String value) {
                        this.value = value;
                }
                public String value(){
                        return this.value;
                }
        }

        public static String AMOUNT = "amt";
        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 ERROR = "Error";
        public static String ERRORTEXT = "ErrorText";
        
        PaymentServiceClient pclient = null;
        TransactionServiceClient tsc = null;
        UserContextServiceClient usc = null;
        
        String amount;
        String trackId;
        String result;
        String postdate;
        String auth;
        String ref;
        String tranId;
        String paymentId;
        String sessionId;
        String errorText;
        String errorNo;
        long txnId;
        long merchantPaymentId;
        List<Order> orders = null;

        /*
         *      pw.println("RESULT="+result);
                        pw.println("PDATE="+postdate);
                        pw.println("TXNID="+tranid);
                        pw.println("AUTH="+auth);
                        pw.println("TRACKID="+trackid);
                        pw.println("REF="+ref);
                        pw.println("AMT="+amount);
                        
                        pw.flush();
                */      
                        
        String message = "Unable to process the payment. PLease try Again.";
        
        public PayResponseController(){
                super();
                try {
                        pclient = new PaymentServiceClient();
                        tsc = new TransactionServiceClient();
                        usc = new UserContextServiceClient();
                } catch (Exception e) {
                        Logger.log("Could not initialize the paymentservice client", this);
                }
        }
        
        public String index() throws IOException, SecurityException{
        if(!userinfo.isLoggedIn()){
                setRedirectUrl();
                return "redirect";
        }

        result = this.request.getParameter(RESULT);
                postdate = this.request.getParameter(POSTDATE);
                tranId = this.request.getParameter(TRANID);
                auth = this.request.getParameter(AUTH);
                ref = this.request.getParameter(REF);
                amount = this.request.getParameter(AMOUNT);
                paymentId = this.request.getParameter(PAYMENTID);
                if (result.equalsIgnoreCase(PaymentReturnStatus.CANCELLED.value())){
                        trackId = this.request.getParameter(TRACKID_CANCELLED);
                }else{
                        trackId = this.request.getParameter(TRACKID);
                }
                merchantPaymentId = Long.parseLong(trackId);
                sessionId = this.request.getSession().getId();
                errorNo = request.getParameter(ERROR);
                errorText = request.getParameter(ERRORTEXT);
                /*
                String udf1=request.getParameter("udf1");
                String udf2=request.getParameter("udf2");
                String udf3=request.getParameter("udf3");
                String udf4=request.getParameter("udf4");
                String udf5=request.getParameter("udf5");
                */
                //update the payment info object
                try {
                        if(result.trim().equals(PaymentReturnStatus.CAPTURED.value())){
                                String message = "Payment successful";
                                pclient.getClient().updatePaymentDetails(merchantPaymentId, paymentId, sessionId, result, message, tranId, auth, ref, errorNo, PaymentStatus.SUCCESS, null);
                                txnId = pclient.getClient().getPayment(merchantPaymentId).getMerchantTxnId();
                                
                                tsc.getClient().changeTransactionStatus(txnId, TransactionStatus.IN_PROCESS, "Payment received for the order");
                                
                        orders = tsc.getClient().getOrdersForTransaction(txnId);
                        Map<Long,Double> items = new HashMap<Long, Double>();
                        
                                for(Order order: orders){
                                        for(LineItem lineitem: order.getLineitems()){
                                                items.put(lineitem.getItem_id(),lineitem.getQuantity());
                                        }
                                }
                                usc.getClient().resetCart(userinfo.getCartId(), items);
                                //TODO Optimize the function to send less number of data over network
                                userinfo.setTotalItems(Utils.getNumberOfItemsInCart(userinfo.getCartId()));

                                return "index";
                        }else{
                                String message = "Unable to process the payment.";
                                addActionError("Previous payment failed. Try again.");
                                
                                txnId = pclient.getClient().getPayment(merchantPaymentId).getMerchantTxnId();
                                tsc.getClient().changeTransactionStatus(txnId, TransactionStatus.FAILED, "Payment failed for the transaction.");
                                
                                pclient.getClient().updatePaymentDetails(merchantPaymentId, paymentId, sessionId, result, errorText, tranId, auth, ref, errorNo, PaymentStatus.FAILED, null);
                                return "shipping-redirect";
                        }
                        
                }
                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) {
                        Logger.log(e.toString(), this);
                } catch (ShoppingCartException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                return "index";
        }
        
        public List<Order> getOrders(){
                return this.orders;
        }
        
        public String getMessage(){
                return this.message;
        }
        
}