-
Notifications
You must be signed in to change notification settings - Fork 62
/
wechat.js
166 lines (153 loc) · 4.67 KB
/
wechat.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
/*
* wechat.js
* Copyright (C) 2018 disoul <disoul@DiSouldeMacBook-Pro.local>
*
* Distributed under terms of the MIT license.
*/
const express = require('express');
const jwt = require('jsonwebtoken');
const auth = require('../auth');
const router = express.Router();
const PILI = require('../pili');
const { User, Group } = require('../models');
const { getStreamKey } = require('../utils/pili');
const { getUserInfo, decryptData } = require('../utils/wechat');
const { getUsersFromGroup } = require('../utils/storage');
router.use('/api', auth);
/**
* 通过微信登录的code获取微信登录session
*/
router.get('/login/:code', async (req, res) => {
const code = req.params.code;
const token = req.get('Authorization');
// 如果用户携带了认证token,就表示session没有过期,直接查表
if (token) {
const decoded = jwt.decode(token);
const user = await User.findOne({ _id: decoded.id });
try {
jwt.verify(token, user.sessionKey);
res.json({ token });
} catch (e) {
res.status(403).json();
}
return;
}
try {
const userInfo = await getUserInfo(code);
await User.findOneAndUpdate(
{ userId: userInfo.openid },
{ userId: userInfo.openid, sessionKey: userInfo.session_key },
{ upsert: true },
);
const user = await User.findOne({ userId: userInfo.openid });
console.log(user);
// 签发JWTtoken,用于之后认证
const token = jwt.sign({ id: user._id }, user.sessionKey);
res.json({ token });
} catch (e) {
console.log(e);
res.status(500).json({ error: e.toString() });
}
});
/**
* 解密微信的加密数据
*/
router.post('/api/decrypt', async (req, res) => {
const { iv, encryptedData } = req.body;
try {
const user = req.user;
const data = decryptData(iv, user.sessionKey, encryptedData);
res.send(data);
} catch (e) {
console.log(e);
res.status(500).json({ error: e.toString() });
}
});
/**
* 将用户加入内部群组
* 如果用户通过微信群打开,就把用户加入这个内部存储的群组
* 方便实现基于群组的逻辑
*/
router.post('/api/group/add_member', async (req, res) => {
const { groupId } = req.body;
try {
const group = await Group.findOneAndUpdate({ groupId }, { groupId }, { upsert: true }) || { groupId, members: [] };
group.members.push(req.user.userId);
group.members = Array.from(new Set(group.members));
await Group.findOneAndUpdate({ groupId }, group, { upsert: true });
res.json({ status: 'ok' });
} catch (e) {
console.log(e);
res.status(500).json({ error: e.toString() });
}
});
/**
* 将微信用户加入内部用户
*/
router.post('/api/user', async (req, res) => {
try {
const user = {
...req.body.userInfo,
userId: req.user.userId,
};
await User.findOneAndUpdate({ userId: req.user.userId }, user, { upsert: true });
res.json({ status: 'ok' });
} catch (e) {
console.log(e);
res.status(500).json({ error: e.toString() });
}
});
/**
* 获取当前正在推流的用户
* 如果传入groupID,就只会搜索群组内部的用户
* (目前Demo没有做groupId的限制
*/
let fetchLock = false;
let cacheUser = [];
let cacheStream = [];
let lastCache = 0;
router.get('/api/activeuser/:groupId?', async (req, res) => {
try {
const groupId = req.params.groupId;
let users;
let activeStream;
// 一个简单的缓存和缓存锁,永远每隔3s更新一次数据,防止重复请求
if (!fetchLock && Date.now() - lastCache > 3000) {
fetchLock = true;
try {
users = groupId ? await getUsersFromGroup(groupId) : await User.find();
const streams = users.map(user => getStreamKey(user.userId, groupId));
activeStream = await PILI.getActiveStream(streams);
cacheUser = users;
cacheStream = activeStream;
lastCache = Date.now();
fetchLock = false;
} catch (e) {
fetchLock = false;
res.status(500).json({ error: e.toString() });
return;
}
} else {
users = cacheUser;
activeStream = cacheStream;
}
let activeStreamMap = {};
activeStream.map(stream => activeStreamMap[stream.key] = stream);
const activeUser = [];
for (let i = 0; i < users.length; i += 1) {
const user = users[i];
const streamKey = getStreamKey(user.userId, groupId);
if (Object.keys(activeStreamMap).indexOf(streamKey) !== -1) {
delete user._doc.sessionKey;
activeUser.push({
...user._doc,
stream: activeStreamMap[streamKey],
});
}
}
res.json({ activeUser });
} catch (e) {
res.status(500).json({ error: e.toString() });
}
})
module.exports = router;