-
Notifications
You must be signed in to change notification settings - Fork 0
/
rest.js
61 lines (58 loc) · 1.73 KB
/
rest.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
/**
* Created by yatesmiao on 2017/8/14.
*/
// middleware
/**
* 这样,任何支持REST的异步函数只需要简单地调用:
* ctx.rest({
data: 123
});
*/
module.exports = {
APIError: function (code, message) {
this.code = code || 'internal:unknown_error';
this.message = message || '';
},
restify: (pathPrefix) => {
// REST API前缀,默认为/api/:
pathPrefix = pathPrefix || '/api/';
return async (ctx, next) => {
// 是否是REST API前缀?
if (ctx.request.path.startsWith(pathPrefix)) {
// 绑定rest()方法:
ctx.rest = (data, token) => {
ctx.set('Access-Control-Allow-Origin', '*');
ctx.set('Access-Control-Expose-Headers', 'Access-Token, Uid, Authorization');
ctx.set('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE');
ctx.set('Access-Control-Allow-Headers', 'Authorization');
ctx.set('Authorization', token);
ctx.type = 'application/json';
ctx.body = data;
}
try {
await next();
} catch (e) {
console.log('错误是', e)
// 返回错误:
if (e.message === 'Authentication Error') {
ctx.response.status = 401;
ctx.response.type = 'application/json';
ctx.response.body = {
code: e.code || 'authError:authError',
message: e.message || ''
};
} else {
ctx.response.status = 400;
ctx.response.type = 'application/json';
ctx.response.body = {
code: e.code || 'internal:unknown_error',
message: e.message || ''
};
}
}
} else {
await next();
}
};
}
};