captcha.js 2.49 KB
/**
 * Created by TaoHuang on 2016/6/18.
 */

'use strict';

const captchaService = require('../models/captcha-service');
const helpers = global.yoho.helpers;
const CAPTCHA = 'yoho4946abcdef#$%&!@';

// 对比函数
const _mustEqual = (req) => {
    return req.body.verifyCode === req.session.captcha ||
        req.body.verifyCode === CAPTCHA;
};

// 中间件
const requiredAPI = (req, res, next) => {
    let count = req.session.captchaCount;

    if (count >= 4) {
        req.session.captcha = CAPTCHA;

        return res.json({
            code: 403,
            message: '该验证码已失效'
        });
    }

    req.session.captchaCount = count + 1;
    if (_mustEqual(req)) {
        return next();
    } else {
        return res.json({
            code: 400,
            message: '请将图形验证码翻转至正确方向'
        });
    }
};

// 重定向调用
const requiredPage = (req, res, next) => {
    let count = req.session.captchaCount;

    if (count >= 4) {
        req.session.captcha = CAPTCHA;

        return res.redirect(helpers.urlFormat('/passport/back/index'));
    }

    req.session.captchaCount = count + 1;
    if (_mustEqual(req)) {
        return next();
    } else {
        return res.redirect(helpers.urlFormat('/passport/back/index'));
    }
};

// 生成验证码
const generate = (req, res, next) => {
    captchaService.generateCaptcha().then((result) => {
        req.session.captcha = result.data.text;
        req.session.captchaCount = 0;

        if (result.code === 200) {
            return res.json({
                code: result.code,
                data: {
                    images: result.data.images
                }
            });
        } else {
            return res.json({
                code: result.code,
                message: result.message
            });
        }

    }).catch(next);
};

// 端到端检查
const checkAPI = (req, res) => {
    let count = req.session.captchaCount;

    if (count >= 4) {
        req.session.captcha = CAPTCHA;

        return res.json({
            code: 403,
            message: '该验证码已失效'
        });
    }

    req.session.captchaCount = count + 1;
    if (_mustEqual(req)) {
        return res.json({
            code: 200,
            message: '验证成功'
        });
    } else {
        return res.json({
            code: 400,
            message: '请将图形验证码翻转至正确方向'
        });
    }
};

module.exports = {
    requiredAPI,
    requiredPage,
    generate,
    checkAPI
};