Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.support.services.RegisteredUsersGenerator;
import in.shop2020.thrift.clients.HelperServiceClient;
import in.shop2020.utils.StatisticsUser;

import java.io.ByteArrayOutputStream;
import java.io.IOException;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.apache.struts2.interceptor.ServletRequestAware;
import org.apache.struts2.interceptor.ServletResponseAware;

public class RegisteredUsersController implements ServletResponseAware, ServletRequestAware {

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

    private final String authsuccess = "authsuccess";
    private final String authfail = "authfail";
    private String message = "";
    
    public RegisteredUsersController() {

    }

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

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

    public String index() {
        if (getSessionUserName() == null) {
            return authfail;
        } else {
            return authsuccess;
        }
    }

    // Handles the POST request (Form Submission)
    public String create() {
        String username = request.getParameter("username");
        String password = request.getParameter("password");
        if(username != null && password != null) {
            try{
                HelperServiceClient hsc = new HelperServiceClient();
                in.shop2020.utils.HelperService.Client client = hsc.getClient();
                StatisticsUser user = client.authenticateStatisticsUser(username, password);
                session.setAttribute("username", user.getUsername());
                return authsuccess;
            }catch(Exception e){
                e.printStackTrace();
                return authfail;
            }
        }
        if (getSessionUserName() == null) {
            return authfail;
        }
        

        RegisteredUsersGenerator usersReportGenerator = new RegisteredUsersGenerator();
        ByteArrayOutputStream baos = usersReportGenerator.generateRegisteredUsersReport();

        if(baos == null) {
            message = "No registered user exists.";;
            return authsuccess;
        }
        // Preparing XLS file for output
        response.setContentType("application/vnd.ms-excel");
        response.setHeader("Content-disposition", "inline; filename=registered-users" + ".xls");
        ServletOutputStream sos;
        try {
            sos = response.getOutputStream();
            baos.writeTo(sos);
            sos.flush();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return authsuccess;
    }

    public String getSessionUserName() {
        return (String) session.getAttribute("username");
    }

    public String getMessage() {
        return message;
    }
}