Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.config.ConfigException;

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

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

import in.shop2020.payments.Attribute;

import in.shop2020.payments.Payment;

import in.shop2020.payments.PaymentException;

import in.shop2020.payments.PaymentStatus;

import in.shop2020.serving.services.CommonPaymentService;

import in.shop2020.thrift.clients.PaymentClient;

import in.shop2020.thrift.clients.TransactionClient;

import in.shop2020.thrift.clients.UserClient;

import in.shop2020.thrift.clients.config.ConfigClient;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.Map;

import java.util.TreeMap;

import javax.servlet.http.HttpServletRequest;

import org.apache.log4j.Logger;

import org.apache.thrift.TException;

@SuppressWarnings("serial")

public class PayuPayResponseController extends BaseController{

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

	private static final String UNMAPPED_STATUS = "UnMappedStatus";

	private static String successUrl;

	private static String errorUrl;

	private static String rechargeResultUri;

	/**

	 * The secret key used to decode RC4 encoded data.

	 */

	private static String accountKey;

	private static String salt;

	private String redirectUrl;

	private String eurl;

	private String surl;

	static{

		try {

			ConfigClient cc = ConfigClient.getClient();

			successUrl = cc.get("payu_success_url");

			errorUrl = cc.get("payu_error_url");

			accountKey = cc.get("payu_account_id");

			salt = cc.get("payu_secret_key");

			rechargeResultUri = ConfigClient.getClient().get("recharge_result_uri");

		} catch (ConfigException e) {

			log.error("Unable to get success and error usr info from config server.");

		}

	}

	private Map<String, String> paymentParams = new TreeMap<String, String>();

	public String create() {

		eurl = errorUrl;

		surl = successUrl;

		updatePaymentParams(request.getParameterMap());

		PaymentClient paymentServiceClient = null;

		TransactionClient transactionServiceClient = null;

		UserClient userServiceClient = null;

		try {

			userServiceClient = new UserClient();

			paymentServiceClient = new PaymentClient();

			transactionServiceClient = new TransactionClient();

		} catch (Exception e) {

			log.error("Unable to initialize one of the clients", e);

		}

		long merchantPaymentId = Long.parseLong(paymentParams.get("txnid"));

		String gatewayPaymentId = paymentParams.get("mihpayid");

		double amount = Double.parseDouble(paymentParams.get("amount"));

		String gatewayTxnStatus = paymentParams.get("status");

		String hash = paymentParams.get("hash");

		String bankRefId = paymentParams.get("bank_ref_num");

		String unmappedStatus  = paymentParams.get("unmappedstatus");

		List<Attribute> attributes = new ArrayList<Attribute>();

		attributes.add(new Attribute(UNMAPPED_STATUS, unmappedStatus));

		Payment payment = null;

		Long txnId = null;

		try {

			payment = paymentServiceClient.getClient().getPayment(merchantPaymentId);

			txnId = payment.getMerchantTxnId();

		} catch (PaymentException e1) {

			log.error("Payment exception. It is serious, check merchant payment id + " + merchantPaymentId, e1);

		} catch (TException e1) {

			log.error("Thrift exception. Check payment id "+ merchantPaymentId, e1);

		}

		if(payment.getStatus() != PaymentStatus.INIT){

			// We have already processed a response for this payment. Processing

			// it again may fail his orders. So, let's ask him to check his

			// account.

			return "ebs-pay-response-maybe";

		}

		if(payment.isIsDigital()){

			eurl = rechargeResultUri;

			surl = rechargeResultUri;

		}

		if(!validatePaymentParams(amount, payment, hash)){

			this.redirectUrl = eurl + "?paymentId=" + merchantPaymentId;

			return "index";

		}

		if(gatewayTxnStatus.equalsIgnoreCase("SUCCESS")){

			try {

				paymentServiceClient.getClient().updatePaymentDetails(merchantPaymentId, gatewayPaymentId,

						"", gatewayTxnStatus, unmappedStatus.equals("captured") ? "Transaction captured at PG":"Transaction Authorized at PG", 

						"", "", bankRefId, "", unmappedStatus.equals("captured") ? PaymentStatus.SUCCESS : PaymentStatus.AUTHORIZED, new SimpleDateFormat().format(new Date()), attributes);

			} catch (PaymentException e) {

				this.redirectUrl = eurl + "?paymentId=" + merchantPaymentId;

				return "index";

			} catch (TException e) {

				this.redirectUrl = eurl + "?paymentId=" + merchantPaymentId;

				return "index";

			}

            if(payment.isIsDigital()){

            	RechargeOrder rechargeOrder = null;

                try {

                    rechargeOrder = transactionServiceClient.getClient().getRechargeOrdersForTransaction(txnId);

                } catch (Exception e1) {

                    log.error("Problem with txn client while getting recharge object", e1);

    				this.redirectUrl = eurl + "?paymentId=" + merchantPaymentId;

    				return "index";

                }

		        try {

		            PaymentClient pcl = new PaymentClient();

		            boolean isCaptured = pcl.getClient().capturePayment(txnId, true);

		            //Retry in case we are not able to capture first time

		            if(!isCaptured){

		            	Thread.sleep(2000);

		            	isCaptured = pcl.getClient().capturePayment(txnId, true);

		            }

		            if(!isCaptured){

		            	Thread.sleep(2000);

		            	isCaptured = pcl.getClient().capturePayment(txnId, true);

		            }

		            if(isCaptured) {

		                transactionServiceClient.getClient().updateRechargeOrderStatus(rechargeOrder.getId(), RechargeOrderStatus.PAYMENT_SUCCESSFUL);

		            } else {

                        transactionServiceClient.getClient().updateRechargeOrderStatus(rechargeOrder.getId(), RechargeOrderStatus.PAYMENT_FAILED);

		            }

		        } catch (Exception e) {

		            log.error("Problem with txn client while trying to recharge", e);

		        }

            } else {

		        CommonPaymentService.processSuccessfulTxn(txnId, userServiceClient, transactionServiceClient, false);

            }

            this.redirectUrl = surl + "?paymentId=" + merchantPaymentId;

		} else{

			try {

				paymentServiceClient.getClient().updatePaymentDetails(merchantPaymentId, gatewayPaymentId,

						"", gatewayTxnStatus, "Payment Failed at PG", "", "", "", "", PaymentStatus.FAILED, new SimpleDateFormat().format(new Date()), attributes);

			} catch (PaymentException e) {

			    log.error("Unable to mark the payment as failed", e);

			} catch (TException e) {

			    log.error("Unable to mark the payment as failed", e);

			}

			if(!payment.isIsDigital()) {

				CommonPaymentService.processFailedTxn(txnId, transactionServiceClient);

			}

			this.redirectUrl = eurl + "?paymentId=" + merchantPaymentId;

		}

		log.info("User will be redirected to: " + surl + "?paymentId=" + merchantPaymentId);

		return "index";

	}

	private boolean validatePaymentParams(double returnedAmount, Payment payment, String hash){

		if(!(payment != null && Math.abs(payment.getAmount() - returnedAmount) <= 0.50 && hash.equals(getSecureHash()))){

			// We did not request this payment or the authorised amount is different.

			log.error("Checks and balance failed on returned data");

			return false;

		}

		return true;

	}

	public String getSecureHash(){

		try{

			String pass = salt + "|" + paymentParams.get("status") + "|||||||||||" + paymentParams.get("email") + "|" +  paymentParams.get("firstname") + "|" + paymentParams.get("productinfo") + "|" + paymentParams.get("amount") + "|" + paymentParams.get("txnid") + "|" + accountKey;

			System.out.println(pass);

			MessageDigest md = MessageDigest.getInstance("SHA-512");

			md.update(pass.getBytes(), 0, pass.getBytes().length);

			byte[] mdbytes = md.digest();

			//	convert the byte to hex format method

			StringBuffer sb = new StringBuffer();

			for (int i = 0; i < mdbytes.length; i++) {

				sb.append(Integer.toString((mdbytes[i] & 0xff) + 0x100, 16).substring(1));

			}

			return sb.toString();

		}catch(NoSuchAlgorithmException nsae){

			log.error("No such algorithm exception");

			return null;

		}

	}

	private void updatePaymentParams(Map map){

		for(Object key : map.keySet()){

		    String keyStr = (String)key;

		    String[] vals = (String[])map.get(keyStr);

		    String value = vals[0];

		    System.out.println("Key " + (String)key + "     :    " + value);

		    paymentParams.put(keyStr, value);

		}

	}

	public String getRedirectUrl(){

		return this.redirectUrl;

	}

	@Override

	public void setServletRequest(HttpServletRequest request) {

		this.request = request;

	}

	public Map<String, String> getPaymentParams() {

		return paymentParams;

	}

}