elysia 2.0.0-exp.47 → 2.0.0-exp.49

package/dist/adapter/bun/index.js

@@ -6,7 +6,6 @@ const require_adapter_web_standard_index = require('../web-standard/index.js');
 const require_generation = require('../../generation.js');
 const require_compile_handler_index = require('../../compile/handler/index.js');
 const require_route_table = require('../../route-table.js');
-const require_ws_route = require('../../ws/route.js');
 
 //#region src/adapter/bun/index.ts
 const nativeStaticMethods = new Set([
@@ -29,6 +28,16 @@ function collectStaticRoutes(app) {
 	if (!table || !length) return;
 	const methods = table.method;
 	const paths = table.path;
+	const handlers = table.handler;
+	let hasCandidate = false;
+	for (let i = 0; i < length; i++) {
+		const h = handlers[i];
+		if (typeof h === "function" || h instanceof Error || h instanceof Promise) continue;
+		if (!nativeStaticMethods.has(methods[i])) continue;
+		hasCandidate = true;
+		break;
+	}
+	if (!hasCandidate) return;
 	const ready = require_utils.nullObject();
 	const strictPath = frozenRoot["~config"]?.strictPath === true;
 	const seen = /* @__PURE__ */ new Map();
@@ -58,6 +67,8 @@ function collectStaticRoutes(app) {
 		const path = paths[i];
 		if (seen.get(method + " " + path) !== i) continue;
 		if (!nativeStaticMethods.has(method)) continue;
+		const h = handlers[i];
+		if (typeof h === "function" || h instanceof Error || h instanceof Promise) continue;
 		const value = require_compile_handler_index.buildNativeStaticResponse(require_route_table.routeRow(table, i), app);
 		if (!value) continue;
 		add(method, path, value);
@@ -98,11 +109,11 @@ const BunAdapter = require_adapter_index.createAdapter({
 			} catch (error) {
 				console.warn("[Elysia] Native static promotion was skipped:", error);
 			}
-			const hasWs = app["~hasWS"];
 			let websocket;
-			if (hasWs) {
-				const defaultConfig = require_generation.frozenRootOf(app)["~config"]?.websocket;
-				websocket = defaultConfig ? Object.assign(require_ws_route.buildGlobalWSHandler(), defaultConfig) : require_ws_route.buildGlobalWSHandler();
+			if (app["~hasWS"]) {
+				const resolved = require_generation.resolvedWsOf(app);
+				if (!resolved) throw new Error("[Elysia] internal: WebSocket routes are present but no capability provider was resolved.");
+				websocket = resolved.config ? Object.assign(resolved.provider.buildGlobalWSHandler(), resolved.config) : resolved.provider.buildGlobalWSHandler();
 			}
 			return {
 				fetch,

package/dist/adapter/bun/index.mjs

@@ -2,10 +2,9 @@ import { createAdapter } from "../index.mjs";
 import { isDynamicRegex, needEncodeRegex } from "../../constants.mjs";
 import { flattenChain, getLoosePath, nullObject } from "../../utils.mjs";
 import { WebStandardAdapter } from "../web-standard/index.mjs";
-import { frozenRootOf } from "../../generation.mjs";
+import { frozenRootOf, resolvedWsOf } from "../../generation.mjs";
 import { buildNativeStaticResponse } from "../../compile/handler/index.mjs";
 import { routeRow } from "../../route-table.mjs";
-import { buildGlobalWSHandler } from "../../ws/route.mjs";
 
 //#region src/adapter/bun/index.ts
 const nativeStaticMethods = new Set([
@@ -28,6 +27,16 @@ function collectStaticRoutes(app) {
 	if (!table || !length) return;
 	const methods = table.method;
 	const paths = table.path;
+	const handlers = table.handler;
+	let hasCandidate = false;
+	for (let i = 0; i < length; i++) {
+		const h = handlers[i];
+		if (typeof h === "function" || h instanceof Error || h instanceof Promise) continue;
+		if (!nativeStaticMethods.has(methods[i])) continue;
+		hasCandidate = true;
+		break;
+	}
+	if (!hasCandidate) return;
 	const ready = nullObject();
 	const strictPath = frozenRoot["~config"]?.strictPath === true;
 	const seen = /* @__PURE__ */ new Map();
@@ -57,6 +66,8 @@ function collectStaticRoutes(app) {
 		const path = paths[i];
 		if (seen.get(method + " " + path) !== i) continue;
 		if (!nativeStaticMethods.has(method)) continue;
+		const h = handlers[i];
+		if (typeof h === "function" || h instanceof Error || h instanceof Promise) continue;
 		const value = buildNativeStaticResponse(routeRow(table, i), app);
 		if (!value) continue;
 		add(method, path, value);
@@ -97,11 +108,11 @@ const BunAdapter = createAdapter({
 			} catch (error) {
 				console.warn("[Elysia] Native static promotion was skipped:", error);
 			}
-			const hasWs = app["~hasWS"];
 			let websocket;
-			if (hasWs) {
-				const defaultConfig = frozenRootOf(app)["~config"]?.websocket;
-				websocket = defaultConfig ? Object.assign(buildGlobalWSHandler(), defaultConfig) : buildGlobalWSHandler();
+			if (app["~hasWS"]) {
+				const resolved = resolvedWsOf(app);
+				if (!resolved) throw new Error("[Elysia] internal: WebSocket routes are present but no capability provider was resolved.");
+				websocket = resolved.config ? Object.assign(resolved.provider.buildGlobalWSHandler(), resolved.config) : resolved.provider.buildGlobalWSHandler();
 			}
 			return {
 				fetch,

package/dist/base.d.ts

@@ -1,12 +1,12 @@
 import { ElysiaStatus } from "./error.js";
-import { WSHandlerResponse, WSLocalHook, WSMessageHandler } from "./ws/types.js";
-import { TraceHandler } from "./trace.js";
+import { TraceCapability, TraceHandler } from "./trace.js";
 import { ListenCallback, Serve, Server } from "./universal/server.js";
 import { AddRoute, AddWSRoute, AfterHandler, AfterResponseHandler, AnyErrorConstructor, AnyLocalHook, BodyHandler, CompiledHandler, CreateEden, DefaultEphemeral, DefaultMetadata, DefaultSingleton, DefinitionBase, DocumentDecoration, ElysiaConfig, ElysiaHandlerToResponseSchemaAmbiguous, EphemeralType, ErrorDefinitionEntry, ErrorHandler, EventScope, ExcludeElysiaResponse, ExtractErrorFromHandle, GlobalHookReturn, GracefulHandler, GuardHookSingleton, GuardLocalHook, HTTPMethod, HistoryEntry, HookContextSchema, HookContextSingleton, InlineHandler, InlineHandlerNonMacro, InputSchema, InputSchemaKey, InternalRoute, IntersectIfObjectSchema, JoinPath, LocalHook, LocalHookReturn, Macro, MacroSchemaChannel, MacroToContext, MacroToProperty, MapResponse, MaybeArray, MaybePromise, MergeElysiaInstances, MergeSchema, MergeScopedSchemas, MetadataBase, NonResolvableMacroKey, ObjectMacroDefs, OptionalHandler, PluginHookReturn, PreHandler, Prettify, PublicRoute, ResolveRouteErrors, RouteBase, ScopedHookReturn, ScopedMapDeriveReturn, SingletonBase, StaticMapAliases, TransformHandler, UnionResponseStatus, UnwrapRoute, WrapFn } from "./types.js";
 import { ChainNode } from "./utils.js";
 import { Context, ErrorContext, LifecycleContext } from "./context.js";
 import { AnySchema } from "./type/types.js";
 import { AotFingerprint, CompilerSession, ProgramId } from "./compile/aot.js";
+import { WSCapability, WSHandlerResponse, WSLocalHook, WSMessageHandler, WSOptions, WSOptionsEntry } from "./ws/types.js";
 import { RouteTable } from "./route-table.js";
 import { Generation } from "./generation.js";
 import Memoirist from "memoirist";
@@ -39,8 +39,18 @@ declare class Elysia<const in out BasePath extends string = '', const in out Sco
     hoc?: WrapFn<any>[];
     setup?: GracefulHandler<any>[];
     cleanup?: GracefulHandler<any>[];
+    capability?: {
+      trace?: {
+        provider: TraceCapability;
+      };
+      ws?: {
+        provider: WSCapability;
+        options?: WSOptionsEntry[];
+      };
+    };
   };
   '~hookChain'?: ChainNode;
+  '~wsConfig'?: WSOptions;
   server?: Server;
   get history(): readonly HistoryEntry[];
   get ['~routes'](): readonly InternalRoute[];
@@ -514,6 +524,21 @@ declare class Elysia<const in out BasePath extends string = '', const in out Sco
     parser: Metadata['parser'];
     response: Metadata['response'] & MacroContext['response'] & ElysiaHandlerToResponseSchemaAmbiguous<BeforeHandle> & ElysiaHandlerToResponseSchemaAmbiguous<AfterHandle> & ElysiaHandlerToResponseSchemaAmbiguous<ErrorHandle>;
   }, {}, Ephemeral, Volatile>) => NewElysia): Elysia<BasePath, Scope, Singleton, Definitions, Metadata, Routes & NewElysia['~Routes'], Ephemeral, Volatile>;
+  /**
+   * Single idempotent capability resolver-accessor (README rev 2.1). Reads
+   * the merged `~ext.capability` channel — never mutated during a build — and
+   * throws when a capability is used but its provider was never registered.
+   * For `trace` the resolved view IS the nearest-root provider (merge already
+   * keeps it); for `ws` it additionally resolves the depth-precedence base
+   * options. Called at the start of every router build and by `compileHandler`
+   * before it reads the frozen root, so direct-compiler paths resolve without
+   * a prior seal.
+   */
+  ['~resolvedCapability'](kind: 'trace'): TraceCapability | undefined;
+  ['~resolvedCapability'](kind: 'ws'): {
+    provider: WSCapability;
+    options: WSOptions | undefined;
+  } | undefined;
   /**
    * ### macro
    * Declare a custom route property: applying it on a route or guard folds

package/dist/base.js

@@ -6,9 +6,9 @@ const require_utils = require('./utils.js');
 const require_type_bridge = require('./type/bridge.js');
 const require_universal_is_production = require('./universal/is-production.js');
 const require_compile_aot = require('./compile/aot.js');
+const require_generation = require('./generation.js');
 const require_compile_handler_index = require('./compile/handler/index.js');
 const require_route_table = require('./route-table.js');
-const require_ws_route = require('./ws/route.js');
 const require_adapter_bun_index = require('./adapter/bun/index.js');
 const require_handler_fetch = require('./handler/fetch.js');
 const require_compile_analysis_cache = require('./compile/analysis-cache.js');
@@ -57,20 +57,20 @@ var Elysia = class Elysia {
 		if (!this.#declaredRoutes?.length) return [];
 		const routes = this.#declaredRoutes;
 		if (!this["~ext"]?.macro && !this["~scopeChildren"]) return routes;
-		return routes.map((route) => {
-			const localHook = route[4];
-			if (!localHook) return route;
-			const resolved = require_compile_handler_index.resolveLocalHook(require_compile_handler_index.localMacroRoot(route[7] ?? route[3] ?? this, this), localHook, this);
-			if (resolved === localHook) return route;
+		return routes.map((r) => {
+			const localHook = r[4];
+			if (!localHook) return r;
+			const resolved = require_compile_handler_index.resolveLocalHook(require_compile_handler_index.localMacroRoot(r[7] ?? r[3] ?? this, this), localHook, this);
+			if (resolved === localHook) return r;
 			return [
-				route[0],
-				route[1],
-				route[2],
-				route[3],
+				r[0],
+				r[1],
+				r[2],
+				r[3],
 				resolved,
-				route[5],
-				route[6],
-				route[7]
+				r[5],
+				r[6],
+				r[7]
 			];
 		});
 	}
@@ -385,6 +385,11 @@ var Elysia = class Elysia {
 			if (src.models) ext.models = Object.assign(require_utils.nullObject(), src.models);
 			if (src.macro) ext.macro = Object.create(src.macro);
 			if (src.parser) ext.parser = Object.assign(require_utils.nullObject(), src.parser);
+			if (src.capability) {
+				const cap = ext.capability = require_utils.nullObject();
+				if (src.capability.trace) cap.trace = src.capability.trace;
+				if (src.capability.ws) cap.ws = src.capability.ws;
+			}
 		}
 		if (isSchema) child.guard({ ...schemaOrRun });
 		callback(child);
@@ -395,6 +400,21 @@ var Elysia = class Elysia {
 	get #ext() {
 		return this["~ext"] ??= require_utils.nullObject();
 	}
+	["~resolvedCapability"](kind) {
+		if (kind === "ws") {
+			const ws = this["~ext"]?.capability?.ws;
+			const provider = ws?.provider;
+			if (this["~hasWS"] && !provider) throw new Error(require_generation.wsCapabilityRequired);
+			if (!provider) return;
+			return {
+				provider,
+				options: this["~hasWS"] && ws.options?.length ? provider.resolveOptions(ws.options) : void 0
+			};
+		}
+		const provider = this["~ext"]?.capability?.trace?.provider;
+		if (this["~hasTrace"] && !provider) throw new Error(require_generation.traceCapabilityRequired);
+		return provider;
+	}
 	#ensureMacroTable() {
 		const ext = this.#ext;
 		if (ext.macro) return ext.macro;
@@ -620,7 +640,7 @@ var Elysia = class Elysia {
 		}
 		const hookChain = app["~hookChain"];
 		if (app["~ext"]) {
-			const { decorator, store, headers, models, parser, macro, error, hoc, setup, cleanup } = app["~ext"];
+			const { decorator, store, headers, models, parser, macro, error, hoc, setup, cleanup, capability } = app["~ext"];
 			const ext = this["~ext"] ??= require_utils.nullObject();
 			if (decorator) {
 				const cloned = require_utils.clonePlainDecorators(decorator);
@@ -678,6 +698,48 @@ var Elysia = class Elysia {
 					ext.cleanup.push(fn);
 				}
 			} else ext.cleanup = cleanup.slice();
+			if (capability) {
+				const target = ext.capability ??= require_utils.nullObject();
+				if (capability.trace) {
+					const incoming = capability.trace.provider;
+					if (!target.trace) target.trace = { provider: incoming };
+					else if (target.trace.provider !== incoming) console.warn(`[Elysia] Duplicate trace capability providers detected:\n  ${target.trace.provider.id}\n  ${incoming.id}\nUsing the first; ensure a single copy of 'elysia/trace' is installed.`);
+				}
+				if (capability.ws) {
+					const incoming = capability.ws.provider;
+					const existing = target.ws;
+					const incomingOptions = capability.ws.options;
+					if (!existing) target.ws = {
+						provider: incoming,
+						options: incomingOptions?.length ? incomingOptions.map((entry) => ({
+							depth: entry.depth + 1,
+							value: entry.value,
+							origin: entry.origin
+						})) : void 0
+					};
+					else {
+						if (existing.provider !== incoming) console.warn(`[Elysia] Duplicate WebSocket capability providers detected:\n  ${existing.provider.id}\n  ${incoming.id}\nUsing the first; ensure a single copy of 'elysia/websocket' is installed.`);
+						if (incomingOptions?.length) {
+							const base = existing.options ?? [];
+							const seen = new Set(base.map((e) => e.origin));
+							let next;
+							for (const entry of incomingOptions) {
+								if (seen.has(entry.origin)) continue;
+								seen.add(entry.origin);
+								(next ??= base.slice()).push({
+									depth: entry.depth + 1,
+									value: entry.value,
+									origin: entry.origin
+								});
+							}
+							if (next) target.ws = {
+								provider: existing.provider,
+								options: next
+							};
+						}
+					}
+				}
+			}
 		}
 		if (app.#hasPlugin || app.#hasGlobal || hookChain) {
 			let pluginEvents;
@@ -1088,9 +1150,11 @@ var Elysia = class Elysia {
 			throw routeError;
 		}
 		this.#compiled[index] = handler;
+		let releasedNow = false;
 		if (this.#jitColdRemaining !== void 0 && --this.#jitColdRemaining === 0) {
 			require_compile_aot.Compiled.release(this["~programId"]);
 			this.#jitColdRemaining = void 0;
+			releasedNow = true;
 		}
 		const aliases = this.#jitAliases?.[index];
 		if (aliases) {
@@ -1098,9 +1162,16 @@ var Elysia = class Elysia {
 			const map = this["~map"][aliases.method] ??= require_utils.nullObject();
 			for (let p = 0; p < aliases.paths.length; p++) map[aliases.paths[p]] = handler;
 		} else this.#saveHandler(materialized[0], materialized[1], handler);
-		if (this.#jitRoute) this.#jitRoute[index] = void 0;
-		if (this.#jitStatic) this.#jitStatic[index] = void 0;
-		if (this.#jitAliases) this.#jitAliases[index] = void 0;
+		if (releasedNow) {
+			this.#jitRoute = void 0;
+			this.#jitStatic = void 0;
+			this.#jitAliases = void 0;
+			this.#jitTable = void 0;
+		} else {
+			if (this.#jitRoute) this.#jitRoute[index] = void 0;
+			if (this.#jitStatic) this.#jitStatic[index] = void 0;
+			if (this.#jitAliases) this.#jitAliases[index] = void 0;
+		}
 		return handler(context);
 	}
 	#saveHandler(method, path, handler) {
@@ -1216,10 +1287,6 @@ var Elysia = class Elysia {
 		const previousJitStatic = this.#jitStatic;
 		const previousJitAliases = this.#jitAliases;
 		const previousHasDynamicWS = this["~hasDynamicWS"];
-		const config = this["~config"];
-		const hadWebsocket = !!config && Object.hasOwn(config, "websocket");
-		const previousWebsocket = config?.websocket;
-		const websocketSnapshot = previousWebsocket && typeof previousWebsocket === "object" ? { ...previousWebsocket } : previousWebsocket;
 		const previousGeneration = this["~generation"];
 		this["~map"] = void 0;
 		this["~router"] = void 0;
@@ -1258,11 +1325,6 @@ var Elysia = class Elysia {
 			this.#jitAliases = previousJitAliases;
 			this["~hasDynamicWS"] = previousHasDynamicWS;
 			this["~generation"] = previousGeneration;
-			if (config) {
-				this["~config"] = config;
-				if (hadWebsocket) config.websocket = websocketSnapshot;
-				else delete config.websocket;
-			} else this["~config"] = void 0;
 			throw error;
 		} finally {
 			require_compile_aot.endCompilerSession(this, compilerSession, !buildSucceeded);
@@ -1278,17 +1340,17 @@ var Elysia = class Elysia {
 			"~hookChain": this["~hookChain"],
 			"~scopeChildren": this["~scopeChildren"],
 			"~applyMacro": this["~applyMacro"].bind(this),
-			"~programId": this["~programId"]
+			"~programId": this["~programId"],
+			"~wsConfig": this["~wsConfig"]
 		};
 		if (require_universal_is_production.isProduction() && !require_compile_aot.Capture.isAotBuildEnv()) {
 			if (this["~config"]?.precompile) require_compile_aot.Compiled.release(this["~programId"]);
 			else {
-				const routeCount = this["~routeTable"]?.length ?? this["~routes"].length;
+				const table = this["~routeTable"];
+				const routeCount = table?.length ?? this["~routes"].length;
 				let cold = routeCount;
 				const compiled = this.#compiled;
-				if (compiled) {
-					for (let i = 0; i < routeCount; i++) if (compiled[i] !== void 0) cold--;
-				}
+				for (let i = 0; i < routeCount; i++) if (table !== void 0 && (table.flags[i] & require_route_table.RouteFlag.WS) !== 0 || compiled?.[i] !== void 0) cold--;
 				if (cold <= 0) require_compile_aot.Compiled.release(this["~programId"]);
 				else this.#jitColdRemaining = cold;
 			}
@@ -1305,6 +1367,9 @@ var Elysia = class Elysia {
 	}
 	#buildRouterUnsafe() {
 		const precompile = this["~config"]?.precompile;
+		this["~resolvedCapability"]("trace");
+		const wsCap = this["~resolvedCapability"]("ws");
+		let wsConfig;
 		this.#initMap();
 		const methods = this["~map"];
 		const table = this["~routeTable"] = require_route_table.buildRouteTable(this["~routes"]);
@@ -1341,9 +1406,10 @@ var Elysia = class Elysia {
 			const routePath = path[i];
 			const routeFlags = flags[i];
 			if ((routeFlags & require_route_table.RouteFlag.WS) !== 0) {
-				const ws = require_ws_route.buildWSRoute(require_route_table.routeRow(table, i), this);
+				const ws = wsCap.provider.buildWSRoute(require_route_table.routeRow(table, i), this);
 				const handler = ws[0];
 				const options = ws[1];
+				if (wsConfig === void 0 && wsCap.options) wsConfig = wsCap.options;
 				if ((routeFlags & require_route_table.RouteFlag.Dynamic) !== 0) {
 					(this["~router"] ??= new memoirist.default({ loosePath: isLoose })).add("WS", routePath, handler);
 					this["~hasDynamicWS"] = true;
@@ -1357,15 +1423,8 @@ var Elysia = class Elysia {
 					}
 				}
 				if (options && require_utils.isNotEmpty(options)) {
-					this["~config"] ??= require_utils.nullObject();
-					const existing = this["~config"].websocket;
-					if (existing && require_universal_constants.isBun) {
-						for (const key in options) if (key in existing && existing[key] !== options[key]) {
-							console.warn(`[Elysia] Conflicting per-route WebSocket option '${key}'\nBun uses one global WebSocket config per server, per-route values are not enforced (the last-registered route wins).`);
-							console.warn((/* @__PURE__ */ new Error()).stack);
-						}
-						Object.assign(existing, options);
-					} else this["~config"].websocket = options;
+					wsConfig ??= require_utils.nullObject();
+					wsCap.provider.accumulateOptions(wsConfig, options, routePath);
 				}
 				continue;
 			}
@@ -1405,6 +1464,7 @@ var Elysia = class Elysia {
 				for (let p = 0; p < paths.length; p++) map[paths[p]] = handler;
 			}
 		}
+		this["~wsConfig"] = wsConfig;
 	}
 	#fetchFn;
 	get fetch() {

package/dist/base.mjs

@@ -4,9 +4,9 @@ import { clonePlainDecorators, clonePlainDeep, coalesceStandaloneSchemas, create
 import { Ref } from "./type/bridge.mjs";
 import { isProduction } from "./universal/is-production.mjs";
 import { Capture, Compiled, beginCompilerSession, createAotFingerprint, createProgramId, endCompilerSession } from "./compile/aot.mjs";
+import { traceCapabilityRequired, wsCapabilityRequired } from "./generation.mjs";
 import { compileHandler, composeRouteHook, localMacroRoot, resolveLocalHook } from "./compile/handler/index.mjs";
 import { RouteFlag, buildRouteTable, routeRow } from "./route-table.mjs";
-import { buildWSRoute } from "./ws/route.mjs";
 import { BunAdapter } from "./adapter/bun/index.mjs";
 import { applyHoc, createFetchHandler } from "./handler/fetch.mjs";
 import { clearAuthoringAnalysisCaches } from "./compile/analysis-cache.mjs";
@@ -54,20 +54,20 @@ var Elysia = class Elysia {
 		if (!this.#declaredRoutes?.length) return [];
 		const routes = this.#declaredRoutes;
 		if (!this["~ext"]?.macro && !this["~scopeChildren"]) return routes;
-		return routes.map((route) => {
-			const localHook = route[4];
-			if (!localHook) return route;
-			const resolved = resolveLocalHook(localMacroRoot(route[7] ?? route[3] ?? this, this), localHook, this);
-			if (resolved === localHook) return route;
+		return routes.map((r) => {
+			const localHook = r[4];
+			if (!localHook) return r;
+			const resolved = resolveLocalHook(localMacroRoot(r[7] ?? r[3] ?? this, this), localHook, this);
+			if (resolved === localHook) return r;
 			return [
-				route[0],
-				route[1],
-				route[2],
-				route[3],
+				r[0],
+				r[1],
+				r[2],
+				r[3],
 				resolved,
-				route[5],
-				route[6],
-				route[7]
+				r[5],
+				r[6],
+				r[7]
 			];
 		});
 	}
@@ -382,6 +382,11 @@ var Elysia = class Elysia {
 			if (src.models) ext.models = Object.assign(nullObject(), src.models);
 			if (src.macro) ext.macro = Object.create(src.macro);
 			if (src.parser) ext.parser = Object.assign(nullObject(), src.parser);
+			if (src.capability) {
+				const cap = ext.capability = nullObject();
+				if (src.capability.trace) cap.trace = src.capability.trace;
+				if (src.capability.ws) cap.ws = src.capability.ws;
+			}
 		}
 		if (isSchema) child.guard({ ...schemaOrRun });
 		callback(child);
@@ -392,6 +397,21 @@ var Elysia = class Elysia {
 	get #ext() {
 		return this["~ext"] ??= nullObject();
 	}
+	["~resolvedCapability"](kind) {
+		if (kind === "ws") {
+			const ws = this["~ext"]?.capability?.ws;
+			const provider = ws?.provider;
+			if (this["~hasWS"] && !provider) throw new Error(wsCapabilityRequired);
+			if (!provider) return;
+			return {
+				provider,
+				options: this["~hasWS"] && ws.options?.length ? provider.resolveOptions(ws.options) : void 0
+			};
+		}
+		const provider = this["~ext"]?.capability?.trace?.provider;
+		if (this["~hasTrace"] && !provider) throw new Error(traceCapabilityRequired);
+		return provider;
+	}
 	#ensureMacroTable() {
 		const ext = this.#ext;
 		if (ext.macro) return ext.macro;
@@ -617,7 +637,7 @@ var Elysia = class Elysia {
 		}
 		const hookChain = app["~hookChain"];
 		if (app["~ext"]) {
-			const { decorator, store, headers, models, parser, macro, error, hoc, setup, cleanup } = app["~ext"];
+			const { decorator, store, headers, models, parser, macro, error, hoc, setup, cleanup, capability } = app["~ext"];
 			const ext = this["~ext"] ??= nullObject();
 			if (decorator) {
 				const cloned = clonePlainDecorators(decorator);
@@ -675,6 +695,48 @@ var Elysia = class Elysia {
 					ext.cleanup.push(fn);
 				}
 			} else ext.cleanup = cleanup.slice();
+			if (capability) {
+				const target = ext.capability ??= nullObject();
+				if (capability.trace) {
+					const incoming = capability.trace.provider;
+					if (!target.trace) target.trace = { provider: incoming };
+					else if (target.trace.provider !== incoming) console.warn(`[Elysia] Duplicate trace capability providers detected:\n  ${target.trace.provider.id}\n  ${incoming.id}\nUsing the first; ensure a single copy of 'elysia/trace' is installed.`);
+				}
+				if (capability.ws) {
+					const incoming = capability.ws.provider;
+					const existing = target.ws;
+					const incomingOptions = capability.ws.options;
+					if (!existing) target.ws = {
+						provider: incoming,
+						options: incomingOptions?.length ? incomingOptions.map((entry) => ({
+							depth: entry.depth + 1,
+							value: entry.value,
+							origin: entry.origin
+						})) : void 0
+					};
+					else {
+						if (existing.provider !== incoming) console.warn(`[Elysia] Duplicate WebSocket capability providers detected:\n  ${existing.provider.id}\n  ${incoming.id}\nUsing the first; ensure a single copy of 'elysia/websocket' is installed.`);
+						if (incomingOptions?.length) {
+							const base = existing.options ?? [];
+							const seen = new Set(base.map((e) => e.origin));
+							let next;
+							for (const entry of incomingOptions) {
+								if (seen.has(entry.origin)) continue;
+								seen.add(entry.origin);
+								(next ??= base.slice()).push({
+									depth: entry.depth + 1,
+									value: entry.value,
+									origin: entry.origin
+								});
+							}
+							if (next) target.ws = {
+								provider: existing.provider,
+								options: next
+							};
+						}
+					}
+				}
+			}
 		}
 		if (app.#hasPlugin || app.#hasGlobal || hookChain) {
 			let pluginEvents;
@@ -1085,9 +1147,11 @@ var Elysia = class Elysia {
 			throw routeError;
 		}
 		this.#compiled[index] = handler;
+		let releasedNow = false;
 		if (this.#jitColdRemaining !== void 0 && --this.#jitColdRemaining === 0) {
 			Compiled.release(this["~programId"]);
 			this.#jitColdRemaining = void 0;
+			releasedNow = true;
 		}
 		const aliases = this.#jitAliases?.[index];
 		if (aliases) {
@@ -1095,9 +1159,16 @@ var Elysia = class Elysia {
 			const map = this["~map"][aliases.method] ??= nullObject();
 			for (let p = 0; p < aliases.paths.length; p++) map[aliases.paths[p]] = handler;
 		} else this.#saveHandler(materialized[0], materialized[1], handler);
-		if (this.#jitRoute) this.#jitRoute[index] = void 0;
-		if (this.#jitStatic) this.#jitStatic[index] = void 0;
-		if (this.#jitAliases) this.#jitAliases[index] = void 0;
+		if (releasedNow) {
+			this.#jitRoute = void 0;
+			this.#jitStatic = void 0;
+			this.#jitAliases = void 0;
+			this.#jitTable = void 0;
+		} else {
+			if (this.#jitRoute) this.#jitRoute[index] = void 0;
+			if (this.#jitStatic) this.#jitStatic[index] = void 0;
+			if (this.#jitAliases) this.#jitAliases[index] = void 0;
+		}
 		return handler(context);
 	}
 	#saveHandler(method, path, handler) {
@@ -1213,10 +1284,6 @@ var Elysia = class Elysia {
 		const previousJitStatic = this.#jitStatic;
 		const previousJitAliases = this.#jitAliases;
 		const previousHasDynamicWS = this["~hasDynamicWS"];
-		const config = this["~config"];
-		const hadWebsocket = !!config && Object.hasOwn(config, "websocket");
-		const previousWebsocket = config?.websocket;
-		const websocketSnapshot = previousWebsocket && typeof previousWebsocket === "object" ? { ...previousWebsocket } : previousWebsocket;
 		const previousGeneration = this["~generation"];
 		this["~map"] = void 0;
 		this["~router"] = void 0;
@@ -1255,11 +1322,6 @@ var Elysia = class Elysia {
 			this.#jitAliases = previousJitAliases;
 			this["~hasDynamicWS"] = previousHasDynamicWS;
 			this["~generation"] = previousGeneration;
-			if (config) {
-				this["~config"] = config;
-				if (hadWebsocket) config.websocket = websocketSnapshot;
-				else delete config.websocket;
-			} else this["~config"] = void 0;
 			throw error;
 		} finally {
 			endCompilerSession(this, compilerSession, !buildSucceeded);
@@ -1275,17 +1337,17 @@ var Elysia = class Elysia {
 			"~hookChain": this["~hookChain"],
 			"~scopeChildren": this["~scopeChildren"],
 			"~applyMacro": this["~applyMacro"].bind(this),
-			"~programId": this["~programId"]
+			"~programId": this["~programId"],
+			"~wsConfig": this["~wsConfig"]
 		};
 		if (isProduction() && !Capture.isAotBuildEnv()) {
 			if (this["~config"]?.precompile) Compiled.release(this["~programId"]);
 			else {
-				const routeCount = this["~routeTable"]?.length ?? this["~routes"].length;
+				const table = this["~routeTable"];
+				const routeCount = table?.length ?? this["~routes"].length;
 				let cold = routeCount;
 				const compiled = this.#compiled;
-				if (compiled) {
-					for (let i = 0; i < routeCount; i++) if (compiled[i] !== void 0) cold--;
-				}
+				for (let i = 0; i < routeCount; i++) if (table !== void 0 && (table.flags[i] & RouteFlag.WS) !== 0 || compiled?.[i] !== void 0) cold--;
 				if (cold <= 0) Compiled.release(this["~programId"]);
 				else this.#jitColdRemaining = cold;
 			}
@@ -1302,6 +1364,9 @@ var Elysia = class Elysia {
 	}
 	#buildRouterUnsafe() {
 		const precompile = this["~config"]?.precompile;
+		this["~resolvedCapability"]("trace");
+		const wsCap = this["~resolvedCapability"]("ws");
+		let wsConfig;
 		this.#initMap();
 		const methods = this["~map"];
 		const table = this["~routeTable"] = buildRouteTable(this["~routes"]);
@@ -1338,9 +1403,10 @@ var Elysia = class Elysia {
 			const routePath = path[i];
 			const routeFlags = flags[i];
 			if ((routeFlags & RouteFlag.WS) !== 0) {
-				const ws = buildWSRoute(routeRow(table, i), this);
+				const ws = wsCap.provider.buildWSRoute(routeRow(table, i), this);
 				const handler = ws[0];
 				const options = ws[1];
+				if (wsConfig === void 0 && wsCap.options) wsConfig = wsCap.options;
 				if ((routeFlags & RouteFlag.Dynamic) !== 0) {
 					(this["~router"] ??= new Memoirist({ loosePath: isLoose })).add("WS", routePath, handler);
 					this["~hasDynamicWS"] = true;
@@ -1354,15 +1420,8 @@ var Elysia = class Elysia {
 					}
 				}
 				if (options && isNotEmpty(options)) {
-					this["~config"] ??= nullObject();
-					const existing = this["~config"].websocket;
-					if (existing && isBun) {
-						for (const key in options) if (key in existing && existing[key] !== options[key]) {
-							console.warn(`[Elysia] Conflicting per-route WebSocket option '${key}'\nBun uses one global WebSocket config per server, per-route values are not enforced (the last-registered route wins).`);
-							console.warn((/* @__PURE__ */ new Error()).stack);
-						}
-						Object.assign(existing, options);
-					} else this["~config"].websocket = options;
+					wsConfig ??= nullObject();
+					wsCap.provider.accumulateOptions(wsConfig, options, routePath);
 				}
 				continue;
 			}
@@ -1402,6 +1461,7 @@ var Elysia = class Elysia {
 				for (let p = 0; p < paths.length; p++) map[paths[p]] = handler;
 			}
 		}
+		this["~wsConfig"] = wsConfig;
 	}
 	#fetchFn;
 	get fetch() {