Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

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

import in.shop2020.thrift.clients.CatalogServiceClient;

import in.shop2020.thrift.clients.HelperServiceClient;

import in.shop2020.utils.LogisticsUser;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileInputStream;

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.ServletContext;

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.util.ServletContextAware;

/**

 * Allows executives of courier companies to login and download courier details

 * report which they then use to upload into their database.

 * 

 * @author Chandranshu

 * 

 */

public class CourierDetailsController implements ServletResponseAware,

		ServletRequestAware, ServletContextAware {

	private String id;

	private int daysToSubtract;

	//FIXME: Read this configuration from the config client

	private String courierDetailsPath = "/CourierDetailReports";

	private ServletContext context;

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

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

			Calendar date = new GregorianCalendar();

			date.add(Calendar.DAY_OF_MONTH, daysToSubtract);

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

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

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

			String fileName = courierDetailsPath + "/courier-details-"+warehouseId+"-"+ providerId + "-"+year+"-"+ month+"-" + day +".xls";

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

			ServletOutputStream sos;

			try {

				ByteArrayOutputStream baos = new ByteArrayOutputStream();

				baos.write(getBytesFromFile(new File(fileName)));

				sos = response.getOutputStream();

				baos.writeTo(sos);

				sos.flush();

			} catch (IOException e) {

				e.printStackTrace();

			}	

			return "authsuccess";

		}catch(NumberFormatException nfe){

			nfe.printStackTrace();

		}

		return "authfail";

	}

	public String dayBefore(){

		daysToSubtract = -2;

		return show();

	}

	public String yesterday(){

		daysToSubtract = -1;

		return show();

	}

	public String today(){

		daysToSubtract = 0;

		return show();

	}

	// Returns the contents of the file in a byte array.

	public static byte[] getBytesFromFile(File file) throws IOException {

	    FileInputStream is = new FileInputStream(file);

	    // Get the size of the file

	    long length = file.length();

	    // You cannot create an array using a long type.

	    // It needs to be an int type.

	    // Before converting to an int type, check

	    // to ensure that file is not larger than Integer.MAX_VALUE.

	    if (length > Integer.MAX_VALUE) {

	        // File is too large

	    }

	    // Create the byte array to hold the data

	    byte[] bytes = new byte[(int)length];

	    // Read in the bytes

	    int offset = 0;

	    int numRead = 0;

	    while (offset < bytes.length

	           && (numRead=is.read(bytes, offset, bytes.length-offset)) >= 0) {

	        offset += numRead;

	    }

	    // Ensure all the bytes have been read in

	    if (offset < bytes.length) {

	        throw new IOException("Could not completely read file "+file.getName());

	    }

	    // Close the input stream and return bytes

	    is.close();

	    return bytes;

	}

	@Override

	public void setServletContext(ServletContext context) {

		this.context = context;

	}

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

	}

	public String getServletContextPath(){

		return context.getContextPath();

	}

}