Subversion Repositories SmartDukaan

package in.shop2020.support.services;

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

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

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

import in.shop2020.thrift.clients.UserContextServiceClient;

import java.io.ByteArrayOutputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.util.Calendar;

import java.util.List;

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.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.thrift.TException;

public class RegisteredUsersGenerator {

    UserContextServiceClient usc;

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

    public RegisteredUsersGenerator() {

        try {

            usc = new UserContextServiceClient();

            uClient = usc.getClient();

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

    /**

     * This method is used in RegisteredUsersController. 

     * If any registered user(s) exist(s), then returns ByteArrayOutputStream containing user(s) data,

     * otherwise returns null

     * @param startDate -- inclusive

     * @param endDate -- inclusive

     * @return

     */

    public ByteArrayOutputStream generateRegisteredUsersReport() {

        List<User> users = null;

        try {

            users = uClient.getAllUsers(UserType.USER);

        } catch (TException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        if(users == null || users.isEmpty()) {

            return null;

        }

        // Preparing XLS file for output

        return getSpreadSheetData(users);

    }

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

    // formatting

    private ByteArrayOutputStream getSpreadSheetData(List<User> users) {

        ByteArrayOutputStream baosXLS = new ByteArrayOutputStream();

        Workbook wb = new HSSFWorkbook();

        Font font = wb.createFont();

        font.setBoldweight(Font.BOLDWEIGHT_BOLD);

        CellStyle style = wb.createCellStyle();

        style.setFont(font);

        CellStyle styleWT = wb.createCellStyle();

        styleWT.setWrapText(true);

        Sheet userSheet = wb.createSheet("User");

        short userSerialNo = 0;

        Row titleRow = userSheet.createRow(userSerialNo++);

        Cell titleCell = titleRow.createCell(0);

        titleCell.setCellValue("Registered Users");

        titleCell.setCellStyle(style);

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

        userSheet.createRow(userSerialNo++);

        Row headerRow = userSheet.createRow(userSerialNo++);

        headerRow.createCell(0).setCellValue("User Id");

        headerRow.createCell(1).setCellValue("Email");

        headerRow.createCell(2).setCellValue("Name");

        headerRow.createCell(3).setCellValue("Communication Email");

        headerRow.createCell(4).setCellValue("Date of Birth");

        headerRow.createCell(5).setCellValue("Sex");

        headerRow.createCell(6).setCellValue("Mobile");

        headerRow.createCell(7).setCellValue("Default Address");

        int addrColWidth = 40;

        userSheet.setColumnWidth(7, 256 * addrColWidth); // set width of address column to 40 characters

        Calendar calendar = Calendar.getInstance();

        Row contentRow;

        float defaultRowHeight = userSheet.getDefaultRowHeightInPoints();

        for (User u : users) {

            userSerialNo++;

            contentRow = userSheet.createRow(userSerialNo);

            contentRow.createCell(0).setCellValue(u.getUserId());

            contentRow.createCell(1).setCellValue(u.getEmail());

            contentRow.createCell(2).setCellValue(u.getName());

            contentRow.createCell(3).setCellValue(u.getCommunicationEmail());

            contentRow.createCell(4).setCellValue(u.getDateOfBirth());

            // calendar.setTimeInMillis

            contentRow.createCell(5).setCellValue(u.getSex().name());

            contentRow.createCell(6).setCellValue(u.getMobileNumber());

            List<Address> addresses = u.getAddresses();

            if(addresses == null || addresses.isEmpty()) {

                continue;

            }

            int i = 0, col, defaultAddrCol = 7, maxAddrLength = 0, len;

            String addressString;

            for(Address a : addresses) {

                if(a.getId() == u.getDefaultAddressId()) {

                    col = defaultAddrCol;

                } else {

                    i++;

                    col = defaultAddrCol + i;

                    headerRow.createCell(col).setCellValue("Address " + i);

                }

                userSheet.setColumnWidth(col, addrColWidth * 256);

                addressString = generateAddressString(a);

                contentRow.createCell(col).setCellValue(addressString);

                contentRow.getCell(col).setCellStyle(styleWT);

                len = addressString.length();

                if(len > maxAddrLength)

                    maxAddrLength = len;

            }

            contentRow.setHeightInPoints((maxAddrLength / addrColWidth + 1) * defaultRowHeight); // Setting Row Height

        }

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

            userSheet.autoSizeColumn(i);

        }

        // Write the workbook to the output stream

        try {

            wb.write(baosXLS);

            baosXLS.close();

        } catch (IOException e) {

            e.printStackTrace();

        }

        return baosXLS;

    }

    private String generateAddressString(Address a) {

        String addrStr = "";

        addrStr += "Name: " + getStringValue(a.getName());

        addrStr += "; Address: " + getStringValue(a.getLine1());

        addrStr += ", " + getStringValue(a.getLine2());

        addrStr += "; City: " + getStringValue(a.getCity());

        addrStr += "; State: " + getStringValue(a.getState());

        addrStr += "; Pin: " + getStringValue(a.getPin());

        addrStr += "; Phone: " + getStringValue(a.getPhone());

        return addrStr;

    }

    private String getStringValue(String str) {

        return str == null ? "" : str.trim();

    }

    public static void main(String[] args) {

        RegisteredUsersGenerator rug = new RegisteredUsersGenerator();

        try {

            String userHome = System.getProperty("user.home");

            FileOutputStream f = new FileOutputStream(userHome + "/registered-users.xls");

            ByteArrayOutputStream baosXLS = rug.generateRegisteredUsersReport();

            if(baosXLS == null) {

                System.out.println("No data");

                return;

            }

            baosXLS.writeTo(f);

            f.close();

        } catch (FileNotFoundException e) {

            e.printStackTrace();

        } catch (IOException e) {

            e.printStackTrace();

        }

        System.out.println("Successfully generated the registered users report");

    }

}