YangHuCost/modules/users/controllers/login_controller.js

/**
 * 登录相关控制器
 *
 * @author CaiAoLin
 * @date 2017/6/8
 * @version
 */
import UserModel from "../models/user_model";
import SettingModel from "../models/setting_model";
import CompilationModel from "../models/compilation_model";
// import Captcha from "../models/captcha";
// 验证码
const Captcha = require("../models/captcha");

class LoginController {

    /**
     * 登录页面
     *
     * @param {object} request
     * @param {object} response
     * @return {void}
     */
    async index(request, response) {
        // 判断是否有带token和ssoID参数
        if (request.query.ssoID !== undefined && request.query.token !== undefined) {
            let ssoID = request.query.ssoID;
            let token = request.query.token;
            let preferenceSetting = {};
            let compilationList = [];
            try {
                let userModel = new UserModel();
                // 调用接口验证登录信息
                let responseData = await userModel.getInfoFromSSO2(ssoID, token);
                // 先判断返回值是否为未激活状态
                if ( responseData === '-3') {
                    throw '因邮箱未完成认证，账号未激活；去<a href="https://sso.smartcost.com.cn" target="_blank">激活</a>。';
                }
                if ( responseData === '-2') {
                    throw 'token已过期，请重新登录Z+获取';
                }
                responseData = JSON.parse(responseData);
                if (typeof responseData !== 'object') {
                    throw 'ssoId错误或token过期';
                }

                if (responseData.length <= 0) {
                    throw '接口返回数据错误';
                }
                let userData = responseData[0];
                let sessionUser = {
                    ssoId: userData.id,
                    username: userData.username,
                    email: userData.useremail,
                    mobile: userData.mobile,
                };

                request.session.sessionUser = sessionUser;
                // 记录用户数据到数据库
                let result = await userModel.markUser(sessionUser, request);

                // 获取偏好设置
                let settingModel = new SettingModel();
                preferenceSetting = await settingModel.getPreferenceSetting(request.session.sessionUser.id);
                if (!result) {
                    throw '标记用户信息失败!';
                }
                let compilationModel = new CompilationModel();
                if (preferenceSetting.login_ask === 1 || preferenceSetting.select_version === ''){
                    preferenceSetting.login_ask = 1;
                    compilationList = await  compilationModel.getList();
                } else {
                    compilationList = [];
                }
                // 获取编办信息
                let sessionCompilation = request.session.sessionCompilation;

                if (preferenceSetting.login_ask === 0 && !sessionCompilation &&
                    preferenceSetting.select_version !== '') {
                    let compilationData = await compilationModel.getCompilationById(preferenceSetting.select_version);
                    // 判断当前用户的是使用免费版还是专业版
                    let compilationVersion = await userModel.getVersionFromUpgrade(sessionUser.ssoId, preferenceSetting.select_version);
                    request.session.compilationVersion = compilationVersion;
                    request.session.sessionCompilation = compilationData;
                    if(request.session.sessionUser.latest_used !== preferenceSetting.select_version) await userModel.updateLatestUsed(request.session.sessionUser.id,preferenceSetting.select_version);
                }
                console.log(`${request.session.sessionUser.real_name}--id:${request.session.sessionUser.id}--登录了系统`);
                if (preferenceSetting.login_ask === 1 || preferenceSetting.select_version === '') {
                    let renderData = {
                        versionData: compilationList,
                    };
                    response.render('users/html/login-ver', renderData);
                } else {
                    return response.redirect("/pm");
                }
            } catch (error) {
                console.log(error)
                return response.redirect("/login");
            }
        } else {
            let sessionUser = request.session.sessionUser;
            if (sessionUser !== undefined && sessionUser.ssoId >= 0) {
                return response.redirect("/pm");
            } else {
                response.render('users/html/login', {});
            }
        }
    }

    /**
     * 登录操作
     *
     * @param {object} request
     * @param {object} response
     * @return {string}
     */
    async login(request, response) {
        let account = request.body.account;
        let password = request.body.pw;
        let preferenceSetting = {};
        let compilationList = [];
        try {
            // 调用接口验证登录信息
            let userModel = new UserModel();
            let responseData = await userModel.getInfoFromSSO(account, password);
            responseData = JSON.parse(responseData);
            if (typeof responseData !== 'object') {
                throw '邮箱/手机 或 密码错误';
            }

            if (responseData.length <= 0) {
                throw '接口返回数据错误';
            }

            // 正确登录后 存入session
            let userData = responseData[0];

            if (userData.mobile === '') {
                return response.json({error: 2,ssoId: userData.id});
            }

            //还要判断account是否是专业版用户
            // let isPro = true;
            // const userInfo = await userModel.findDataByAccount(account);
            //
            // if (userInfo && userInfo.upgrade_list !== undefined) {
            //     for (const ul of userInfo.upgrade_list) {
            //         if (ul.isUpgrade === true) {
            //             isPro = true;
            //             break;
            //         }
            //     }
            // }
            // 专业版短信验证码验证
            // if (isPro) {
            //     const codeMsg = request.session.code;
            //     if (codeMsg !== undefined && request.body.code !== '') {
            //         const code = codeMsg.split('_')[0];
            //         const time = codeMsg.split('_')[1];
            //         console.log(code);
            //         console.log(request.body.code);
            //         if (Date.parse(new Date())/1000 > time+60*5 || request.body.code !== code) {
            //             return response.json({error: 3, msg: '验证码错误。'});
            //         } else {
            //             delete request.session.code;
            //         }
            //     } else {
            //         return response.json({error: 3, msg: '验证码错误。'});
            //     }
            // }

            // 判断极验验证码是否通过
            const captcha = new Captcha();
            const captchResult = await captcha.validate(request);
            if (!captchResult) {
                throw '极验验证码错误';
            }

            let sessionUser = {
                ssoId: userData.id,
                username: userData.username,
                email: userData.useremail,
                mobile: userData.mobile,
            };

            request.session.sessionUser = sessionUser;
            // 记录用户数据到数据库
            let result = await userModel.markUser(sessionUser, request);

            // 获取偏好设置
            let settingModel = new SettingModel();
            preferenceSetting = await settingModel.getPreferenceSetting(request.session.sessionUser.id);
            if (!result) {
                throw '标记用户信息失败!';
            }
            let compilationModel = new CompilationModel();
            if(preferenceSetting.login_ask === 1 || preferenceSetting.select_version === ''){
                preferenceSetting.login_ask = 1;
                compilationList = await  compilationModel.getList();
            }
            else{
                compilationList = [];
            }
            // 获取编办信息
            let sessionCompilation = request.session.sessionCompilation;

            if (preferenceSetting.login_ask === 0 && !sessionCompilation &&
                preferenceSetting.select_version !== '') {
                let compilationData = await compilationModel.getCompilationById(preferenceSetting.select_version);
                // 判断当前用户的是使用免费版还是专业版
                let compilationVersion = await userModel.getVersionFromUpgrade(sessionUser.ssoId, preferenceSetting.select_version);
                request.session.compilationVersion = compilationVersion;
                request.session.sessionCompilation = compilationData;
                if(request.session.sessionUser.latest_used !== preferenceSetting.select_version) await userModel.updateLatestUsed(request.session.sessionUser.id,preferenceSetting.select_version);
            }

        } catch (error) {
            console.log(error);
            return response.json({error: 1, msg: error});
        }
        console.log(`${request.session.sessionUser.real_name}--id:${request.session.sessionUser.id}--登录了系统`);
        response.json({
            error: 0,
            msg: '',
            login_ask: preferenceSetting.login_ask,
            compilation_list: JSON.stringify(compilationList),
            last_page: request.session.lastPage
        });
    }

    /**
     * 验证码注册
     *
     * @param {object} request
     * @param {object} response
     * @return {string}
     */
    async captcha(request, response) {
        const captcha = new Captcha();
        const res = await captcha.register(request);
        response.json(res);
    }

    /**
     * 判断用户是否是专业版用户
     * @param request
     * @param response
     * @returns {Promise<void>}
     */
    // async accountIsPro(request, response) {
    //     let res = {
    //         error: 0,
    //         msg: '',
    //         result: false,
    //     };
    //     try{
    //         const account = request.body.account;
    //         const password = request.body.pw;
    //
    //         // 根据邮箱或手机号获取账号信息
    //         let userModel = new UserModel();
    //         // 调用接口验证登录信息
    //         let responseData = await userModel.getInfoFromSSO(account, password);
    //         console.log(responseData);
    //         // 先判断返回值是否为未激活状态
    //         if ( responseData === '-3') {
    //             throw '因邮箱未完成认证，账号未激活；去<a href="https://sso.smartcost.com.cn" target="_blank">激活</a>。';
    //         }
    //         responseData = JSON.parse(responseData);
    //         if (typeof responseData !== 'object') {
    //             throw '邮箱/手机 或 密码错误';
    //         }
    //
    //         if (responseData.length <= 0) {
    //             throw '接口返回数据错误';
    //         }
    //
    //         // 正确登录后 存入session
    //         let userData = responseData[0];
    //
    //         if (userData.mobile === '') {
    //             return response.json({error: 2,ssoId: userData.id});
    //         }
    //
    //         // const userInfo = await userModel.findDataByAccount(account);
    //         // if (userInfo && userInfo.upgrade_list !== undefined) {
    //         //     for (const ul of userInfo.upgrade_list) {
    //         //         if (ul.isUpgrade === true) {
    //         //             res.result = true;
    //         //             res.data = userInfo.mobile;
    //         //             break;
    //         //         }
    //         //     }
    //         // } else {
    //         //     res.msg = '当前未存在此用户';
    //         // }
    //         res.result = true;
    //         // if (userInfo) {
    //         //     res.data = userInfo.mobile;
    //         // } else {
    //             res.data = userData.mobile;
    //         // }
    //     } catch (err) {
    //         res.error = 1;
    //         res.msg = err;
    //     }
    //
    //     response.json(res);
    // }

}

export default LoginController;