Subversion Repositories SmartDukaan

package in.shop2020.inventory.controllers;

import in.shop2020.thrift.clients.HelperClient;
import in.shop2020.thrift.clients.WarehouseClient;
import in.shop2020.utils.HelperService;
import in.shop2020.warehouse.WarehouseService;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Map;

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

import org.apache.log4j.Logger;
import org.apache.shiro.SecurityUtils;
import org.apache.struts2.interceptor.ServletRequestAware;
import org.apache.struts2.interceptor.ServletResponseAware;
import org.apache.struts2.interceptor.SessionAware;

import com.opensymphony.xwork2.ValidationAwareSupport;

/**
 * Base class for all user action handlers i.e. controllers
 * 
 * @author Vikas
 */
public abstract class BaseController extends ValidationAwareSupport implements
        ServletResponseAware, ServletRequestAware, SessionAware
{
    private static final long serialVersionUID = 3339523094497219816L;
    protected static Logger log = Logger.getLogger(BaseController.class);
    
    //TODO get it from db
    public static final List<Long>PHYSICAL_WAREHOUSES = new ArrayList<Long>(Arrays.asList(7573L,7678L, 7681L, 8468L, 7720L, 8889L, 8947L, 9203L, 10010L, 9213L, 9349L, 9470L, 9513L,9514L));
    
    private String output;
    
    
    
    public String getOutput() {
                return output;
        }

        public void setOutput(String output) {
                this.output = output;
        }

        protected static final String INPUT = "input";
    protected static final String INDEX = "index";
    protected static final String EDIT_NEW = "editNew";
    protected static final String EDIT = "edit";
    protected static final String SHOW = "show";
    protected static final String EXCEPTION = "exception";
    protected static final String OUTPUT = "output";
    public static final String SESSION_WAREHOUSE_IDS = "allowedWarehouseIds";
    public static final String UNAUTHORIZED_ACCESS_ERROR = "Unauthorized Access ";
    protected final SimpleDateFormat SDF = new SimpleDateFormat("dd MMM, yyyy hh:mm a");

    protected HttpServletResponse response;
    protected HttpServletRequest request;
    protected HttpSession session;
    protected Map<String, Object> sessionMap;

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

    public void setServletRequest(HttpServletRequest request) {
        this.request = request;
    }
    
    @SuppressWarnings("unchecked")
        public void setSession(Map<String, Object> sessionMap) {
        this.session = request.getSession();
        List<Long> allowedWarehouseIds = null;
        if(this.session==null || this.session.getAttribute(SESSION_WAREHOUSE_IDS)==null) {
                try {
                HelperService.Client helperClient = new HelperClient().getClient();
                log.info("SecurityUtils.getSubject().getPrincipal().toString()  " + SecurityUtils.getSubject().getPrincipal().toString() );
                allowedWarehouseIds = helperClient.getWarehouseIdsForAgent(SecurityUtils.getSubject().getPrincipal().toString());
                if(allowedWarehouseIds.contains(0L)) {
                        allowedWarehouseIds = PHYSICAL_WAREHOUSES;
                }
                this.session.setAttribute(SESSION_WAREHOUSE_IDS, allowedWarehouseIds);
                } catch (Exception e) {
                        e.printStackTrace();
                }
        } else {
                allowedWarehouseIds = (List<Long>) this.session.getAttribute(SESSION_WAREHOUSE_IDS);
        }
        this.sessionMap = sessionMap;
    }
    
    /**
     * Utility method to convert a date to a readable format 
     */
    public String convertDate(Long date) {
        if (date == null || date == 0) {
            return "N/A";
        }

        return SDF.format(new Date(date));
    }

    public String index() {
        return INDEX;
    }

    public String editNew() {
        return EDIT_NEW;
    }

    public String edit() {
        return EDIT;
    }

    public boolean isPermitted(String permission) {
        log.info("Permission === " + permission + ", " +SecurityUtils.getSubject().isPermitted(permission));
        return SecurityUtils.getSubject().isPermitted(permission);
    }
    
    @SuppressWarnings("unchecked")
        public boolean isAutorizedToAccessWarehouse(Long warehouseId) {
        List<Long> warehouseIds;
                warehouseIds = (List<Long>)this.session.getAttribute(SESSION_WAREHOUSE_IDS);
                
                if(warehouseIds == null || warehouseIds.size() == 0){
                        return false;
                } else if(warehouseIds.contains(0l)){
                        return true; 
                } else {
                        if(warehouseId == null) {
                                return false;
                        } else {
                                if(warehouseIds.contains(warehouseId)){
                                        return true;
                                } else {
                                        return false;
                                }
                        }
                }
    }
    
    @SuppressWarnings("unchecked")
        public List<Long> getAuthorizedWarehousesForCurrentUser(){
        return (List<Long>)this.session.getAttribute(SESSION_WAREHOUSE_IDS);
    }
}