Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

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

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

import in.shop2020.model.v1.order.TransactionService.Client;

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

import in.shop2020.thrift.clients.TransactionClient;

import java.util.ArrayList;

import java.util.Calendar;

import java.util.List;

import javax.servlet.ServletContext;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.struts2.util.ServletContextAware;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class PickupStoreController implements ServletResponseAware, ServletRequestAware, ServletContextAware {

    private static Logger logger = LoggerFactory.getLogger(PickupStoreController.class);

    private HttpServletRequest request;

    private HttpServletResponse response;

    private ServletContext context;

    private String storeIdString;

    private long storeId;

    private String id;

    private String error = "";

    @Override

    public void setServletRequest(HttpServletRequest req) {

        this.request = req;

    }

    public HttpServletRequest getServletRequest() {

        return request;

    }

    @Override

    public void setServletResponse(HttpServletResponse res) {

        this.response = res;

    }

    public HttpServletResponse getServletResponse() {

        return response;

    }

    @Override

    public void setServletContext(ServletContext context) {

        this.context = context;

    }

    public String getServletContextPath() {

        return context.getContextPath();

    }

    public String index(){

    	storeIdString = (String) request.getSession().getAttribute("STORE_ID");

    	if(storeIdString == null){

    		storeIdString = request.getParameter("storeid");

    		if(storeIdString == null){

    			return "authfail";

    		}else{

    			request.getSession().setAttribute("STORE_ID", storeIdString);

    		}

    	}

    	storeId = Long.parseLong(storeIdString);

    	return "index";

    }

    public String show(){

    	storeIdString = (String) request.getSession().getAttribute("STORE_ID");

    	if(storeIdString == null){

    		storeIdString = request.getParameter("storeid");

    		if(storeIdString == null){

    			return "authfail";

    		}else{

    			request.getSession().setAttribute("STORE_ID", storeIdString);

    		}

    	}

    	storeId = Long.parseLong(storeIdString);

    	return "show";

    }

    // Handles the POST request (Form Submission)

    public String create() {

    	storeIdString = (String) request.getSession().getAttribute("STORE_ID");

    	if(storeIdString == null){

    		return "authfail";

    	}

    	storeId = Long.parseLong(storeIdString);

    	String orderIdString = request.getParameter("orderid");

    	String action = request.getParameter("action");

    	long orderId = Long.parseLong(orderIdString);

    	logger.info("orderIdString" + orderIdString);

    	if(orderIdString != null){

    		try {

				Client client = (new TransactionClient()).getClient();

				if(action.equals("markreceived")){

					client.markOrderAsReceivedAtStore(Long.parseLong(orderIdString), Calendar.getInstance().getTimeInMillis());	

				}else if(action.equals("markdelivered")){

					String receiver = request.getParameter("receiver");

					String secretCode = request.getParameter("secretcode");

					Order order = client.getOrder(orderId);

					List<Attribute> attributes = client.getAllAttributesForOrderId(orderId);

					for(Attribute attribute: attributes){

						if(attribute.getName().equals("SECRET_CODE")){

							if(!attribute.getValue().equalsIgnoreCase(secretCode)){

								error = "Secret code does not match";

							}

						}

					}

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

					if(order.isCod()){

						double amount = Double.parseDouble(request.getParameter("amount"));

						if(order.getTotal_amount() != amount){

							error = "<br>Amount does not match";

						}

					}

					if(!error.equalsIgnoreCase("")){

						setId("deliver");

						return show();

					}else{

						if(order.isCod()){

							String paymentType =  request.getParameter("paymenttype");

							newAttributes.add(new Attribute("PAYMENT_MODE", paymentType));

							client.setOrderAttributes(orderId, newAttributes);

						}

					}

					client.markOrderAsDelivered(Long.parseLong(orderIdString), Calendar.getInstance().getTimeInMillis(), receiver);

				}else if(action.equals("markreturned")){

					String orderIdsString = request.getParameter("orderIds");

					long providerId = Long.parseLong(request.getParameter("providerID"));

					List<Long> orderIds = new ArrayList<Long>();

					for(String orderIdString1: orderIdsString.split(":")){

						orderIds.add(Long.parseLong(orderIdString1));

					}

					client.markOrdersAsReturnedFromStore(providerId, orderIds);

				}

				return "index";

			} catch (TTransportException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			} catch (NumberFormatException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			} catch (TransactionServiceException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			} catch (TException e) {

				// TODO Auto-generated catch block

				e.printStackTrace();

			}

    	}

        return "index";

    }

    public List<Order> getReceivePendingOrders() throws TException{

    	Client client = (new TransactionClient()).getClient();

    	return client.getReceivePendingOrders(storeId);

    }

    public List<Order> getReceivedAtStoreOrders() throws TException{

    	Client client = (new TransactionClient()).getClient();

    	return client.getReceivedAtStoreOrders(storeId);

    }

	public void setId(String id) {

		this.id = id;

	}

	public String getId() {

		return id;

	}

	public void setError(String error) {

		this.error = error;

	}

	public String getError() {

		return error;

	}

	public long getStoreId(){

		return storeId;

	}

}