UNPKG

phecda-server

Version:

server framework that provide IOC/type-reuse/http&rpc-adaptor

172 lines (170 loc) 6.61 kB
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 };