elysia 2.0.0-beta.2 → 2.0.0-beta.21

package/dist/compile/handler/utils.js

@@ -1,4 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_universal_file = require('../../universal/file.js');
 const require_utils = require('../../utils.js');
 const require_type_constants = require('../../type/constants.js');
 const require_adapter_skip_clone = require('../../adapter/skip-clone.js');
@@ -10,7 +11,6 @@ const require_adapter_origin = require('../../adapter/origin.js');
 const require_compile_lexer = require('../lexer.js');
 
 //#region src/compile/handler/utils.ts
-const childName = (fn) => fn?.name && typeof fn.name === "string" ? fn.name : "anonymous";
 const noTrace = {
 	begin: "",
 	end: () => ""
@@ -23,6 +23,16 @@ function cloneResponse(r) {
 	}
 	return r;
 }
+function cloneStaticValue(value) {
+	if (require_universal_file.ElysiaFile.isElysiaFile(value)) return value;
+	try {
+		const cloned = structuredClone(value);
+		if (Object.getPrototypeOf(value) === require_error.ElysiaStatus.prototype) Object.setPrototypeOf(cloned, require_error.ElysiaStatus.prototype);
+		return cloned;
+	} catch {
+		return value;
+	}
+}
 function hasRequestBody(request) {
 	const length = request.headers.get("content-length");
 	if (length !== null) return length !== "0";
@@ -42,28 +52,102 @@ function armEntryAbort(context) {
 	if (context.request === require_adapter_origin.origin.request) return false;
 	return (context["~sig"] = context.request.signal).aborted;
 }
-const trace = (report, fn) => report?.resolveChild(childName(fn)) ?? noTrace;
+const trace = (report, fn) => report?.resolveChild(fn?.name && typeof fn.name === "string" ? fn.name : "anonymous") ?? noTrace;
 const toArray = (v) => Array.isArray(v) ? v : [v];
-const mapTransform = /* @__PURE__ */ map((i, fn, [isAsync, report]) => {
+const mapTransform = /* @__PURE__ */ map((i, fn, [isAsync, report, arm]) => {
 	const t = trace(report, fn);
-	const guard = awaitGuard(fn, isAsync, "_tf");
-	const call = guard ? `_tf=tf${at(i)}(c)\n${guard}` : `${Await(fn)}tf${at(i)}(c)\n`;
+	const call = isAsync ? awaitSite(`tf${at(i)}(c)`, fn, isAsync, "_tf", arm) : `tf${at(i)}(c)\n`;
 	return t.begin + call + t.end();
 });
 const deriveKeyCache = /* @__PURE__ */ new WeakMap();
 function extractDeriveKeys(fn) {
 	const cached = deriveKeyCache.get(fn);
 	if (cached !== void 0) return cached;
-	const result = scanDeriveKeys(fn);
+	let src;
+	try {
+		src = Function.prototype.toString.call(fn);
+	} catch {
+		src = void 0;
+	}
+	const result = src === void 0 || src.includes("[native code]") || src.includes("...") ? null : returnedObjectKeys(src);
 	deriveKeyCache.set(fn, result);
 	return result;
 }
+function returnedObjectKeys(src) {
+	const tokens = require_compile_lexer.scanTokens(src);
+	if (!tokens) return null;
+	const is = (i, value) => tokens[i]?.k === "p" && tokens[i].value === value;
+	const blank = (i) => {
+		const token = tokens[i];
+		return /^[ \t\n\r]*$/.test(src.slice(token.at + token.value.length + (token.k === "s" ? 2 : 0), tokens[i + 1]?.at));
+	};
+	let open = -1;
+	let parens = 0;
+	let depth = 0;
+	for (let i = 0; i < tokens.length; i++) {
+		const { k, value } = tokens[i];
+		if (k === "i") {
+			if (depth === 0 && value === "function") break;
+		} else if (k !== "p") continue;
+		else if (value === "(" || value === "[" || value === "{") depth++;
+		else if (value === ")" || value === "]" || value === "}") depth--;
+		else if (depth === 0 && value === "=>") {
+			if (!blank(i)) return null;
+			if (is(i + 1, "(")) {
+				if (!blank(i + 1) || !is(i + 2, "{")) return null;
+				open = i + 2;
+				parens = 1;
+			} else if (!is(i + 1, "{")) return null;
+			break;
+		}
+	}
+	if (open === -1) {
+		let at = -1;
+		for (let i = 0; i < tokens.length; i++) {
+			const { k, value } = tokens[i];
+			if (k !== "i") continue;
+			if (value === "return") {
+				if (at !== -1) return null;
+				at = i;
+			} else if (value.includes("return") && /[^\w$]/.test(value)) return null;
+		}
+		if (at === -1) return null;
+		for (; blank(at) && is(at + 1, "("); at++) parens++;
+		if (!blank(at) || !is(at + 1, "{")) return null;
+		open = at + 1;
+	}
+	const keys = [];
+	let i = open + 1;
+	while (!is(i, "}")) {
+		const key = tokens[i];
+		if (!key || (key.k === "s" ? key.value.includes("\\") : key.k !== "i" || !/^[\w$]+$/.test(key.value)) || !blank(i) || !is(i + 1, ":")) return null;
+		keys.push(key.value);
+		for (depth = 0, i += 2;; i++) {
+			const token = tokens[i];
+			if (!token) return null;
+			if (token.k !== "p") continue;
+			const { value } = token;
+			if (value === "(" || value === "[" || value === "{") depth++;
+			else if (value === ")" || value === "]" || value === "}") {
+				if (depth === 0) {
+					if (value !== "}") return null;
+					break;
+				}
+				depth--;
+			} else if (value === "," && depth === 0) break;
+		}
+		if (is(i, ",")) i++;
+	}
+	for (; parens; parens--) if (!is(++i, ")")) return null;
+	return i + 1 === tokens.length || is(i + 1, ";") || is(i + 1, "}") ? keys : null;
+}
 function replaceDeriveContext(context, derivative) {
 	const next = Object.create(Object.getPrototypeOf(context));
-	Object.assign(next, derivative);
+	require_utils.assignOwn(next, derivative);
 	next.request = context.request;
 	next["~sig"] = context["~sig"];
 	next["~afterResponse"] = context["~afterResponse"];
+	next["~dispose"] = context["~dispose"];
 	next.store = context.store;
 	next.set = context.set;
 	next.body = context.body;
@@ -83,21 +167,19 @@ function replaceDeriveContext(context, derivative) {
 	next.redirect = context.redirect;
 	return next;
 }
-function deriveModeQueues(entries) {
-	if (!entries?.length) return;
-	const queues = /* @__PURE__ */ new Map();
-	for (let i = 0; i < entries.length; i++) {
-		const entry = entries[i];
-		const fn = require_utils.deriveEntryFn(entry);
-		const queue = queues.get(fn);
-		const mode = require_utils.isMapDeriveEntry(entry);
-		if (queue) queue.push(mode);
-		else queues.set(fn, [mode]);
-	}
-	return queues;
-}
 function deriveModes(hooks, entries) {
-	const queues = deriveModeQueues(entries);
+	let queues;
+	if (entries?.length) {
+		queues = /* @__PURE__ */ new Map();
+		for (let i = 0; i < entries.length; i++) {
+			const entry = entries[i];
+			const fn = require_utils.deriveEntryFn(entry);
+			const queue = queues.get(fn);
+			const mode = require_utils.isMapDeriveEntry(entry);
+			if (queue) queue.push(mode);
+			else queues.set(fn, [mode]);
+		}
+	}
 	if (!queues) return;
 	let found = false;
 	const modes = Array(hooks.length);
@@ -110,236 +192,39 @@ function deriveModes(hooks, entries) {
 	}
 	return found ? modes : void 0;
 }
-function scanDeriveKeys(fn) {
-	let src;
-	try {
-		src = Function.prototype.toString.call(fn);
-	} catch {
-		return null;
-	}
-	if (src.includes("[native code]")) return null;
-	if (src.includes("...")) return null;
-	const objStart = findReturnedObjectStart(src);
-	if (objStart === -1) return null;
-	return scanObjectLiteralKeys(src, objStart);
-}
-function findReturnedObjectStart(src) {
-	const arrow = topLevelArrowIndex(src);
-	if (arrow !== -1) {
-		let i = arrow + 2;
-		while (i < src.length && require_compile_lexer.isSpace(src[i])) i++;
-		if (src[i] === "(") {
-			let j = i + 1;
-			while (j < src.length && require_compile_lexer.isSpace(src[j])) j++;
-			if (src[j] === "{") return j;
-			return -1;
-		}
-		if (src[i] !== "{") return -1;
-	}
-	const { count: returns, firstIndex: idx } = scanReturns(src);
-	if (returns !== 1) return -1;
-	if (idx === -1) return -1;
-	let i = idx + 6;
-	while (i < src.length && require_compile_lexer.isSpace(src[i])) i++;
-	while (i < src.length && src[i] === "(") {
-		i++;
-		while (i < src.length && require_compile_lexer.isSpace(src[i])) i++;
-	}
-	return src[i] === "{" ? i : -1;
-}
-function topLevelArrowIndex(src) {
-	let depth = 0;
-	for (let i = 0; i < src.length;) {
-		const ch = src[i];
-		if (ch === "\"" || ch === "'" || ch === "`") {
-			i = require_compile_lexer.skipString(src, i);
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "/") {
-			const nl = src.indexOf("\n", i);
-			if (nl === -1) return -1;
-			i = nl + 1;
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "*") {
-			const end = src.indexOf("*/", i);
-			if (end === -1) return -1;
-			i = end + 2;
-			continue;
-		}
-		if (depth === 0 && ch === "f" && src.startsWith("function", i) && !require_compile_lexer.isIdentChar(src[i - 1] ?? " ") && !require_compile_lexer.isIdentChar(src[i + 8] ?? " ")) return -1;
-		if (ch === "(" || ch === "[" || ch === "{") {
-			depth++;
-			i++;
-			continue;
-		}
-		if (ch === ")" || ch === "]" || ch === "}") {
-			depth--;
-			i++;
-			continue;
-		}
-		if (depth === 0 && ch === "=" && src[i + 1] === ">") return i;
-		i++;
-	}
-	return -1;
-}
-function scanReturns(src) {
-	let count = 0;
-	let firstIndex = -1;
-	for (let i = 0; i < src.length;) {
-		const ch = src[i];
-		if (ch === "\"" || ch === "'" || ch === "`") {
-			i = require_compile_lexer.skipString(src, i);
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "/") {
-			const nl = src.indexOf("\n", i);
-			if (nl === -1) break;
-			i = nl;
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "*") {
-			const end = src.indexOf("*/", i);
-			if (end === -1) break;
-			i = end + 2;
-			continue;
-		}
-		if (ch === "r" && src.startsWith("return", i) && !require_compile_lexer.isIdentChar(src[i - 1] ?? " ") && !require_compile_lexer.isIdentChar(src[i + 6] ?? " ")) {
-			if (firstIndex === -1) firstIndex = i;
-			count++;
-			i += 6;
-			continue;
-		}
-		i++;
-	}
-	return {
-		count,
-		firstIndex
-	};
-}
-function scanObjectLiteralKeys(src, open) {
-	const keys = [];
-	let i = open + 1;
-	let expectKey = true;
-	while (i < src.length) {
-		const ch = src[i];
-		if (require_compile_lexer.isSpace(ch)) {
-			i++;
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "/") {
-			const nl = src.indexOf("\n", i);
-			if (nl === -1) return null;
-			i = nl + 1;
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "*") {
-			const end = src.indexOf("*/", i);
-			if (end === -1) return null;
-			i = end + 2;
-			continue;
-		}
-		if (ch === "}") return keys;
-		if (expectKey) {
-			if (ch === "[") return null;
-			let key;
-			if (ch === "\"" || ch === "'") {
-				const end = require_compile_lexer.skipString(src, i);
-				key = src.slice(i + 1, end - 1);
-				if (key.includes("\\")) return null;
-				i = end;
-			} else if (require_compile_lexer.isIdentChar(ch) && !(ch >= "0" && ch <= "9")) {
-				const start = i;
-				while (i < src.length && require_compile_lexer.isIdentChar(src[i])) i++;
-				key = src.slice(start, i);
-			} else return null;
-			let j = i;
-			while (j < src.length && require_compile_lexer.isSpace(src[j])) j++;
-			if (src[j] !== ":") return null;
-			keys.push(key);
-			i = j + 1;
-			i = skipValue(src, i);
-			if (i === -1) return null;
-			expectKey = false;
-			continue;
-		}
-		if (ch === ",") {
-			expectKey = true;
-			i++;
-			continue;
-		}
-		return null;
-	}
-	return null;
-}
-function skipValue(src, i) {
-	let depth = 0;
-	while (i < src.length) {
-		const ch = src[i];
-		if (ch === "\"" || ch === "'" || ch === "`") {
-			i = require_compile_lexer.skipString(src, i);
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "/") {
-			const nl = src.indexOf("\n", i);
-			if (nl === -1) return -1;
-			i = nl + 1;
-			continue;
-		}
-		if (ch === "/" && src[i + 1] === "*") {
-			const end = src.indexOf("*/", i);
-			if (end === -1) return -1;
-			i = end + 2;
-			continue;
-		}
-		if (ch === "{" || ch === "(" || ch === "[") {
-			depth++;
-			i++;
-			continue;
-		}
-		if (ch === "}" || ch === ")" || ch === "]") {
-			if (depth === 0) {
-				if (ch === "}") return i;
-				return -1;
-			}
-			depth--;
-			i++;
-			continue;
-		}
-		if (ch === "," && depth === 0) return i;
-		i++;
-	}
-	return -1;
-}
-function mapBeforeHandle(_hooks, derive, link, isAsync, report, abortGuard) {
+function mapBeforeHandle(_hooks, derive, link, isAsync, report, abortGuard, arm) {
 	const hooks = toArray(_hooks);
 	const modes = deriveModes(hooks, derive);
+	const tail = asyncTail(isAsync);
 	let code = "";
 	let depth = 0;
 	let needsEs = false;
 	for (let i = 0; i < hooks.length; i++) {
 		const fn = hooks[i];
-		if (i > 0) {
-			code += `if(${abortGuard ? `!${abortGuard}&&` : ""}_r===undefined){\n`;
+		const guard = i > 0 ? `${abortGuard ? `!${abortGuard}&&` : ""}_r===undefined` : void 0;
+		if (tail) code += tailStage(tail, guard);
+		else if (guard) {
+			code += `if(${guard}){\n`;
 			depth++;
 		}
 		const t = trace(report, fn);
 		code += t.begin;
-		code += `tmp=${Await(fn)}bf${at(i)}(c)\n`;
-		code += awaitGuard(fn, isAsync, "tmp");
+		code += awaitSite(`bf${at(i)}(c)`, fn, isAsync, "tmp", arm);
 		if (modes?.[i] !== void 0) {
 			needsEs = true;
+			link(require_handler_utils.registerDeriveDisposable, "dsp");
 			if (modes[i]) {
 				link(replaceDeriveContext, "rdc");
-				code += "if(tmp instanceof es)_r=tmp\nelse if(tmp){if(typeof tmp==='object'||typeof tmp==='function')c=rdc(c,tmp);tmp=undefined}\n";
+				code += "if(tmp instanceof es)_r=tmp\nelse if(tmp){if(typeof tmp==='object'||typeof tmp==='function'){const _pc=c;c=rdc(c,tmp);for(const _k of Object.keys(tmp))dsp(c,c[_k],_pc)}tmp=undefined}\n";
 			} else {
 				const keys = extractDeriveKeys(fn);
-				const merge = keys && keys.length ? keys.map((k) => `c[${JSON.stringify(k)}]=tmp[${JSON.stringify(k)}]`).join(";") : "Object.assign(c,tmp)";
+				const merge = !!keys && !!keys.length && !keys.includes("__proto__") ? keys.map((k) => `_v=tmp[${JSON.stringify(k)}];dsp(c,_v);c[${JSON.stringify(k)}]=_v`).join(";") : "if(Object.hasOwn(tmp,'__proto__')){for(const _k of Object.keys(tmp))dsp(c,tmp[_k]);Object.defineProperties(c,Object.getOwnPropertyDescriptors(tmp))}else{_v=Object.assign({},tmp);for(const _k of Reflect.ownKeys(_v)){dsp(c,_v[_k]);c[_k]=_v[_k]}}";
 				code += `if(tmp instanceof es)_r=tmp
 else if(tmp){${merge};tmp=undefined}\n`;
 			}
 		} else code += "if(tmp!==undefined)_r=tmp\n";
 		code += t.end("tmp");
+		if (tail) code += "}\n";
 	}
 	code += "}".repeat(depth);
 	if (needsEs) link(require_error.ElysiaStatus, "es");
@@ -372,7 +257,7 @@ async function runBeforeHandlePrefixAsync(prefix, context, abort) {
 			if (!first && abort && context["~sig"]?.aborted) return;
 			first = false;
 			let result = values[j](context);
-			if (result instanceof Promise) {
+			if (typeof result?.then === "function") {
 				result = await result;
 				if (abort) context["~sig"] ??= context.request.signal;
 			}
@@ -380,52 +265,58 @@ async function runBeforeHandlePrefixAsync(prefix, context, abort) {
 		}
 	}
 }
-function mapChainHook(hooks, prefix, isAsync, report, abortGuard) {
+function mapChainHook(_hooks, prefix, isAsync, report, abortGuard, arm) {
+	const hooks = toArray(_hooks);
+	const tail = asyncTail(isAsync);
 	let code = "";
 	let depth = 0;
 	for (let i = 0; i < hooks.length; i++) {
 		const fn = hooks[i];
-		if (i > 0) {
-			code += `if(${abortGuard ? `!${abortGuard}&&` : ""}tmp===undefined){\n`;
+		const guard = i > 0 ? `${abortGuard ? `!${abortGuard}&&` : ""}tmp===undefined` : void 0;
+		if (tail) code += tailStage(tail, guard);
+		else if (guard) {
+			code += `if(${guard}){\n`;
 			depth++;
 		}
 		const t = trace(report, fn);
 		code += t.begin;
-		code += `tmp=${Await(fn)}${prefix}${at(i)}(c)\n`;
-		code += awaitGuard(fn, isAsync, "tmp");
+		code += awaitSite(`${prefix}${at(i)}(c)`, fn, isAsync, "tmp", arm);
 		code += t.end("tmp");
+		if (tail) code += "}\n";
 	}
 	code += "}".repeat(depth);
-	code += `if(tmp!==undefined)_r=c.responseValue=tmp\n`;
+	code += tailPlain(isAsync, `if(tmp!==undefined)_r=c.responseValue=tmp\n`);
 	return code;
 }
-const mapAfterHandle = (_hooks, isAsync, report, abortGuard) => mapChainHook(toArray(_hooks), "af", isAsync, report, abortGuard);
-const mapMapResponse = (_hooks, isAsync, report, abortGuard) => mapChainHook(toArray(_hooks), "mr", isAsync, report, abortGuard);
 const mapAfterResponse = /* @__PURE__ */ map((i, fn, [report]) => {
 	const t = trace(report, fn);
-	const call = require_compile_utils.isAsyncFunction(fn) ? `await ar${at(i)}(c)\n` : `let _ar=ar${at(i)}(c)\nif(_ar instanceof Promise)await _ar\n`;
-	return `try{` + t.begin + call + t.end() + `}catch(_e){` + t.end("_e") + `}\n`;
+	const call = require_compile_utils.isAsyncFunction(fn) ? `await ar${at(i)}(c)\n` : `let _ar=ar${at(i)}(c)\nif(typeof _ar?.then==='function')await _ar\n`;
+	return `try{${t.begin}${call}${t.end()}}catch(_e){${t.end("_e")}console.error(_e)}\n`;
 });
-const mapError = /* @__PURE__ */ map((i, fn, [map, link, mapResponse, schedule, sign, isAsync]) => {
+const mapError = /* @__PURE__ */ map((i, fn, [map, link, mapResponse, schedule, sign, isAsync, arm]) => {
 	link(mapResponse, "rm");
 	link(require_handler_error.adoptErrorType, "aet");
-	return `_r=${Await(fn)}er${at(i)}(c)\n` + awaitGuard(fn, isAsync, "_r") + "if(_r!==undefined){\nif(_r instanceof Response)c.set.status=_r.status\nelse if(c.set.status===undefined||c.set.status===200)c.set.status=500\n" + schedule + sign + `return _em(c,${map}(aet(_r,e),c.set,c.request,true))\n}\n`;
+	return awaitSite(`er${at(i)}(c)`, fn, isAsync, "_r", arm) + "if(_r!==undefined){\nif(_r instanceof Response)c.set.status=_r.status\nelse if(c.set.status===undefined||c.set.status===200)c.set.status=500\n" + schedule + sign + `return _em(c,${map}(aet(_r,e),c.set,c.request,true))\n}\n`;
 });
 function map(map) {
-	return function(event, rest, abortGuard) {
+	return function(event, rest, abortGuard, tail) {
 		if (Array.isArray(event)) {
 			let code = "";
 			let depth = 0;
 			for (let i = 0; i < event.length; i++) {
-				if (i > 0 && abortGuard) {
-					code += `if(!${abortGuard}){\n`;
+				const guard = i > 0 && abortGuard ? `!${abortGuard}` : void 0;
+				if (tail) code += tailStage(tail, guard);
+				else if (guard) {
+					code += `if(${guard}){\n`;
 					depth++;
 				}
 				code += map(i, event[i], rest);
+				if (tail) code += "}\n";
 			}
 			code += "}".repeat(depth);
 			return code;
-		} else return map(void 0, event, rest);
+		} else if (tail) return tailStage(tail) + map(void 0, event, rest) + "}\n";
+		else return map(void 0, event, rest);
 	};
 }
 const at = (index) => index === void 0 ? "" : `[${index}]`;
@@ -495,23 +386,38 @@ function getQueryParseChannels(querySchema) {
 	queryParseChannelsCache.set(querySchema, result);
 	return result ?? void 0;
 }
-const Await = (fn) => require_compile_utils.isAsyncFunction(fn) ? "await " : "";
-const awaitGuard = (fn, isAsync, target) => isAsync && !require_compile_utils.isAsyncFunction(fn) ? `if(${target} instanceof Promise)${target}=await ${target}\n` : "";
+const asyncTail = (mode) => typeof mode === "object" && mode.async ? mode : void 0;
+function awaitSite(call, fn, isAsync, target, arm = "") {
+	if (typeof isAsync !== "object") return `${target}=${call}\n${awaitGuard(fn, isAsync, target, arm)}`;
+	const k = isAsync.n++;
+	isAsync.sites.push(`${isAsync.e ? "e:" : ""}${target}=${call}`);
+	return isAsync.async ? `${target}=_rk===${k}?_y:${call}\nif(_rk===${k}||typeof ${target}?.then==='function'){${arm ? `;${arm}\n` : ""}${target}=await ${target}}\n` : `${target}=${call}\nif(typeof ${target}?.then==='function')return _t(c,${k},${target},${isAsync.r ? "_r" : "undefined"},${isAsync.e ? "e" : "undefined"}${isAsync.live})\n`;
+}
+const tailStage = (mode, guard) => `if(_rk<=${mode.n}${guard ? `&&(_rk===${mode.n}||${guard})` : ""}){\n`;
+const tailPlain = (mode, code) => !code || !asyncTail(mode) ? code : mode.n ? `if(_rk<${mode.n}){\n${code}}\n` : "";
+function awaitGuard(fn, isAsync, target, arm = "") {
+	if (!isAsync) return "";
+	const code = `${arm ? `;${arm}\n` : ""}${target}=await ${target}\n`;
+	return require_compile_utils.isAsyncFunction(fn) ? code : `if(typeof ${target}?.then==='function'){${code}}\n`;
+}
 
 //#endregion
 exports.armEntryAbort = armEntryAbort;
+exports.asyncTail = asyncTail;
+exports.awaitGuard = awaitGuard;
+exports.awaitSite = awaitSite;
 exports.cloneResponse = cloneResponse;
+exports.cloneStaticValue = cloneStaticValue;
 exports.deriveModes = deriveModes;
-exports.emptyResponse = require_handler_utils.emptyResponse;
-exports.extractDeriveKeys = extractDeriveKeys;
 exports.getQueryParseChannels = getQueryParseChannels;
 exports.hasRequestBody = hasRequestBody;
-exports.mapAfterHandle = mapAfterHandle;
 exports.mapAfterResponse = mapAfterResponse;
 exports.mapBeforeHandle = mapBeforeHandle;
+exports.mapChainHook = mapChainHook;
 exports.mapError = mapError;
-exports.mapMapResponse = mapMapResponse;
 exports.mapTransform = mapTransform;
 exports.replaceDeriveContext = replaceDeriveContext;
 exports.runBeforeHandlePrefix = runBeforeHandlePrefix;
-exports.runBeforeHandlePrefixAsync = runBeforeHandlePrefixAsync;
+exports.runBeforeHandlePrefixAsync = runBeforeHandlePrefixAsync;
+exports.tailPlain = tailPlain;
+exports.tailStage = tailStage;