'use strict'; const assert = require('assert'); const _ = require('lodash'); const uuid = require('uuid'); const urljoin = require('url-join'); const stringRandom = require('string-random'); const Controller = require('egg').Controller; const { BusinessError, ErrorCode } = require('naf-core').Error; /** * 微信认证,获得openid和用户信息,生成微信Jwt */ class WeixinController extends Controller { /** * 认证流程 * 1. 缓存原始请求地址,生成state和认证回调地址 * 2. 通过wxapi认证,获得认证code * 3. 通过code获得openid,通过openid,查询绑定用户,创建jwt * 4. jwt写入redis,返回认证code * 5. 通过code获取jwt */ // GET 请求认证 // response_type: // code - url带上code参数重定向到原始地址 // store - 默认,认证结果写入sessionStore,然后重定向回请求页面(要求请求页面和认证服务在同一域名下) // token - url带上token参数重定向到原始地址 async auth() { const { redirect_uri, code, test, type, response_type = 'store', uid, qrcode, msgid, objid } = this.ctx.query; if (test) { return await this.authTest(); } if (code) { return await this.authBack(this.ctx.query); } this.ctx.logger.debug(`[auth] reditect_uri - ${redirect_uri}`); // assert(redirect_uri, '回调地址不能为空'); // TODO: 保存原始请求地址 // const { config } = this.app; // console.log('ctx.host: ', this.ctx.host); // console.log('config.hostHeaders: ', config.hostHeaders); // TODO: 保存原始请求地址 const state = uuid(); const key = `visit:auth:state:${state}`; const val = JSON.stringify({ redirect_uri, type, uid, qrcode, msgid, objid }); await this.app.redis.set(key, val, 'EX', 600); // TODO: 生成回调地址 const { wxapi, authUrl = this.ctx.path } = this.app.config; const backUrl = encodeURI(`${this.app.config.baseUrl}${this.config.authUrl}?state=${state}`); const to_uri = encodeURI(`${wxapi.baseUrl}/api/auth?appid=${wxapi.appid}&response_type=code&redirect_uri=${backUrl}#wechat`); // const backUrl = encodeURI(`${this.app.config.baseUrl}${this.config.authUrl}/`); // const to_uri = `https://open.weixin.qq.com/connect/oauth2/authorize?appid=${wxapi.appid}&response_type=code&scope=snsapi_base&redirect_uri=${backUrl}&state=${state}&connect_redirect=1#wechat_redirect`; this.ctx.redirect(to_uri); } // GET 认证回调 async authBack({ code, state }) { // const { code, state, type, redirecturi } = this.ctx.query; this.ctx.logger.debug(`[auth-back] code - ${code}, state - ${state}`); assert(code, 'code不能为空'); assert(state, 'state不能为空'); const { weixin } = this.ctx.service; const key = `visit:auth:state:${state}`; const val = await this.app.redis.get(key); let openid; try { ({ openid } = await weixin.fetch(code)); } catch (err) { this.ctx.logger.error(err); console.error(err); await this.ctx.render('error.njk', { title: err.message, message: err.details }); return; } if (openid) { const { redirect_uri, type, uid, qrcode, msgid, objid } = JSON.parse(val); const user = await this.ctx.service.user.findByOpenid(openid); if (type === '0') { // 通过openid取得用户信息 if (user) { const getToken = async user => await this.ctx.service.login.createJwt(user); if (user.type === '4') { // 学生 // 检查学生是否已退出,学生表的isComming是不是2,如果是2 就跳到别的地方提示您已退出 const stu = await this.ctx.service.student.fetch({ id: user.uid }); if (stu) { const { isComming } = stu; if (isComming && isComming === '2') { const refuse = `${this.app.config.baseUrl}/msgconfirm/isleave`; this.ctx.redirect(refuse); } else { const to_uri = urljoin(redirect_uri, `?token=${await getToken(user)}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } } else { const touri = `${this.app.config.baseUrl}/student/bind`; const to_uri = urljoin(touri, `?openid=${openid}`); this.ctx.redirect(to_uri); } } else if (user.type === '1') { // 班主任 const touri = `${this.app.config.baseUrl}/mobiledirtea`; const to_uri = urljoin(touri, `?token=${await getToken(user)}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } else if (user.type === '3') { // 教师 const touri = `${this.app.config.baseUrl}/mobiledirtea`; const to_uri = urljoin(touri, `?token=${await getToken(user)}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } } else { // const resunionid = await weixin.fetchUnionID(openid); const touri = `${this.app.config.baseUrl}/student/bind`; const to_uri = urljoin(touri, `?openid=${openid}`); // &unionid=${resunionid.unionid} // TODO: 重定性页面 this.ctx.redirect(to_uri); } } else if (type === '1') { // 绑定微信 const to_uri = urljoin(redirect_uri, `?openid=${openid}&uid=${uid}&type=${type}&qrcode=${qrcode}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } else if (type === '2') { // 学生绑定 const to_uri = urljoin(redirect_uri, `?openid=${openid}&type=${type}&qrcode=${qrcode}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } else if (type === '9') { // 查看消息 const to_uri = urljoin(redirect_uri, `?openid=${openid}&type=${type}&msgid=${msgid}&objid=${objid}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } } } // GET 用户授权内部测试接口 async authTest() { const { redirect_uri, type, uid, qrcode, openid, msgid, objid } = this.ctx.query; // const openid = stringRandom(); this.ctx.logger.debug(`[auth-test] reditect_uri - ${redirect_uri}, openid - ${openid}`); assert(redirect_uri, '回调地址不能为空'); assert(openid, 'openid不能为空'); const user = await this.ctx.service.user.findByOpenid(openid); if (type === '0') { // 通过openid取得用户信息 if (user) { const token = await this.ctx.service.login.createJwt(user); if (user.type === '4') { const to_uri = urljoin(redirect_uri, `?token=${token}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } else if (user.type === '1') { const touri = `${this.app.config.baseUrl}/mobiledirtea`; const to_uri = urljoin(touri, `?token=${token}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } else if (user.type === '3') { const touri = `${this.app.config.baseUrl}/mobiledirtea`; const to_uri = urljoin(touri, `?token=${token}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } } else { console.log('rrr0000--->' + redirect_uri); const resunionid = await this.ctx.service.weixin.fetchUnionID(openid); const touri = `${this.app.config.baseUrl}/student/bind`; const to_uri = urljoin(touri, `?openid=${openid}&unionid=${resunionid.unionid}`); // TODO: 重定性页面 this.ctx.redirect(to_uri); } } else if (type === '1') { const to_uri = urljoin(redirect_uri, `?openid=${openid}&uid=${uid}&type=${type}&qrcode=${qrcode}`); // TODO: 重定性页面 console.log('1111---?' + to_uri); this.ctx.redirect(to_uri); } else if (type === '2') { const to_uri = urljoin(redirect_uri, `?openid=${openid}&type=${type}&qrcode=${qrcode}`); // TODO: 重定性页面 console.log('22222---?' + to_uri); this.ctx.redirect(to_uri); } else if (type === '9') { // const token = await this.ctx.service.login.createJwt(user); const to_uri = urljoin(redirect_uri, `?openid=${openid}&type=${type}&msgid=${msgid}&objid=${objid}`); // TODO: 重定性页面 console.log('99999---?' + to_uri); this.ctx.redirect(to_uri); } } // GET 签到小程序请求用户 async appAuth() { try { const { js_code, login = true } = this.ctx.query; console.log(js_code); const appopenid = await this.ctx.service.weixin.appAuth(js_code); if (!login) this.ctx.ok(appopenid); console.warn(`openid=>${appopenid}`); // const state = uuid(); // const key = `appinfo:${state}`; // await this.app.redis.set(key, appopenid, 'EX', 600); const res = await this.ctx.service.user.findByAppOpenid(appopenid); const obj = { user: res, openid: appopenid }; this.ctx.ok(obj); } catch (error) { throw new BusinessError(ErrorCode.NETWORK, '获取小程序信息失败,请尝试重新进入'); } } // 微信网页端登录(班主任,学生) async openidLogin() { const res = await this.ctx.service.login.openidLogin(this.ctx.request.body); this.ctx.ok({ data: res }); } } module.exports = WeixinController;