Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.model.v1.user.User;
import in.shop2020.model.v1.user.UserCommunication;
import in.shop2020.model.v1.user.UserCommunicationException;
import in.shop2020.model.v1.user.UserContextException;
import in.shop2020.model.v1.user.UserContextService.Client;
import in.shop2020.thrift.clients.UserClient;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Date;
import java.util.List;

import javax.servlet.ServletOutputStream;
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.ServletResponseAware;
import org.apache.thrift.TException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class UserCommunicationsController implements ServletResponseAware{
    
    private static Logger logger = LoggerFactory.getLogger(UserCommunicationsController.class);

        private HttpServletResponse response;
        
        private String errorMsg = "";
        
        @Override
        public void setServletResponse(HttpServletResponse res) {
                this.response = res;
        }
        
        public String index()   {
            try   {
                UserClient userContextServiceClient = new UserClient();
            Client userClient = userContextServiceClient.getClient();

            //Retrieving all user communications
            List<UserCommunication> allCommunications = userClient.getAllUserCommunications();
            
            // Preparing XLS file for output
            response.setContentType("application/vnd.ms-excel");
            
            response.setHeader("Content-disposition", "inline; filename=user-communications" + ".xls");
            
            ServletOutputStream sos;
            try {
                ByteArrayOutputStream baos = getSpreadSheetData(allCommunications, userClient);
                sos = response.getOutputStream();
                baos.writeTo(sos);
                sos.flush();
            } catch (IOException e) {
                logger.error("Unable to stream the user communications report");
                errorMsg = "Failed to write to response.";
            }

        } catch (UserCommunicationException e) {
            logger.error("Error while getting user communications", e);
            errorMsg = e.getMessage();
        } catch (TException e) {
            logger.error("Unable to get user communications", e);
            errorMsg = e.getMessage();
        } catch (Exception e) {
            logger.error("Unexpected exception", e);
            errorMsg = e.getMessage();
        }
        return null;
        }
        
        // Prepares the XLS worksheet object and fills in the data with proper formatting
        private ByteArrayOutputStream getSpreadSheetData(List<UserCommunication> allCommunications,
                        Client userClient) 
        {
            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"));
            
            createAllUserCommunicationSheet(allCommunications, userClient, wb, style, dateCellStyle);
                        
                // Write the workbook to the output stream
                try {
                        wb.write(baosXLS);
                        baosXLS.close();
                } catch (IOException e) {
                        logger.error("Error while converting the user communications report to a byte array", e);
                }               
                return baosXLS;
        }
        
        private void createAllUserCommunicationSheet(
                        List<UserCommunication> allCommunications, 
                        Client userClient,
                        Workbook wb, 
                        CellStyle style, CellStyle dateCellStyle) 
        {
                // USER COMMUNICATION SHEET
            Sheet commSheet = wb.createSheet("All Users Communications");
            short commSerialNo = 0;

            Row commTitleRow = commSheet.createRow(commSerialNo ++);
            Cell commTitleCell = commTitleRow.createCell(0);
            commTitleCell.setCellValue("All Users Communications");
            commTitleCell.setCellStyle(style);
            
            commSheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 6));
            
            commSheet.createRow(commSerialNo ++);
                
            Row commHeaderRow = commSheet.createRow(commSerialNo++);
            commHeaderRow.createCell(0).setCellValue("User Email");
            commHeaderRow.createCell(1).setCellValue("User Name");
            commHeaderRow.createCell(2).setCellValue("Communication Email");
            commHeaderRow.createCell(3).setCellValue("Date of Birth");
            commHeaderRow.createCell(4).setCellValue("Mobile Number");
            commHeaderRow.createCell(5).setCellValue("User Id");
            commHeaderRow.createCell(6).setCellValue("Sex");
            commHeaderRow.createCell(7).setCellValue("Order Id");
            commHeaderRow.createCell(8).setCellValue("Communication Type");
            commHeaderRow.createCell(9).setCellValue("AirwayBill No");
            commHeaderRow.createCell(10).setCellValue("TimeStamp");
            commHeaderRow.createCell(11).setCellValue("Product Name");
            commHeaderRow.createCell(12).setCellValue("Reply To");
            commHeaderRow.createCell(13).setCellValue("Subject");
            commHeaderRow.createCell(14).setCellValue("Message");
            for (int i=0; i<15 ;i++) {
                commHeaderRow.getCell(i).setCellStyle(style);
            }
            
                for( UserCommunication userComm : allCommunications) {
                        commSerialNo++;
                        Row commContentRow = commSheet.createRow(commSerialNo);
                        
                        try {
                                User user = userClient.getUserById(userComm.getUserId());
                                if (user.getUserId() != -1) {
                                        commContentRow.createCell(0).setCellValue(user.getEmail());
                                        commContentRow.createCell(1).setCellValue(user.getName());
                                        commContentRow.createCell(2).setCellValue(user.getCommunicationEmail());
                                        commContentRow.createCell(3).setCellValue(user.getDateOfBirth());
                                        commContentRow.createCell(4).setCellValue(user.getMobileNumber());
                                        commContentRow.createCell(5).setCellValue(user.getUserId());
                                        if (user.getSex() != null) {
                                                commContentRow.createCell(6).setCellValue(user.getSex().name());
                                        }
                                }
                        } catch (UserContextException e) {
                                logger.error("Error while getting user data", e);
                        } catch (TException e) {
                                logger.error("Unable to get user data", e);
                        }
                        commContentRow.createCell(7).setCellValue(userComm.getOrderId());
                        if (userComm.getCommunicationType() != null) {
                                commContentRow.createCell(8).setCellValue(
                                                userComm.getCommunicationType().name());
                        }
                        commContentRow.createCell(9).setCellValue(userComm.getAirwaybillNo());
                        commContentRow.createCell(10).setCellValue(new Date(userComm.getCommunication_timestamp()));
                        commContentRow.getCell(10).setCellStyle(dateCellStyle);
                        commContentRow.createCell(11).setCellValue(userComm.getProductName());
                        commContentRow.createCell(12).setCellValue(userComm.getReplyTo());
                        commContentRow.createCell(13).setCellValue(userComm.getSubject());
                        commContentRow.createCell(14).setCellValue(userComm.getMessage());
                }
        }

        public String getErrorMsg() {
                return errorMsg;
        }
}