phecda-server
Version:
server framework that provide IOC/type-reuse/http&rpc-adaptor
172 lines (170 loc) • 6.61 kB
JavaScript
import {
BadRequestException,
Context,
createControllerMetaMap,
detectAopDep,
joinUrl
} from "../../chunk-WST6E6MQ.mjs";
import {
HMR,
__name
} from "../../chunk-URKOYTBU.mjs";
// src/http/koa/bind.ts
import Router from "@koa/router";
import Debug from "debug";
var debug = Debug("phecda-server/koa");
function bind(router, data, opts = {}) {
const { globalGuards, parallelRoute, globalAddons = [], parallelAddons = [], globalFilter, globalPipe, dynamic = false } = opts;
const { moduleMap, meta } = data;
const originStack = router.stack.slice(0, router.stack.length);
const metaMap = createControllerMetaMap(meta, (meta2) => {
const { controller, http, method, tag } = meta2.data;
if (controller === "http" && http?.method) {
debug(`register method "${method}" in module "${tag}"`);
return true;
}
});
detectAopDep(meta, {
addons: [
...globalAddons,
...parallelAddons
],
guards: globalGuards
});
registerRoute();
function registerRoute() {
Context.applyAddons(globalAddons, router, "koa");
if (parallelRoute) {
const subRouter = new Router();
Context.applyAddons(parallelAddons, subRouter, "koa");
subRouter.post(parallelRoute, async (ctx, next) => {
const { body } = ctx.request;
async function errorHandler(e) {
const error = await Context.filterRecord.default(e);
ctx.status = error.status;
ctx.body = error;
}
__name(errorHandler, "errorHandler");
if (!Array.isArray(body)) return errorHandler(new BadRequestException("data format should be an array"));
try {
return Promise.all(body.map((item, i) => {
return new Promise(async (resolve) => {
if (!item) return resolve(null);
const { tag, method } = item;
debug(`(parallel)invoke method "${method}" in module "${tag}"`);
if (!metaMap.has(tag)) return resolve(await Context.filterRecord.default(new BadRequestException(`module "${tag}" doesn't exist`)));
const meta2 = metaMap.get(tag)[method];
if (!meta2) return resolve(await Context.filterRecord.default(new BadRequestException(`"${method}" in "${tag}" doesn't exist`)));
const aop = Context.getAop(meta2, {
globalGuards,
globalFilter,
globalPipe
});
const contextData = {
type: "koa",
category: "http",
index: i,
ctx,
meta: meta2,
moduleMap,
parallel: true,
next,
app: router,
...item,
getCookie: /* @__PURE__ */ __name((key) => ctx.cookies.get(key), "getCookie"),
setCookie: /* @__PURE__ */ __name((key, value, opts2) => ctx.cookies.set(key, value, opts2), "setCookie"),
delCookie: /* @__PURE__ */ __name((key) => ctx.cookies.set(key, "", {
expires: /* @__PURE__ */ new Date(0)
}), "delCookie"),
redirect: /* @__PURE__ */ __name((url) => ctx.redirect(url), "redirect"),
setResHeaders: /* @__PURE__ */ __name((headers) => ctx.set(headers), "setResHeaders"),
setResStatus: /* @__PURE__ */ __name((status) => ctx.status = status, "setResStatus"),
getRequest: /* @__PURE__ */ __name(() => ctx.req, "getRequest"),
getResponse: /* @__PURE__ */ __name(() => ctx.res, "getResponse")
};
const context = new Context(contextData);
context.run(aop, resolve, resolve);
});
})).then((ret) => {
ctx.body = ret;
});
} catch (e) {
return errorHandler(e);
}
});
router.use(subRouter.routes()).use(subRouter.allowedMethods());
}
for (const [tag, record] of metaMap) {
for (const method in record) {
const meta2 = metaMap.get(tag)[method];
const { data: { addons, http } } = meta2;
if (!http?.method) continue;
let aop;
if (!dynamic) {
aop = Context.getAop(meta2, {
globalFilter,
globalGuards,
globalPipe
});
}
const subRouter = new Router();
Context.applyAddons(addons, subRouter, "koa");
router[http.method](joinUrl(http.prefix, http.route), async (ctx, next) => {
debug(`invoke method "${method}" in module "${tag}"`);
const contextData = {
type: "koa",
app: router,
ctx,
meta: meta2,
moduleMap,
tag,
method,
query: ctx.query,
params: ctx.params,
category: "http",
body: ctx.request.body,
headers: ctx.headers,
next,
getCookie: /* @__PURE__ */ __name((key) => ctx.cookies.get(key), "getCookie"),
setCookie: /* @__PURE__ */ __name((key, value, opts2) => ctx.cookies.set(key, value, opts2), "setCookie"),
delCookie: /* @__PURE__ */ __name((key) => ctx.cookies.set(key, "", {
expires: /* @__PURE__ */ new Date(0)
}), "delCookie"),
redirect: /* @__PURE__ */ __name((url) => ctx.redirect(url), "redirect"),
setResHeaders: /* @__PURE__ */ __name((headers) => ctx.set(headers), "setResHeaders"),
setResStatus: /* @__PURE__ */ __name((status) => ctx.status = status, "setResStatus"),
getRequest: /* @__PURE__ */ __name(() => ctx.req, "getRequest"),
getResponse: /* @__PURE__ */ __name(() => ctx.res, "getResponse")
};
const context = new Context(contextData);
if (http.headers) ctx.set(http.headers);
if (dynamic) {
aop = Context.getAop(meta2, {
globalFilter,
globalGuards,
globalPipe
});
}
await context.run(aop, (returnData) => {
if (ctx.res.writableEnded) return;
ctx.body = returnData;
}, (err) => {
if (ctx.res.writableEnded) return;
ctx.status = err.status;
ctx.body = err;
});
});
router.use(subRouter.routes()).use(subRouter.allowedMethods());
}
}
}
__name(registerRoute, "registerRoute");
HMR(async () => {
router.stack = originStack;
registerRoute();
});
}
__name(bind, "bind");
export {
bind
};