Subversion Repositories SmartDukaan

package in.shop2020.recharge.controllers;

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

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

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

import in.shop2020.thrift.clients.TransactionClient;

import java.text.SimpleDateFormat;

import java.util.Date;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

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

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

    public static Map<Long, HotspotStore> hotspotStores = new HashMap<Long, HotspotStore>();

    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;

    protected Long storeId;

    public static final String HEADER_X_FORWARDED_FOR = "X-FORWARDED-FOR";

    public static Map<Long, String> mobileProvidersMap;

    public static Map<Long, String> dthProvidersMap;

    public static Map<Long, List<RechargePlan>> operatorPlanMap;

    public static Map<String, Long> operatorNametoValueMap = new HashMap<String, Long>();

    static {

        TransactionClient tcl;

        try {

            operatorPlanMap = new HashMap<Long, List<RechargePlan>>();

            tcl = new TransactionClient();

            mobileProvidersMap = tcl.getClient().getServiceProviders(RechargeType.MOBILE, true);

            dthProvidersMap = tcl.getClient().getServiceProviders(RechargeType.DTH, true);

            for (Long operatorId : mobileProvidersMap.keySet()) {

                List<RechargePlan> plans = tcl.getClient().getPlansForOperator(operatorId);

                if (!plans.isEmpty()) {

                    operatorPlanMap.put(operatorId, plans);

                }

            }

            for (Entry<Long, String> entry : mobileProvidersMap.entrySet()) {

                operatorNametoValueMap.put(entry.getValue(), entry.getKey());

            }

        } catch(Exception e) {

            log.error("Could not get all plans", e);

        }

    }

    public void setServletResponse(HttpServletResponse response) {

        this.response = response;

    }

    public void setServletRequest(HttpServletRequest request) {

        this.request = request;

        this.session = request.getSession();

    }

    public void setSession(Map<String, Object> sessionMap) {

        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 remoteAddr(HttpServletRequest request) {

        String remoteAddr = "";

        String x;

        x = request.getHeader(HEADER_X_FORWARDED_FOR);

        if (x != null && !x.isEmpty()) {

            remoteAddr = x;

            int idx = remoteAddr.lastIndexOf(',');

            if (idx > -1) {

                remoteAddr = remoteAddr.substring(idx + 1).trim();

            }

        } else {

            remoteAddr = request.getRemoteAddr();

        }

        return remoteAddr;

    }

    public String index() {

        return INDEX;

    }

    public String editNew() {

        return EDIT_NEW;

    }

    public String edit() {

        return EDIT;

    }

    public boolean isPermitted(String permission) {

        return SecurityUtils.getSubject().isPermitted(permission);

    }

    public String getStoreCode(){

        return hotspotStores.get(storeId).getHotspotId();   

    }

    public String getCollectedAmount(){

        return hotspotStores.get(storeId).getCollectedAmount()+"";

    }

    public Long getStoreId() {

        return storeId;

    }

    public void setStoreId(Long storeId) {

        this.storeId = storeId;

    }

    public static void setDthProvidersMap(Map<Long, String> dthProvidersMap) {

        BaseController.dthProvidersMap = dthProvidersMap;

    }

    public static Map<Long, String> getDthProvidersMap() {

        return dthProvidersMap;

    }

    public static void setMobileProvidersMap(Map<Long, String> mobileProvidersMap) {

        BaseController.mobileProvidersMap = mobileProvidersMap;

    }

    public static Map<Long, String> getMobileProvidersMap() {

        return mobileProvidersMap;

    }

}