Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.support.services.RegisteredUsersGenerator;

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

import javax.servlet.ServletOutputStream;

import org.apache.struts2.convention.annotation.InterceptorRef;
import org.apache.struts2.convention.annotation.InterceptorRefs;
import org.apache.struts2.convention.annotation.Result;


@InterceptorRefs({
    @InterceptorRef("myDefault"),
    @InterceptorRef("login")
})

public class RegisteredUsersController extends ReportsController {

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

    }

    public String index() {
        if(!canAccessReport()) {
            return "exception";
        }
        return authsuccess;
    }

    // Handles the POST request (Form Submission)
    public String create() {
        RegisteredUsersGenerator usersReportGenerator = new RegisteredUsersGenerator();
        ByteArrayOutputStream baos = usersReportGenerator.generateRegisteredUsersReport();

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

    public String getMessage() {
        return message;
    }
}