img-captcha.js 1.84 KB
/**
 * Created by TaoHuang on 2016/6/18.
 */

/**
 * 验证码的形式: xxxx|page
 */

const CaptchaServiceModel = require('../models/captcha-img-service');
const request = require('request');
const {CAPTCHA_SWITCH, CAPTCHA_TYPE} = require('../models/captcha-type');

// 对比函数
const _mustEqualAsync = (req) => {
    let id = req.session.id;
    let verifyCode = req.body.verifyCode;

    if (!id || !verifyCode) {
        return Promise.resolve({
            code: 405,
            message: '请将所有图片点击翻转至正向朝上'
        });
    }

    return req.ctx(CaptchaServiceModel).check(id, verifyCode);
};

// 中间件
const requiredAPI = (req, res, next) => {
    next();
};

// 验证码
const generate = (req, res, next) => {
    req.ctx(CaptchaServiceModel).generateCaptcha(req.session.id).then((result) => {
        res.type('png');
        if (result.code === 200) {
            request({
                url: result.data.url,
                headers: Object.assign({
                    'X-request-ID': req.reqID || '',
                    'X-YOHO-IP': req.yoho.clientIp || '',
                    'X-Forwarded-For': req.yoho.clientIp || '',
                    'User-Agent': 'yoho/nodejs'
                }, result.headers || {})
            }).pipe(res);
        }

    }).catch(next);
};

// 端到端检查
const checkAPI = (req, res, next) => {
    _mustEqualAsync(req).then(result => {
        res.json(result);
    }).catch(next);
};

const trySwitch = (req, res) => {
    return req.ctx(CaptchaServiceModel).try().then((result) => {
        const captcha = {
            type: CAPTCHA_TYPE.image,
            value: result ? CAPTCHA_SWITCH.on : CAPTCHA_SWITCH.off
        };

        req.session.captcha = captcha;
        res.locals.captcha = captcha;
    });
};

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