Subversion Repositories SmartDukaan

package in.shop2020.support.services;

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

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

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

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

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

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

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

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

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

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

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.TransactionClient;

import in.shop2020.thrift.clients.UserClient;

import java.io.ByteArrayOutputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.text.DateFormat;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.Calendar;

import java.util.Date;

import java.util.List;

import java.util.TimeZone;

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;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class RegisteredUsersGenerator {

    private static Logger logger = LoggerFactory.getLogger(RegisteredUsersGenerator.class);

	UserClient usc;

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

    TransactionClient tsc;

    in.shop2020.model.v1.order.TransactionService.Client tClient;

    public RegisteredUsersGenerator() {

        try {

            usc = new UserClient();

            uClient = usc.getClient();

            tsc = new TransactionClient();

            tClient = tsc.getClient();

        } catch (Exception e) {

            logger.error("Error initializing connection to user or order service", e);

        }

    }

    /**

     * 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(long startDate, long endDate) {

        List<User> users = null;

        try {

            users = uClient.getAllUsers(UserType.USER, startDate, endDate);

        } catch (TException e) {

            logger.error("Unable to get all users", e);

        }

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

        createRegisteredUserSheet(users, wb, style, styleWT);

        // Write the workbook to the output stream

        try {

            wb.write(baosXLS);

            baosXLS.close();

        } catch (IOException e) {

            logger.error("Unable to write the registered orders report to the byte array", e);

        }

        return baosXLS;

    }

    private void createRegisteredUserSheet(List<User> users, Workbook wb,

            CellStyle style, CellStyle styleWT) {

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

        short userSerialNo = 0;

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

        Cell titleCell = titleRow.createCell(0);

        titleCell.setCellValue("Registered Users");

        titleCell.setCellStyle(style);

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

        Row headerRow = registeredUserSheet.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("Registered On");

        headerRow.createCell(8).setCellValue("Last Login");

        headerRow.createCell(9).setCellValue("Successful Orders");

        headerRow.createCell(10).setCellValue("Pending/Failed Orders");

        headerRow.createCell(11).setCellValue("Items in Cart");

        registeredUserSheet.setColumnWidth(11, 20 * 256); // set width of items in cart column to 20 chars

        Calendar calendar = Calendar.getInstance();

        Row contentRow;

        float defaultRowHeight = registeredUserSheet.getDefaultRowHeightInPoints();

        UserState uState;

        DateFormat formatter = new SimpleDateFormat("EEE, dd-MMM-yyyy hh:mm a");

        Date currentDate = new Date();

        for (User u : users) {

            userSerialNo++;

            contentRow = registeredUserSheet.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());

            try {

                uState = uClient.getUserState(u.getUserId());

                calendar.setTimeInMillis(uState.getActiveSince());

                contentRow.createCell(7).setCellValue(formatter.format(calendar.getTime()));

                calendar.setTimeInMillis(uState.getLastLogin());

                contentRow.createCell(8).setCellValue(formatter.format(calendar.getTime()));

            } catch (UserContextException e) {

                logger.error("Error while getting user state", e);

            } catch (TException e) {

                logger.error("Unable to get user state", e);

            }

            try {

                List<Order> orders = tClient.getOrdersForCustomer(u.getUserId(), 0, currentDate.getTime(), null);

                int noOfFailedOrders = 0, noOfValidOrders = 0;

                for(Order order : orders) {

                    if(order.getStatus().getValue() < 3) {

                        noOfFailedOrders++;

                    } else {

                        noOfValidOrders++;

                    }

                }

                contentRow.createCell(9).setCellValue(noOfValidOrders);

                contentRow.createCell(10).setCellValue(noOfFailedOrders);

            } catch (TransactionServiceException e) {

                logger.error("Error while getting orders for the customer", e);

                e.printStackTrace();

            } catch (TException e) {

                logger.error("Unable to get orders for the customer", e);

                e.printStackTrace();

            }

            String itemList = getCartItems(u.getUserId());

            contentRow.createCell(11).setCellValue(itemList);

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

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

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

                continue;

            }

            int i = 0, col, defaultAddrCol = 12, maxAddrLength = 0, len, addrColWidth = 40;

            String addressString;

            for(Address a : addresses) {

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

                    col = defaultAddrCol;

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

                } else {

                    i++;

                    col = defaultAddrCol + i;

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

                }

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

                addressString = generateAddressString(a);

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

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

                len = addressString.length();

                if(len > maxAddrLength)

                    maxAddrLength = len;

            }

            if((len = itemList.length()) > maxAddrLength) {

                maxAddrLength = len;

            }

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

        }

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

            registeredUserSheet.autoSizeColumn(i);

        }

    }

    private String getCartItems(long userId) {

        String itemList = "";

        try {

            Cart cart = uClient.getCurrentCart(userId);

            if(cart == null) {

                return "";

            }

            List<Line> lines = cart.getLines();

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

                return "";

            }

            boolean firstItem = true;

            for(Line line : lines) {

                if(firstItem) {

                    firstItem = false;

                    itemList += line.getItemId();

                    continue;

                }

                itemList += ", " + line.getItemId();;

            }

        } catch (ShoppingCartException e) {

            logger.error("Error while getting the cart of the user", e);

        } catch (TException e) {

            logger.error("Unable to get the cart of the user", e);

        }

        return itemList;

    }

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

            String startDateStr = "01/11/2011";

            String endDateStr = "05/11/2011";

            DateFormat df = new SimpleDateFormat("MM/dd/yyyy");

            df.setTimeZone(TimeZone.getTimeZone("IST"));

            Date startDate = null;

            Date endDate = null;

            try {

                startDate = df.parse(startDateStr);

                endDate = df.parse(endDateStr);

                //Calendar cal = Calendar.getInstance();

                //cal.setTime(endDate);

                //endDate.setTime(cal.getTimeInMillis());

            } catch (ParseException pe) {

               System.out.println("Please enter start and end dates in format MM/dd/yyyy");

            }

            ByteArrayOutputStream baosXLS = rug.generateRegisteredUsersReport(startDate.getTime(), endDate.getTime());

            if(baosXLS == null) {

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

                return;

            }

            baosXLS.writeTo(f);

            f.close();

        } catch (FileNotFoundException e) {

            logger.error("Unable to create the registered users report", e);

        } catch (IOException e) {

            logger.error("IO Error while creating the registered users report", e);

        }

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

    }

}