Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.support.services.RegisteredUsersGenerator;
import in.shop2020.support.utils.ReportsUtils;
import in.shop2020.thrift.clients.UserClient;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.TimeZone;

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.convention.annotation.InterceptorRef;
import org.apache.struts2.convention.annotation.InterceptorRefs;
import org.apache.struts2.convention.annotation.Result;
import org.apache.struts2.convention.annotation.Results;
import org.apache.struts2.interceptor.ServletRequestAware;
import org.apache.struts2.interceptor.ServletResponseAware;
import org.apache.struts2.util.ServletContextAware;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@InterceptorRefs({
    @InterceptorRef("defaultStack"),
    @InterceptorRef("login")
})
@Results({
    @Result(name="authfail", type="redirectAction", params = {"actionName" , "reports"})
})
public class RegisteredUsersController implements ServletRequestAware, ServletResponseAware, ServletContextAware {
    
    private static Logger logger = LoggerFactory.getLogger(RegisteredUsersController.class);

    private HttpServletRequest request;
    private HttpSession session;
    private HttpServletResponse response;
    private ServletContext context;

    private final String authsuccess = "authsuccess";
    private String message = "";

    private Date endDate;
    private Date startDate;

    private String errorMsg;
    
    public RegisteredUsersController() {

    }

    public String index() {
        if(!ReportsUtils.canAccessReport((Long)session.getAttribute(ReportsUtils.ROLE), request.getServletPath())) {
            return "authfail";
        }
        return authsuccess;
    }

    // Handles the POST request (Form Submission)
    public String create() {
        String startDateStr = request.getParameter("startDate");
        String endDateStr = request.getParameter("endDate");

        DateFormat df = new SimpleDateFormat("MM/dd/yyyy");
        df.setTimeZone(TimeZone.getTimeZone("IST"));
        try {
            startDate = df.parse(startDateStr);
            endDate = df.parse(endDateStr);
        } catch (ParseException pe) {
            errorMsg = "Please enter start and end dates in format MM/dd/yyyy";
            return authsuccess;
        }

        UserClient usc;
        in.shop2020.model.v1.user.UserContextService.Client uClient;
        
        List<String> emails = new ArrayList<String>();
        
        try {
            usc = new UserClient();
            uClient = usc.getClient();
            
            emails = uClient.getUserEmails(startDate.getTime(), endDate.getTime());
        } catch (Exception e) {
            logger.error("Error initializing connection to user or order service", e);
        }
        
        // Preparing XLS file for output
                response.setContentType("application/vnd.ms-excel");
                response.setHeader("Content-disposition", "attachment; filename=email-addresses.xls");
                
        ServletOutputStream sos;
        try {
                StringBuilder builder = new StringBuilder();
                
                for (String email: emails)      {
                        builder.append(email);
                        builder.append("\n");
                }
                
                        sos = response.getOutputStream();
                        sos.write(builder.toString().getBytes());
                        sos.flush();
                        sos.close();
        } catch (IOException e) {
            logger.error("Error streaming registered users report");
        }
        return authsuccess;
    }

    public String getMessage() {
        return message;
    }

    @Override
    public void setServletRequest(HttpServletRequest req) {
        this.request = req;
        this.session = req.getSession();        
    }

    @Override
    public void setServletResponse(HttpServletResponse res) {
        this.response = res;
    }
    
    @Override
    public void setServletContext(ServletContext context) {
        this.context = context;
    }

    public String getServletContextPath() {
        return context.getContextPath();
    }

    public void setStartDate(Date startDate) {
        this.startDate = startDate;
    }

    public String getErrorMsg() {
        return errorMsg;
    }

    public void setErrorMsg(String errorMsg) {
        this.errorMsg = errorMsg;
    }
}