Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

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

import in.shop2020.support.services.CourierDetailsGenerator;

import in.shop2020.thrift.clients.CatalogServiceClient;

import in.shop2020.thrift.clients.HelperServiceClient;

import in.shop2020.utils.HelperService;

import in.shop2020.utils.LogisticsUser;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.util.Calendar;

import java.util.GregorianCalendar;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.struts2.rest.DefaultHttpHeaders;

import org.apache.struts2.rest.HttpHeaders;

public class CourierDetailsController implements ServletResponseAware,

		ServletRequestAware {

	private String id;

	//private long warehouseId;

	//private long providerId;

	private HttpServletRequest request;

	private HttpServletResponse response;

	private HttpSession session;

	public String index(){

		if(getSessionUserName()==null)

			return "authfail";

		else

			return "authsuccess";

	}

	// Handler for POST /courier-details

	public String create(){

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

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

		try{

			HelperServiceClient helperServiceClient = new HelperServiceClient();

			in.shop2020.utils.HelperService.Client client = helperServiceClient.getClient();

			LogisticsUser user = client.authenticateLogisticsUser(username, password);

			session.setAttribute("username", user.getUsername());

			session.setAttribute("providerId", Long.valueOf(user.getProviderId()));

		}catch(Exception e){

			e.printStackTrace();

			return "authfail";

		}

		return "authsuccess";

	}

	// Handler for GET /courier-details/<warehouseId>

	public String show(){

		try {

			long warehouseId = Long.parseLong(getId());

			long providerId = ((Long)session.getAttribute("providerId")).longValue();

			System.out.println("Warehouse Id is:  " + warehouseId);

			System.out.println("Provider Id is: " + providerId);

			CourierDetailsGenerator courierDetailsGenerator = new CourierDetailsGenerator();

			ByteArrayOutputStream baos = courierDetailsGenerator.generateCourierDetails(warehouseId, providerId);

			response.setContentType("application/vnd.ms-excel");

			Calendar date = new GregorianCalendar();

			int year = date.get(Calendar.YEAR);

			int month = date.get(Calendar.MONTH) +1;

			int day = date.get(Calendar.DAY_OF_MONTH);

			response.setHeader("Content-disposition", "inline; filename=courier-details-"+warehouseId+"-"+ providerId + "-"+year+"-"+ month+"-" + day +".xls" );

			ServletOutputStream sos;

			try {

				sos = response.getOutputStream();

				baos.writeTo(sos);

				sos.flush();

			} catch (IOException e) {

				e.printStackTrace();

			}	

			return "authsuccess";

		}catch(NumberFormatException nfe){

			nfe.printStackTrace();

		}

		return "authfail";

	}

	@Override

	public void setServletResponse(HttpServletResponse response) {

		this.response  = response;

	}

	@Override

	public void setServletRequest(HttpServletRequest request) {

		this.request = request;

		this.session = request.getSession();

	}

	public String getId(){

		return id;

	}

	public void setId(String id){

		this.id = id;

	}

	public String getSessionUserName(){

		return (String) session.getAttribute("username");

	}

	public Map<Long, String> getWarehouses(){

		Map<Long, String> warehouseMap = new HashMap<Long, String>();

		try{

			CatalogServiceClient csc = new CatalogServiceClient();

			in.shop2020.model.v1.catalog.InventoryService.Client catalogClient= csc.getClient();

			List<Warehouse> warehouses = catalogClient.getAllWarehouses(true);

			for(Warehouse warehouse : warehouses){

				warehouseMap.put(warehouse.getId(), warehouse.getDisplayName());

			}

		}catch(Exception e){

			e.printStackTrace();

		}

		return warehouseMap;

	}

}