check.js 2.32 KB
'use strict';

const _ = require('lodash');
const logger = global.yoho.logger;
const Geetest = require('geetest');
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 key = `pc:limiter:${remoteIp}`;
            let key10m = `pc:limiter:10m:${remoteIp}`;
            let dels = [
                cache.delAsync(key),
                cache.delAsync(key10m)
            ];

            if (req.body.pid) {
                dels.push(cache.delAsync(req.body.pid));
            }

            yield Promise.all(dels).catch(e => {
                console.log(`check.js err: ${e.message}`);
            });

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

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