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 = "";

    private Client client;

    public PickupStoreController() throws TTransportException{

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

    }

    @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");

		try {

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

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

				long orderId = Long.parseLong(orderIdString);

				client.markOrderAsReceivedAtStore(orderId, Calendar.getInstance().getTimeInMillis());	

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

				long orderId = Long.parseLong(orderIdString);

		    	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 String getSecretCode(long orderId) throws TException{

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

    	for(Attribute attribute: attributes){

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

    			return attribute.getValue();

    		}

    	}

    	return "";

    }

	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;

	}

}