shiro （三） spring结合 LoginController

 

package com.miv.shiro.login.controller;

 

import java.util.Map;

 

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

 

import org.apache.shiro.SecurityUtils;

import org.apache.shiro.subject.Subject;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Controller;

import org.springframework.ui.ModelMap;

import org.springframework.web.bind.annotation.PathVariable;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.ResponseBody;

import org.springframework.web.servlet.mvc.support.RedirectAttributes;

 

import com.miv.agencyportal.menu.home.service.AgencyHomeService;

import com.miv.common.constant.AgencyModules;

import com.miv.common.constant.CallCenterModules;

import com.miv.common.constant.UserModules;

import com.miv.common.utils.APSysLogger;

import com.miv.common.utils.CPSysLogger;

import com.miv.common.utils.UPSysLogger;

import com.miv.core.constant.DatabaseConstants;

import com.miv.core.controller.WebBaseController;

import com.miv.core.json.JsonResponse;

import com.miv.core.utils.MessageUtils;

import com.miv.entity.User;

import com.miv.form.LoginView;

import com.miv.shiro.common.ShiroEncryption;

import com.miv.shiro.login.service.LoginService;

 

@RequestMapping("/login")

@Controller

public class LoginController extends WebBaseController {

 

    final Logger logger = LoggerFactory.getLogger(LoginController.class);

 

    @Autowired

    private LoginService loginService;

    @Autowired

    private AgencyHomeService agencyHomeService;

    @Autowired

    private APSysLogger apSysLogger;

    @Autowired

    private CPSysLogger cpSysLogger;

    @Autowired

    private UPSysLogger upSysLogger;

 

    /**

     * 返回地址

     */

    public String getLoaction(Integer ROLE, String role_code) throws Exception {

        Subject subject = SecurityUtils.getSubject();

        if (subject.getPrincipal() != null) {// 记住我功能

            String loginName = subject.getPrincipal().toString();

            boolean flag = loginService.checkDisableStatus(loginName) || loginService.checkPassword(loginName);// 密码修改返回true,否则false；

            if (flag) {// 修改密码后

                return ShiroEncryption.decryptionURL(ROLE, ShiroEncryption.FAILURE);

            } else if (subject.hasRole(role_code)) {// 普通用户

                if (ROLE.equals(DatabaseConstants.ROLE_CODE_AGENCY)) {

                    apSysLogger.log(MessageUtils.getMessage("log_ap_login"), getSessionSubject().getId(),

                            AgencyModules.Login.loginManage,

                            MessageUtils.getMessage("log_login_data", new Object[] { loginName, role_code, true }));

                }

                if (ROLE.equals(DatabaseConstants.ROLE_CODE_CALL_CENTER)) {

                    cpSysLogger.log(MessageUtils.getMessage("log_cp_login"), getSessionSubject().getId(),

                            CallCenterModules.Login.loginManage,

                            MessageUtils.getMessage("log_login_data", new Object[] { loginName, role_code, true }));

                }

                if (ROLE.equals(DatabaseConstants.ROLE_CODE_USER)) {

                    upSysLogger.log(MessageUtils.getMessage("log_up_login"), getSessionSubject().getId(),

                            UserModules.Login.loginManage,

                            MessageUtils.getMessage("log_login_data", new Object[] { loginName, role_code, true }));

                }

                return ShiroEncryption.decryptionURL(ROLE, ShiroEncryption.SUCCESS);

            } else {// 其他

                return ShiroEncryption.decryptionURL(ROLE, ShiroEncryption.FAILURE);

            }

        } else {// 未记住我

            return ShiroEncryption.decryptionURL(ROLE, ShiroEncryption.FAILURE);

        }

    }

 

    /**

     * 判定权限不足时跳转对应登录页

     * 

     * @param role_code

     * @return

     */

    @RequestMapping("/init")

    public String init(HttpServletRequest request) throws Exception {

        String[] role_code = (String[]) SecurityUtils.getSubject().getSession().getAttribute("role_code");

        if (role_code == null || role_code[0].trim().equals("")) {// 非法进来的就普通用户

            request.setAttribute("_cmd", ShiroEncryption.encryption(DatabaseConstants.ROLE_CODE_USER));

            return ShiroEncryption.decryptionURL(DatabaseConstants.ROLE_CODE_USER, ShiroEncryption.FAILURE);

        }

        Integer ROLE_CODE = ShiroEncryption.decryption(role_code[0]);

        request.setAttribute("_cmd", ShiroEncryption.encryption(ROLE_CODE));

 

        return ShiroEncryption.decryptionURL(ROLE_CODE, ShiroEncryption.FAILURE);

    }

 

    /**

     * 登录处理普通用户登录地址

     * 

     * @param ModelMap

     * @param HttpServletRequest

     * @return String

     * @throws Exception

     */

    @RequestMapping("/user")

    public String user(ModelMap map, HttpServletRequest request) throws Exception {

        request.setAttribute("_cmd", ShiroEncryption.encryption(DatabaseConstants.ROLE_CODE_USER));

        return this.getLoaction(DatabaseConstants.ROLE_CODE_USER, "user");

    }

 

    /**

     * 登录处理经销商登录地址

     * 

     * @param ModelMap

     * @param HttpServletRequest

     * @return String

     * @throws Exception

     */

    @RequestMapping("/agency")

    public String agency(ModelMap map, HttpServletRequest request, HttpServletResponse response, HttpSession session)

            throws Exception {

        request.setAttribute("_cmd", ShiroEncryption.encryption(DatabaseConstants.ROLE_CODE_AGENCY));

        String returnHome = this.getLoaction(DatabaseConstants.ROLE_CODE_AGENCY, "agency");

        String successHome = ShiroEncryption.decryptionURL(DatabaseConstants.ROLE_CODE_AGENCY, ShiroEncryption.SUCCESS);

        if (returnHome.equalsIgnoreCase(successHome)) {

            Map<String, String> urlM = agencyHomeService.findPlots(super.getSessionSubject(), response, session);

            request.setAttribute("urlM", urlM);

        }

        return returnHome;

    }

 

    /**

     * 登录处理客户中心登录地址

     * 

     * @param ModelMap

     * @param HttpServletRequest

     * @return String

     * @throws Exception

     */

    @RequestMapping("/call")

    public String call(ModelMap map, HttpServletRequest request) throws Exception {

        request.setAttribute("_cmd", ShiroEncryption.encryption(DatabaseConstants.ROLE_CODE_CALL_CENTER));

        return this.getLoaction(DatabaseConstants.ROLE_CODE_CALL_CENTER, "callCenter");

    }

 

    /**

     * 登录处理mapping

     * 

     * @param Model

     * @param BindingResult

     * @return String

     * @throws Exception

     */

    @ResponseBody

    @RequestMapping(value = "/index")

    public JsonResponse login(LoginView user, String _cmd, HttpServletRequest request) throws Exception {

        JsonResponse jsonResponse = null;

        user.setLoginName(user.getLoginName().trim());// 去除前后空格

        boolean remember = false;

        if (user.getRememberMe() != null && (!user.getRememberMe().trim().equals(""))) {

            remember = true;

        } else {

            remember = false;

        }

        user.set_cmd(_cmd);

        jsonResponse = loginService.findLogin(remember, user);

        if (jsonResponse.isSuccess()) {

            Integer ROLE_CODE = ShiroEncryption.decryption(user.get_cmd());

            if (ROLE_CODE.equals(DatabaseConstants.ROLE_CODE_AGENCY)) {

                apSysLogger.log(MessageUtils.getMessage("log_ap_login"), getSessionSubject().getId(),

                        AgencyModules.Login.loginManage,

                        MessageUtils.getMessage("log_login_data", new Object[] { user.getLoginName(), _cmd, false }));

            }

            if (ROLE_CODE.equals(DatabaseConstants.ROLE_CODE_CALL_CENTER)) {

                cpSysLogger.log(MessageUtils.getMessage("log_cp_login"), getSessionSubject().getId(),

                        CallCenterModules.Login.loginManage,

                        MessageUtils.getMessage("log_login_data", new Object[] { user.getLoginName(), _cmd, false }));

            }

            if (ROLE_CODE.equals(DatabaseConstants.ROLE_CODE_USER)) {

                upSysLogger.log(MessageUtils.getMessage("log_up_login"), getSessionSubject().getId(),

                        UserModules.Login.loginManage,

                        MessageUtils.getMessage("log_login_data", new Object[] { user.getLoginName(), _cmd, false }));

            }

 

        }

        return jsonResponse;

    }

 

    /**

     * 登出处理

     * 

     * @param ModelMap

     * @param BindingResult

     * @return

     * @throws Exception

     */

    @RequestMapping("/logout/{roleCode}")

    public String logout(@PathVariable String roleCode, ModelMap map, RedirectAttributes redirectAttributes,

            HttpServletRequest request) throws Exception {

        Integer ROLE_CODE = ShiroEncryption.decryption(roleCode);

        request.setAttribute("_cmd", ShiroEncryption.encryption(ROLE_CODE));

        User user = null;

        try {

            user = super.getSessionSubject();

        } catch (Exception e) {

            return ShiroEncryption.decryptionURL(ROLE_CODE, ShiroEncryption.FAILURE);

        }

 

        if (ROLE_CODE.equals(DatabaseConstants.ROLE_CODE_AGENCY)) {

            apSysLogger.log(MessageUtils.getMessage("log_ap_loginout"), getSessionSubject().getId(),

                    AgencyModules.Login.loginManage,

                    MessageUtils.getMessage("log_loginout_data", new Object[] { user.getLoginName() }));

        }

        if (ROLE_CODE.equals(DatabaseConstants.ROLE_CODE_CALL_CENTER)) {

            cpSysLogger.log(MessageUtils.getMessage("log_cp_loginout"), getSessionSubject().getId(),

                    CallCenterModules.Login.loginManage,

                    MessageUtils.getMessage("log_loginout_data", new Object[] { user.getLoginName() }));

        }

        if (ROLE_CODE.equals(DatabaseConstants.ROLE_CODE_USER)) {

            upSysLogger.log(MessageUtils.getMessage("log_up_loginout"), getSessionSubject().getId(),

                    UserModules.Login.loginManage,

                    MessageUtils.getMessage("log_loginout_data", new Object[] { user.getLoginName() }));

        }

        try {

            SecurityUtils.getSubject().logout();

        } catch (Exception e) {

            logger.warn("Session失效注销");

        }

 

        return ShiroEncryption.decryptionURL(ROLE_CODE, ShiroEncryption.FAILURE);

    }

 

    @RequestMapping("/home")

    public String loginHome(HttpServletRequest request) throws Exception {

        Subject subject = SecurityUtils.getSubject();

        String url = ShiroEncryption.decryptionURL(DatabaseConstants.ROLE_CODE_USER, ShiroEncryption.FAILURE);

        if (subject.isAuthenticated()) {

            User user = loginService.findAllByLoginName(subject.getPrincipal().toString());

            url = "login/login_goto";

            request.setAttribute("url",

                    ShiroEncryption.decryptionURL(Integer.valueOf(user.getRole().getId().toString())));

        }

        request.setAttribute("_cmd", ShiroEncryption.encryption(DatabaseConstants.ROLE_CODE_USER));

        return url;

    }

 

    /**** 找回密码 *****/

    /**

     * 提交申请转向JSP

     * 

     * @return

     */

    @RequestMapping("/applyPasswordJsp")

    public String applyPassword() {

        return "login/apply_password";

    }

 

    /**

     * 获得申请key，并发送邮件

     * 

     * @param loginName

     * @param email

     * @return

     * @throws Exception

     */

    @ResponseBody

    @RequestMapping("/applyPassword")

    public JsonResponse applyPassword(String loginName, String email) throws Exception {

        JsonResponse jsonResponse = null;

        boolean flag = loginService.insertApplyPassword(loginName, email);

        jsonResponse = ajaxSuccess(flag);

        return jsonResponse;

    }

 

    /**

     * 登录找回密码JSP

     * 

     * @param request

     * @param applyKey

     * @return

     */

    @RequestMapping("/retrievePasswordJsp/{applyKey}")

    public String retrievePassword(HttpServletRequest request, @PathVariable String applyKey) {

        String url = "";

        boolean refresh = request.getSession().getAttribute("applyPassword") != null

                && ("REFRESH").equals(request.getSession().getAttribute("applyPassword").toString());

        if (refresh) {

            url = "login/retrieve_password";

            return url;

        }

        boolean flag = loginService.updateApplyPassword(applyKey);

 

        if (!flag) {

            request.getSession().setAttribute("applyPassword", "NO");

            url = "login/apply_key_error";

        } else {

            request.getSession().setAttribute("applyPassword", "REFRESH");

            url = "login/retrieve_password";

        }

        return url;

    }

 

    /**

     * 找回密码--新密码更改操作

     * 

     * @param applyKey

     * @return

     */

    @ResponseBody

    @RequestMapping("/retrievePassword")

    public JsonResponse retrievePassword(String applyKey, String newPassword, String confirmPassword) {

        JsonResponse jsonResponse = null;

        if (!newPassword.equals(confirmPassword)) {

            jsonResponse = ajaxFailure("确认密码与新密码不一致");

            return jsonResponse;

        }

        boolean flag = loginService.updateApplyPassword(applyKey, newPassword);

        if (flag) {

            jsonResponse = ajaxSuccess();

            User user = loginService.findUserByApplyKey(applyKey);

            upSysLogger.log(MessageUtils.getMessage("log_up_changePassword"), user.getId(),

                    UserModules.MyInfo.CHANGE_PASSWORD,

                    MessageUtils.getMessage("log_up_changePassword_data", new Object[] { user.getLoginName() }));

        } else {

            jsonResponse = ajaxFailure("修改失败");

        }

        return jsonResponse;

    }

 

    /**

     * 此为测试JSP

     * */

    @RequestMapping("/test/{jsp}")

    public String test(@PathVariable String jsp, ModelMap map, RedirectAttributes redirectAttributes,

            HttpServletRequest request) throws Exception {

        // System.out.println(jsp);

        return "sections/" + jsp;

    }

}