check.js 2.21 KB
'use strict';

const _ = require('lodash');
const logger = global.yoho.logger;
const Geetest = require('geetest');
const config = global.yoho.config;
const co = Promise.coroutine;
const cache = global.yoho.cache.master;
const captcha = new Geetest({
    geetest_id: 'bce95d796bc3058615fdf2ec2c0aef29',
    geetest_key: '124c41d3a22aa48f36ae3cd609c51db5'
});

exports.index = (req, res) => {
    req.yoho.captchaShow = false;
    res.locals.useGeetest = true;

    if (_.has(res, 'locals.loadJs')) {
        res.locals.loadJs.push({
            src: global.yoho.config.geetestJs
        });
    } else {
        res.locals.loadJs = [
            {
                src: global.yoho.config.geetestJs
            }
        ];
    }
    res.render('check', {
        width750: true,
        localCss: true
    });
};

exports.submit = (req, res) => {
    co(function * () {
        let challenge = req.body.geetest_challenge,
            validate = req.body.geetest_validate,
            seccode = req.body.geetest_seccode;

        let errRes = {
            code: 400,
            message: '验证码错误',
            captchaShow: true,
            changeCaptcha: true
        };

        if (!challenge || !validate || !seccode) {
            return res.json(errRes);
        }

        let geetestRes = yield captcha.validate({
            challenge,
            validate,
            seccode
        });

        if (geetestRes) {
            logger.info('geetest success');

            let remoteIp = req.get('X-Forwarded-For') || req.ip;

            if (remoteIp.indexOf(',') > 0) {
                let arr = remoteIp.split(',');

                remoteIp = arr[0];
            }

            let operations = [cache.delAsync(`${config.app}:limiter:${remoteIp}`)];

            _.forEach(config.REQUEST_LIMIT, (val, key) => {
                operations.push(cache.delAsync(`${config.app}:limiter:${key}:max:${remoteIp}`));
            });

            yield Promise.all(operations);

            return res.json({
                code: 200
            });
        } else {
            logger.info('geetest faild');
            return res.json(errRes);
        }

    })().catch(() => {
        return res.json({
            code: 400
        });
    });
};