@taserjs/router 0.0.2 → 0.0.4

package/src/types/index.ts

@@ -1,6 +1,5 @@
-import type { Awaitable, TaserCookieJar, TaserHeaders } from '@taserjs/router-core'
-import type { ReplyResult } from '@taserjs/router-utils'
-import type { RouterRegister } from '../register.js'
+import type { Awaitable, TaserCookieJar, TaserHeaders } from "@taserjs/router-core";
+import type { RouterRegister } from "../register.js";
 import type {
   MergeMiddlewareField,
   MergeMiddlewareInputField,
@@ -8,30 +7,33 @@ import type {
   RequestShape,
   RuntimeContextFields,
   Simplify,
-} from './type-utils.js'
-import type { ReturnsMap, ValidHandlerReply } from './returns.js'
-import type { Schema } from './schema.js'
+  UnionToIntersection,
+} from "./type-utils.js";
+import type { ReturnsMap, ValidHandlerReply } from "./returns.js";
+import type { Schema } from "./schema.js";
 import type {
   AppContext,
+  ExtractState,
   HandlerContext,
   HandlerUnit,
+  InlineMiddlewareOptions,
   Method,
   MiddlewareDefinition,
   MiddlewareReturnFromParts,
   MiddlewareUnit,
   NativeContext,
   ValidatorParts,
-} from './units.js'
+} from "./units.js";
 
-export type { Awaitable } from '@taserjs/router-core'
+export type { Awaitable } from "@taserjs/router-core";
 export type {
   SchemaRequestShape,
   InferInput,
   InferOutput,
   Schema,
   StandardSchemaV1,
-} from './schema.js'
-export type { RouterRegister } from '../register.js'
+} from "./schema.js";
+export type { RouterRegister } from "../register.js";
 export type {
   MergeMiddlewareField,
   MergeMiddlewareInputField,
@@ -39,7 +41,7 @@ export type {
   RequestShape,
   Simplify,
   UnwrapPart,
-} from './type-utils.js'
+} from "./type-utils.js";
 export type {
   EmptyReturns,
   EnforceHandlerReply,
@@ -50,112 +52,132 @@ export type {
   ReturnsMap,
   StatusCode,
   ValidHandlerReply,
-} from './returns.js'
+} from "./returns.js";
 export type {
   AppContext,
+  ExtractState,
   HandlerContext,
   HandlerUnit,
   HttpMethod,
   InlineMiddlewareOptions,
+  IsUnknown,
   Method,
   MiddlewareDefinition,
   MiddlewareNext,
   MiddlewareReturnFromParts,
   MiddlewareUnit,
   NativeContext,
+  NextFn,
+  NextResult,
   StandaloneMiddlewareContext,
+  StateBrand,
   ValidatorParts,
-} from './units.js'
+} from "./units.js";
 
-export type RoutePath = RouterRegister extends { RoutePath: infer P } ? P : never
-export type LayoutId = RouterRegister extends { LayoutId: infer L } ? L : never
-export type LayoutTree = RouterRegister extends { LayoutTree: infer T } ? T : Record<never, never>
+export type RoutePath = RouterRegister extends { RoutePath: infer P } ? P : never;
+export type LayoutId = RouterRegister extends { LayoutId: infer L } ? L : never;
+export type LayoutTree = RouterRegister extends { LayoutTree: infer T } ? T : Record<never, never>;
 export type RouteByPathMethod = RouterRegister extends { RouteByPathMethod: infer R }
   ? R
-  : Record<never, never>
-
-type ParseParam<Segment extends string>
-  = Segment extends '*' ? { _splat: string }
-    : Segment extends `:${infer Name extends string}`
-      ? Name extends '' ? {} : { [K in Name]: string }
-      : {}
-
-type ParseParamsFromSegments<S extends string>
-  = S extends `${infer Segment}/${infer Rest}`
-    ? ParseParam<Segment> & ParseParamsFromSegments<Rest>
-    : ParseParam<S>
-
-export type PathParams<Path extends string>
-  = Path extends `/${infer Rest}` ? ParseParamsFromSegments<Rest> : {}
-
-type LayoutParent<Layout extends LayoutId>
-  = LayoutTree[Layout] extends { parent: infer Parent extends LayoutId | null }
-    ? Parent
-    : null
+  : Record<never, never>;
+
+type ParseParam<Segment extends string> = Segment extends "*"
+  ? { _splat: string }
+  : Segment extends `:${infer Name extends string}`
+    ? Name extends ""
+      ? {}
+      : { [K in Name]: string }
+    : {};
+
+type ParseParamsFromSegments<S extends string> = S extends `${infer Segment}/${infer Rest}`
+  ? ParseParam<Segment> & ParseParamsFromSegments<Rest>
+  : ParseParam<S>;
+
+export type PathParams<Path extends string> = Path extends `/${infer Rest}`
+  ? ParseParamsFromSegments<Rest>
+  : {};
+
+type LayoutParent<Layout extends LayoutId> = LayoutTree[Layout] extends {
+  parent: infer Parent extends LayoutId | null;
+}
+  ? Parent
+  : null;
 
-type ResolveLayoutMiddlewares<Layout extends LayoutId | null>
-  = Layout extends null ? readonly []
-    : Layout extends LayoutId
-      ? readonly [
+type ResolveLayoutMiddlewares<Layout extends LayoutId | null> = Layout extends null
+  ? readonly []
+  : Layout extends LayoutId
+    ? readonly [
         ...ResolveLayoutMiddlewares<LayoutParent<Layout>>,
-        ...LayoutTree[Layout]['middlewares'],
+        ...LayoutTree[Layout]["middlewares"],
       ]
-      : readonly []
+    : readonly [];
+
+export type ResolveLayoutIdChain<L extends LayoutId | null> = L extends null
+  ? readonly []
+  : L extends LayoutId
+    ? readonly [...ResolveLayoutIdChain<LayoutParent<L>>, L]
+    : readonly [];
+
+export type ResolveLayoutMiddlewaresState<Layout extends LayoutId | null> = Layout extends LayoutId
+  ? MergeMiddlewareField<ResolveLayoutMiddlewares<Layout>, "state">
+  : {};
+
+export type ResolveLayoutsState<Layouts extends LayoutId | readonly LayoutId[]> =
+  Layouts extends LayoutId
+    ? ResolveLayoutMiddlewaresState<Layouts>
+    : Layouts extends readonly LayoutId[]
+      ? UnionToIntersection<
+          Layouts[number] extends infer L extends LayoutId
+            ? ResolveLayoutMiddlewaresState<L>
+            : never
+        >
+      : {};
 
 /** Fold layout middleware Acc from a route's layoutChain (shallow → deep). */
-type ResolveLayoutChainMiddlewares<Chain extends readonly LayoutId[]>
-  = Chain extends readonly [infer Head extends LayoutId, ...infer Rest]
-    ? readonly [
-      ...LayoutTree[Head]['middlewares'],
-      ...ResolveLayoutChainMiddlewares<
-        Rest extends readonly LayoutId[] ? Rest : readonly []
-      >,
+type ResolveLayoutChainMiddlewares<Chain extends readonly LayoutId[]> = Chain extends readonly [
+  infer Head extends LayoutId,
+  ...infer Rest,
+]
+  ? readonly [
+      ...LayoutTree[Head]["middlewares"],
+      ...ResolveLayoutChainMiddlewares<Rest extends readonly LayoutId[] ? Rest : readonly []>,
     ]
-    : readonly []
+  : readonly [];
 
-type RouteEntry<
-  Path extends RoutePath,
-  TMethod extends Method,
-> = RouteByPathMethod[Path] extends { [K in TMethod]: infer Entry } ? Entry : never
+type RouteEntry<Path extends RoutePath, TMethod extends Method> = RouteByPathMethod[Path] extends {
+  [K in TMethod]: infer Entry;
+}
+  ? Entry
+  : never;
 
-type RouteParent<Path extends RoutePath, TMethod extends Method>
-  = RouteEntry<Path, TMethod> extends { parent: infer Parent extends LayoutId | null }
+type RouteParent<Path extends RoutePath, TMethod extends Method> =
+  RouteEntry<Path, TMethod> extends { parent: infer Parent extends LayoutId | null }
     ? Parent
-    : null
+    : null;
 
-type RouteLayoutChain<Path extends RoutePath, TMethod extends Method>
-  = RouteEntry<Path, TMethod> extends { layoutChain: infer Chain extends readonly LayoutId[] }
+type RouteLayoutChain<Path extends RoutePath, TMethod extends Method> =
+  RouteEntry<Path, TMethod> extends { layoutChain: infer Chain extends readonly LayoutId[] }
     ? Chain
     : RouteParent<Path, TMethod> extends LayoutId
       ? readonly [RouteParent<Path, TMethod>]
-      : readonly []
+      : readonly [];
 
 type RouteResolvedField<
   Path extends RoutePath,
   TMethod extends Method,
   Acc extends readonly unknown[],
-  Field extends 'query' | 'params' | 'body' | 'state' | 'ctx',
-> = MergeMiddlewareField<
-  ResolveLayoutChainMiddlewares<RouteLayoutChain<Path, TMethod>>,
-  Field
-> & MergeMiddlewareField<Acc, Field>
-
-type MiddlewareCtxField<
-  Layout extends LayoutId,
-  Field extends 'query' | 'params' | 'body' | 'state' | 'ctx',
-> = MergeMiddlewareField<
-  ResolveLayoutMiddlewares<LayoutParent<Layout>>,
-  Field
->
+  Field extends "query" | "params" | "body" | "state",
+> = MergeMiddlewareField<ResolveLayoutChainMiddlewares<RouteLayoutChain<Path, TMethod>>, Field> &
+  MergeMiddlewareField<Acc, Field>;
 
 type RouteChainField<
   Path extends RoutePath,
   TMethod extends Method,
   Acc extends readonly unknown[],
-  Field extends 'query' | 'params' | 'body' | 'state' | 'ctx',
-> = RouteResolvedField<Path, TMethod, Acc, Field>
+  Field extends "query" | "params" | "body" | "state",
+> = RouteResolvedField<Path, TMethod, Acc, Field>;
 
-type UnitRuntimeContext = Omit<RuntimeContextFields<NativeContext>, 'var'>
+type UnitRuntimeContext = Omit<RuntimeContextFields<NativeContext>, "var">;
 
 export type RouteChainContext<
   Path extends RoutePath,
@@ -166,104 +188,144 @@ export type RouteChainContext<
   TBody,
   TAppContext extends Record<string, unknown> = AppContext,
 > = Simplify<
-  TAppContext
-  & UnitRuntimeContext
-  & RouteChainField<Path, TMethod, Acc, 'ctx'>
-  & {
-    query: Simplify<MergePart<TQuery, RouteChainField<Path, TMethod, Acc, 'query'>>>
-    params: Simplify<PathParams<Path> & MergePart<TParams, RouteChainField<Path, TMethod, Acc, 'params'>>>
-    body: Simplify<MergePart<TBody, RouteChainField<Path, TMethod, Acc, 'body'>>>
-    state: Simplify<RouteChainField<Path, TMethod, Acc, 'state'>>
-    headers: TaserHeaders
-    cookies: TaserCookieJar
-  }
->
+  TAppContext &
+    UnitRuntimeContext & {
+      query: Simplify<MergePart<TQuery, RouteChainField<Path, TMethod, Acc, "query">>>;
+      params: Simplify<
+        PathParams<Path> & MergePart<TParams, RouteChainField<Path, TMethod, Acc, "params">>
+      >;
+      body: Simplify<MergePart<TBody, RouteChainField<Path, TMethod, Acc, "body">>>;
+      state: Simplify<RouteChainField<Path, TMethod, Acc, "state">>;
+      headers: TaserHeaders;
+      cookies: TaserCookieJar;
+    }
+>;
 
 export type RouteHandleContext<
   Path extends RoutePath,
   TMethod extends Method,
   Acc extends readonly unknown[],
   Validators extends {
-    query?: unknown
-    params?: unknown
-    body?: unknown
+    query?: unknown;
+    params?: unknown;
+    body?: unknown;
   },
   TAppContext extends Record<string, unknown> = AppContext,
 > = Simplify<
-  TAppContext
-  & UnitRuntimeContext
-  & RouteResolvedField<Path, TMethod, Acc, 'ctx'>
-  & {
-    path: Path
-    method: TMethod
-    query: Simplify<MergePart<
-      Validators extends { query?: infer Q } ? Q : unknown,
-      RouteResolvedField<Path, TMethod, Acc, 'query'>
-    >>
-    params: Simplify<PathParams<Path> & MergePart<
-      Validators extends { params?: infer P } ? P : unknown,
-      RouteResolvedField<Path, TMethod, Acc, 'params'>
-    >>
-    body: Simplify<MergePart<
-      Validators extends { body?: infer B } ? B : unknown,
-      RouteResolvedField<Path, TMethod, Acc, 'body'>
-    >>
-    state: Simplify<RouteResolvedField<Path, TMethod, Acc, 'state'>>
-    headers: TaserHeaders
-    cookies: TaserCookieJar
-  }
->
+  TAppContext &
+    UnitRuntimeContext & {
+      path: Path;
+      method: TMethod;
+      query: Simplify<
+        MergePart<
+          Validators extends { query?: infer Q } ? Q : unknown,
+          RouteResolvedField<Path, TMethod, Acc, "query">
+        >
+      >;
+      params: Simplify<
+        PathParams<Path> &
+          MergePart<
+            Validators extends { params?: infer P } ? P : unknown,
+            RouteResolvedField<Path, TMethod, Acc, "params">
+          >
+      >;
+      body: Simplify<
+        MergePart<
+          Validators extends { body?: infer B } ? B : unknown,
+          RouteResolvedField<Path, TMethod, Acc, "body">
+        >
+      >;
+      state: Simplify<RouteResolvedField<Path, TMethod, Acc, "state">>;
+      headers: TaserHeaders;
+      cookies: TaserCookieJar;
+    }
+>;
 
 export type RouteHandleContextWithoutBody<
   Path extends RoutePath,
   TMethod extends Method,
   Acc extends readonly unknown[],
   Validators extends {
-    query?: unknown
-    params?: unknown
+    query?: unknown;
+    params?: unknown;
   },
   TAppContext extends Record<string, unknown> = AppContext,
 > = Simplify<
-  TAppContext
-  & UnitRuntimeContext
-  & RouteResolvedField<Path, TMethod, Acc, 'ctx'>
-  & {
-    path: Path
-    method: TMethod
-    query: Simplify<MergePart<
-      Validators extends { query?: infer Q } ? Q : unknown,
-      RouteResolvedField<Path, TMethod, Acc, 'query'>
-    >>
-    params: Simplify<PathParams<Path> & MergePart<
-      Validators extends { params?: infer P } ? P : unknown,
-      RouteResolvedField<Path, TMethod, Acc, 'params'>
-    >>
-    body: never
-    state: Simplify<RouteResolvedField<Path, TMethod, Acc, 'state'>>
-    headers: TaserHeaders
-    cookies: TaserCookieJar
-  }
->
+  TAppContext &
+    UnitRuntimeContext & {
+      path: Path;
+      method: TMethod;
+      query: Simplify<
+        MergePart<
+          Validators extends { query?: infer Q } ? Q : unknown,
+          RouteResolvedField<Path, TMethod, Acc, "query">
+        >
+      >;
+      params: Simplify<
+        PathParams<Path> &
+          MergePart<
+            Validators extends { params?: infer P } ? P : unknown,
+            RouteResolvedField<Path, TMethod, Acc, "params">
+          >
+      >;
+      body: never;
+      state: Simplify<RouteResolvedField<Path, TMethod, Acc, "state">>;
+      headers: TaserHeaders;
+      cookies: TaserCookieJar;
+    }
+>;
 
 type RouteResolvedInputField<
   Path extends RoutePath,
   TMethod extends Method,
   Acc extends readonly unknown[],
-  Field extends 'query' | 'params' | 'body',
+  Field extends "query" | "params" | "body",
 > = MergeMiddlewareInputField<
   ResolveLayoutChainMiddlewares<RouteLayoutChain<Path, TMethod>>,
   Field
-> & MergeMiddlewareInputField<Acc, Field>
-
-type IsNever<T> = [T] extends [never] ? true : false
-
-type IsEmptyObject<T> = [keyof T] extends [never] ? true : false
-
-type InputFacet<T, K extends string> = IsNever<T> extends true
-  ? {}
-  : IsEmptyObject<T> extends true
-    ? {}
-    : { [Key in K]: T }
+> &
+  MergeMiddlewareInputField<Acc, Field>;
+
+type IsNever<T> = [T] extends [never] ? true : false;
+
+type IsEmptyObject<T> = [keyof T] extends [never] ? true : false;
+
+export type IsLayoutAllowed<TRequiredLayouts, TChain extends readonly LayoutId[]> = [
+  TRequiredLayouts,
+] extends [never]
+  ? true
+  : [unknown] extends [TRequiredLayouts]
+    ? true
+    : [null] extends [TRequiredLayouts]
+      ? true
+      : [undefined] extends [TRequiredLayouts]
+        ? true
+        : TRequiredLayouts extends LayoutId
+          ? TRequiredLayouts extends TChain[number]
+            ? true
+            : false
+          : TRequiredLayouts extends readonly LayoutId[]
+            ? [TRequiredLayouts[number] & TChain[number]] extends [never]
+              ? false
+              : true
+            : true;
+
+export type IsStateSatisfied<TActualState, TRequiredState> = [TRequiredState] extends [never]
+  ? true
+  : [unknown] extends [TRequiredState]
+    ? true
+    : [null] extends [TRequiredState]
+      ? true
+      : [undefined] extends [TRequiredState]
+        ? true
+        : [IsEmptyObject<TRequiredState>] extends [true]
+          ? true
+          : [TActualState] extends [TRequiredState]
+            ? true
+            : false;
+
+type InputFacet<T, K extends string> =
+  IsNever<T> extends true ? {} : IsEmptyObject<T> extends true ? {} : { [Key in K]: T };
 
 type RouteQueryInput<
   Path extends RoutePath,
@@ -276,9 +338,9 @@ type RouteQueryInput<
       Validators extends { queryIn?: infer QI } ? QI : unknown,
       Validators extends { query?: infer Q } ? Q : unknown
     >,
-    RouteResolvedInputField<Path, TMethod, Acc, 'query'>
+    RouteResolvedInputField<Path, TMethod, Acc, "query">
   >
->
+>;
 
 type RouteParamsInput<
   Path extends RoutePath,
@@ -286,14 +348,15 @@ type RouteParamsInput<
   Acc extends readonly unknown[],
   Validators extends ValidatorParts,
 > = Simplify<
-  PathParams<Path> & MergePart<
-    RequestShape<
-      Validators extends { paramsIn?: infer PI } ? PI : unknown,
-      Validators extends { params?: infer P } ? P : unknown
-    >,
-    RouteResolvedInputField<Path, TMethod, Acc, 'params'>
-  >
->
+  PathParams<Path> &
+    MergePart<
+      RequestShape<
+        Validators extends { paramsIn?: infer PI } ? PI : unknown,
+        Validators extends { params?: infer P } ? P : unknown
+      >,
+      RouteResolvedInputField<Path, TMethod, Acc, "params">
+    >
+>;
 
 type RouteBodyInput<
   Path extends RoutePath,
@@ -306,9 +369,9 @@ type RouteBodyInput<
       Validators extends { bodyIn?: infer BI } ? BI : unknown,
       Validators extends { body?: infer B } ? B : unknown
     >,
-    RouteResolvedInputField<Path, TMethod, Acc, 'body'>
+    RouteResolvedInputField<Path, TMethod, Acc, "body">
   >
->
+>;
 
 export type RouteHandleInputWithoutBody<
   Path extends RoutePath,
@@ -316,9 +379,9 @@ export type RouteHandleInputWithoutBody<
   Acc extends readonly unknown[],
   Validators extends ValidatorParts = {},
 > = Simplify<
-  InputFacet<RouteQueryInput<Path, TMethod, Acc, Validators>, 'query'>
-  & InputFacet<RouteParamsInput<Path, TMethod, Acc, Validators>, 'params'>
->
+  InputFacet<RouteQueryInput<Path, TMethod, Acc, Validators>, "query"> &
+    InputFacet<RouteParamsInput<Path, TMethod, Acc, Validators>, "params">
+>;
 
 export type RouteHandleInput<
   Path extends RoutePath,
@@ -326,22 +389,23 @@ export type RouteHandleInput<
   Acc extends readonly unknown[],
   Validators extends ValidatorParts = {},
 > = Simplify<
-  RouteHandleInputWithoutBody<Path, TMethod, Acc, Validators>
-  & InputFacet<RouteBodyInput<Path, TMethod, Acc, Validators>, 'body'>
->
+  RouteHandleInputWithoutBody<Path, TMethod, Acc, Validators> &
+    InputFacet<RouteBodyInput<Path, TMethod, Acc, Validators>, "body">
+>;
 
-export type InferRouteInput<
-  Path extends RoutePath,
-  TMethod extends Method,
-> = RouteEntry<Path, TMethod> extends { route: infer Route }
-  ? Route extends { $Infer: { Input: infer I } } ? I : never
-  : never
+export type InferRouteInput<Path extends RoutePath, TMethod extends Method> =
+  RouteEntry<Path, TMethod> extends { route: infer Route }
+    ? Route extends { $Infer: { Input: infer I } }
+      ? I
+      : never
+    : never;
 
 type MiddlewareChainField<
   Layout extends LayoutId,
   Acc extends readonly unknown[],
-  Field extends 'query' | 'params' | 'body' | 'state' | 'ctx',
-> = MiddlewareCtxField<Layout, Field> & MergeMiddlewareField<Acc, Field>
+  Field extends "query" | "params" | "body" | "state",
+> = MergeMiddlewareField<ResolveLayoutMiddlewares<LayoutParent<Layout>>, Field> &
+  MergeMiddlewareField<Acc, Field>;
 
 export type MiddlewareChainContext<
   Layout extends LayoutId,
@@ -351,227 +415,44 @@ export type MiddlewareChainContext<
   TBody,
   TAppContext extends Record<string, unknown> = AppContext,
 > = Simplify<
-  TAppContext
-  & UnitRuntimeContext
-  & MiddlewareChainField<Layout, Acc, 'ctx'>
-  & {
-    query: Simplify<MergePart<TQuery, MiddlewareChainField<Layout, Acc, 'query'>>>
-    params: Simplify<MergePart<TParams, MiddlewareChainField<Layout, Acc, 'params'>>>
-    body: Simplify<MergePart<TBody, MiddlewareChainField<Layout, Acc, 'body'>>>
-    state: Simplify<MiddlewareChainField<Layout, Acc, 'state'>>
-    headers: TaserHeaders
-    cookies: TaserCookieJar
-  }
->
-
-type StatefulUseOptions<
-  Ctx,
-  TState,
-  TQuery,
-  TParams,
-  TBody,
-  TReturns extends ReturnsMap = {},
-  TQueryIn = unknown,
-  TParamsIn = unknown,
-  TBodyIn = unknown,
-> = {
-  state: Schema<TState>
-  query?: Schema<TQuery, TQueryIn>
-  params?: Schema<TParams, TParamsIn>
-  body?: Schema<TBody, TBodyIn>
-  returns?: TReturns
-  handler: (
-    ctx: Ctx,
-    next: (args: { state: TState }) => Promise<ReplyResult>,
-  ) => Awaitable<ReplyResult | Response | unknown>
-}
-
-type CtxUseOptions<
-  Ctx,
-  TCtx,
-  TQuery,
-  TParams,
-  TBody,
-  TReturns extends ReturnsMap = {},
-  TQueryIn = unknown,
-  TParamsIn = unknown,
-  TBodyIn = unknown,
-> = {
-  ctx: Schema<TCtx>
-  query?: Schema<TQuery, TQueryIn>
-  params?: Schema<TParams, TParamsIn>
-  body?: Schema<TBody, TBodyIn>
-  returns?: TReturns
-  handler: (
-    ctx: Ctx,
-    next: (args: { ctx: TCtx }) => Promise<ReplyResult>,
-  ) => Awaitable<ReplyResult | Response | unknown>
-}
-
-type StateAndCtxUseOptions<
-  Ctx,
-  TState,
-  TCtx,
-  TQuery,
-  TParams,
-  TBody,
-  TReturns extends ReturnsMap = {},
-  TQueryIn = unknown,
-  TParamsIn = unknown,
-  TBodyIn = unknown,
-> = {
-  state: Schema<TState>
-  ctx: Schema<TCtx>
-  query?: Schema<TQuery, TQueryIn>
-  params?: Schema<TParams, TParamsIn>
-  body?: Schema<TBody, TBodyIn>
-  returns?: TReturns
-  handler: (
-    ctx: Ctx,
-    next: (args: { state: TState, ctx: TCtx }) => Promise<ReplyResult>,
-  ) => Awaitable<ReplyResult | Response | unknown>
-}
-
-type StatelessUseOptions<
-  Ctx,
-  TQuery,
-  TParams,
-  TBody,
-  TReturns extends ReturnsMap = {},
-  TQueryIn = unknown,
-  TParamsIn = unknown,
-  TBodyIn = unknown,
-> = {
-  query?: Schema<TQuery, TQueryIn>
-  params?: Schema<TParams, TParamsIn>
-  body?: Schema<TBody, TBodyIn>
-  returns?: TReturns
-  handler: (
-    ctx: Ctx,
-    next: () => Promise<ReplyResult>,
-  ) => Awaitable<ReplyResult | Response | unknown>
-}
+  TAppContext &
+    UnitRuntimeContext & {
+      query: Simplify<MergePart<TQuery, MiddlewareChainField<Layout, Acc, "query">>>;
+      params: Simplify<MergePart<TParams, MiddlewareChainField<Layout, Acc, "params">>>;
+      body: Simplify<MergePart<TBody, MiddlewareChainField<Layout, Acc, "body">>>;
+      state: Simplify<MiddlewareChainField<Layout, Acc, "state">>;
+      headers: TaserHeaders;
+      cookies: TaserCookieJar;
+    }
+>;
 
 export type MiddlewareBuilder<
   Layout extends LayoutId,
   Acc extends readonly unknown[] = readonly [],
   TAppContext extends Record<string, unknown> = AppContext,
 > = Acc & {
-  readonly layout: Layout
-  readonly middlewares: readonly MiddlewareDefinition[]
+  readonly layout: Layout;
+  readonly middlewares: readonly MiddlewareDefinition[];
   readonly $Infer: {
-    Context: MiddlewareChainContext<Layout, Acc, unknown, unknown, unknown, TAppContext>
-  }
-  use<TAcc, TReturns extends ReturnsMap = {}>(
-    unit: MiddlewareUnit<TAcc, TReturns>,
-  ): MiddlewareBuilder<Layout, readonly [...Acc, TAcc], TAppContext>
-  use<
-    TState,
-    TCtx,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    TReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: StateAndCtxUseOptions<
-      MiddlewareChainContext<Layout, Acc, TQuery, TParams, TBody, TAppContext>,
-      TState,
-      TCtx,
-      TQuery,
-      TParams,
-      TBody,
-      TReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): MiddlewareBuilder<
-    Layout,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      TState,
-      TCtx,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    TAppContext
-  >
-  use<
-    TState,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    TReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: StatefulUseOptions<
-      MiddlewareChainContext<Layout, Acc, TQuery, TParams, TBody, TAppContext>,
-      TState,
-      TQuery,
-      TParams,
-      TBody,
-      TReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): MiddlewareBuilder<
-    Layout,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      TState,
-      unknown,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    TAppContext
-  >
-  use<
-    TCtx,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    TReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: CtxUseOptions<
-      MiddlewareChainContext<Layout, Acc, TQuery, TParams, TBody, TAppContext>,
-      TCtx,
-      TQuery,
-      TParams,
-      TBody,
-      TReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): MiddlewareBuilder<
-    Layout,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      {},
-      TCtx,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    TAppContext
-  >
+    Context: MiddlewareChainContext<Layout, Acc, unknown, unknown, unknown, TAppContext>;
+  };
+  use<TAcc, TReturns extends ReturnsMap = {}, TRequiredLayouts = unknown, TRequiredState = unknown>(
+    unit: MiddlewareUnit<TAcc, TReturns, TRequiredLayouts, TRequiredState>,
+    ..._assert: [
+      IsLayoutAllowed<TRequiredLayouts, ResolveLayoutIdChain<Layout>>,
+      IsStateSatisfied<MiddlewareChainField<Layout, Acc, "state">, TRequiredState>,
+    ] extends [true, true]
+      ? []
+      : [
+          {
+            error: "Middleware cannot be attached to this layout";
+            requiredLayouts: TRequiredLayouts;
+            layoutChain: ResolveLayoutIdChain<Layout>;
+            requiredState: TRequiredState;
+            actualState: MiddlewareChainField<Layout, Acc, "state">;
+          },
+        ]
+  ): MiddlewareBuilder<Layout, readonly [...Acc, TAcc], TAppContext>;
   use<
     TQuery = unknown,
     TParams = unknown,
@@ -580,8 +461,9 @@ export type MiddlewareBuilder<
     TQueryIn = unknown,
     TParamsIn = unknown,
     TBodyIn = unknown,
+    R = unknown,
   >(
-    options: StatelessUseOptions<
+    options: InlineMiddlewareOptions<
       MiddlewareChainContext<Layout, Acc, TQuery, TParams, TBody, TAppContext>,
       TQuery,
       TParams,
@@ -589,23 +471,26 @@ export type MiddlewareBuilder<
       TReturns,
       TQueryIn,
       TParamsIn,
-      TBodyIn
+      TBodyIn,
+      R
     >,
   ): MiddlewareBuilder<
     Layout,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      {},
-      unknown,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
+    readonly [
+      ...Acc,
+      MiddlewareReturnFromParts<
+        TQuery,
+        TParams,
+        TBody,
+        ExtractState<R>,
+        TQueryIn,
+        TParamsIn,
+        TBodyIn
+      >,
+    ],
     TAppContext
-  >
-}
+  >;
+};
 
 export type RouteExport<
   Path extends RoutePath,
@@ -614,27 +499,27 @@ export type RouteExport<
   TReturns extends ReturnsMap = {},
   TOutput = Response,
 > = Acc & {
-  readonly path: Path
-  readonly method: TMethod | 'ANY' | 'ALL'
-  readonly methods?: readonly Method[]
-  readonly middlewares: readonly MiddlewareDefinition[]
-  readonly handlerMiddlewares: readonly MiddlewareDefinition[]
-  readonly returns?: TReturns
-  handler: (ctx: unknown) => Awaitable<Response>
+  readonly path: Path;
+  readonly method: TMethod | "ANY" | "ALL";
+  readonly methods?: readonly Method[];
+  readonly middlewares: readonly MiddlewareDefinition[];
+  readonly handlerMiddlewares: readonly MiddlewareDefinition[];
+  readonly returns?: TReturns;
+  handler: (ctx: unknown) => Awaitable<Response>;
   readonly $Infer: {
-    Context: TMethod extends 'GET' | 'DELETE' | 'HEAD' | 'OPTIONS'
+    Context: TMethod extends "GET" | "DELETE" | "HEAD" | "OPTIONS"
       ? RouteHandleContextWithoutBody<Path, TMethod, Acc, {}>
-      : RouteHandleContext<Path, TMethod, Acc, {}>
+      : RouteHandleContext<Path, TMethod, Acc, {}>;
     /** Concrete handler reply (`ReplyOf` union) for client inference. */
-    Output: TOutput
-  }
-  query?: Schema<unknown>
-  params?: Schema<unknown>
-  body?: Schema<unknown>
-  handlerQuery?: Schema<unknown>
-  handlerParams?: Schema<unknown>
-  handlerBody?: Schema<unknown>
-}
+    Output: TOutput;
+  };
+  query?: Schema<unknown>;
+  params?: Schema<unknown>;
+  body?: Schema<unknown>;
+  handlerQuery?: Schema<unknown>;
+  handlerParams?: Schema<unknown>;
+  handlerBody?: Schema<unknown>;
+};
 
 type RouteHandleResult<
   Path extends RoutePath,
@@ -644,17 +529,17 @@ type RouteHandleResult<
   TReturns extends ReturnsMap = {},
   TOutput = Response,
   TAppContext extends Record<string, unknown> = AppContext,
-> = Omit<RouteExport<Path, TMethod, Acc, TReturns, TOutput>, '$Infer'> & {
+> = Omit<RouteExport<Path, TMethod, Acc, TReturns, TOutput>, "$Infer"> & {
   readonly $Infer: {
-    Context: TMethod extends 'GET' | 'DELETE' | 'HEAD' | 'OPTIONS'
+    Context: TMethod extends "GET" | "DELETE" | "HEAD" | "OPTIONS"
       ? RouteHandleContextWithoutBody<Path, TMethod, Acc, Validators, TAppContext>
-      : RouteHandleContext<Path, TMethod, Acc, Validators, TAppContext>
-    Input: TMethod extends 'GET' | 'DELETE' | 'HEAD' | 'OPTIONS'
+      : RouteHandleContext<Path, TMethod, Acc, Validators, TAppContext>;
+    Input: TMethod extends "GET" | "DELETE" | "HEAD" | "OPTIONS"
       ? RouteHandleInputWithoutBody<Path, TMethod, Acc, Validators>
-      : RouteHandleInput<Path, TMethod, Acc, Validators>
-    Output: TOutput
-  }
-}
+      : RouteHandleInput<Path, TMethod, Acc, Validators>;
+    Output: TOutput;
+  };
+};
 
 export type RouteBuilder<
   Path extends RoutePath,
@@ -664,18 +549,32 @@ export type RouteBuilder<
   TReturns extends ReturnsMap = {},
   TAppContext extends Record<string, unknown> = AppContext,
 > = {
-  readonly path: Path
-  readonly method: TMethod
+  readonly path: Path;
+  readonly method: TMethod;
   readonly $Infer: {
-    Context: TMethod extends 'GET' | 'DELETE' | 'HEAD' | 'OPTIONS'
+    Context: TMethod extends "GET" | "DELETE" | "HEAD" | "OPTIONS"
       ? RouteHandleContextWithoutBody<Path, TMethod, Acc, Validators, TAppContext>
-      : RouteHandleContext<Path, TMethod, Acc, Validators, TAppContext>
-  }
+      : RouteHandleContext<Path, TMethod, Acc, Validators, TAppContext>;
+  };
   returns<const M extends ReturnsMap>(
     map: M,
-  ): RouteBuilder<Path, TMethod, Acc, Validators, Omit<TReturns, keyof M> & M, TAppContext>
-  use<TAcc, UReturns extends ReturnsMap = {}>(
-    unit: MiddlewareUnit<TAcc, UReturns>,
+  ): RouteBuilder<Path, TMethod, Acc, Validators, Omit<TReturns, keyof M> & M, TAppContext>;
+  use<TAcc, UReturns extends ReturnsMap = {}, TRequiredLayouts = unknown, TRequiredState = unknown>(
+    unit: MiddlewareUnit<TAcc, UReturns, TRequiredLayouts, TRequiredState>,
+    ..._assert: [
+      IsLayoutAllowed<TRequiredLayouts, RouteLayoutChain<Path, TMethod>>,
+      IsStateSatisfied<RouteResolvedField<Path, TMethod, Acc, "state">, TRequiredState>,
+    ] extends [true, true]
+      ? []
+      : [
+          {
+            error: "Middleware cannot be attached to this route";
+            requiredLayouts: TRequiredLayouts;
+            routeLayoutChain: RouteLayoutChain<Path, TMethod>;
+            requiredState: TRequiredState;
+            actualState: RouteResolvedField<Path, TMethod, Acc, "state">;
+          },
+        ]
   ): RouteBuilder<
     Path,
     TMethod,
@@ -683,10 +582,8 @@ export type RouteBuilder<
     Validators,
     Omit<TReturns, keyof UReturns> & UReturns,
     TAppContext
-  >
+  >;
   use<
-    TState,
-    TCtx,
     TQuery = unknown,
     TParams = unknown,
     TBody = unknown,
@@ -694,169 +591,53 @@ export type RouteBuilder<
     TQueryIn = unknown,
     TParamsIn = unknown,
     TBodyIn = unknown,
+    R = unknown,
   >(
-    options: StateAndCtxUseOptions<
+    options: InlineMiddlewareOptions<
       RouteChainContext<Path, TMethod, Acc, TQuery, TParams, TBody, TAppContext>,
-      TState,
-      TCtx,
       TQuery,
       TParams,
       TBody,
       UReturns,
       TQueryIn,
       TParamsIn,
-      TBodyIn
+      TBodyIn,
+      R
     >,
   ): RouteBuilder<
     Path,
     TMethod,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      TState,
-      TCtx,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    Validators,
-    Omit<TReturns, keyof UReturns> & UReturns,
-    TAppContext
-  >
-  use<
-    TState,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    UReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: StatefulUseOptions<
-      RouteChainContext<Path, TMethod, Acc, TQuery, TParams, TBody, TAppContext>,
-      TState,
-      TQuery,
-      TParams,
-      TBody,
-      UReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): RouteBuilder<
-    Path,
-    TMethod,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      TState,
-      unknown,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    Validators,
-    Omit<TReturns, keyof UReturns> & UReturns,
-    TAppContext
-  >
-  use<
-    TCtx,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    UReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: CtxUseOptions<
-      RouteChainContext<Path, TMethod, Acc, TQuery, TParams, TBody, TAppContext>,
-      TCtx,
-      TQuery,
-      TParams,
-      TBody,
-      UReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): RouteBuilder<
-    Path,
-    TMethod,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      {},
-      TCtx,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    Validators,
-    Omit<TReturns, keyof UReturns> & UReturns,
-    TAppContext
-  >
-  use<
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    UReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: StatelessUseOptions<
-      RouteChainContext<Path, TMethod, Acc, TQuery, TParams, TBody, TAppContext>,
-      TQuery,
-      TParams,
-      TBody,
-      UReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): RouteBuilder<
-    Path,
-    TMethod,
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      {},
-      unknown,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
+    readonly [
+      ...Acc,
+      MiddlewareReturnFromParts<
+        TQuery,
+        TParams,
+        TBody,
+        ExtractState<R>,
+        TQueryIn,
+        TParamsIn,
+        TBodyIn
+      >,
+    ],
     Validators,
     Omit<TReturns, keyof UReturns> & UReturns,
     TAppContext
-  >
+  >;
   handler<R extends Response = Response>(
     fn: (
-      ctx: TMethod extends 'GET' | 'DELETE' | 'HEAD' | 'OPTIONS'
+      ctx: TMethod extends "GET" | "DELETE" | "HEAD" | "OPTIONS"
         ? RouteHandleContextWithoutBody<Path, TMethod, Acc, Validators, TAppContext>
         : RouteHandleContext<Path, TMethod, Acc, Validators, TAppContext>,
     ) => Awaitable<R>,
     ..._assert: [R] extends [ValidHandlerReply<R, TReturns>]
       ? []
-      : [{
-          expected: ValidHandlerReply<R, TReturns>
-          actual: R
-        }]
-  ): RouteHandleResult<
-    Path,
-    TMethod,
-    Acc,
-    Validators,
-    TReturns,
-    R,
-    TAppContext
-  >
+      : [
+          {
+            expected: ValidHandlerReply<R, TReturns>;
+            actual: R;
+          },
+        ]
+  ): RouteHandleResult<Path, TMethod, Acc, Validators, TReturns, R, TAppContext>;
   handler<
     HandlerAcc extends readonly unknown[] = readonly [],
     HandlerValidators extends ValidatorParts = ValidatorParts,
@@ -872,8 +653,8 @@ export type RouteBuilder<
     Omit<TReturns, keyof HReturns> & HReturns,
     HOutput,
     TAppContext
-  >
-}
+  >;
+};
 
 export type HandlerBuilder<
   Acc extends readonly unknown[] = readonly [],
@@ -883,18 +664,27 @@ export type HandlerBuilder<
 > = {
   returns<const M extends ReturnsMap>(
     map: M,
-  ): HandlerBuilder<Acc, Validators, Omit<TReturns, keyof M> & M, TAppContext>
-  use<TAcc, UReturns extends ReturnsMap = {}>(
-    unit: MiddlewareUnit<TAcc, UReturns>,
+  ): HandlerBuilder<Acc, Validators, Omit<TReturns, keyof M> & M, TAppContext>;
+  use<TAcc, UReturns extends ReturnsMap = {}, TRequiredLayouts = unknown, TRequiredState = unknown>(
+    unit: MiddlewareUnit<TAcc, UReturns, TRequiredLayouts, TRequiredState>,
+    ..._assert: [IsStateSatisfied<MergeMiddlewareField<Acc, "state">, TRequiredState>] extends [
+      true,
+    ]
+      ? []
+      : [
+          {
+            error: "Middleware cannot be attached to this handler";
+            requiredState: TRequiredState;
+            actualState: MergeMiddlewareField<Acc, "state">;
+          },
+        ]
   ): HandlerBuilder<
     readonly [...Acc, TAcc],
     Validators,
     Omit<TReturns, keyof UReturns> & UReturns,
     TAppContext
-  >
+  >;
   use<
-    TState,
-    TCtx,
     TQuery = unknown,
     TParams = unknown,
     TBody = unknown,
@@ -902,163 +692,45 @@ export type HandlerBuilder<
     TQueryIn = unknown,
     TParamsIn = unknown,
     TBodyIn = unknown,
+    R = unknown,
   >(
-    options: StateAndCtxUseOptions<
-      HandlerContext<
-        Acc,
-        { query: TQuery, params: TParams, body: TBody },
-        TAppContext
-      >,
-      TState,
-      TCtx,
+    options: InlineMiddlewareOptions<
+      HandlerContext<Acc, { query: TQuery; params: TParams; body: TBody }, TAppContext>,
       TQuery,
       TParams,
       TBody,
       UReturns,
       TQueryIn,
       TParamsIn,
-      TBodyIn
+      TBodyIn,
+      R
     >,
   ): HandlerBuilder<
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      TState,
-      TCtx,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    Validators,
-    Omit<TReturns, keyof UReturns> & UReturns,
-    TAppContext
-  >
-  use<
-    TState,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    UReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: StatefulUseOptions<
-      HandlerContext<
-        Acc,
-        { query: TQuery, params: TParams, body: TBody },
-        TAppContext
-      >,
-      TState,
-      TQuery,
-      TParams,
-      TBody,
-      UReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): HandlerBuilder<
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      TState,
-      unknown,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    Validators,
-    Omit<TReturns, keyof UReturns> & UReturns,
-    TAppContext
-  >
-  use<
-    TCtx,
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    UReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: CtxUseOptions<
-      HandlerContext<
-        Acc,
-        { query: TQuery, params: TParams, body: TBody },
-        TAppContext
+    readonly [
+      ...Acc,
+      MiddlewareReturnFromParts<
+        TQuery,
+        TParams,
+        TBody,
+        ExtractState<R>,
+        TQueryIn,
+        TParamsIn,
+        TBodyIn
       >,
-      TCtx,
-      TQuery,
-      TParams,
-      TBody,
-      UReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): HandlerBuilder<
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      {},
-      TCtx,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
-    Validators,
-    Omit<TReturns, keyof UReturns> & UReturns,
-    TAppContext
-  >
-  use<
-    TQuery = unknown,
-    TParams = unknown,
-    TBody = unknown,
-    UReturns extends ReturnsMap = {},
-    TQueryIn = unknown,
-    TParamsIn = unknown,
-    TBodyIn = unknown,
-  >(
-    options: StatelessUseOptions<
-      HandlerContext<
-        Acc,
-        { query: TQuery, params: TParams, body: TBody },
-        TAppContext
-      >,
-      TQuery,
-      TParams,
-      TBody,
-      UReturns,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >,
-  ): HandlerBuilder<
-    readonly [...Acc, MiddlewareReturnFromParts<
-      TQuery,
-      TParams,
-      TBody,
-      {},
-      unknown,
-      TQueryIn,
-      TParamsIn,
-      TBodyIn
-    >],
+    ],
     Validators,
     Omit<TReturns, keyof UReturns> & UReturns,
     TAppContext
-  >
+  >;
   handler<R extends Response = Response>(
     fn: (ctx: HandlerContext<Acc, Validators, TAppContext>) => Awaitable<R>,
     ..._assert: [R] extends [ValidHandlerReply<R, TReturns>]
       ? []
-      : [{
-          expected: ValidHandlerReply<R, TReturns>
-          actual: R
-        }]
-  ): HandlerUnit<Acc, Validators, TReturns, R>
-}
+      : [
+          {
+            expected: ValidHandlerReply<R, TReturns>;
+            actual: R;
+          },
+        ]
+  ): HandlerUnit<Acc, Validators, TReturns, R>;
+};