Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

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

import in.shop2020.model.v1.user.Affiliate;

import in.shop2020.model.v1.user.MasterAffiliate;

import in.shop2020.model.v1.user.TrackLog;

import in.shop2020.model.v1.user.UserContextService.Client;

import in.shop2020.thrift.clients.UserContextServiceClient;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.text.ParseException;

import java.util.Date;

import java.util.HashMap;

import java.util.LinkedList;

import java.util.List;

import java.util.Map;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import org.apache.poi.ss.usermodel.Cell;

import org.apache.poi.ss.usermodel.CellStyle;

import org.apache.poi.ss.usermodel.CreationHelper;

import org.apache.poi.ss.usermodel.Font;

import org.apache.poi.ss.usermodel.Row;

import org.apache.poi.ss.usermodel.Sheet;

import org.apache.poi.ss.usermodel.Workbook;

import org.apache.poi.ss.util.CellRangeAddress;

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

public class AffiliateController implements ServletResponseAware, ServletRequestAware{

	private HttpServletResponse response;

	private HttpServletRequest request;

	private String errorMsg = "";

	private List<MasterAffiliate> masterAffiliates;

	@Override

	public void setServletResponse(HttpServletResponse res) {

		this.response = res;

	}

	@Override

    public void setServletRequest(HttpServletRequest req) {

	    this.request = req;

    }

	public String index()  {

        try {

            UserContextServiceClient userContextServiceClient = new UserContextServiceClient();

            Client userClient = userContextServiceClient.getClient();

            masterAffiliates = userClient.getAllMasterAffiliates();

        } catch (Exception e) {

            e.printStackTrace();

        }

        return "report";

    }

	public String create()	{

	    try   {

	        long mAfId = Long.parseLong(request.getParameter("masterAffiliate"));

            UserContextServiceClient userContextServiceClient = new UserContextServiceClient();

            Client userClient = userContextServiceClient.getClient();

            List<Map<String, String>> contentRows = new LinkedList<Map<String,String>>(); 

            MasterAffiliate mAffiliate = userClient.getMasterAffiliateById(mAfId);

            for (Affiliate aff : userClient.getAffiliatesByMasterAffiliate(mAfId)) {

                for (TrackLog tracklog : userClient.getTrackLogsByAffiliate(aff.getId())) {

                    Map<String, String> contentRow = new HashMap<String, String>();

                    contentRow.put("mAffiliate", mAffiliate.getName());

                    contentRow.put("affiliate", aff.getName());

                    contentRow.put("affiliateUrl", aff.getUrl());

                    contentRow.put("date",String.valueOf(tracklog.getAddedOn()));

                    contentRow.put("event", tracklog.getEvent());

                    contentRow.put("url", tracklog.getUrl());

                    contentRow.put("data", tracklog.getData());

                    contentRows.add(contentRow);

                }

            }

            // Preparing XLS file for output

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

            response.setHeader("Content-disposition", "inline; filename=affiliate-report" + ".xls");

            ServletOutputStream sos;

            try {

                ByteArrayOutputStream baos = getSpreadSheetData(contentRows, mAffiliate);

                sos = response.getOutputStream();

                baos.writeTo(sos);

                sos.flush();

            } catch (IOException e) {

                errorMsg = "Failed to write to response.";

                e.printStackTrace();

            }

        } catch (ParseException e)  {

            errorMsg = e.getMessage();

            e.printStackTrace();

        } catch (TransactionServiceException e) {

            errorMsg = e.getMessage();

            e.printStackTrace();

        } catch (Exception e)   {

            errorMsg = e.getMessage();

            e.printStackTrace();

        }

        return null;

	}

	// Prepares the XLS worksheet object and fills in the data with proper formatting

	private ByteArrayOutputStream getSpreadSheetData(List<Map<String, String>> contentRows, MasterAffiliate mAffiliate)

	{

	    ByteArrayOutputStream baosXLS = new ByteArrayOutputStream();

		Workbook wb = new HSSFWorkbook();

	    Font font = wb.createFont();

	    font.setBoldweight(Font.BOLDWEIGHT_BOLD);

	    CellStyle style = wb.createCellStyle();

	    style.setFont(font);

	    CreationHelper createHelper = wb.getCreationHelper();

	    CellStyle dateCellStyle = wb.createCellStyle();

	    dateCellStyle.setDataFormat(createHelper.createDataFormat().getFormat("DD/MM/YYYY HH:MM"));

	    createAffiliateSheet(contentRows, mAffiliate, wb, style, dateCellStyle);

		// Write the workbook to the output stream

		try {

			wb.write(baosXLS);

			baosXLS.close();

		} catch (IOException e) {

			e.printStackTrace();

		}		

		return baosXLS;

	}

	private void createAffiliateSheet(

	        List<Map<String, String>> contentRows, 

			MasterAffiliate mAffiliate, Workbook wb, 

			CellStyle style, CellStyle dateCellStyle) 

	{

		// Affiliate SHEET

	    Sheet affSheet = wb.createSheet("Affiliates Report");

	    short affSerialNo = 0;

	    Row affTitleRow = affSheet.createRow(affSerialNo ++);

	    Cell affTitleCell = affTitleRow.createCell(0);

	    affTitleCell.setCellValue(mAffiliate.getName() + " Affiliates Report");

	    affTitleCell.setCellStyle(style);

	    affSheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 6));

	    affSheet.createRow(affSerialNo ++);

	    Row affHeaderRow = affSheet.createRow(affSerialNo++);

	    affHeaderRow.createCell(0).setCellValue("Affiliate");

	    affHeaderRow.createCell(1).setCellValue("Affiliate Url");

	    affHeaderRow.createCell(2).setCellValue("Date");

	    affHeaderRow.createCell(3).setCellValue("Event");

	    affHeaderRow.createCell(4).setCellValue("Url");

	    affHeaderRow.createCell(5).setCellValue("Data");

	    for (int i=0; i<6 ;i++) {

	        affHeaderRow.getCell(i).setCellStyle(style);

	    }

		for( Map<String, String> contentRow : contentRows) {

			affSerialNo++;

			Row commContentRow = affSheet.createRow(affSerialNo);

			commContentRow.createCell(0).setCellValue(contentRow.get("affiliate"));

			commContentRow.createCell(1).setCellValue(contentRow.get("affiliateUrl"));

			commContentRow.createCell(2).setCellValue(new Date(Long.parseLong(contentRow.get("date"))));

			commContentRow.getCell(2).setCellStyle(dateCellStyle);

            commContentRow.createCell(3).setCellValue(contentRow.get("event"));

			commContentRow.createCell(4).setCellValue(contentRow.get("url"));

			commContentRow.createCell(5).setCellValue(contentRow.get("data"));

		}

        for (int i = 0; i<6; i++) {

            affSheet.autoSizeColumn(i);

        }

	}

	public String getErrorMsg() {

		return errorMsg;

	}

	public List<MasterAffiliate> getMasterAffiliates() {

        return masterAffiliates;

    }

}