ConstructionCost/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) {
        let sessionUser = request.session.sessionUser;
        if (sessionUser !== undefined && sessionUser.ssoId >= 0) {
            return response.redirect("/pm");
        }
        // 判断是否有带token和ssoID参数
        if (request.query.ssoID !== undefined && request.query.token !== undefined) {
            let ssoID = request.query.ssoID;
            let token = request.query.token;
            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];
                //还要判断account是否是专业版用户
                let isPro = false;
                const userInfo = await userModel.findDataByAccount(userData.mobile);
                if (userInfo && userInfo.upgrade_list !== undefined) {
                    for (const ul of userInfo.upgrade_list) {
                        if (ul.isUpgrade === true) {
                            isPro = true;
                            break;
                        }
                    }
                }
                if (isPro) {
                    let renderData = {
                      ssoID: ssoID,
                      token: token,
                      mobile: userInfo.mobile,
                      showMobile: userInfo.mobile.substr(0, 3) + '****' + userInfo.mobile.substr(7, 11)
                    };
                    response.render('users/html/login-sms', renderData);
                    return
                } else {

                }
            } catch (error) {
                console.log(error)
                return response.redirect("/login");
            }
        }
        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);
            // 先判断返回值是否为未激活状态
            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});
            }

            //还要判断account是否是专业版用户
            // let isPro = false;
            // 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
        });
    }

    /**
     * Z+平台专业版用户登录方式
     */
    async startupLogin(request, response) {
        let ssoID = request.body.ssoID;
        let token = request.body.token;
        let preferenceSetting = {};
        let compilationList = [];
        try {
            let userModel = new UserModel();

            // 调用接口验证登录信息
            let responseData = await userModel.getInfoFromSSO2(ssoID, token);
            if ( responseData === '-2') {
                throw 'token已过期，请重新登录Z+获取';
            }
            responseData = JSON.parse(responseData);
            if (typeof responseData !== 'object') {
                throw 'ssoId错误或token过期';
            }

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

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

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

            // 专业版短信验证码验证
            const codeMsg = request.session.code;
            if (codeMsg !== undefined && request.body.code !== '') {
                const code = codeMsg.split('_')[0];
                const time = codeMsg.split('_')[1];
                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: '验证码错误。'});
            }

            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);
    //         // 先判断返回值是否为未激活状态
    //         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 = '当前未存在此用户';
    //         }
    //     } catch (err) {
    //         res.error = 1;
    //         res.msg = err;
    //     }
    //
    //     response.json(res);
    // }

}

export default LoginController;