27 lines
486 B
JavaScript
27 lines
486 B
JavaScript
// @flow
|
|
import stream from 'stream';
|
|
import { type Context } from 'koa';
|
|
|
|
export default function apiWrapper() {
|
|
return async function apiWrapperMiddleware(
|
|
ctx: Context,
|
|
next: () => Promise<*>
|
|
) {
|
|
await next();
|
|
|
|
const ok = ctx.status < 400;
|
|
|
|
if (
|
|
typeof ctx.body !== 'string' &&
|
|
!(ctx.body instanceof stream.Readable)
|
|
) {
|
|
// $FlowFixMe
|
|
ctx.body = {
|
|
...ctx.body,
|
|
status: ctx.status,
|
|
ok,
|
|
};
|
|
}
|
|
};
|
|
}
|