2019独角兽企业重金招聘Python工程师标准>>> 
首先在nodejs里的config下创建一个wechat.js文件，里面配置好appid和appSecret。
let config = exports;
config.production = {
appId: 'wxc464e493a4e946b6',
appSecret: 'ab86e8264ed8c8a91df958a98a7928b4',
// appId: 'wxe27bcb44cb1e4753',
// appSecret: '920841d36776734b772f778cfa8c8b77',
};
config.testing = Object.assign({}, config.production);
config.development = Object.assign({}, config.production);
config.qa = Object.assign({}, config.testing, {});
config.nc = Object.assign({}, config.production, {});
config.sjhl = Object.assign({}, config.production, {});
config.gm = Object.assign({}, config.production, {});然后在libs下创建wechat.js文件，
里面的代码为
var fs = require('fs');
var OAuth = require('wechat-oauth');
class Wechat {
//拿到配置文件里的appid和appSecret
constructor(appId, appSecret) {
this.appId = appId;
this.appSecret = appSecret;
this.client = new OAuth(this.appId, this.appSecret, this.getToken, this.setToken);
}
//从缓存中获取用户的assets token
getToken(openid, callback) {
let file = `runtime/access_token_${openid}.txt`;
if (!fs.existsSync(file)) {
return callback(null);
}
fs.readFile(file, 'utf8', function (err, txt) {
if (err) {
console.log('[wechat] get token err', err);
return callback(err);
}
callback(null, JSON.parse(txt));
});
}
//如果没获取到token就把token设置到缓存中
setToken(openid, token, callback) {
let file = `runtime/access_token_${openid}.txt`;
fs.writeFile(file, JSON.stringify(token), callback);
}
//传入url让用户点击授权
getAuthorizeURL(url) {
let redirectUrl = this.client.getAuthorizeURL(url, 'qiongtao.li', 'snsapi_userinfo');
return redirectUrl;
}
//获取用户的accessToken
async getAccessToken(code) {
return new Promise((resolve, reject) => {
this.client.getAccessToken(code, (err, rst) => {
if (err) {
console.log('[wechat] get access token err', err);
resolve();
}
console.log('[wechat] access token:', rst.data);
resolve(rst.data);
})
})
}
//通过openid获取到用户信息
async getUser(openid) {
return new Promise((resolve, reject) => {
this.client.getUser(openid, (err, rst) => {
if (err) {
console.log('[wechat] get user info err', err);
resolve();
}
console.log('[wechat] userinfo:', rst);
resolve(rst);
})
})
}
}
module.exports = Wechat最后在toutes里写action，代码为
const Wechat = require('../../libs/Wechat');
const Trans = require('../../libs/database/Mysql');
const Model = require('../../libs/Model');
const RecordLog = require('../../libs/Log');
const Request = require('../../libs/Request');
const redis = require('../../libs/Redis');
const wechat = new Wechat(_.config.wechat.appId, _.config.wechat.appSecret);
module.exports = {
'/npi/wechat/login': {
get: async (ctx, next) => {
let echostr = ctx.query.echostr || '';
let code = ctx.query.code || '';
if (echostr != '') { //证明你是开发者
return echostr;
}
//授权认证
if (code == '') {
let domain = ctx.protocol + '://' + ctx.host; //
let url = wechat.getAuthorizeURL(domain + ctx.url);
ctx.set('Referrer', ctx.headers['Referrer'] || `${domain}/npi/test/succ`)
ctx.redirect('back', url);
return '';
}
//获得userInfo
let rst = await wechat.getAccessToken(code);
let userInfo = await wechat.getUser(rst.openid);
console.log('111111111111111111111111111', userInfo);
//开始判断登录或者注册
let openid = userInfo.data.openid;
let gender = '';
if (userInfo.data.sex == '1') {
gender = 'MAN';
} else {
gender = 'WOMEN';
}
let row = await Model.tblGyUserSocial().where('openid', openid).where('status', 'ACTIVE').getOne();
//登录
if (row.id) {
let user = await Model.tblGyUserInfo().where('id', row.user_id).getOne();
let token = _.makeToken({
userId: user.id,
userType: 'PERSON',
}, ctx);
let str = {
userId: user.id,
userType: 'PERSON',
};
let JsStr = JSON.stringify(str);
await redis.setex("gy:app:u:token:" + token, 7200, JsStr);
return _.succ('登录成功！', {
token: token,
});
}
//注册
let data = {
businessId: 1,
datacenterId: 1,
serverId: 1,
};
let uid = await new Request().getJSON(_.config.settings.getId, data);
Trans.transBegin();
let res = await Model.tblGyUserInfo()
.set('status', 'ACTIVE')
.set('update_time', new Date())
.set('create_time', new Date())
.set('id', uid)
.set('nick_name', row.data.nickname)
.set('gender', gender)
.set('icon', row.data.headimgurl)
.insert();
if (res === false) {
Trans.transRollback();
return _.err('失败！', 711);
}
let res1 = await Model.tblGyUserSocial()
.set('status', 'ACTIVE')
.set('update_time', new Date())
.set('create_time', new Date())
.set('user_id', uid)
.set('platform', uid)
.set('openid', openid)
.set('nick_name', row.data.nickname)
.set('gender', gender)
.set('province', row.data.province)
.set('city', row.data.city)
.set('country', row.data.country)
.set('icon', row.data.headimgurl)
.insert();
if (res1 === false) {
Trans.transRollback();
return _.err('失败！', 711);
}
let token = _.makeToken({
userId: uid,
userType: 'PERSON',
}, ctx);
let str = {
userId: uid,
userType: 'PERSON',
};
let JsStr = JSON.stringify(str);
await redis.setex("gy:app:u:token:" + token, 7200, JsStr);
ctx.status = 301;
ctx.redirect(_.config.settings.wechatRedirectUrl + `?token=${token}`);
// return _.succ(userInfo);
}
},
}