28 lines
508 B
JavaScript
28 lines
508 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 = {
|
|
// $FlowFixMe
|
|
...ctx.body,
|
|
status: ctx.status,
|
|
ok,
|
|
};
|
|
}
|
|
};
|
|
}
|