@trpc/server 10.28.2 → 10.29.0

package/dist/resolveHTTPResponse-a9be3d96.mjs

@@ -0,0 +1,282 @@
+import { b as callProcedure } from './config-65c5b6d1.mjs';
+import { T as TRPCError, a as getTRPCErrorFromUnknown } from './TRPCError-2b10c8d2.mjs';
+import { t as transformTRPCResponse, g as getErrorShape } from './transformTRPCResponse-3dca0b20.mjs';
+import { g as getJsonContentTypeInputs } from './contentType-acc3be52.mjs';
+import { b as getHTTPStatusCode } from './index-044a193b.mjs';
+
+const HTTP_METHOD_PROCEDURE_TYPE_MAP = {
+    GET: 'query',
+    POST: 'mutation'
+};
+const fallbackContentTypeHandler = {
+    getInputs: getJsonContentTypeInputs
+};
+function initResponse(initOpts) {
+    const { ctx , paths , type , responseMeta , untransformedJSON , errors =[] ,  } = initOpts;
+    let status = untransformedJSON ? getHTTPStatusCode(untransformedJSON) : 200;
+    const headers = {
+        'Content-Type': 'application/json'
+    };
+    const eagerGeneration = !untransformedJSON;
+    const data = eagerGeneration ? [] : Array.isArray(untransformedJSON) ? untransformedJSON : [
+        untransformedJSON
+    ];
+    const meta = responseMeta?.({
+        ctx,
+        paths,
+        type,
+        data,
+        errors,
+        eagerGeneration
+    }) ?? {};
+    for (const [key, value] of Object.entries(meta.headers ?? {})){
+        headers[key] = value;
+    }
+    if (meta.status) {
+        status = meta.status;
+    }
+    return {
+        status,
+        headers
+    };
+}
+async function inputToProcedureCall(procedureOpts) {
+    const { opts , ctx , type , input , path  } = procedureOpts;
+    try {
+        const data = await callProcedure({
+            procedures: opts.router._def.procedures,
+            path,
+            rawInput: input,
+            ctx,
+            type
+        });
+        return {
+            result: {
+                data
+            }
+        };
+    } catch (cause) {
+        const error = getTRPCErrorFromUnknown(cause);
+        opts.onError?.({
+            error,
+            path,
+            input,
+            ctx,
+            type: type,
+            req: opts.req
+        });
+        return {
+            error: getErrorShape({
+                config: opts.router._def._config,
+                error,
+                type,
+                path,
+                input,
+                ctx
+            })
+        };
+    }
+}
+function caughtErrorToData(cause, errorOpts) {
+    const { router , req , onError  } = errorOpts.opts;
+    const error = getTRPCErrorFromUnknown(cause);
+    onError?.({
+        error,
+        path: errorOpts.path,
+        input: errorOpts.input,
+        ctx: errorOpts.ctx,
+        type: errorOpts.type,
+        req
+    });
+    const untransformedJSON = {
+        error: getErrorShape({
+            config: router._def._config,
+            error,
+            type: errorOpts.type,
+            path: errorOpts.path,
+            input: errorOpts.input,
+            ctx: errorOpts.ctx
+        })
+    };
+    const transformedJSON = transformTRPCResponse(router._def._config, untransformedJSON);
+    const body = JSON.stringify(transformedJSON);
+    return {
+        error,
+        untransformedJSON,
+        body
+    };
+}
+// implementation
+async function resolveHTTPResponse(opts) {
+    const { router , req , onHead , onChunk  } = opts;
+    if (req.method === 'HEAD') {
+        // can be used for lambda warmup
+        const headResponse = {
+            status: 204
+        };
+        onHead?.(headResponse);
+        onChunk?.([
+            -1,
+            ''
+        ]);
+        return headResponse;
+    }
+    const contentTypeHandler = opts.contentTypeHandler ?? fallbackContentTypeHandler;
+    const batchingEnabled = opts.batching?.enabled ?? true;
+    const type = HTTP_METHOD_PROCEDURE_TYPE_MAP[req.method] ?? 'unknown';
+    let ctx = undefined;
+    let paths;
+    const isBatchCall = !!req.query.get('batch');
+    const isStreamCall = isBatchCall && onHead && onChunk && req.headers['trpc-batch-mode'] === 'stream';
+    try {
+        if (opts.error) {
+            throw opts.error;
+        }
+        if (isBatchCall && !batchingEnabled) {
+            throw new Error(`Batching is not enabled on the server`);
+        }
+        /* istanbul ignore if -- @preserve */ if (type === 'subscription') {
+            throw new TRPCError({
+                message: 'Subscriptions should use wsLink',
+                code: 'METHOD_NOT_SUPPORTED'
+            });
+        }
+        if (type === 'unknown') {
+            throw new TRPCError({
+                message: `Unexpected request method ${req.method}`,
+                code: 'METHOD_NOT_SUPPORTED'
+            });
+        }
+        const inputs = await contentTypeHandler.getInputs({
+            isBatchCall,
+            req,
+            router,
+            preprocessedBody: opts.preprocessedBody ?? false
+        });
+        paths = isBatchCall ? opts.path.split(',') : [
+            opts.path
+        ];
+        ctx = await opts.createContext();
+        const promises = paths.map((path, index)=>inputToProcedureCall({
+                opts,
+                ctx,
+                type,
+                input: inputs[index],
+                path
+            }));
+        if (!isStreamCall) {
+            /**
+       * Non-streaming response:
+       * - await all responses in parallel, blocking on the slowest one
+       * - create headers with known response body
+       * - return a complete HTTPResponse
+       */ const untransformedJSON = await Promise.all(promises);
+            const errors = untransformedJSON.flatMap((response)=>'error' in response ? [
+                    response.error
+                ] : []);
+            const headResponse1 = initResponse({
+                ctx,
+                paths,
+                type,
+                responseMeta: opts.responseMeta,
+                untransformedJSON,
+                errors
+            });
+            onHead?.(headResponse1);
+            // return body stuff
+            const result = isBatchCall ? untransformedJSON : untransformedJSON[0]; // eslint-disable-line @typescript-eslint/no-non-null-assertion -- `untransformedJSON` should be the length of `paths` which should be at least 1 otherwise there wouldn't be a request at all
+            const transformedJSON = transformTRPCResponse(router._def._config, result);
+            const body = JSON.stringify(transformedJSON);
+            onChunk?.([
+                -1,
+                body
+            ]);
+            return {
+                status: headResponse1.status,
+                headers: headResponse1.headers,
+                body
+            };
+        }
+        /**
+     * Streaming response:
+     * - block on none, call `onChunk` as soon as each response is ready
+     * - create headers with minimal data (cannot know the response body in advance)
+     * - return void
+     */ const headResponse2 = initResponse({
+            ctx,
+            paths,
+            type,
+            responseMeta: opts.responseMeta
+        });
+        onHead(headResponse2);
+        const indexedPromises = new Map(promises.map((promise, index)=>[
+                index,
+                promise.then((r)=>[
+                        index,
+                        r
+                    ])
+            ]));
+        for(let i = 0; i < paths.length; i++){
+            const [index, untransformedJSON1] = await Promise.race(indexedPromises.values());
+            indexedPromises.delete(index);
+            try {
+                const transformedJSON1 = transformTRPCResponse(router._def._config, untransformedJSON1);
+                const body1 = JSON.stringify(transformedJSON1);
+                onChunk([
+                    index,
+                    body1
+                ]);
+            } catch (cause) {
+                const path = paths[index];
+                const input = inputs[index];
+                const { body: body2  } = caughtErrorToData(cause, {
+                    opts,
+                    ctx,
+                    type,
+                    path,
+                    input
+                });
+                onChunk([
+                    index,
+                    body2
+                ]);
+            }
+        }
+        return;
+    } catch (cause1) {
+        // we get here if
+        // - batching is called when it's not enabled
+        // - `createContext()` throws
+        // - `router._def._config.transformer.output.serialize()` throws
+        // - post body is too large
+        // - input deserialization fails
+        // - `errorFormatter` return value is malformed
+        const { error , untransformedJSON: untransformedJSON2 , body: body3  } = caughtErrorToData(cause1, {
+            opts,
+            ctx,
+            type
+        });
+        const headResponse3 = initResponse({
+            ctx,
+            paths,
+            type,
+            responseMeta: opts.responseMeta,
+            untransformedJSON: untransformedJSON2,
+            errors: [
+                error
+            ]
+        });
+        onHead?.(headResponse3);
+        onChunk?.([
+            -1,
+            body3
+        ]);
+        return {
+            status: headResponse3.status,
+            headers: headResponse3.headers,
+            body: body3
+        };
+    }
+}
+
+export { resolveHTTPResponse as r };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trpc/server",
-  "version": "10.28.2",
+  "version": "10.29.0",
   "description": "The tRPC server library",
   "author": "KATT",
   "license": "MIT",
@@ -143,7 +143,7 @@
     "@types/express": "^4.17.17",
     "@types/hash-sum": "^1.0.0",
     "@types/node": "^18.16.16",
-    "@types/react": "^18.2.6",
+    "@types/react": "^18.2.8",
     "@types/react-dom": "^18.2.4",
     "@types/ws": "^8.2.0",
     "@web3-storage/multipart-parser": "^1.0.0",
@@ -164,7 +164,7 @@
     "superstruct": "^1.0.0",
     "tslib": "^2.5.0",
     "tsx": "^3.12.7",
-    "typescript": "^5.0.4",
+    "typescript": "^5.1.3",
     "vitest": "^0.28.5",
     "ws": "^8.0.0",
     "yup": "^1.0.0",
@@ -173,5 +173,5 @@
   "funding": [
     "https://trpc.io/sponsor"
   ],
-  "gitHead": "055c27724a3163ee9fde633f92f30756caf7690b"
+  "gitHead": "49b0a55b74b57bbf7451dad5976d315d9fb7785c"
 }

package/src/adapters/fastify/fastifyRequestHandler.ts

@@ -1,6 +1,12 @@
+import { Readable } from 'node:stream';
 import { FastifyReply, FastifyRequest } from 'fastify';
 import { AnyRouter, inferRouterContext } from '../../core';
-import { HTTPBaseHandlerOptions, HTTPRequest } from '../../http';
+import {
+  getBatchStreamFormatter,
+  HTTPBaseHandlerOptions,
+  HTTPRequest,
+} from '../../http';
+import { HTTPResponse, ResponseChunk } from '../../http/internals/types';
 import { resolveHTTPResponse } from '../../http/resolveHTTPResponse';
 import { NodeHTTPCreateContextOption } from '../node-http';
 
@@ -43,7 +49,48 @@ export async function fastifyRequestHandler<
     body: opts.req.body ?? 'null',
   };
 
-  const result = await resolveHTTPResponse({
+  let resolve: (value: FastifyReply) => void;
+  const promise = new Promise<FastifyReply>((r) => (resolve = r));
+
+  const onHead = (head: HTTPResponse) => {
+    if (!opts.res.statusCode || opts.res.statusCode === 200) {
+      opts.res.statusCode = head.status;
+    }
+    for (const [key, value] of Object.entries(head.headers ?? {})) {
+      /* istanbul ignore if -- @preserve */
+      if (typeof value === 'undefined') {
+        continue;
+      }
+      void opts.res.header(key, value);
+    }
+  };
+
+  let isStream = false;
+  let stream: Readable;
+  const formatter = getBatchStreamFormatter();
+  const onChunk = ([index, string]: ResponseChunk) => {
+    if (index === -1) {
+      // full response, no streaming
+      resolve(opts.res.send(string));
+      return;
+    }
+    if (!isStream) {
+      void opts.res.header('Transfer-Encoding', 'chunked');
+      void opts.res.header(
+        'Vary',
+        opts.res.hasHeader('Vary')
+          ? 'trpc-batch-mode, ' + opts.res.getHeader('Vary')
+          : 'trpc-batch-mode',
+      );
+      stream = new Readable();
+      stream._read = () => {}; // eslint-disable-line @typescript-eslint/no-empty-function -- https://github.com/fastify/fastify/issues/805#issuecomment-369172154
+      resolve(opts.res.send(stream));
+      isStream = true;
+    }
+    stream.push(formatter(index, string));
+  };
+
+  resolveHTTPResponse({
     req,
     createContext,
     path: opts.path,
@@ -53,20 +100,20 @@ export async function fastifyRequestHandler<
     onError(o) {
       opts?.onError?.({ ...o, req: opts.req });
     },
-  });
-
-  const { res } = opts;
-
-  if ('status' in result && (!res.statusCode || res.statusCode === 200)) {
-    res.statusCode = result.status;
-  }
-  for (const [key, value] of Object.entries(result.headers ?? {})) {
-    /* istanbul ignore if -- @preserve */
-    if (typeof value === 'undefined') {
-      continue;
-    }
+    onHead,
+    onChunk,
+  })
+    .then(() => {
+      if (isStream) {
+        stream.push(formatter.end());
+        stream.push(null); // https://github.com/fastify/fastify/issues/805#issuecomment-369172154
+      }
+    })
+    .catch(() => {
+      if (isStream) {
+        stream.push(null);
+      }
+    });
 
-    void res.header(key, value);
-  }
-  await res.send(result.body);
+  return promise;
 }

package/src/adapters/fetch/fetchRequestHandler.ts

@@ -1,5 +1,6 @@
 import { AnyRouter } from '../../core';
-import { HTTPRequest } from '../../http';
+import { getBatchStreamFormatter, HTTPRequest } from '../../http';
+import { HTTPResponse, ResponseChunk } from '../../http/internals/types';
 import { resolveHTTPResponse } from '../../http/resolveHTTPResponse';
 import { FetchHandlerOptions } from './types';
 
@@ -29,7 +30,61 @@ export async function fetchRequestHandler<TRouter extends AnyRouter>(
         : '',
   };
 
-  const result = await resolveHTTPResponse({
+  let resolve: (value: Response) => void;
+  const promise = new Promise<Response>((r) => (resolve = r));
+  let status = 200;
+
+  const onHead = (head: HTTPResponse) => {
+    for (const [key, value] of Object.entries(head.headers ?? {})) {
+      /* istanbul ignore if -- @preserve */
+      if (typeof value === 'undefined') {
+        continue;
+      }
+      if (typeof value === 'string') {
+        resHeaders.set(key, value);
+        continue;
+      }
+      for (const v of value) {
+        resHeaders.append(key, v);
+      }
+    }
+    status = head.status;
+  };
+
+  let isStream = false;
+  let controller: ReadableStreamController<any>;
+  let encoder: TextEncoder;
+  const formatter = getBatchStreamFormatter();
+  const onChunk = ([index, string]: ResponseChunk) => {
+    if (index === -1) {
+      // full response, no streaming
+      const response = new Response(string || null, {
+        status,
+        headers: resHeaders,
+      });
+      resolve(response);
+      return;
+    }
+    if (!isStream) {
+      resHeaders.set('Transfer-Encoding', 'chunked');
+      resHeaders.append('Vary', 'trpc-batch-mode');
+      const stream = new ReadableStream({
+        start(c) {
+          controller = c;
+        },
+      });
+      const response = new Response(stream, {
+        status,
+        headers: resHeaders,
+      });
+      resolve(response);
+      encoder = new TextEncoder();
+      isStream = true;
+    }
+    controller.enqueue(encoder.encode(formatter(index, string)));
+  };
+
+  resolveHTTPResponse({
     req,
     createContext,
     path,
@@ -39,28 +94,20 @@ export async function fetchRequestHandler<TRouter extends AnyRouter>(
     onError(o) {
       opts?.onError?.({ ...o, req: opts.req });
     },
-  });
-
-  for (const [key, value] of Object.entries(result.headers ?? {})) {
-    /* istanbul ignore if -- @preserve */
-    if (typeof value === 'undefined') {
-      continue;
-    }
-
-    if (typeof value === 'string') {
-      resHeaders.set(key, value);
-      continue;
-    }
-
-    for (const v of value) {
-      resHeaders.append(key, v);
-    }
-  }
-
-  const res = new Response(result.body, {
-    status: result.status,
-    headers: resHeaders,
-  });
+    onHead,
+    onChunk,
+  })
+    .then(() => {
+      if (isStream) {
+        controller.enqueue(encoder.encode(formatter.end()));
+        controller.close();
+      }
+    })
+    .catch(() => {
+      if (isStream) {
+        controller.close();
+      }
+    });
 
-  return res;
+  return promise;
 }

package/src/adapters/node-http/content-type/form-data/streamSlice.ts

@@ -1,28 +1,24 @@
 import { Transform, TransformCallback } from 'node:stream';
 
 class SliceStream extends Transform {
-  #start: number;
-  #end: number;
-  #offset = 0;
-  #emitUp = false;
-  #emitDown = false;
+  private indexOffset = 0;
+  private emitUp = false;
+  private emitDown = false;
 
-  constructor(start = 0, end = Infinity) {
+  constructor(private startIndex = 0, private endIndex = Infinity) {
     super();
-    this.#start = start;
-    this.#end = end;
   }
 
   _transform(chunk: any, _: BufferEncoding, done: TransformCallback): void {
-    this.#offset += chunk.length;
+    this.indexOffset += chunk.length;
 
-    if (!this.#emitUp && this.#offset >= this.#start) {
-      this.#emitUp = true;
-      const start = chunk.length - (this.#offset - this.#start);
+    if (!this.emitUp && this.indexOffset >= this.startIndex) {
+      this.emitUp = true;
+      const start = chunk.length - (this.indexOffset - this.startIndex);
 
-      if (this.#offset > this.#end) {
-        const end = chunk.length - (this.#offset - this.#end);
-        this.#emitDown = true;
+      if (this.indexOffset > this.endIndex) {
+        const end = chunk.length - (this.indexOffset - this.endIndex);
+        this.emitDown = true;
         this.push(chunk.slice(start, end));
       } else {
         this.push(chunk.slice(start, chunk.length));
@@ -31,10 +27,12 @@ class SliceStream extends Transform {
       return done();
     }
 
-    if (this.#emitUp && !this.#emitDown) {
-      if (this.#offset >= this.#end) {
-        this.#emitDown = true;
-        this.push(chunk.slice(0, chunk.length - (this.#offset - this.#end)));
+    if (this.emitUp && !this.emitDown) {
+      if (this.indexOffset >= this.endIndex) {
+        this.emitDown = true;
+        this.push(
+          chunk.slice(0, chunk.length - (this.indexOffset - this.endIndex)),
+        );
       } else {
         this.push(chunk);
       }
@@ -46,6 +44,6 @@ class SliceStream extends Transform {
   }
 }
 
-export function streamSlice(start = 0, end = Infinity): SliceStream {
-  return new SliceStream(start, end);
+export function streamSlice(startIndex = 0, endIndex = Infinity): SliceStream {
+  return new SliceStream(startIndex, endIndex);
 }

package/src/adapters/node-http/nodeHTTPRequestHandler.ts

@@ -1,7 +1,8 @@
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
 import { AnyRouter } from '../../core';
 import { inferRouterContext } from '../../core/types';
-import { HTTPRequest } from '../../http';
+import { getBatchStreamFormatter, HTTPRequest } from '../../http';
+import { HTTPResponse, ResponseChunk } from '../../http/internals/types';
 import { resolveHTTPResponse } from '../../http/resolveHTTPResponse';
 import { nodeHTTPJSONContentTypeHandler } from './content-type/json';
 import { NodeHTTPContentTypeHandler } from './internals/contentType';
@@ -63,7 +64,47 @@ export async function nodeHTTPRequestHandler<
       body: bodyResult.ok ? bodyResult.data : undefined,
     };
 
-    const result = await resolveHTTPResponse({
+    const onHead = (head: HTTPResponse) => {
+      if (
+        'status' in head &&
+        (!opts.res.statusCode || opts.res.statusCode === 200)
+      ) {
+        opts.res.statusCode = head.status;
+      }
+      for (const [key, value] of Object.entries(head.headers ?? {})) {
+        /* istanbul ignore if -- @preserve */
+        if (typeof value === 'undefined') {
+          continue;
+        }
+        opts.res.setHeader(key, value);
+      }
+    };
+
+    const formatter = getBatchStreamFormatter();
+    let isStream = false;
+    const onChunk = ([index, string]: ResponseChunk) => {
+      if (index === -1) {
+        /**
+         * Full response, no streaming. This can happen
+         * - if the response is an error
+         * - if response is empty (HEAD request)
+         */
+        opts.res.end(string);
+        return;
+      }
+      if (!isStream) {
+        opts.res.setHeader('Transfer-Encoding', 'chunked');
+        const vary = opts.res.getHeader('Vary');
+        opts.res.setHeader(
+          'Vary',
+          vary ? 'trpc-batch-mode, ' + vary : 'trpc-batch-mode',
+        );
+        isStream = true;
+      }
+      opts.res.write(formatter(index, string));
+    };
+
+    await resolveHTTPResponse({
       batching: opts.batching,
       responseMeta: opts.responseMeta,
       path: opts.path,
@@ -79,18 +120,16 @@ export async function nodeHTTPRequestHandler<
         });
       },
       contentTypeHandler,
+      onHead,
+      onChunk,
     });
 
-    const { res } = opts;
-    if ('status' in result && (!res.statusCode || res.statusCode === 200)) {
-      res.statusCode = result.status;
+    if (!isStream) {
+      return opts.res;
     }
-    for (const [key, value] of Object.entries(result.headers ?? {})) {
-      if (typeof value === 'undefined') {
-        continue;
-      }
-      res.setHeader(key, value);
-    }
-    res.end(result.body);
+
+    opts.res.write(formatter.end());
+    opts.res.end();
+    return opts.res;
   });
 }